qontract-reconcile 0.10.1rc992__py3-none-any.whl → 0.10.1rc993__py3-none-any.whl

{qontract_reconcile-0.10.1rc992.dist-info → qontract_reconcile-0.10.1rc993.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: qontract-reconcile
-Version: 0.10.1rc992
+Version: 0.10.1rc993
 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.1rc992.dist-info → qontract_reconcile-0.10.1rc993.dist-info}/RECORD

@@ -740,7 +740,7 @@ reconcile/utils/clusterhealth/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5N
 reconcile/utils/clusterhealth/providerbase.py,sha256=DXomGYogckBLqWtXn0PXU0hWYxB6K0F7ernldrkHhVY,1140
 reconcile/utils/clusterhealth/telemeter.py,sha256=PllSLsJXvGNatmTF4mxCNPVbDrpr_MPk0m5pWj-LT6g,1534
 reconcile/utils/dynatrace/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-reconcile/utils/dynatrace/client.py,sha256=CsjhxyCxum2X_cl8bAe701rUvVu6PflAwIR0nBEQ2yU,1697
+reconcile/utils/dynatrace/client.py,sha256=siVDdCQeLaWm4W1T-QKnFwlotPmXmZfpyoe3pXowBt8,2760
 reconcile/utils/glitchtip/__init__.py,sha256=FT6iBhGqoe7KExFdbgL8AYUb64iW_4snF5__Dcl7yt0,258
 reconcile/utils/glitchtip/client.py,sha256=ovh4tx-ajlihjvcq6nyY4chulbuMJYvzDPv9j9CuAKM,7867
 reconcile/utils/glitchtip/models.py,sha256=uHbCK9-RWgxNYUAkEHXRAZRDDZW7jkOFFt9MdlqN4bU,6481
@@ -854,8 +854,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.1rc992.dist-info/METADATA,sha256=DRFJMM8ZzS7zsrDv77wTXC9PngnJiNg-SphE3U7qjrQ,2262
-qontract_reconcile-0.10.1rc992.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
-qontract_reconcile-0.10.1rc992.dist-info/entry_points.txt,sha256=GKQqCl2j2X1BJQ69een6rHcR26PmnxnONLNOQB-nRjY,491
-qontract_reconcile-0.10.1rc992.dist-info/top_level.txt,sha256=l5ISPoXzt0SdR4jVdkfa7RPSKNc8zAHYWAnR-Dw8Ey8,24
-qontract_reconcile-0.10.1rc992.dist-info/RECORD,,
+qontract_reconcile-0.10.1rc993.dist-info/METADATA,sha256=D2wGMe7boygHgZavBj6P-78L2Mx4C9FZxKzi-rHUTYc,2262
+qontract_reconcile-0.10.1rc993.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
+qontract_reconcile-0.10.1rc993.dist-info/entry_points.txt,sha256=GKQqCl2j2X1BJQ69een6rHcR26PmnxnONLNOQB-nRjY,491
+qontract_reconcile-0.10.1rc993.dist-info/top_level.txt,sha256=l5ISPoXzt0SdR4jVdkfa7RPSKNc8zAHYWAnR-Dw8Ey8,24
+qontract_reconcile-0.10.1rc993.dist-info/RECORD,,

reconcile/utils/dynatrace/client.py

@@ -3,6 +3,7 @@ from __future__ import annotations
 from collections.abc import Iterable
 
 from dynatrace import Dynatrace
+from dynatrace.environment_v2.tokens_api import ApiTokenUpdate
 from pydantic import BaseModel
 
 
@@ -10,6 +11,10 @@ class DynatraceTokenCreationError(Exception):
     pass
 
 
+class DynatraceTokenUpdateError(Exception):
+    pass
+
+
 class DynatraceTokenRetrievalError(Exception):
     pass
 
@@ -23,6 +28,11 @@ class DynatraceAPITokenCreated(BaseModel):
     id: str
 
 
+class DynatraceAPIToken(BaseModel):
+    id: str
+    scopes: list[str]
+
+
 class DynatraceClient:
     def __init__(self, environment_url: str, api: Dynatrace) -> None:
         self._environment_url = environment_url
@@ -48,6 +58,29 @@ class DynatraceClient:
             ) from e
         return [token.id for token in dt_tokens if token.name.startswith(prefix)]
 
+    def get_token_by_id(self, token_id: str) -> DynatraceAPIToken:
+        try:
+            token = self._api.tokens.get(token_id=token_id)
+        except Exception as e:
+            raise DynatraceTokenRetrievalError(
+                f"{self._environment_url=} Failed to retrieve token for {token_id=}", e
+            ) from e
+        return DynatraceAPIToken(id=token.id, scopes=token.scopes)
+
+    def update_token_scopes(self, token_id: str, scopes: list[str]) -> None:
+        try:
+            self._api.tokens.put(
+                token_id=token_id,
+                api_token=ApiTokenUpdate(
+                    scopes=scopes,
+                ),
+            )
+        except Exception as e:
+            raise DynatraceTokenUpdateError(
+                f"{self._environment_url=} Failed to update token scopes for {token_id=}",
+                e,
+            ) from e
+
     @staticmethod
     def create(
         environment_url: str, token: str | None, api: Dynatrace | None