qontract-reconcile 0.10.2.dev130__py3-none-any.whl → 0.10.2.dev132__py3-none-any.whl

{qontract_reconcile-0.10.2.dev130.dist-info → qontract_reconcile-0.10.2.dev132.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: qontract-reconcile
-Version: 0.10.2.dev130
+Version: 0.10.2.dev132
 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.dev130.dist-info → qontract_reconcile-0.10.2.dev132.dist-info}/RECORD

@@ -94,7 +94,7 @@ reconcile/quay_mirror.py,sha256=0KtQFwrvMNtlsPJ9F_-ICaVIjgIUjFxqipvAPcvyg3Q,1533
 reconcile/quay_mirror_org.py,sha256=tXKuF6JtmaNRwu8_g_65U_Vpd6sFBYeXmJA-flVhylE,10764
 reconcile/quay_permissions.py,sha256=9KOutS1w4RFQqkvMSy54VtsKNx56-phzP6yI_rEW-B8,4244
 reconcile/quay_repos.py,sha256=cuEYG0HUe0ut5yvLdEwOF5-CmccpXQHRb_wDazvDrvQ,6895
-reconcile/queries.py,sha256=3nqVO5zv_TsnCK4jsJfF8N4Z7ADfgtUaTD2nv8EzlMs,51506
+reconcile/queries.py,sha256=ICAkBP8atJEZlPCniMJljb2hJAZdvSZdT1-ciLqVGl8,51560
 reconcile/query_validator.py,sha256=MSh5pKLBksws4AqfuvT8nrIGucIbqX-IOzYyPYTLO7k,1491
 reconcile/requests_sender.py,sha256=914iluuF4UVgG3VyxxtnHOu4yf6YKS2fIy6PViSsFTQ,3875
 reconcile/resource_scraper.py,sha256=znXCHrU7YwPfKuxGBiUrV7T1tYtn4vlz9qmZlfy6Flg,2307
@@ -121,13 +121,13 @@ reconcile/terraform_vpc_peerings.py,sha256=0N7uxG6Afe2oNrUyw-apjV-fX2P9TbI5A3xjX
 reconcile/vault_replication.py,sha256=YD70ROZXC1dYnUcS3q1WgnAoIDSdoZVdZsUGc5fanNU,17556
 reconcile/vpc_peerings_validator.py,sha256=aESqrhm1tpkc2iqSL1UV5to_HjNgjRSffD0crb_q49g,7113
 reconcile/aus/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-reconcile/aus/advanced_upgrade_service.py,sha256=NF3UQ02MdUpVPup50EqnTbTn3dRHGoXoSfFLZxOjr8Q,23744
-reconcile/aus/aus_label_source.py,sha256=2Rktyvs7ZvCwsRwmbGUhddycTBTq0jHq8wpBaiDHAMA,4165
-reconcile/aus/base.py,sha256=2h_8GuwEHJBLmiwTwaFT9EjiCHcmIEXF8Zo7rZ974kQ,48832
+reconcile/aus/advanced_upgrade_service.py,sha256=lt684trHbKvVDLwwuNVz3Wu_MnytFSbS_7MZTIITh9k,23969
+reconcile/aus/aus_label_source.py,sha256=o0S2f0qwcII_8nzhHZhRQ83gEZ1DrSXyO4xzSwLebuU,4382
+reconcile/aus/base.py,sha256=K-CLPUDpaFSfFsf_UgcvaUPCgqTLTGkx_aoUH0VWhv8,50431
 reconcile/aus/cluster_version_data.py,sha256=VZWbUEIbrDKO-sroMpQtiWCTqDraTMd8tssKV0HyTQ0,7140
 reconcile/aus/healthchecks.py,sha256=jR9c-syh9impnkV0fd6XW3Bnk7iRN5zv8oCRYM-yIRY,2700
 reconcile/aus/metrics.py,sha256=nKT4m2zGT-QOMR0c-z-npVNKWsNMubzdffpU_f9n4II,3927
