qontract-reconcile 0.10.2.dev159__tar.gz → 0.10.2.dev160__tar.gz

{qontract_reconcile-0.10.2.dev159 → qontract_reconcile-0.10.2.dev160}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: qontract-reconcile
-Version: 0.10.2.dev159
+Version: 0.10.2.dev160
 Summary: Collection of tools to reconcile services with their desired state as defined in the app-interface DB.
 Project-URL: homepage, https://github.com/app-sre/qontract-reconcile
 Project-URL: repository, https://github.com/app-sre/qontract-reconcile

{qontract_reconcile-0.10.2.dev159 → qontract_reconcile-0.10.2.dev160}/reconcile/aws_cloudwatch_log_retention/integration.py

@@ -5,13 +5,22 @@ from collections import defaultdict
 from collections.abc import Iterable
 from datetime import UTC, datetime, timedelta
 from enum import Enum
-from typing import TYPE_CHECKING
+from typing import (
+    TYPE_CHECKING,
+)
 
 from botocore.exceptions import ClientError
 from pydantic import BaseModel
 
 from reconcile import queries
-from reconcile.queries import get_aws_accounts
+from reconcile.gql_definitions.aws_cloudwatch_log_retention.aws_accounts import (
+    AWSAccountCleanupOptionCloudWatchV1,
+    AWSAccountV1,
+)
+from reconcile.typed_queries.aws_cloudwatch_log_retention.aws_accounts import (
+    get_aws_accounts,
+)
+from reconcile.utils import gql
 from reconcile.utils.aws_api import AWSApi
 
 if TYPE_CHECKING:
