Lambda function storage check (#20)

* add new check for lambda function and layer storage

Author: mgreschl

CHANGELOG.md

@@ -5,6 +5,12 @@ All notable changes to this project will be documented in this file.
 The format is based on [Keep a Changelog](https://keepachangelog.com/en/1.0.0/),
 and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0.html).
 
+## Unreleased
+
+### Added
+
+- new check: lambda_function_storage
+
 ## [1.6.0] - 2021-04-30
 
 ### Added

aws_quota/check/__init__.py

@@ -10,6 +10,7 @@
 from .elasticbeanstalk import *
 from .elb import *
 from .iam import *
+from .lambdas import *
 from .route53 import *
 from .route53resolver import *
 from .s3 import *
@@ -18,9 +19,15 @@
 from .vpc import *
 from .quota_check import QuotaCheck
 
+
 def __all_subclasses(cls):
     return set(cls.__subclasses__()).union(
-        [s for c in cls.__subclasses__() for s in __all_subclasses(c)])
+        [s for c in cls.__subclasses__() for s in __all_subclasses(c)]
+    )
+
 
-ALL_CHECKS = sorted([clazz for clazz in __all_subclasses(QuotaCheck) if clazz != InstanceQuotaCheck], key=lambda clz: clz.key)
+ALL_CHECKS = sorted(
+    [clazz for clazz in __all_subclasses(QuotaCheck) if clazz != InstanceQuotaCheck],
+    key=lambda clz: clz.key,
+)
 ALL_INSTANCE_SCOPED_CHECKS = filter(lambda check: check.scope == QuotaScope.INSTANCE, ALL_CHECKS)

aws_quota/check/lambdas.py

@@ -0,0 +1,18 @@
+from .quota_check import QuotaCheck, QuotaScope
+
+
+class FunctionAndLayerStorageCheck(QuotaCheck):
+    key = "lambda_function_storage"
+    description = "Lambda function and layer storage"
+    scope = QuotaScope.REGION
+    service_code = 'lambda'
+    quota_code = 'L-2ACBD22F'
+
+    @property
+    def current(self):
+        return (
+            self.boto_session.client('lambda').get_account_settings()['AccountUsage'][
+                'TotalCodeSize'
+            ]
+            / 1000000000
+        )