-reconcile/aus/models.py,sha256=R72G-2_mx4KHvdIFOzOd2neubu-BBlw5i7hqKVk7W3s,8149
+reconcile/aus/models.py,sha256=qLjWLDJe5PGXPPtJ5PI01IVEYaSGweu9dkAgf0ZM2hk,8297
 reconcile/aus/node_pool_spec.py,sha256=FkMggklG-4BgQwud2Swp2m3AAAKzZmeaXgohl9uwxZ8,1138
 reconcile/aus/ocm_addons_upgrade_scheduler_org.py,sha256=SHbFEEq9ETtkkuzBBaNuwSDze8fj6DWHeA_XNgOyTbo,10308
 reconcile/aus/ocm_upgrade_scheduler.py,sha256=2uPn13y3QGCHLoKwCc1Z7q9wQsoQf_F1HATMYUbl53s,3695
@@ -227,14 +227,14 @@ reconcile/glitchtip_project_alerts/integration.py,sha256=BgMx-NyV9mTuv7Sotb2OioC
 reconcile/glitchtip_project_dsn/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 reconcile/glitchtip_project_dsn/integration.py,sha256=2iugub-kHYkHNK33n0v9_TeWonuxCPah_VkoTPvaajE,8077
 reconcile/gql_definitions/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-reconcile/gql_definitions/introspection.json,sha256=0begChFQIAwAkSL-tmYKbh3fJmtvgxoyHm0FRP3HPGY,2283286
+reconcile/gql_definitions/introspection.json,sha256=mzkjTtBMjRDoC00njCLGylyb85atOvRRPzQo4GhKti8,2283808
 reconcile/gql_definitions/acs/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 reconcile/gql_definitions/acs/acs_instances.py,sha256=L91WW9LbhJbBSrECqShQpFtjoBOsmNIYLRpMbx1io5o,2181
 reconcile/gql_definitions/acs/acs_policies.py,sha256=bN5i4mks10Z23KJSj7jqp966Osq2dps4d-sPH9gjxEA,7008
 reconcile/gql_definitions/acs/acs_rbac.py,sha256=cZsIlCWliPQdQHgmBsIMx54fJNOtkdRXLzmOKZmJNHk,3009
 reconcile/gql_definitions/advanced_upgrade_service/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-reconcile/gql_definitions/advanced_upgrade_service/aus_clusters.py,sha256=RpOrRYJnC6qgf5AvfB80Knp2Ww9R0viuxYVg_LetglY,4366
-reconcile/gql_definitions/advanced_upgrade_service/aus_organization.py,sha256=zU-WJ9CASV1Ok-1jUro6K426v3ug5YNR1XoXmV7SwQ8,3364
+reconcile/gql_definitions/advanced_upgrade_service/aus_clusters.py,sha256=230uwBoBaaVz686d_WNZT9eAw6BX2VpoQX0mKhf5UcM,4390
+reconcile/gql_definitions/advanced_upgrade_service/aus_organization.py,sha256=9uWc_zpTrl06ysfbrkqZsIGuCoVTZONDyj0V2UkkX7o,3388
 reconcile/gql_definitions/app_interface_metrics_exporter/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 reconcile/gql_definitions/app_interface_metrics_exporter/onboarding_status.py,sha256=uVEEqU6YYmKsNTo6EWlFnoVmqha2rvBDx-wiD64VmG0,1679
 reconcile/gql_definitions/app_sre_tekton_access_revalidation/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -276,7 +276,7 @@ reconcile/gql_definitions/common/app_quay_repos_escalation_policies.py,sha256=ck
 reconcile/gql_definitions/common/apps.py,sha256=lC7uZjyHAvJXhCDPJJ-dQJInBJkUCNZoe129XMa67wo,1868
 reconcile/gql_definitions/common/aws_vpc_requests.py,sha256=2lpyy-QSfHPSTvV_zbnqzqUoFzVNl-CJDmoqd_zVSX4,2366
 reconcile/gql_definitions/common/aws_vpcs.py,sha256=Dss9dQ3xagnz3Ltg1e9mtG2PAmQGBbUzKCmmzvuN28s,1892