@@ -39,31 +48,33 @@ DEFAULT_AWS_CLOUDWATCH_CLEANUP_OPTION = AWSCloudwatchCleanupOption(
 
 
 def get_desired_cleanup_options_by_region(
-    account: dict,
+    account: AWSAccountV1,
 ) -> dict[str, list[AWSCloudwatchCleanupOption]]:
-    default_region = account["resourcesDefaultRegion"]
+    default_region = account.resources_default_region
     result = defaultdict(list)
-    if cleanup := account.get("cleanup"):
-        for cleanup_option in cleanup:
-            if cleanup_option["provider"] == "cloudwatch":
-                region = cleanup_option.get("region") or default_region
-                result[region].append(
-                    AWSCloudwatchCleanupOption(
-                        regex=re.compile(cleanup_option["regex"]),
-                        retention_in_days=cleanup_option["retention_in_days"],
-                        delete_empty_log_group=bool(
-                            cleanup_option["delete_empty_log_group"]
-                        ),
-                    )
+    for cleanup_option in account.cleanup or []:
+        if isinstance(cleanup_option, AWSAccountCleanupOptionCloudWatchV1):
+            region = cleanup_option.region or default_region
+            result[region].append(
+                AWSCloudwatchCleanupOption(
+                    regex=re.compile(cleanup_option.regex),
+                    retention_in_days=cleanup_option.retention_in_days,
+                    delete_empty_log_group=bool(cleanup_option.delete_empty_log_group),
                 )
+            )
     if not result:
         result[default_region].append(DEFAULT_AWS_CLOUDWATCH_CLEANUP_OPTION)
     return result
 
 
-def create_awsapi_client(accounts: list, thread_pool_size: int) -> AWSApi:
+def create_awsapi_client(accounts: list[AWSAccountV1], thread_pool_size: int) -> AWSApi:
     settings = queries.get_secret_reader_settings()
-    return AWSApi(thread_pool_size, accounts, settings=settings, init_users=False)
+    return AWSApi(
+        thread_pool_size,
+        [account.dict(by_alias=True) for account in accounts],
+        settings=settings,
+        init_users=False,
+    )
 
 
 def is_empty(log_group: LogGroupTypeDef) -> bool:
@@ -192,10 +203,10 @@ def _find_desired_cleanup_option(
 
 def _reconcile_log_groups(
     dry_run: bool,
-    aws_account: dict,
+    aws_account: AWSAccountV1,
     awsapi: AWSApi,
 ) -> None:
-    account_name = aws_account["name"]
+    account_name = aws_account.name
     desired_cleanup_options_by_region = get_desired_cleanup_options_by_region(
         aws_account
     )
@@ -230,12 +241,15 @@ def _reconcile_log_groups(
             )
 
 
-def get_active_aws_accounts() -> list[dict]:
+def get_active_aws_accounts() -> list[AWSAccountV1]:
     return [
-        a
-        for a in get_aws_accounts(cleanup=True)
-        if "aws-cloudwatch-log-retention"
-        not in (a.get("disable") or {}).get("integrations", [])
+        account
+        for account in get_aws_accounts(gql.get_api())
+        if not (
+            account.disable
+            and account.disable.integrations
+            and "aws-cloudwatch-log-retention" in account.disable.integrations
+        )
     ]
 
 

qontract_reconcile-0.10.2.dev160/reconcile/gql_definitions/aws_cloudwatch_log_retention/aws_accounts.py

@@ -0,0 +1,158 @@
+"""
+Generated by qenerate plugin=pydantic_v1. DO NOT MODIFY MANUALLY!
+"""
+from collections.abc import Callable  # noqa: F401 # pylint: disable=W0611
+from datetime import datetime  # noqa: F401 # pylint: disable=W0611
+from enum import Enum  # noqa: F401 # pylint: disable=W0611
+from typing import (  # noqa: F401 # pylint: disable=W0611
+    Any,
+    Optional,
+    Union,
+)
+
+from pydantic import (  # noqa: F401 # pylint: disable=W0611
+    BaseModel,
+    Extra,
+    Field,
+    Json,
+)
+
+
+DEFINITION = """
+query AWSAccountsCloudwatchLogRetentionCleanup {
+  accounts: awsaccounts_v1
+  {
+    path
+    name
+    uid
+    terraformUsername
+    consoleUrl
+    resourcesDefaultRegion
+    supportedDeploymentRegions
+    providerVersion
+    accountOwners {
+      name
+      email
+    }
+    automationToken {
+      path
+      field
+      version
+      format
+    }
+    garbageCollection
+    enableDeletion
+    deletionApprovals {
+      type
+      name
+      expiration
+    }
+    disable {
+      integrations
+    }
+    deleteKeys
+    premiumSupport
+    ecrs {
+      region
+    }
+    partition
+    cleanup {
+      provider
+      ... on AWSAccountCleanupOptionCloudWatch_v1 {
+        regex
+        retention_in_days
+        delete_empty_log_group
+        region
+      }
+    }
+  }
+}
+"""
+
+
+class ConfiguredBaseModel(BaseModel):
+    class Config:
+        smart_union=True
+        extra=Extra.forbid
+
+
+class OwnerV1(ConfiguredBaseModel):
+    name: str = Field(..., alias="name")
+    email: str = Field(..., alias="email")
+
+
+class VaultSecretV1(ConfiguredBaseModel):
+    path: str = Field(..., alias="path")
+    field: str = Field(..., alias="field")
+    version: Optional[int] = Field(..., alias="version")
+    q_format: Optional[str] = Field(..., alias="format")
+
+
+class DeletionApprovalV1(ConfiguredBaseModel):
+    q_type: str = Field(..., alias="type")
+    name: str = Field(..., alias="name")
+    expiration: str = Field(..., alias="expiration")
+
+
+class DisableClusterAutomationsV1(ConfiguredBaseModel):
+    integrations: Optional[list[str]] = Field(..., alias="integrations")
+
+
+class AWSECRV1(ConfiguredBaseModel):
+    region: str = Field(..., alias="region")
+
+
+class AWSAccountCleanupOptionV1(ConfiguredBaseModel):
+    provider: str = Field(..., alias="provider")
+
+
+class AWSAccountCleanupOptionCloudWatchV1(AWSAccountCleanupOptionV1):
+    regex: str = Field(..., alias="regex")
+    retention_in_days: int = Field(..., alias="retention_in_days")
+    delete_empty_log_group: Optional[bool] = Field(..., alias="delete_empty_log_group")
+    region: Optional[str] = Field(..., alias="region")
+
+
+class AWSAccountV1(ConfiguredBaseModel):
+    path: str = Field(..., alias="path")
+    name: str = Field(..., alias="name")
+    uid: str = Field(..., alias="uid")
+    terraform_username: Optional[str] = Field(..., alias="terraformUsername")
+    console_url: str = Field(..., alias="consoleUrl")
+    resources_default_region: str = Field(..., alias="resourcesDefaultRegion")
+    supported_deployment_regions: Optional[list[str]] = Field(..., alias="supportedDeploymentRegions")
+    provider_version: str = Field(..., alias="providerVersion")
+    account_owners: list[OwnerV1] = Field(..., alias="accountOwners")
+    automation_token: VaultSecretV1 = Field(..., alias="automationToken")
+    garbage_collection: Optional[bool] = Field(..., alias="garbageCollection")
+    enable_deletion: Optional[bool] = Field(..., alias="enableDeletion")
+    deletion_approvals: Optional[list[DeletionApprovalV1]] = Field(..., alias="deletionApprovals")
+    disable: Optional[DisableClusterAutomationsV1] = Field(..., alias="disable")
+    delete_keys: Optional[list[str]] = Field(..., alias="deleteKeys")
+    premium_support: bool = Field(..., alias="premiumSupport")
+    ecrs: Optional[list[AWSECRV1]] = Field(..., alias="ecrs")
+    partition: Optional[str] = Field(..., alias="partition")
+    cleanup: Optional[list[Union[AWSAccountCleanupOptionCloudWatchV1, AWSAccountCleanupOptionV1]]] = Field(..., alias="cleanup")
+
+
+class AWSAccountsCloudwatchLogRetentionCleanupQueryData(ConfiguredBaseModel):
+    accounts: Optional[list[AWSAccountV1]] = Field(..., alias="accounts")
+
+
+def query(query_func: Callable, **kwargs: Any) -> AWSAccountsCloudwatchLogRetentionCleanupQueryData:
+    """
+    This is a convenience function which queries and parses the data into
+    concrete types. It should be compatible with most GQL clients.
+    You do not have to use it to consume the generated data classes.
+    Alternatively, you can also mime and alternate the behavior
+    of this function in the caller.
+
+    Parameters:
+        query_func (Callable): Function which queries your GQL Server
+        kwargs: optional arguments that will be passed to the query function
+
+    Returns:
+        AWSAccountsCloudwatchLogRetentionCleanupQueryData: queried data parsed into generated classes
+    """
+    raw_data: dict[Any, Any] = query_func(DEFINITION, **kwargs)
+    return AWSAccountsCloudwatchLogRetentionCleanupQueryData(**raw_data)

qontract_reconcile-0.10.2.dev160/reconcile/typed_queries/aws_cloudwatch_log_retention/aws_accounts.py

@@ -0,0 +1,12 @@
+from reconcile.gql_definitions.aws_cloudwatch_log_retention.aws_accounts import (
+    AWSAccountV1,
+    query,
+)
+from reconcile.utils.gql import GqlApi
+
+
+def get_aws_accounts(
+    gql_api: GqlApi,
+) -> list[AWSAccountV1]:
+    data = query(query_func=gql_api.query)
+    return data.accounts or []