qontract-reconcile 0.10.1rc879__py3-none-any.whl → 0.10.1rc894__py3-none-any.whl

reconcile/closedbox_endpoint_monitoring_base.py

@@ -3,10 +3,7 @@ import logging
 from collections import defaultdict
 from collections.abc import Callable
 from dataclasses import field
-from typing import (
-    Any,
-    Optional,
-)
+from typing import Any
 from urllib.parse import urlparse
 
 from pydantic.dataclasses import dataclass
@@ -49,14 +46,14 @@ class EndpointMonitoringProvider:
     name: str
     provider: str
     description: str
-    timeout: Optional[str] = None
-    checkInterval: Optional[str] = None
-    blackboxExporter: Optional[BlackboxMonitoringProvider] = None
-    signalFx: Optional[SignalfxMonitoringProvier] = None
-    metricLabels: Optional[str] = None
+    timeout: str | None = None
+    checkInterval: str | None = None
+    blackboxExporter: BlackboxMonitoringProvider | None = None
+    signalFx: SignalfxMonitoringProvier | None = None
+    metricLabels: str | None = None
 
     @property
-    def namespace(self) -> Optional[dict[str, Any]]:
+    def namespace(self) -> dict[str, Any] | None:
         if self.blackboxExporter:
             return self.blackboxExporter.namespace
 