-reconcile/gql_definitions/common/clusters.py,sha256=Dr5AsSsTuqjAxkI9fU0fdiaP6u5qkmRpkkCcYDnU584,21868
+reconcile/gql_definitions/common/clusters.py,sha256=j646LA-vlHbA131OrCF13g8rPPrBPWL3PLrvqWLZs58,21979
 reconcile/gql_definitions/common/clusters_minimal.py,sha256=JYrJV_aStmryiiGKyiXhj47qpF_8KilCqy-d9CofBCo,4635
 reconcile/gql_definitions/common/clusters_with_dms.py,sha256=GJ53P8tgMLh1NfVkaV9_AmaqF9pNUqJZcDkcKzKzUy0,2242
 reconcile/gql_definitions/common/clusters_with_peering.py,sha256=B1Hi3u6rZZsl4bDDPMLIcSRI5lNFuh29lPVVTOrRRpQ,11929
@@ -321,7 +321,7 @@ reconcile/gql_definitions/external_resources/fragments/external_resources_module
 reconcile/gql_definitions/fleet_labeler/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 reconcile/gql_definitions/fleet_labeler/fleet_labels.py,sha256=XPk1YFmiCtGlwrldxdABwYaT-wgZdTjpDFHmUnq6xcM,4349
 reconcile/gql_definitions/fragments/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-reconcile/gql_definitions/fragments/aus_organization.py,sha256=uBKbTuBa3CZmTXR5HOcGhRcu2U9kM93KbYmoWTxcpB0,4767
+reconcile/gql_definitions/fragments/aus_organization.py,sha256=21L_d35HCGbn-PaZSzIGzIVm3Wt-IPTdBzftxK5Y0h4,4850
 reconcile/gql_definitions/fragments/aws_account_common.py,sha256=3-7ZAP6GSff7Z2Syz2VQCLY4IySqBOSVmceaRiVNQpw,2385
 reconcile/gql_definitions/fragments/aws_account_managed.py,sha256=V_9pH0wVza2sPyq8ckkabNMYIRZt6VW4Nixw_jAxaMc,1892
 reconcile/gql_definitions/fragments/aws_account_sso.py,sha256=ITR3PLz4Iq1SiWAoYGWPDuHJnAmTyZ0QQqs2Zsi8pxA,979
@@ -769,7 +769,7 @@ tools/sd_app_sre_alert_report.py,sha256=jQpJdXVID68bSNtJNOGDh0-ei1CfEUS4Itr4MAaB
 tools/template_validation.py,sha256=qpKYaTgk0GOPGa2Ct5_5sKdwIHtCAKIBGzsMPuJU5fw,3371
 tools/cli_commands/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 tools/cli_commands/container_images_report.py,sha256=SXh6sZ1dXXzd-2R5eeBufCtW3J4-Y55--0MZLdo4lr8,5409
-tools/cli_commands/erv2.py,sha256=qojYCTope8ao-NBQVITs5Kd4mJBjRLNZwfvrV-Eov9w,23770
+tools/cli_commands/erv2.py,sha256=5EekfLBoA0spsdUHrPH47_x7bxxe_q7dTWW7nDw3UxM,23833
 tools/cli_commands/gpg_encrypt.py,sha256=JWwds_Qg7KhSJMIGUh8TfI5-Jf17iUtmaEi4kWJxfVE,4907
 tools/cli_commands/systems_and_tools.py,sha256=EMHOF1AtUDaoSk0bbjl6oUKYAz4rTZjIBaF-6E6GspM,16816
 tools/cli_commands/cost_report/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -791,7 +791,7 @@ tools/saas_promotion_state/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJ
 tools/saas_promotion_state/saas_promotion_state.py,sha256=UfwwRLS5Ya4_Nh1w5n1dvoYtchQvYE9yj1VANt2IKqI,3925
 tools/sre_checkpoints/__init__.py,sha256=CDaDaywJnmRCLyl_NCcvxi-Zc0hTi_3OdwKiFOyS39I,145
 tools/sre_checkpoints/util.py,sha256=zEDbGr18ZeHNQwW8pUsr2JRjuXIPz--WAGJxZo9sv_Y,894
