qontract-reconcile 0.10.1rc948__py3-none-any.whl → 0.10.1rc949__py3-none-any.whl

{qontract_reconcile-0.10.1rc948.dist-info → qontract_reconcile-0.10.1rc949.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: qontract-reconcile
-Version: 0.10.1rc948
+Version: 0.10.1rc949
 Summary: Collection of tools to reconcile services with their desired state as defined in the app-interface DB.
 Home-page: https://github.com/app-sre/qontract-reconcile
 Author: Red Hat App-SRE Team

{qontract_reconcile-0.10.1rc948.dist-info → qontract_reconcile-0.10.1rc949.dist-info}/RECORD

@@ -732,6 +732,8 @@ reconcile/utils/cloud_resource_best_practice/aws_rds.py,sha256=EvE6XKLsrZ531MJpt
 reconcile/utils/clusterhealth/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 reconcile/utils/clusterhealth/providerbase.py,sha256=DXomGYogckBLqWtXn0PXU0hWYxB6K0F7ernldrkHhVY,1140
 reconcile/utils/clusterhealth/telemeter.py,sha256=plZ2xInKJ1ZzgZGPMJzPYmPAWCG2YUxs8j_9IPsqsH8,1426
+reconcile/utils/dynatrace/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+reconcile/utils/dynatrace/client.py,sha256=CsjhxyCxum2X_cl8bAe701rUvVu6PflAwIR0nBEQ2yU,1697
 reconcile/utils/glitchtip/__init__.py,sha256=FT6iBhGqoe7KExFdbgL8AYUb64iW_4snF5__Dcl7yt0,258
 reconcile/utils/glitchtip/client.py,sha256=EXJVIc0aUqkfApU4IIDoc89841XgUL7jJNyb0gLDpMI,7736
 reconcile/utils/glitchtip/models.py,sha256=i3DsGoRJkAeOVsdCEqjlMRjJcYBMXf9Qtx5VpG9xgdA,6391
@@ -844,8 +846,8 @@ tools/test/test_qontract_cli.py,sha256=_D61RFGAN5x44CY1tYbouhlGXXABwYfxKSWSQx3Jr
 tools/test/test_saas_promotion_state.py,sha256=dy4kkSSAQ7bC0Xp2CociETGN-2aABEfL6FU5D9Jl00Y,6056
 tools/test/test_sd_app_sre_alert_report.py,sha256=v363r9zM7__0kR5K6mvJoGFcM9BvE33fWAayrqkpojA,2116
 tools/test/test_sre_checkpoints.py,sha256=SKqPPTl9ua0RFdSSofnoQX-JZE6dFLO3LRhfQzqtfh8,2607
-qontract_reconcile-0.10.1rc948.dist-info/METADATA,sha256=EANO3eIGME2obDm2HO01Vd2LeR5lLaf6_JBfi6ZJEew,2262
-qontract_reconcile-0.10.1rc948.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
-qontract_reconcile-0.10.1rc948.dist-info/entry_points.txt,sha256=GKQqCl2j2X1BJQ69een6rHcR26PmnxnONLNOQB-nRjY,491
-qontract_reconcile-0.10.1rc948.dist-info/top_level.txt,sha256=l5ISPoXzt0SdR4jVdkfa7RPSKNc8zAHYWAnR-Dw8Ey8,24
-qontract_reconcile-0.10.1rc948.dist-info/RECORD,,
+qontract_reconcile-0.10.1rc949.dist-info/METADATA,sha256=SaOcWUK9s8vt6wWla8KbHay5iD3vZYVIDq89zJMrRYA,2262
+qontract_reconcile-0.10.1rc949.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
+qontract_reconcile-0.10.1rc949.dist-info/entry_points.txt,sha256=GKQqCl2j2X1BJQ69een6rHcR26PmnxnONLNOQB-nRjY,491
+qontract_reconcile-0.10.1rc949.dist-info/top_level.txt,sha256=l5ISPoXzt0SdR4jVdkfa7RPSKNc8zAHYWAnR-Dw8Ey8,24
+qontract_reconcile-0.10.1rc949.dist-info/RECORD,,

reconcile/utils/dynatrace/client.py

@@ -0,0 +1,57 @@
+from __future__ import annotations
+
+from collections.abc import Iterable
+
+from dynatrace import Dynatrace
+from pydantic import BaseModel
+
+
+class DynatraceTokenCreationError(Exception):
+    pass
+
+
+class DynatraceTokenRetrievalError(Exception):
+    pass
+
+
+class DynatraceAPITokenCreated(BaseModel):
+    """
+    Wrap dynatrace.ApiTokenCreated for decoupling
+    """
+
+    token: str
+    id: str
+
+
+class DynatraceClient:
+    def __init__(self, environment_url: str, api: Dynatrace) -> None:
+        self._environment_url = environment_url
+        self._api = api
+
+    def create_api_token(
+        self, name: str, scopes: Iterable[str]
+    ) -> DynatraceAPITokenCreated:
+        try:
+            token = self._api.tokens.create(name=name, scopes=scopes)
+        except Exception as e:
+            raise DynatraceTokenCreationError(
+                f"{self._environment_url=} Failed to create token for {name=}", e
+            ) from e
+        return DynatraceAPITokenCreated(token=token.token, id=token.id)
+
+    def get_token_ids_for_name_prefix(self, prefix: str) -> list[str]:
+        try:
+            dt_tokens = self._api.tokens.list()
+        except Exception as e:
+            raise DynatraceTokenRetrievalError(
+                f"{self._environment_url=} Failed to retrieve tokens for {prefix=}", e
+            ) from e
+        return [token.id for token in dt_tokens if token.name.startswith(prefix)]
+
+    @staticmethod
+    def create(
+        environment_url: str, token: str | None, api: Dynatrace | None
+    ) -> DynatraceClient:
+        if not api:
+            api = Dynatrace(base_url=environment_url, token=token)
+        return DynatraceClient(environment_url=environment_url, api=api)