Add a deployment-specific option to list_lambdas()

Author: dsotirho-ucsc

src/azul/lambdas.py

@@ -101,20 +101,40 @@ def _lambda(self):
         return aws.lambda_
 
     def list_lambdas(self) -> list[Lambda]:
+        """
+        Return a list of all the AWS Lambda functions in the AWS account. Only
+        the latest version of each AWS Lambda function will be included in the
+        list.
+        """
         return [
             Lambda.from_response(function)
             for response in self._lambda.get_paginator('list_functions').paginate()
             for function in response['Functions']
         ]
 
-    def manage_lambdas(self, enabled: bool):
+    def list_deployment_lambdas(self) -> list[Lambda]:
+        """
+        Return a list of all AWS Lambda functions in the current deployment. All
+        versions (including '$LATEST') of each AWS Lambda function will be
+        included in the list.
+        """
         paginator = self._lambda.get_paginator('list_functions')
-        lambda_prefixes = [config.qualified_resource_name(lambda_infix) for lambda_infix in config.lambda_names()]
+        lambda_prefixes = [
+            config.qualified_resource_name(lambda_infix)
+            for lambda_infix in config.lambda_names()
+        ]
         assert all(lambda_prefixes)
-        for lambda_page in paginator.paginate(FunctionVersion='ALL', MaxItems=500):
-            for lambda_name in [metadata['FunctionName'] for metadata in lambda_page['Functions']]:
-                if any(lambda_name.startswith(prefix) for prefix in lambda_prefixes):
-                    self.manage_lambda(lambda_name, enabled)
+        return [
+            Lambda.from_response(function)
+            for response in paginator.paginate(FunctionVersion='ALL', MaxItems=500)
+            for function in response['Functions']
+            if any(function['FunctionName'].startswith(prefix)
+                   for prefix in lambda_prefixes)
+        ]
+
+    def manage_lambdas(self, enabled: bool):
+        for lambda_ in self.list_deployment_lambdas():
+            self.manage_lambda(lambda_.name, enabled)
 
     def manage_lambda(self, lambda_name: str, enable: bool):
         lambda_settings = self._lambda.get_function(FunctionName=lambda_name)