-qontract_reconcile-0.10.2.dev130.dist-info/METADATA,sha256=2CCwcz__a7QKk0BgiDB2-tPBCrscTGVp8QHheIuF5Fo,24566
-qontract_reconcile-0.10.2.dev130.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-qontract_reconcile-0.10.2.dev130.dist-info/entry_points.txt,sha256=5i9l54La3vQrDLAdwDKQWC0iG4sV9RRfOb1BpvzOWLc,698
-qontract_reconcile-0.10.2.dev130.dist-info/RECORD,,
+qontract_reconcile-0.10.2.dev132.dist-info/METADATA,sha256=NGyCyUtUnoFRVQrSNesrYtCpD5s1naAW4KIK676EcK0,24566
+qontract_reconcile-0.10.2.dev132.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+qontract_reconcile-0.10.2.dev132.dist-info/entry_points.txt,sha256=5i9l54La3vQrDLAdwDKQWC0iG4sV9RRfOb1BpvzOWLc,698
+qontract_reconcile-0.10.2.dev132.dist-info/RECORD,,

reconcile/aus/advanced_upgrade_service.py

@@ -289,6 +289,10 @@ class OrganizationLabelSet(BaseModel):
 
     blocked_versions: CSV | None = Field(alias=aus_label_key("blocked-versions"))
 
+    sector_max_parallel_upgrades: dict[str, str] = labelset_groupfield(
+        group_prefix=aus_label_key("sector-max-parallel-upgrades.")
+    )
+
     sector_deps: dict[str, CSV] = labelset_groupfield(
         group_prefix=aus_label_key("sector-deps.")
     )
