datamasque-python 1.2.4__py3-none-any.whl → 1.2.5__py3-none-any.whl

datamasque/client/models/connection.py

@@ -289,6 +289,22 @@ class SnowflakeConnectionConfig(ConnectionConfig):
         return data
 
 
+# Engines that can sit behind an Amazon RDS, Aurora or Redshift endpoint, and so can carry a role
+# for DataMasque to assume when tagging the resource. Aurora needs no entry of its own: it uses
+# these same engines with a cluster endpoint.
+_AWS_TAGGABLE_DATABASE_TYPES = frozenset(
+    {
+        DatabaseType.postgres,
+        DatabaseType.mysql,
+        DatabaseType.mariadb,
+        DatabaseType.oracle,
+        DatabaseType.mssql,
+        DatabaseType.db2_luw,
+        DatabaseType.redshift,
+    }
+)
+
+
 class DatabaseConnectionConfig(ConnectionConfig):
     """
     Connection configuration for a SQL database.
@@ -309,6 +325,11 @@ class DatabaseConnectionConfig(ConnectionConfig):
     is_read_only: bool = False
     s3_bucket_name: Optional[str] = None
     s3_redshift_iam_role: Optional[str] = None
+    # An IAM role for DataMasque to assume when tagging this database's AWS resource after a run.
+    # Only the engines in `_AWS_TAGGABLE_DATABASE_TYPES` can be an RDS instance, Aurora cluster or
+    # Redshift cluster, so it is pruned for the rest. Distinct from `s3_redshift_iam_role`, which is
+    # the role the Redshift cluster itself uses to reach S3.
+    iam_role_arn: Optional[str] = None
 
     @model_validator(mode="after")
     def _reject_special_engines(self) -> "DatabaseConnectionConfig":
@@ -348,6 +369,8 @@ class DatabaseConnectionConfig(ConnectionConfig):
         if db_type is not DatabaseType.redshift:
             d.pop("s3_bucket_name", None)
             d.pop("s3_redshift_iam_role", None)
+        if db_type not in _AWS_TAGGABLE_DATABASE_TYPES:
+            d.pop("iam_role_arn", None)
         if not d.get("engine_options"):
             d.pop("engine_options", None)
         return d

{datamasque_python-1.2.4.dist-info → datamasque_python-1.2.5.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.5
 Name: datamasque-python
-Version: 1.2.4
+Version: 1.2.5
 Summary: Official Python client for the DataMasque data-masking API.
 Project-URL: Homepage, https://datamasque.com/
 Project-URL: Documentation, https://datamasque-python.readthedocs.io/

{datamasque_python-1.2.4.dist-info → datamasque_python-1.2.5.dist-info}/RECORD

@@ -19,7 +19,7 @@ datamasque/client/spcs.py,sha256=geHrBQ6WLWsMGpSMV8hmK5SFAwftrOU0DXkrR2orOmo,753
 datamasque/client/table_references.py,sha256=dPyB59Sz4iLuF1lc7xffgf9qeNTFA5srT55E8TnFLvk,6092
 datamasque/client/users.py,sha256=VCUo2CJyOw4-aO_3mp_w_0BcSoa6-h1HcReMcAMyumw,3701
 datamasque/client/models/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-datamasque/client/models/connection.py,sha256=loLy4RCiIfY9kvBFspSqJSRhHqhcJxm_mQzXwMRS4VU,17529
+datamasque/client/models/connection.py,sha256=w4-mAja3nSnKkmCs_99RkDLIKvwPzgbB_bnfWEJv-wc,18528
 datamasque/client/models/data_selection.py,sha256=9hRpkQQIQlKGFYDYQCrreggsv96dh6DelppCpCE4S2U,2725
 datamasque/client/models/discovery.py,sha256=J_cAXuIpoQTzHh_-1zvBwXsFnbKI2hRywALb5Yfc5vI,17955
 datamasque/client/models/discovery_config.py,sha256=8sqs9x_X0H_hR1fJDEFmOdJyGRDwgbdb3eddEAe98WE,2420
@@ -38,7 +38,7 @@ datamasque/client/models/safe_data_preview.py,sha256=BiI9d13KdQrTin2JfbjiRiODjns
 datamasque/client/models/status.py,sha256=DXcucL_jRsOQ0vaV0t8CJt6crvCIOLBn07KuwpwW-4A,3247
 datamasque/client/models/table_reference.py,sha256=55fKdHjC2TQnUmbp0XiWqfeFGXqcq4wXjbWgMaN3p3c,3835
 datamasque/client/models/user.py,sha256=UGAUzgJkf78m24_zFXXoA99zdut48BXkX_ivV8yq1Vc,2043
-datamasque_python-1.2.4.dist-info/METADATA,sha256=JfLuGmshJx1zHMoPocIbxfuO3H3jMKAJEa5hPTnKPNM,4497
-datamasque_python-1.2.4.dist-info/WHEEL,sha256=zOwg4jB6zX2kU910N-cMawjivD6tO8NEWvE12je1bVk,87
-datamasque_python-1.2.4.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-datamasque_python-1.2.4.dist-info/RECORD,,
+datamasque_python-1.2.5.dist-info/METADATA,sha256=kfyTtmM4E4_WDXVh_8vrzyIzMOyn21gvvGaJGqSfqLw,4497
+datamasque_python-1.2.5.dist-info/WHEEL,sha256=zOwg4jB6zX2kU910N-cMawjivD6tO8NEWvE12je1bVk,87
+datamasque_python-1.2.5.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+datamasque_python-1.2.5.dist-info/RECORD,,