First quicksight codes. 🚀

Author: igorborgest

awswrangler/__init__.py

@@ -7,7 +7,7 @@
 
 import logging
 
-from awswrangler import athena, catalog, cloudwatch, db, emr, exceptions, s3  # noqa
+from awswrangler import athena, catalog, cloudwatch, db, emr, exceptions, quicksight, s3  # noqa
 from awswrangler.__metadata__ import __description__, __license__, __title__, __version__  # noqa
 from awswrangler._utils import get_account_id  # noqa
 

awswrangler/_data_types.py

@@ -114,6 +114,34 @@ def athena2redshift(  # pylint: disable=too-many-branches,too-many-return-statem
     raise exceptions.UnsupportedType(f"Unsupported Athena type: {dtype}")  # pragma: no cover
 
 
+def athena2quicksight(dtype: str) -> str:  # pylint: disable=too-many-branches,too-many-return-statements
+    """Athena to Quicksight data types conversion."""
+    dtype = dtype.lower()
+    if dtype == "smallint":
+        return "INTEGER"
+    if dtype in ("int", "integer"):
+        return "INTEGER"
+    if dtype == "bigint":
+        return "INTEGER"
+    if dtype == "float":
+        return "DECIMAL"
+    if dtype == "double":
+        return "DECIMAL"
+    if dtype in ("boolean", "bool"):
+        return "BOOLEAN"
+    if dtype in ("string", "char", "varchar"):
+        return "STRING"
+    if dtype == "timestamp":
+        return "DATETIME"
+    if dtype == "date":
+        return "DATETIME"
+    if dtype.startswith("decimal"):
+        return "DECIMAL"
+    if dtype in ("binary" or "varbinary"):
+        return "BIT"
+    raise exceptions.UnsupportedType(f"Unsupported Athena type: {dtype}")  # pragma: no cover
+
+
 def pyarrow2athena(dtype: pa.DataType) -> str:  # pylint: disable=too-many-branches,too-many-return-statements
     """Pyarrow to Athena data types conversion."""
     if pa.types.is_int8(dtype):

awswrangler/quicksight/__init__.py

@@ -0,0 +1,9 @@
+"""Amazon QuickSight Module."""
+
+from awswrangler.quicksight._cancel import *  # noqa
+from awswrangler.quicksight._create import *  # noqa
+from awswrangler.quicksight._delete import *  # noqa
+from awswrangler.quicksight._describe import *  # noqa
+from awswrangler.quicksight._get import *  # noqa
+from awswrangler.quicksight._list import *  # noqa
+from awswrangler.quicksight._utils import list_ingestions  # noqa

awswrangler/quicksight/_cancel.py

@@ -0,0 +1,58 @@
+"""Amazon QuickSight Cancel Module."""
+
+import logging
+from typing import Optional
+
+import boto3  # type: ignore
+
+from awswrangler import _utils, exceptions
+from awswrangler.quicksight import _get
+
+_logger: logging.Logger = logging.getLogger(__name__)
+
+
+def cancel_ingestion(
+    ingestion_id: str = None,
+    dataset_name: Optional[str] = None,
+    dataset_id: Optional[str] = None,
+    account_id: Optional[str] = None,
+    boto3_session: Optional[boto3.Session] = None,
+) -> None:
+    """Cancel an ongoing ingestion of data into SPICE.
+
+    Note
+    ----
+    You must pass a not None value for ``dataset_name`` or ``dataset_id`` argument.
+
+    Parameters
+    ----------
+    ingestion_id : str
+        Ingestion ID.
+    dataset_name : str, optional
+        Dataset name.
+    dataset_id : str, optional
+        Dataset ID.
+    account_id : str, optional
+        If None, the account ID will be inferred from your boto3 session.
+    boto3_session : boto3.Session(), optional
+        Boto3 Session. The default boto3 session will be used if boto3_session receive None.
+
+    Returns
+    -------
+    None
+        None.
+
+    Examples
+    --------
+    >>> import awswrangler as wr
+    >>>  wr.quicksight.cancel_ingestion(ingestion_id="...", dataset_name="...")
+    """
+    if (dataset_name is None) and (dataset_id is None):
+        raise exceptions.InvalidArgument("You must pass a not None name or dataset_id argument.")
+    session: boto3.Session = _utils.ensure_session(session=boto3_session)
+    if account_id is None:
+        account_id = _utils.get_account_id(boto3_session=session)
+    if (dataset_id is None) and (dataset_name is not None):
+        dataset_id = _get.get_dataset_id(name=dataset_name, account_id=account_id, boto3_session=session)
+    client: boto3.client = _utils.client(service_name="quicksight", session=session)
+    client.cancel_ingestion(IngestionId=ingestion_id, AwsAccountId=account_id, DataSetId=dataset_id)