@@ -131,7 +128,7 @@ def fill_desired_state(
 def run_for_provider(
     provider: str,
     probe_builder: Callable[
-        [EndpointMonitoringProvider, list[Endpoint]], Optional[OpenshiftResource]
+        [EndpointMonitoringProvider, list[Endpoint]], OpenshiftResource | None
     ],
     integration: str,
     integration_version: str,

reconcile/cluster_deployment_mapper.py

@@ -1,9 +1,6 @@
 import logging
 import sys
-from typing import (
-    Any,
-    Optional,
-)
+from typing import Any
 
 from reconcile.status import ExitCodes
 from reconcile.typed_queries.app_interface_vault_settings import (
@@ -18,7 +15,7 @@ from reconcile.utils.vault import VaultClient
 QONTRACT_INTEGRATION = "cluster-deployment-mapper"
 
 
-def run(dry_run: bool, vault_output_path: Optional[str]) -> None:
+def run(dry_run: bool, vault_output_path: str | None) -> None:
     """Get Hive ClusterDeployments from clusters and save mapping to Vault"""
     if not vault_output_path:
         logging.error("must supply vault output path")

reconcile/cna/assets/asset.py

@@ -9,10 +9,7 @@ from dataclasses import (
     field,
 )
 from enum import Enum
-from typing import (
-    Any,
-    Optional,
-)
+from typing import Any
 
 
 class AssetError(Exception):
@@ -31,9 +28,9 @@ class AssetStatus(Enum):
 
 @dataclass(frozen=True)
 class Asset(ABC):
-    uuid: Optional[str] = field(compare=False, hash=True)
-    href: Optional[str] = field(compare=False, hash=True)
-    status: Optional[AssetStatus] = field(compare=False, hash=True)
+    uuid: str | None = field(compare=False, hash=True)
+    href: str | None = field(compare=False, hash=True)
+    status: AssetStatus | None = field(compare=False, hash=True)
     name: str
     kind: AssetType
 

reconcile/cna/assets/null.py

@@ -2,10 +2,7 @@ from __future__ import annotations
 
 from collections.abc import Mapping
 from dataclasses import dataclass
-from typing import (
-    Any,
-    Optional,
-)
+from typing import Any
 
 from reconcile.cna.assets.asset import (
     Asset,
@@ -18,7 +15,7 @@ from reconcile.gql_definitions.cna.queries.cna_resources import CNANullAssetV1
 
 @dataclass(frozen=True)
 class NullAsset(Asset):
-    addr_block: Optional[str]
+    addr_block: str | None
 
     def api_payload(self) -> dict[str, Any]:
         return {

reconcile/cna/integration.py

@@ -3,7 +3,6 @@ from collections.abc import (
     Iterable,
     Mapping,
 )
-from typing import Optional
 
 from reconcile.cna.assets.asset_factory import asset_factory_from_schema
 from reconcile.cna.client import CNAClient
@@ -45,8 +44,8 @@ class CNAIntegration:
         self,
         cna_clients: Mapping[str, CNAClient],
         namespaces: Iterable[NamespaceV1],
-        desired_states: Optional[Mapping[str, State]] = None,
-        current_states: Optional[Mapping[str, State]] = None,
+        desired_states: Mapping[str, State] | None = None,
+        current_states: Mapping[str, State] | None = None,
     ):
         self._cna_clients = cna_clients
         self._namespaces = namespaces

reconcile/cna/state.py

@@ -4,10 +4,7 @@ from collections.abc import (
     Iterable,
     Mapping,
 )
-from typing import (
-    Any,
-    Optional,
-)
+from typing import Any
 
 from reconcile.cna.assets.asset import (
     Asset,
@@ -29,7 +26,7 @@ class State:
     deletions and updates to reach another state.
     """
 
-    def __init__(self, assets: Optional[dict[AssetType, dict[str, Asset]]] = None):
+    def __init__(self, assets: dict[AssetType, dict[str, Asset]] | None = None):
         self._assets: dict[AssetType, dict[str, Asset]] = {}
         for kind in AssetType:
             self._assets[kind] = {}

reconcile/dashdotdb_base.py

@@ -3,10 +3,7 @@ import os
 from base64 import b64encode
 from collections.abc import Mapping
 from dataclasses import dataclass
-from typing import (
-    Any,
-    Optional,
-)
+from typing import Any
 from urllib.parse import urljoin
 
 import requests
@@ -23,8 +20,8 @@ LOG = logging.getLogger(__name__)
 class DashdotDBSecret:
     path: str
     field: str
-    q_format: Optional[str]
-    version: Optional[int]
+    q_format: str | None
+    version: int | None
 
 
 DASHDOTDB_SECRET = DashdotDBSecret(
@@ -53,7 +50,7 @@ class DashdotdbBase:
         self.dashdotdb_pass = self.secret_content["password"]
         self.logmarker = marker
         self.scope = scope
-        self.dashdotdb_token = Optional[str]
+        self.dashdotdb_token: str | None
 
     def _get_token(self) -> None:
         if self.dry_run:
@@ -134,10 +131,10 @@ class DashdotdbBase:
     def _promget(
         self,
         url: str,
-        params: Optional[Mapping[Any, Any]],
-        token: Optional[str] = None,
-        username: Optional[str] = None,
-        password: Optional[str] = None,
+        params: Mapping[Any, Any] | None,
+        token: str | None = None,
+        username: str | None = None,
+        password: str | None = None,
         ssl_verify: bool = True,
         uri: str = "api/v1/query",
     ) -> dict[Any, Any]:

reconcile/dashdotdb_cso.py

@@ -1,7 +1,4 @@
-from typing import (
-    Any,
-    Optional,
-)
+from typing import Any
 
 import requests
 from sretoolbox.utils import threaded
@@ -44,7 +41,7 @@ class DashdotdbCSO(DashdotdbBase):
             secret_reader=secret_reader,
         )
 
-    def _post(self, manifest: dict[Any, Any]) -> Optional[requests.Response]:
+    def _post(self, manifest: dict[Any, Any]) -> requests.Response | None:
         if manifest is None:
             return None
 
@@ -70,7 +67,7 @@ class DashdotdbCSO(DashdotdbBase):
         return response
 
     @staticmethod
-    def _get_imagemanifestvuln(cluster: str, oc_map: OCMap) -> Optional[dict[str, Any]]:
+    def _get_imagemanifestvuln(cluster: str, oc_map: OCMap) -> dict[str, Any] | None:
         LOG.info("%s processing %s", LOGMARKER, cluster)
         oc = oc_map.get(cluster)
         if isinstance(oc, OCLogMsg):

reconcile/dashdotdb_dora.py

@@ -1,17 +1,15 @@
 import functools
 import os
 from collections import defaultdict
+from collections.abc import Iterable, Mapping
 from dataclasses import dataclass
 from datetime import (
+    UTC,
     datetime,
     timedelta,
-    timezone,
 )
 from typing import (
     Any,
-    Iterable,
-    Mapping,
-    Optional,
     Self,
 )
 
@@ -44,8 +42,8 @@ QONTRACT_INTEGRATION = "dashdotdb-dora"
 class DeploymentDBSecret:
     path: str
     field: str
-    q_format: Optional[str]
-    version: Optional[int]
+    q_format: str | None
+    version: int | None
 
 
 @dataclass(eq=True, frozen=True)
@@ -76,9 +74,9 @@ class SaasTarget:
 
 @dataclass(eq=True, frozen=True)
 class RepoChanges:
-    repo_url: Optional[str]
-    ref_from: Optional[str]
-    ref_to: Optional[str]
+    repo_url: str | None
+    ref_from: str | None
+    ref_to: str | None
 
 
 @dataclass(eq=True, frozen=True)
@@ -162,12 +160,10 @@ class Commit:
         finish_timestamp_tzaware = finish_timestamp
 
         if commit_date_tzaware.tzinfo is None:
-            commit_date_tzaware = commit_date_tzaware.replace(tzinfo=timezone.utc)
+            commit_date_tzaware = commit_date_tzaware.replace(tzinfo=UTC)
 
         if finish_timestamp_tzaware.tzinfo is None:
-            finish_timestamp_tzaware = finish_timestamp_tzaware.replace(
-                tzinfo=timezone.utc
-            )
+            finish_timestamp_tzaware = finish_timestamp_tzaware.replace(tzinfo=UTC)
 
         return int((finish_timestamp_tzaware - commit_date_tzaware).total_seconds())
 
@@ -397,7 +393,7 @@ class DashdotdbDORA(DashdotdbBase):
 
     def get_repo_ref_for_sha(
         self, saastarget: SaasTarget, sha: str
-    ) -> tuple[Optional[str], Optional[str]]:
+    ) -> tuple[str | None, str | None]:
         try:
             saas_file_yaml = self.gl_app_interface_get_file(
                 saastarget.path, ref=sha

reconcile/dashdotdb_dvo.py

@@ -4,10 +4,7 @@ from collections.abc import (
     Sequence,
 )
 from dataclasses import dataclass
-from typing import (
-    Any,
-    Optional,
-)
+from typing import Any
 
 import requests
 from requests import Response
@@ -67,7 +64,7 @@ class DashdotdbDVO(DashdotdbBase):
         for i in range(0, len(data), int(size)):
             yield data[i : i + int(size)]
 
-    def _post(self, deploymentvalidation: DVOPayload) -> Optional[Response]:
+    def _post(self, deploymentvalidation: DVOPayload) -> Response | None:
         if deploymentvalidation is None:
             return
         cluster_name = deploymentvalidation.cluster_name
@@ -130,7 +127,7 @@ class DashdotdbDVO(DashdotdbBase):
 
     def _get_deploymentvalidation(
         self, metrics: list[str], cluster: ClusterV1
-    ) -> Optional[DVOPayload]:
+    ) -> DVOPayload | None:
         prom_info = self._get_prometheus_info(cluster)
         if not prom_info:
             return None
@@ -162,8 +159,8 @@ class DashdotdbDVO(DashdotdbBase):
     # via startswith and return only those that match.
     # Returns a map of {cluster: cluster_name, data: [metric_names]}
     def _get_validation_names(
-        self, cluster: ClusterV1, filter: Optional[str] = None
-    ) -> Optional[ClusterValidationMetrics]:
+        self, cluster: ClusterV1, filter: str | None = None
+    ) -> ClusterValidationMetrics | None:
         prom_info = self._get_prometheus_info(cluster)
         if not prom_info:
             return None
@@ -202,7 +199,7 @@ class DashdotdbDVO(DashdotdbBase):
             metrics=deploymentvalidation["data"],
         )
 
-    def _get_prometheus_info(self, cluster: ClusterV1) -> Optional[PrometheusInfo]:
+    def _get_prometheus_info(self, cluster: ClusterV1) -> PrometheusInfo | None:
         if not cluster.automation_token:
             LOG.error(
                 "%s cluster %s does not have an automation token",
@@ -217,14 +214,14 @@ class DashdotdbDVO(DashdotdbBase):
         )
 
     @staticmethod
-    def _get_clusters(name: Optional[str] = None) -> list[ClusterV1]:
+    def _get_clusters(name: str | None = None) -> list[ClusterV1]:
         return [
             c for c in get_clusters_minimal(name=name) if c.ocm and c.prometheus_url
         ]
 
-    def run(self, cname: Optional[str] = None) -> None:
+    def run(self, cname: str | None = None) -> None:
         clusters = self._get_clusters(name=cname)
-        validation_list: list[Optional[ClusterValidationMetrics]] = threaded.run(
+        validation_list: list[ClusterValidationMetrics | None] = threaded.run(
             func=self._get_validation_names,
             iterable=clusters,
             thread_pool_size=self.thread_pool_size,
@@ -258,7 +255,7 @@ class DashdotdbDVO(DashdotdbBase):
 def run(
     dry_run: bool = False,
     thread_pool_size: int = 10,
-    cluster_name: Optional[str] = None,
+    cluster_name: str | None = None,
 ) -> None:
     vault_settings = get_app_interface_vault_settings()
     secret_reader = create_secret_reader(use_vault=vault_settings.vault)

reconcile/dashdotdb_slo.py

@@ -1,9 +1,6 @@
 from collections.abc import Iterable
 from dataclasses import dataclass
-from typing import (
-    Any,
-    Optional,
-)
+from typing import Any
 
 import jinja2
 import requests
@@ -72,7 +69,7 @@ class DashdotdbSLO(DashdotdbBase):
             secret_reader=secret_reader,
         )
 
-    def _post(self, service_slos: Iterable[ServiceSLO]) -> Optional[Response]:
+    def _post(self, service_slos: Iterable[ServiceSLO]) -> Response | None:
         for item in service_slos:
             LOG.debug(f"About to POST SLO JSON item to dashdotDB:\n{item}\n")
 
@@ -106,9 +103,9 @@ class DashdotdbSLO(DashdotdbBase):
                 continue
 
             ns = namespace_access.namespace
-            promtoken: Optional[str] = None
-            username: Optional[str] = None
-            password: Optional[str] = None
+            promtoken: str | None = None
+            username: str | None = None
+            password: str | None = None
             if namespace_access.prometheus_access:
                 promurl = namespace_access.prometheus_access.url
                 username = self.secret_reader.read_secret(

reconcile/database_access_manager.py

@@ -1,5 +1,6 @@
 import base64
 import logging
+from collections.abc import Callable
 from random import choices
 from string import (
     ascii_letters,
@@ -7,8 +8,6 @@ from string import (
 )
 from typing import (
     Any,
-    Callable,
-    Optional,
     TypedDict,
     cast,
 )
@@ -57,7 +56,7 @@ JOB_PSQL_ENGINE_VERSION = "15.4-alpine"
 
 
 def get_database_access_namespaces(
-    query_func: Optional[Callable] = None,
+    query_func: Callable | None = None,
 ) -> list[NamespaceV1]:
     if not query_func:
         query_func = gql.get_api().query
@@ -74,7 +73,7 @@ class DatabaseConnectionParameters(BaseModel):
 
 class PSQLScriptGenerator(BaseModel):
     db_access: DatabaseAccessV1
-    current_db_access: Optional[DatabaseAccessV1]
+    current_db_access: DatabaseAccessV1 | None
     connection_parameter: DatabaseConnectionParameters
     admin_connection_parameter: DatabaseConnectionParameters
     engine: str
@@ -426,7 +425,7 @@ def _populate_resources(
     settings: dict[Any, Any],
     user_connection: DatabaseConnectionParameters,
     admin_connection: DatabaseConnectionParameters,
-    current_db_access: Optional[DatabaseAccessV1] = None,
+    current_db_access: DatabaseAccessV1 | None = None,
 ) -> list[DBAMResource]:
     if user_connection.database == admin_connection.database:
         raise ValueError(f"Can not use default database {admin_connection.database}")
@@ -594,7 +593,7 @@ def _process_db_access(
     if not _db_access_acccess_is_valid(db_access):
         raise ValueError("Duplicate schema in access list.")
 
-    current_db_access: Optional[DatabaseAccessV1] = None
+    current_db_access: DatabaseAccessV1 | None = None
     if state.exists(db_access.name):
         current_state = state.get(db_access.name)
         if current_state == db_access.dict(by_alias=True):

reconcile/dynatrace_token_provider/integration.py

@@ -1,16 +1,14 @@
 import base64
 import logging
 import sys
-from collections.abc import Iterable
+from collections.abc import Iterable, Mapping
 from datetime import timedelta
-from typing import (
-    Any,
-    Mapping,
-)
+from typing import Any
 
 from dynatrace import Dynatrace
 from dynatrace.environment_v2.tokens_api import ApiTokenCreated
 
+from reconcile.dynatrace_token_provider.meta import QONTRACT_INTEGRATION
 from reconcile.dynatrace_token_provider.metrics import (
     DTPClustersManagedGauge,
     DTPOrganizationErrorRate,
@@ -55,7 +53,6 @@ from reconcile.utils.runtime.integration import (
 )
 from reconcile.utils.secret_reader import SecretReaderBase
 
-QONTRACT_INTEGRATION = "dynatrace-token-provider"
 SYNCSET_ID = "ext-dynatrace-tokens-dtp"
 SECRET_NAME = "dynatrace-token-dtp"
 SECRET_NAMESPACE = "dynatrace"

reconcile/dynatrace_token_provider/integration_v2.py

@@ -0,0 +1,20 @@
+from reconcile.dynatrace_token_provider.meta import QONTRACT_INTEGRATION
+from reconcile.utils.runtime.integration import (
+    PydanticRunParams,
+    QontractReconcileIntegration,
+)
+
+
+class DynatraceTokenProviderIntegrationParamsV2(PydanticRunParams):
+    ocm_organization_ids: set[str] | None = None
+
+
+class DynatraceTokenProviderIntegrationV2(
+    QontractReconcileIntegration[DynatraceTokenProviderIntegrationParamsV2]
+):
+    @property
+    def name(self) -> str:
+        return QONTRACT_INTEGRATION
+
+    def run(self, dry_run: bool) -> None:
+        pass

reconcile/dynatrace_token_provider/meta.py

@@ -0,0 +1 @@
+QONTRACT_INTEGRATION = "dynatrace-token-provider"

reconcile/external_resources/integration.py

@@ -1,5 +1,5 @@
 import logging
-from typing import Callable
+from collections.abc import Callable
 
 from reconcile.external_resources.manager import (
     ExternalResourcesInventory,

reconcile/external_resources/manager.py

@@ -1,8 +1,8 @@
 import json
 import logging
 from collections.abc import Iterable
-from datetime import datetime, timezone
-from enum import Enum
+from datetime import UTC, datetime
+from enum import StrEnum
 
 from sretoolbox.utils import threaded
 
@@ -73,7 +73,7 @@ def setup_factories(
     return of
 
 
-class ReconcileAction(str, Enum):
+class ReconcileAction(StrEnum):
     NOOP = "NOOP"
     APPLY_NOT_EXISTS = "Resource does not exist"
     APPLY_ERROR = "Resource status in ERROR state"
@@ -260,7 +260,7 @@ class ExternalResourcesManager:
         return need_secret_sync
 
     def _update_state(self, r: Reconciliation, state: ExternalResourceState) -> None:
-        state.ts = datetime.now(timezone.utc)
+        state.ts = datetime.now(UTC)
         if r.action == Action.APPLY:
             state.resource_status = ResourceStatus.IN_PROGRESS
         elif r.action == Action.DESTROY:

reconcile/external_resources/model.py

@@ -9,7 +9,7 @@ from collections.abc import (
     Iterator,
     MutableMapping,
 )
-from enum import Enum
+from enum import StrEnum
 from typing import Any
 
 from pydantic import BaseModel
@@ -82,7 +82,7 @@ class ExternalResourcesInventory(MutableMapping):
             for (p, ns) in desired_providers
             for r in p.resources
             if isinstance(
-                r, (NamespaceTerraformResourceRDSV1, NamespaceTerraformResourceRoleV1)
+                r, NamespaceTerraformResourceRDSV1 | NamespaceTerraformResourceRoleV1
             )
             and r.managed_by_erv2
         ]
@@ -123,7 +123,7 @@ class ExternalResourcesInventory(MutableMapping):
             raise FetchResourceError(msg)
 
 
-class Action(str, Enum):
+class Action(StrEnum):
     DESTROY: str = "Destroy"
     APPLY: str = "Apply"
 

reconcile/external_resources/secrets_sync.py

@@ -3,9 +3,9 @@ import json
 import logging
 from abc import abstractmethod
 from collections.abc import Iterable, Mapping
-from datetime import datetime, timezone
+from datetime import UTC, datetime
 from hashlib import shake_128
-from typing import Any, Optional
+from typing import Any
 
 from pydantic import BaseModel
 from sretoolbox.utils import threaded
@@ -45,8 +45,8 @@ class VaultSecret(BaseModel):
 
     path: str
     field: str
-    version: Optional[int]
-    q_format: Optional[str]
+    version: int | None
+    q_format: str | None
 
 
 class SecretHelper:
@@ -324,7 +324,7 @@ class InClusterSecretsReconciler(SecretsReconciler):
                 else:
                     data[k] = decoded
 
-            spec.metadata[SECRET_UPDATED_AT] = datetime.now(timezone.utc).strftime(
+            spec.metadata[SECRET_UPDATED_AT] = datetime.now(UTC).strftime(
                 SECRET_UPDATED_AT_TIMEFORMAT
             )
             spec.secret = data

reconcile/external_resources/state.py

@@ -1,7 +1,7 @@
 import logging
 from collections.abc import Mapping
-from datetime import datetime, timezone
-from enum import Enum
+from datetime import UTC, datetime
+from enum import StrEnum
 from typing import Any
 
 from pydantic import BaseModel
@@ -20,14 +20,14 @@ class StateNotFoundError(Exception):
     pass
 
 
-class ReconcileStatus(str, Enum):
+class ReconcileStatus(StrEnum):
     SUCCESS: str = "SUCCESS"
     ERROR: str = "ERROR"
     IN_PROGRESS: str = "IN_PROGRESS"
     NOT_EXISTS: str = "NOT_EXISTS"
 
 
-class ResourceStatus(str, Enum):
+class ResourceStatus(StrEnum):
     CREATED: str = "CREATED"
     DELETED: str = "DELETED"
     ABANDONED: str = "ABANDONED"
@@ -198,7 +198,7 @@ class ExternalResourcesStateDynamoDB:
         else:
             return ExternalResourceState(
                 key=key,
-                ts=datetime.now(timezone.utc),
+                ts=datetime.now(UTC),
                 resource_status=ResourceStatus.NOT_EXISTS,
                 reconciliation=Reconciliation(key=key),
                 reconciliation_errors=0,

reconcile/gabi_authorized_users.py

@@ -10,10 +10,7 @@ from datetime import (
     date,
     datetime,
 )
-from typing import (
-    Any,
-    Optional,
-)
+from typing import Any
 
 import reconcile.openshift_base as ob
 from reconcile import queries
@@ -114,9 +111,9 @@ def fetch_desired_state(
 def run(
     dry_run: bool,
     thread_pool_size: int = 10,
-    internal: Optional[bool] = None,
+    internal: bool | None = None,
     use_jump_host: bool = True,
-    defer: Optional[Callable] = None,
+    defer: Callable | None = None,
 ) -> None:
     gabi_instances = queries.get_gabi_instances()
     if not gabi_instances:

reconcile/gcr_mirror.py

@@ -241,7 +241,7 @@ class QuayMirror:
         control_file_name = "qontract-reconcile-gcr-mirror.timestamp"
         control_file_path = os.path.join(tempfile.gettempdir(), control_file_name)
         try:
-            with open(control_file_path, "r", encoding="locale") as file_obj:
+            with open(control_file_path, encoding="locale") as file_obj:
                 last_deep_sync = float(file_obj.read())
         except FileNotFoundError:
             self._record_timestamp(control_file_path)