qontract-reconcile 0.10.2.dev201__py3-none-any.whl → 0.10.2.dev203__py3-none-any.whl

{qontract_reconcile-0.10.2.dev201.dist-info → qontract_reconcile-0.10.2.dev203.dist-info}/METADATA

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

@@ -597,7 +597,7 @@ reconcile/unleash_feature_toggles/integration.py,sha256=nx7BhtzCsTfPbOp60vI5MkNw
 reconcile/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 reconcile/utils/aggregated_list.py,sha256=_9UeaS1TWbJsGIESvXlzzK-omPI2lMMcCsoqc9LBclc,4022
 reconcile/utils/amtool.py,sha256=Ng5VVNCiPYEK67eDjIwfuuTLs5JsfltCwt6w5UfXbcY,2289
-reconcile/utils/aws_api.py,sha256=7G90_TaUNoUVH3tX3_G3U80m6wj9Hy_DDhiDYq0wyz0,80521
+reconcile/utils/aws_api.py,sha256=Xrt4uukct0u5kkQBvl0azquF8JEbrb_aFtlBNEL2G2E,81488
 reconcile/utils/aws_helper.py,sha256=8PvDR17ntAGX3bBzlTIxDuENl2rkK-RECsNYKm2_DZw,2955
 reconcile/utils/batches.py,sha256=TtEm64a8lWhFuNbUVpFEmXVdU2Q0sTBrP_I0Cjbgh7g,320
 reconcile/utils/binary.py,sha256=lSIevhilMeoGMePPHD7A-pxe45LVpBT0LksecYbM-EA,2477
@@ -809,7 +809,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.dev201.dist-info/METADATA,sha256=3ouikri05CeE-k7JL5J2T38DzVrLe8SHWQkgeta4XRs,24555
-qontract_reconcile-0.10.2.dev201.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-qontract_reconcile-0.10.2.dev201.dist-info/entry_points.txt,sha256=5i9l54La3vQrDLAdwDKQWC0iG4sV9RRfOb1BpvzOWLc,698
-qontract_reconcile-0.10.2.dev201.dist-info/RECORD,,
+qontract_reconcile-0.10.2.dev203.dist-info/METADATA,sha256=jLhZJ7x1qlr15Wil2EAOm2Aq2kAZBKMRm6RgIQhGAoM,24555
+qontract_reconcile-0.10.2.dev203.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+qontract_reconcile-0.10.2.dev203.dist-info/entry_points.txt,sha256=5i9l54La3vQrDLAdwDKQWC0iG4sV9RRfOb1BpvzOWLc,698
+qontract_reconcile-0.10.2.dev203.dist-info/RECORD,,

reconcile/utils/aws_api.py

@@ -42,6 +42,7 @@ if TYPE_CHECKING:
         FilterTypeDef,
         ImageTypeDef,
         LaunchPermissionModificationsTypeDef,
+        NetworkInterfaceTypeDef,
         RouteTableTypeDef,
         SubnetTypeDef,
         TagTypeDef,
@@ -52,6 +53,10 @@ if TYPE_CHECKING:
     )
     from mypy_boto3_ecr import ECRClient
     from mypy_boto3_elb import ElasticLoadBalancingClient
+    from mypy_boto3_elb.type_defs import (
+        LoadBalancerDescriptionTypeDef,
+        TagDescriptionTypeDef,
+    )
     from mypy_boto3_iam import IAMClient, IAMServiceResource
     from mypy_boto3_iam.type_defs import AccessKeyMetadataTypeDef
     from mypy_boto3_logs import CloudWatchLogsClient
@@ -92,7 +97,9 @@ else:
         TransitGatewayTypeDef
     ) = TransitGatewayVpcAttachmentTypeDef = UpgradeTargetTypeDef = (
         VpcEndpointTypeDef
-    ) = VpcTypeDef = object
+    ) = VpcTypeDef = NetworkInterfaceTypeDef = LoadBalancerDescriptionTypeDef = (
+        TagDescriptionTypeDef
+    ) = object
 
 
 class InvalidResourceTypeError(Exception):
@@ -194,9 +201,9 @@ class AWSApi:
         self.get_vpc_route_tables = lru_cache()(self.get_vpc_route_tables)  # type: ignore[method-assign]
         self.get_vpc_subnets = lru_cache()(self.get_vpc_subnets)  # type: ignore[method-assign]
         self._get_vpc_endpoints = lru_cache()(self._get_vpc_endpoints)  # type: ignore[method-assign]
-        self.get_alb_network_interface_ips = lru_cache()(  # type: ignore[method-assign]
-            self.get_alb_network_interface_ips
-        )
+        self.get_network_interfaces = lru_cache()(self.get_network_interfaces)  # type: ignore[method-assign]
+        self.get_load_balancers = lru_cache()(self.get_load_balancers)  # type: ignore[method-assign]
+        self.get_load_balancer_tags = lru_cache()(self.get_load_balancer_tags)  # type: ignore[method-assign]
 
         if init_users:
             self.init_users()
@@ -1584,14 +1591,14 @@ class AWSApi:
             client_type="elb",
         )
         service_tag = {"Key": "kubernetes.io/service-name", "Value": service_name}
-        nis = ec2_client.describe_network_interfaces()["NetworkInterfaces"]
-        lbs = elb_client.describe_load_balancers()["LoadBalancerDescriptions"]
+        nis = self.get_network_interfaces(ec2_client)
+        lbs = self.get_load_balancers(elb_client)
         result_ips = set()
         for lb in lbs:
             lb_name = lb["LoadBalancerName"]
-            tag_descriptions = elb_client.describe_tags(LoadBalancerNames=[lb_name])[
-                "TagDescriptions"
-            ]
+            tag_descriptions = self.get_load_balancer_tags(
+                elb=elb_client, lb_name=lb_name
+            )
             for td in tag_descriptions:
                 tags = td["Tags"]
                 if service_tag not in tags:
@@ -1609,6 +1616,22 @@ class AWSApi:
 
         return result_ips
 
+    @staticmethod
+    def get_network_interfaces(ec2: EC2Client) -> list[NetworkInterfaceTypeDef]:
+        return ec2.describe_network_interfaces()["NetworkInterfaces"]
+
+    @staticmethod
+    def get_load_balancers(
+        elb: ElasticLoadBalancingClient,
+    ) -> list[LoadBalancerDescriptionTypeDef]:
+        return elb.describe_load_balancers()["LoadBalancerDescriptions"]
+
+    @staticmethod
+    def get_load_balancer_tags(
+        elb: ElasticLoadBalancingClient, lb_name: str
+    ) -> list[TagDescriptionTypeDef]:
+        return elb.describe_tags(LoadBalancerNames=[lb_name])["TagDescriptions"]
+
     @staticmethod
     def get_vpc_default_sg_id(vpc_id: str, ec2: EC2Client) -> str | None:
         vpc_security_groups = ec2.describe_security_groups(