qontract-reconcile 0.9.1rc298__py3-none-any.whl → 0.10.1.dev1203__py3-none-any.whl

reconcile/gql_definitions/advanced_upgrade_service/aus_clusters.py

@@ -18,12 +18,13 @@ from pydantic import (  # noqa: F401 # pylint: disable=W0611
 )
 
 from reconcile.gql_definitions.fragments.aus_organization import AUSOCMOrganization
-from reconcile.gql_definitions.fragments.upgrade_policy import ClusterUpgradePolicy
+from reconcile.gql_definitions.fragments.upgrade_policy import ClusterUpgradePolicyV1
 
 
 DEFINITION = """
 fragment AUSOCMOrganization on OpenShiftClusterManager_v1 {
   name
+  labels
   environment {
     ... OCMEnvironment
   }
@@ -33,6 +34,9 @@ fragment AUSOCMOrganization on OpenShiftClusterManager_v1 {
   accessTokenClientSecret {
     ... VaultSecret
   }
+  disable {
+    ... DisableAutomations
+  }
   blockedVersions
   addonManagedUpgrades
   addonUpgradeTests {
@@ -49,9 +53,16 @@ fragment AUSOCMOrganization on OpenShiftClusterManager_v1 {
   }
   inheritVersionData {
     name
-    publishVersionData {
+    orgId
+    environment {
       name
     }
+    publishVersionData {
+      ... MinimalOCMOrganization
+    }
+  }
+  publishVersionData {
+    ... MinimalOCMOrganization
   }
   sectors {
     name
@@ -65,24 +76,45 @@ fragment AUSOCMOrganization on OpenShiftClusterManager_v1 {
   upgradePolicyAllowedWorkloads
   upgradePolicyClusters {
     name
+    serverUrl
+    spec {
+      id
+    }
     upgradePolicy {
-      ... ClusterUpgradePolicy
+      ... ClusterUpgradePolicyV1
     }
   }
+  ausClusterHealthChecks {
+    provider
+    enforced
+  }
 }
 
-fragment ClusterUpgradePolicy on ClusterUpgradePolicy_v1 {
+fragment ClusterUpgradePolicyV1 on ClusterUpgradePolicy_v1 {
   workloads
   schedule
+  versionGateApprovals
   conditions {
     mutexes
     soakDays
     sector
+    blockedVersions
   }
 }
 
+fragment DisableAutomations on DisableClusterAutomations_v1 {
+  integrations
+}
+
+fragment MinimalOCMOrganization on OpenShiftClusterManager_v1 {
+  name
+  orgId
+}
+
 fragment OCMEnvironment on OpenShiftClusterManagerEnvironment_v1 {
     name
+    description
+    labels
     url
     accessTokenClientId
     accessTokenUrl
@@ -105,10 +137,13 @@ query AUSClusters($name: String) {
         ... AUSOCMOrganization
     }
     upgradePolicy {
-        ... ClusterUpgradePolicy
+        ... ClusterUpgradePolicyV1
     }
     spec {
         product
+        id
+        external_id
+        version
     }
     disable {
         integrations
@@ -120,12 +155,15 @@ query AUSClusters($name: String) {
 
 class ConfiguredBaseModel(BaseModel):
     class Config:
-        smart_union = True
-        extra = Extra.forbid
+        smart_union=True
+        extra=Extra.forbid
 
 
 class ClusterSpecV1(ConfiguredBaseModel):
     product: str = Field(..., alias="product")
+    q_id: Optional[str] = Field(..., alias="id")
+    external_id: Optional[str] = Field(..., alias="external_id")
+    version: str = Field(..., alias="version")
 
 
 class DisableClusterAutomationsV1(ConfiguredBaseModel):
@@ -135,7 +173,7 @@ class DisableClusterAutomationsV1(ConfiguredBaseModel):
 class ClusterV1(ConfiguredBaseModel):
     name: str = Field(..., alias="name")
     ocm: Optional[AUSOCMOrganization] = Field(..., alias="ocm")
-    upgrade_policy: Optional[ClusterUpgradePolicy] = Field(..., alias="upgradePolicy")
+    upgrade_policy: Optional[ClusterUpgradePolicyV1] = Field(..., alias="upgradePolicy")
     spec: Optional[ClusterSpecV1] = Field(..., alias="spec")
     disable: Optional[DisableClusterAutomationsV1] = Field(..., alias="disable")
 

reconcile/gql_definitions/advanced_upgrade_service/aus_organization.py

@@ -23,6 +23,7 @@ from reconcile.gql_definitions.fragments.aus_organization import AUSOCMOrganizat
 DEFINITION = """
 fragment AUSOCMOrganization on OpenShiftClusterManager_v1 {
   name
+  labels
   environment {
     ... OCMEnvironment
   }
@@ -32,6 +33,9 @@ fragment AUSOCMOrganization on OpenShiftClusterManager_v1 {
   accessTokenClientSecret {
     ... VaultSecret
   }
+  disable {
+    ... DisableAutomations
+  }
   blockedVersions
   addonManagedUpgrades
   addonUpgradeTests {
@@ -48,9 +52,16 @@ fragment AUSOCMOrganization on OpenShiftClusterManager_v1 {
   }
   inheritVersionData {
     name
-    publishVersionData {
+    orgId
+    environment {
       name
     }
+    publishVersionData {
+      ... MinimalOCMOrganization
+    }
+  }
+  publishVersionData {
+    ... MinimalOCMOrganization
   }
   sectors {
     name
@@ -64,24 +75,45 @@ fragment AUSOCMOrganization on OpenShiftClusterManager_v1 {
   upgradePolicyAllowedWorkloads
   upgradePolicyClusters {
     name
+    serverUrl
+    spec {
+      id
+    }
     upgradePolicy {
-      ... ClusterUpgradePolicy
+      ... ClusterUpgradePolicyV1
     }
   }
+  ausClusterHealthChecks {
+    provider
+    enforced
+  }
 }
 
-fragment ClusterUpgradePolicy on ClusterUpgradePolicy_v1 {
+fragment ClusterUpgradePolicyV1 on ClusterUpgradePolicy_v1 {
   workloads
   schedule
+  versionGateApprovals
   conditions {
     mutexes
     soakDays
     sector
+    blockedVersions
   }
 }
 
+fragment DisableAutomations on DisableClusterAutomations_v1 {
+  integrations
+}
+
+fragment MinimalOCMOrganization on OpenShiftClusterManager_v1 {
+  name
+  orgId
+}
+
 fragment OCMEnvironment on OpenShiftClusterManagerEnvironment_v1 {
     name
+    description
+    labels
     url
     accessTokenClientId
     accessTokenUrl
@@ -107,14 +139,12 @@ query AUSOrganizations($name: String) {
 
 class ConfiguredBaseModel(BaseModel):
     class Config:
-        smart_union = True
-        extra = Extra.forbid
+        smart_union=True
+        extra=Extra.forbid
 
 
 class AUSOrganizationsQueryData(ConfiguredBaseModel):
-    organizations: Optional[list[AUSOCMOrganization]] = Field(
-        ..., alias="organizations"
-    )
+    organizations: Optional[list[AUSOCMOrganization]] = Field(..., alias="organizations")
 
 
 def query(query_func: Callable, **kwargs: Any) -> AUSOrganizationsQueryData:

reconcile/gql_definitions/app_interface_metrics_exporter/onboarding_status.py

@@ -0,0 +1,61 @@
+"""
+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 OnboardingStatus {
+  apps: apps_v1
+  {
+    onboardingStatus
+  }
+}
+"""
+
+
+class ConfiguredBaseModel(BaseModel):
+    class Config:
+        smart_union=True
+        extra=Extra.forbid
+
+
+class AppV1(ConfiguredBaseModel):
+    onboarding_status: str = Field(..., alias="onboardingStatus")
+
+
+class OnboardingStatusQueryData(ConfiguredBaseModel):
+    apps: Optional[list[AppV1]] = Field(..., alias="apps")
+
+
+def query(query_func: Callable, **kwargs: Any) -> OnboardingStatusQueryData:
+    """
+    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:
+        OnboardingStatusQueryData: queried data parsed into generated classes
+    """
+    raw_data: dict[Any, Any] = query_func(DEFINITION, **kwargs)
+    return OnboardingStatusQueryData(**raw_data)

reconcile/gql_definitions/aws_account_manager/aws_accounts.py

@@ -0,0 +1,177 @@
+"""
+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,
+)
+
+from reconcile.gql_definitions.fragments.aws_account_managed import AWSAccountManaged
+from reconcile.gql_definitions.fragments.vault_secret import VaultSecret
+
+
+DEFINITION = """
+fragment AWSAccountManaged on AWSAccount_v1 {
+  name
+  uid
+  alias
+  premiumSupport
+  organization {
+    ou
+    tags
+  }
+  quotaLimits {
+    name
+    quotas {
+      serviceCode
+      quotaCode
+      value
+    }
+  }
+  securityContact {
+    name
+    title
+    email
+    phoneNumber
+  }
+}
+
+fragment VaultSecret on VaultSecret_v1 {
+    path
+    field
+    version
+    format
+}
+
+query AWSAccountManagerAccounts {
+  accounts: awsaccounts_v1 {
+    ... AWSAccountManaged
+    resourcesDefaultRegion
+    automationToken {
+      ...VaultSecret
+    }
+    disable {
+      integrations
+    }
+    automationRole {
+      awsAccountManager
+    }
+    # for the requests via "payer account"
+    account_requests {
+      path
+      name
+      description
+      accountOwner {
+        name
+        email
+      }
+      organization {
+        ou
+        tags
+        payerAccount {
+          path
+        }
+      }
+      quotaLimits {
+        path
+      }
+      resourcesDefaultRegion
+      supportedDeploymentRegions
+      uid
+      additionalFeatures
+    }
+    organization_accounts {
+      ... AWSAccountManaged
+    }
+  }
+}
+"""
+
+
+class ConfiguredBaseModel(BaseModel):
+    class Config:
+        smart_union=True
+        extra=Extra.forbid
+
+
+class DisableClusterAutomationsV1(ConfiguredBaseModel):
+    integrations: Optional[list[str]] = Field(..., alias="integrations")
+
+
+class AWSAutomationRoleV1(ConfiguredBaseModel):
+    aws_account_manager: Optional[str] = Field(..., alias="awsAccountManager")
+
+
+class OwnerV1(ConfiguredBaseModel):
+    name: str = Field(..., alias="name")
+    email: str = Field(..., alias="email")
+
+
+class AWSOrganizationV1_AWSAccountV1(ConfiguredBaseModel):
+    path: str = Field(..., alias="path")
+
+
+class AWSOrganizationV1(ConfiguredBaseModel):
+    ou: str = Field(..., alias="ou")
+    tags: Json = Field(..., alias="tags")
+    payer_account: AWSOrganizationV1_AWSAccountV1 = Field(..., alias="payerAccount")
+
+
+class AWSQuotaLimitsV1(ConfiguredBaseModel):
+    path: str = Field(..., alias="path")
+
+
+class AWSAccountRequestV1(ConfiguredBaseModel):
+    path: str = Field(..., alias="path")
+    name: str = Field(..., alias="name")
+    description: str = Field(..., alias="description")
+    account_owner: OwnerV1 = Field(..., alias="accountOwner")
+    organization: AWSOrganizationV1 = Field(..., alias="organization")
+    quota_limits: Optional[list[AWSQuotaLimitsV1]] = Field(..., alias="quotaLimits")
+    resources_default_region: str = Field(..., alias="resourcesDefaultRegion")
+    supported_deployment_regions: Optional[list[str]] = Field(..., alias="supportedDeploymentRegions")
+    uid: Optional[str] = Field(..., alias="uid")
+    additional_features: Optional[Json] = Field(..., alias="additionalFeatures")
+
+
+class AWSAccountV1(AWSAccountManaged):
+    resources_default_region: str = Field(..., alias="resourcesDefaultRegion")
+    automation_token: VaultSecret = Field(..., alias="automationToken")
+    disable: Optional[DisableClusterAutomationsV1] = Field(..., alias="disable")
+    automation_role: Optional[AWSAutomationRoleV1] = Field(..., alias="automationRole")
+    account_requests: Optional[list[AWSAccountRequestV1]] = Field(..., alias="account_requests")
+    organization_accounts: Optional[list[AWSAccountManaged]] = Field(..., alias="organization_accounts")
+
+
+class AWSAccountManagerAccountsQueryData(ConfiguredBaseModel):
+    accounts: Optional[list[AWSAccountV1]] = Field(..., alias="accounts")
+
+
+def query(query_func: Callable, **kwargs: Any) -> AWSAccountManagerAccountsQueryData:
+    """
+    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:
+        AWSAccountManagerAccountsQueryData: queried data parsed into generated classes
+    """
+    raw_data: dict[Any, Any] = query_func(DEFINITION, **kwargs)
+    return AWSAccountManagerAccountsQueryData(**raw_data)

reconcile/gql_definitions/aws_ami_cleanup/aws_accounts.py

@@ -0,0 +1,161 @@
+"""
+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,
+)
+
+from reconcile.gql_definitions.fragments.aws_account_common import AWSAccountCommon
+from reconcile.gql_definitions.fragments.terraform_state import TerraformState
+
+
+DEFINITION = """
+fragment AWSAccountCommon on AWSAccount_v1 {
+  path
+  name
+  description
+  uid
+  terraformUsername
+  consoleUrl
+  resourcesDefaultRegion
+  supportedDeploymentRegions
+  providerVersion
+  accountOwners {
+    name
+    email
+  }
+  automationToken {
+    ... VaultSecret
+  }
+  garbageCollection
+  enableDeletion
+  deletionApprovals {
+    type
+    name
+    expiration
+  }
+  disable {
+    integrations
+  }
+  deleteKeys
+  premiumSupport
+  partition
+}
+
+fragment TerraformState on TerraformStateAWS_v1 {
+  provider
+  bucket
+  region
+  integrations {
+    key
+    integration
+  }
+}
+
+fragment VaultSecret on VaultSecret_v1 {
+    path
+    field
+    version
+    format
+}
+
+query AWSAccountsAmiCleanup {
+  accounts: awsaccounts_v1 {
+    ...AWSAccountCommon
+    terraformState {
+      ...TerraformState
+    }
+    cleanup {
+      provider
+      ... on AWSAccountCleanupOptionAMI_v1 {
+        regex
+        age
+        region
+      }
+    }
+    sharing {
+      provider
+      account {
+        name
+        supportedDeploymentRegions
+        resourcesDefaultRegion
+      }
+      ... on AWSAccountSharingOptionAMI_v1 {
+        region
+      }
+    }
+  }
+}
+"""
+
+
+class ConfiguredBaseModel(BaseModel):
+    class Config:
+        smart_union=True
+        extra=Extra.forbid
+
+
+class AWSAccountCleanupOptionV1(ConfiguredBaseModel):
+    provider: str = Field(..., alias="provider")
+
+
+class AWSAccountCleanupOptionAMIV1(AWSAccountCleanupOptionV1):
+    regex: str = Field(..., alias="regex")
+    age: str = Field(..., alias="age")
+    region: Optional[str] = Field(..., alias="region")
+
+
+class AWSAccountSharingOptionV1_AWSAccountV1(ConfiguredBaseModel):
+    name: str = Field(..., alias="name")
+    supported_deployment_regions: Optional[list[str]] = Field(..., alias="supportedDeploymentRegions")
+    resources_default_region: str = Field(..., alias="resourcesDefaultRegion")
+
+
+class AWSAccountSharingOptionV1(ConfiguredBaseModel):
+    provider: str = Field(..., alias="provider")
+    account: AWSAccountSharingOptionV1_AWSAccountV1 = Field(..., alias="account")
+
+
+class AWSAccountSharingOptionAMIV1(AWSAccountSharingOptionV1):
+    region: Optional[str] = Field(..., alias="region")
+
+
+class AWSAccountV1(AWSAccountCommon):
+    terraform_state: Optional[TerraformState] = Field(..., alias="terraformState")
+    cleanup: Optional[list[Union[AWSAccountCleanupOptionAMIV1, AWSAccountCleanupOptionV1]]] = Field(..., alias="cleanup")
+    sharing: Optional[list[Union[AWSAccountSharingOptionAMIV1, AWSAccountSharingOptionV1]]] = Field(..., alias="sharing")
+
+
+class AWSAccountsAmiCleanupQueryData(ConfiguredBaseModel):
+    accounts: Optional[list[AWSAccountV1]] = Field(..., alias="accounts")
+
+
+def query(query_func: Callable, **kwargs: Any) -> AWSAccountsAmiCleanupQueryData:
+    """
+    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:
+        AWSAccountsAmiCleanupQueryData: queried data parsed into generated classes
+    """
+    raw_data: dict[Any, Any] = query_func(DEFINITION, **kwargs)
+    return AWSAccountsAmiCleanupQueryData(**raw_data)