@@ -311,6 +315,7 @@ class OrganizationLabelSet(BaseModel):
         return [
             OpenShiftClusterManagerSectorV1(
                 name=s,
+                maxParallelUpgrades=self.sector_max_parallel_upgrades.get(s),
                 dependencies=[
                     OpenShiftClusterManagerSectorDependenciesV1(name=d, ocm=None)
                     for d in self.sector_deps.get(s, [])

reconcile/aus/aus_label_source.py

@@ -88,8 +88,12 @@ class AUSOrganizationLabelSource(LabelSource):
             )
         # sector dependencies
         for sector in organization.sectors or []:
-            if sector.dependencies:
-                labels[aus_label_key(f"sectors.{sector.name}")] = ",".join(
+            if sector.max_parallel_upgrades:
+                labels[aus_label_key(f"sector-max-parallel-upgrades.{sector.name}")] = (
+                    sector.max_parallel_upgrades
+                )
+            if sector.dependencies is not None:
+                labels[aus_label_key(f"sector-deps.{sector.name}")] = ",".join(
                     sorted([dep.name for dep in sector.dependencies])
                 )
         # version-data sharing

reconcile/aus/base.py

@@ -5,6 +5,7 @@ from abc import (
     ABC,
     abstractmethod,
 )
+from collections import defaultdict
 from collections.abc import Callable, Sequence
 from datetime import (
     datetime,
@@ -982,6 +983,47 @@ def verify_lock_should_skip(
     return False
 
 
+def verify_max_upgrades_should_skip(
+    desired: ClusterUpgradeSpec,
+    sector_upgrades: dict[str, set[str]],
+    sector: Sector | None,
+) -> bool:
+    if sector is None:
+        return False
+
+    current_upgrades = sector_upgrades[sector.name]
+    # Allow at least one upgrade
+    if len(current_upgrades) == 0:
+        return False
+
+    # if sector.max_parallel_upgrades is not set, we allow all upgrades
+    if sector.max_parallel_upgrades is None:
+        return False
+
+    sector_cluster_count = len(sector.specs)
+
+    if sector.max_parallel_upgrades.endswith("%"):
+        max_parallel_upgrades_percent = int(sector.max_parallel_upgrades[:-1])
+        max_parallel_upgrades = round(
+            sector_cluster_count * max_parallel_upgrades_percent / 100
+        )
+    else:
+        max_parallel_upgrades = int(sector.max_parallel_upgrades)
+
+    # we allow at least one upgrade
+    if max_parallel_upgrades == 0:
+        max_parallel_upgrades = 1
+
+    if len(current_upgrades) >= max_parallel_upgrades:
+        logging.debug(
+            f"[{desired.org.org_id}/{desired.org.name}/{desired.cluster.name}] skipping cluster: "
+            f"sector '{sector.name}' has reached max parallel upgrades {sector.max_parallel_upgrades}"
+        )
+        return True
+
+    return False
+
+
 def _create_upgrade_policy(
     next_schedule: str, spec: ClusterUpgradeSpec, version: str
 ) -> AbstractUpgradePolicy:
@@ -1042,25 +1084,32 @@ def calculate_diff(
         list: upgrade policies to be applied
     """
 
-    def set_mutex(
-        locked: dict[str, str], cluster_id: str, mutexes: set[str] | None = None
+    locked: dict[str, str] = {}
+    sector_upgrades: dict[str, set[str]] = defaultdict(set)
+
+    def set_upgrading(
+        cluster_id: str, mutexes: set[str] | None, sector_name: str | None
     ) -> None:
         for mutex in mutexes or set():
             locked[mutex] = cluster_id
+        if sector_name:
+            sector_upgrades[sector_name].add(cluster_id)
 
     diffs: list[UpgradePolicyHandler] = []
 
     # all clusters IDs with a current upgradePolicy are considered locked
-    locked: dict[str, str] = {}
     for spec in desired_state.specs:
         if spec.cluster.id in [s.cluster.id for s in current_state]:
-            for mutex in spec.effective_mutexes:
-                locked[mutex] = spec.cluster.id
+            sector_name = spec.upgrade_policy.conditions.sector
+            set_upgrading(spec.cluster.id, spec.effective_mutexes, sector_name)
 
     addon_service = init_addon_service(desired_state.org.environment)
     now = datetime.utcnow()
     gates = get_version_gates(ocm_api)
     for spec in desired_state.specs:
+        sector_name = spec.upgrade_policy.conditions.sector
+        sector = desired_state.sectors[sector_name] if sector_name else None
+
         # Upgrading node pools, only required for Hypershift clusters
         # do this in the same loop, to skip cluster on node pool upgrade
         if spec.cluster.is_rosa_hypershift():
@@ -1070,7 +1119,7 @@ def calculate_diff(
             node_pool_update = _calculate_node_pool_diffs(spec, now)
             if node_pool_update:  # node pool update policy not yet created
                 diffs.append(node_pool_update)
-                set_mutex(locked, spec.cluster.id, spec.effective_mutexes)
+                set_upgrading(spec.cluster.id, spec.effective_mutexes, sector_name)
                 continue
 
         # ignore clusters with an existing upgrade policy
@@ -1089,10 +1138,9 @@ def calculate_diff(
         if verify_lock_should_skip(spec, locked):
             continue
 
-        sector_name = spec.upgrade_policy.conditions.sector
-        sector = None
-        if sector_name:
-            sector = desired_state.sectors[sector_name]
+        if verify_max_upgrades_should_skip(spec, sector_upgrades, sector):
+            continue
+
         version = upgradeable_version(spec, version_data, sector)
         if version:
             if addon_id:
@@ -1139,7 +1187,7 @@ def calculate_diff(
                         policy=_create_upgrade_policy(next_schedule, spec, version),
                     )
                 )
-            set_mutex(locked, spec.cluster.id, spec.effective_mutexes)
+            set_upgrading(spec.cluster.id, spec.effective_mutexes, sector_name)
 
     return diffs
 

reconcile/aus/models.py

@@ -137,7 +137,11 @@ class OrganizationUpgradeSpec(BaseModel):
 
         # extract sectors
         self._sectors = {
-            s.name: Sector(org_id=self.org.org_id, name=s.name)
+            s.name: Sector(
+                org_id=self.org.org_id,
+                name=s.name,
+                max_parallel_upgrades=s.max_parallel_upgrades,
+            )
             for s in self.org.sectors or []
         }
 
@@ -223,6 +227,7 @@ class SectorConfigError(Exception):
 
 class Sector(BaseModel):
     name: str
+    max_parallel_upgrades: str | None
     dependencies: list[Sector] = Field(default_factory=list)
     _specs: dict[str, ClusterUpgradeSpec] = PrivateAttr(default_factory=dict)
 

reconcile/gql_definitions/advanced_upgrade_service/aus_clusters.py

@@ -66,6 +66,7 @@ fragment AUSOCMOrganization on OpenShiftClusterManager_v1 {
   }
   sectors {
     name
+    maxParallelUpgrades
     dependencies {
       name
       ocm {

reconcile/gql_definitions/advanced_upgrade_service/aus_organization.py

@@ -65,6 +65,7 @@ fragment AUSOCMOrganization on OpenShiftClusterManager_v1 {
   }
   sectors {
     name
+    maxParallelUpgrades
     dependencies {
       name
       ocm {

reconcile/gql_definitions/common/clusters.py

@@ -148,6 +148,7 @@ query Clusters($name: String) {
       }
       sectors {
         name
+        maxParallelUpgrades
         dependencies {
           name
           ocm {
@@ -388,6 +389,7 @@ class OpenShiftClusterManagerSectorDependenciesV1(ConfiguredBaseModel):
 
 class OpenShiftClusterManagerSectorV1(ConfiguredBaseModel):
     name: str = Field(..., alias="name")
+    max_parallel_upgrades: Optional[str] = Field(..., alias="maxParallelUpgrades")
     dependencies: Optional[list[OpenShiftClusterManagerSectorDependenciesV1]] = Field(..., alias="dependencies")
 
 

reconcile/gql_definitions/fragments/aus_organization.py

@@ -67,6 +67,7 @@ class OpenShiftClusterManagerSectorDependenciesV1(ConfiguredBaseModel):
 
 class OpenShiftClusterManagerSectorV1(ConfiguredBaseModel):
     name: str = Field(..., alias="name")
+    max_parallel_upgrades: Optional[str] = Field(..., alias="maxParallelUpgrades")
     dependencies: Optional[list[OpenShiftClusterManagerSectorDependenciesV1]] = Field(..., alias="dependencies")
 
 

reconcile/gql_definitions/introspection.json

@@ -8079,6 +8079,18 @@
                             "isDeprecated": false,
                             "deprecationReason": null
                         },
+                        {
+                            "name": "maxParallelUpgrades",
+                            "description": null,
+                            "args": [],
+                            "type": {
+                                "kind": "SCALAR",
+                                "name": "String",
+                                "ofType": null
+                            },
+                            "isDeprecated": false,
+                            "deprecationReason": null
+                        },
                         {
                             "name": "dependencies",
                             "description": null,

reconcile/queries.py

@@ -739,6 +739,7 @@ CLUSTERS_QUERY = """
       }
       sectors {
         name
+        maxParallelUpgrades
         dependencies {
           name
         }
@@ -1348,6 +1349,7 @@ OCM_QUERY = """
     }
     sectors {
       name
+      maxParallelUpgrades
       dependencies {
         name
         ocm {

tools/cli_commands/erv2.py

@@ -222,7 +222,7 @@ class Erv2Cli:
             raise
 
     def enter_shell(self, credentials: Path) -> None:
-        """Run the CDKTF container and enter the shell."""
+        """Run the ERv2 container and enter the shell."""
         input_file = self.temp / "input.json"
         input_file.write_text(self.input_data)
 
@@ -242,6 +242,8 @@ class Erv2Cli:
                     f"{credentials!s}:/credentials:Z",
                     "-e",
                     "AWS_SHARED_CREDENTIALS_FILE=/credentials",
+                    "-e",
+                    "WORK=/tmp/work",
                     "--entrypoint",
                     "/bin/bash",
                     self.image,