fixup! Add a deployment-specific option to list_lambdas()

Author: dsotirho-ucsc

src/azul/lambdas.py

@@ -100,40 +100,41 @@ class Lambdas:
     def _lambda(self):
         return aws.lambda_
 
-    def list_lambdas(self) -> list[Lambda]:
+    def list_lambdas(self,
+                     deployment: str | None = None,
+                     all_versions: bool = False
+                     ) -> 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']
-        ]
+        Return a list of AWS Lambda functions
 
-    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.
+        :param deployment: Limit output to the specified deployment stage. If
+                           `None`, functions from all deployments will be
+                           returned.
+
+        :param all_versions: If `True`, return all versions of each AWS Lambda
+                             function (including '$LATEST'). If `False`, return
+                             only the latest version of each function.
         """
         paginator = self._lambda.get_paginator('list_functions')
         lambda_prefixes = [
-            config.qualified_resource_name(lambda_infix)
+            config.qualified_resource_name(lambda_infix, stage=deployment)
             for lambda_infix in config.lambda_names()
+            if deployment is not None
         ]
         assert all(lambda_prefixes)
+        params = {'FunctionVersion': 'ALL'} if all_versions else {}
         return [
             Lambda.from_response(function)
-            for response in paginator.paginate(FunctionVersion='ALL', MaxItems=500)
+            for response in paginator.paginate(**params)
             for function in response['Functions']
-            if any(function['FunctionName'].startswith(prefix)
-                   for prefix in lambda_prefixes)
+            if deployment is None or any(
+                function['FunctionName'].startswith(prefix)
+                for prefix in lambda_prefixes
+            )
         ]
 
     def manage_lambdas(self, enabled: bool):
-        for lambda_ in self.list_deployment_lambdas():
+        for lambda_ in self.list_lambdas(deployment=config.deployment_stage):
             self.manage_lambda(lambda_.name, enabled)
 
     def manage_lambda(self, lambda_name: str, enable: bool):