ruff format

data-dot-all · Sep 10, 2024 · 3d4c86a · 3d4c86a
1 parent e76479e
commit 3d4c86a
Show file tree

Hide file tree

Showing 6 changed files with 23 additions and 12 deletions.
diff --git a/deploy/stacks/albfront_stack.py b/deploy/stacks/albfront_stack.py
@@ -323,7 +323,7 @@ def create_log_group(self, envname, resource_prefix, log_group_name):
             f'ECSLogGroup{log_group_name}{envname}',
             log_group_name=f'/{resource_prefix}/{envname}/ecs/{log_group_name}',
             removal_policy=RemovalPolicy.DESTROY,
-            retention=getattr(logs.RetentionDays,  self.log_retention_duration)
+            retention=getattr(logs.RetentionDays, self.log_retention_duration),
         )
         return log_group
 

diff --git a/deploy/stacks/container.py b/deploy/stacks/container.py
@@ -103,7 +103,9 @@ def __init__(
                         ecs.CfnTaskDefinition.KeyValuePairProperty(name='envname', value=envname),
                         ecs.CfnTaskDefinition.KeyValuePairProperty(name='LOGLEVEL', value='DEBUG'),
                         ecs.CfnTaskDefinition.KeyValuePairProperty(name='config_location', value='/config.json'),
-                        ecs.CfnTaskDefinition.KeyValuePairProperty(name='LOG_RETENTION', value=self.log_retention_duration),
+                        ecs.CfnTaskDefinition.KeyValuePairProperty(
+                            name='LOG_RETENTION', value=self.log_retention_duration
+                        ),
                     ],
                     essential=True,
                     log_configuration=ecs.CfnTaskDefinition.LogConfigurationProperty(
@@ -717,7 +719,7 @@ def create_log_group(self, envname, resource_prefix, log_group_name):
             f'ECSLogGroup{log_group_name}{envname}',
             log_group_name=f'/{resource_prefix}/{envname}/ecs/{log_group_name}',
             removal_policy=RemovalPolicy.DESTROY,
-            retention=getattr(logs.RetentionDays,  self.log_retention_duration)
+            retention=getattr(logs.RetentionDays, self.log_retention_duration),
         )
         return log_group
 

diff --git a/deploy/stacks/lambda_api.py b/deploy/stacks/lambda_api.py
@@ -108,7 +108,10 @@ def __init__(
             'ElasticSearchProxyHandler',
             function_name=f'{resource_prefix}-{envname}-esproxy',
             log_group=logs.LogGroup(
-                self, 'esproxyloggroup', log_group_name=f'/aws/lambda/{resource_prefix}-{envname}-backend-esproxy', retention=getattr(logs.RetentionDays,  self.log_retention_duration)
+                self,
+                'esproxyloggroup',
+                log_group_name=f'/aws/lambda/{resource_prefix}-{envname}-backend-esproxy',
+                retention=getattr(logs.RetentionDays, self.log_retention_duration),
             ),
             description='dataall es search function',
             role=self.create_function_role(envname, resource_prefix, 'esproxy', pivot_role_name, vpc),
@@ -145,7 +148,10 @@ def __init__(
             'LambdaGraphQL',
             function_name=f'{resource_prefix}-{envname}-graphql',
             log_group=logs.LogGroup(
-                self, 'graphqlloggroup', log_group_name=f'/aws/lambda/{resource_prefix}-{envname}-backend-graphql', retention=getattr(logs.RetentionDays,  self.log_retention_duration)
+                self,
+                'graphqlloggroup',
+                log_group_name=f'/aws/lambda/{resource_prefix}-{envname}-backend-graphql',
+                retention=getattr(logs.RetentionDays, self.log_retention_duration),
             ),
             description='dataall graphql function',
             role=self.create_function_role(envname, resource_prefix, 'graphql', pivot_role_name, vpc),
@@ -176,7 +182,10 @@ def __init__(
             'AWSWorker',
             function_name=f'{resource_prefix}-{envname}-awsworker',
             log_group=logs.LogGroup(
-                self, 'awsworkerloggroup', log_group_name=f'/aws/lambda/{resource_prefix}-{envname}-backend-awsworker', retention=getattr(logs.RetentionDays,  self.log_retention_duration)
+                self,
+                'awsworkerloggroup',
+                log_group_name=f'/aws/lambda/{resource_prefix}-{envname}-backend-awsworker',
+                retention=getattr(logs.RetentionDays, self.log_retention_duration),
             ),
             description='dataall aws worker for aws asynchronous tasks function',
             role=self.create_function_role(envname, resource_prefix, 'awsworker', pivot_role_name, vpc),
@@ -248,7 +257,7 @@ def __init__(
                     self,
                     'customauthorizerloggroup',
                     log_group_name=f'/aws/lambda/{resource_prefix}-{envname}-custom-authorizer',
-                    retention=getattr(logs.RetentionDays,  self.log_retention_duration)
+                    retention=getattr(logs.RetentionDays, self.log_retention_duration),
                 ),
                 handler='custom_authorizer_lambda.lambda_handler',
                 code=_lambda.Code.from_asset(
@@ -766,7 +775,7 @@ def set_up_graphql_api_gateway(
             f'{resource_prefix}/{envname}/apigateway',
             log_group_name=f'{resource_prefix}/{envname}/apigateway',
             removal_policy=RemovalPolicy.DESTROY,
-            retention=getattr(logs.RetentionDays,  self.log_retention_duration),
+            retention=getattr(logs.RetentionDays, self.log_retention_duration),
         )
 
         iam_policy = iam.PolicyDocument(

diff --git a/deploy/stacks/opensearch.py b/deploy/stacks/opensearch.py
@@ -68,7 +68,7 @@ def __init__(
             id='EsAppLogGroup',
             log_group_name=f'/{resource_prefix}/{envname}/opensearch',
             removal_policy=RemovalPolicy.DESTROY,
-            retention=getattr(logs.RetentionDays,  log_retention_duration),
+            retention=getattr(logs.RetentionDays, log_retention_duration),
         )
 
         self.domain = opensearch.Domain(

diff --git a/deploy/stacks/pipeline.py b/deploy/stacks/pipeline.py
@@ -652,7 +652,7 @@ def set_backend_stage(self, target_env, repository_name):
                 custom_waf_rules=target_env.get('custom_waf_rules', None),
                 with_approval_tests=target_env.get('with_approval_tests', False),
                 allowed_origins=target_env.get('allowed_origins', '*'),
-                log_retention_duration=target_env.get('log_retention_duration', 'TWO_YEARS')
+                log_retention_duration=target_env.get('log_retention_duration', 'TWO_YEARS'),
             )
         )
         return backend_stage

diff --git a/deploy/stacks/vpc.py b/deploy/stacks/vpc.py
@@ -27,7 +27,7 @@ def __init__(
         **kwargs,
     ):
         super().__init__(scope, id, **kwargs)
-        self.log_retention_duration=log_retention_duration
+        self.log_retention_duration = log_retention_duration
 
         if vpc_id:
             self.vpc = ec2.Vpc.from_lookup(self, 'vpc', vpc_id=vpc_id)
@@ -181,7 +181,7 @@ def create_new_vpc(self, cidr, envname, resource_prefix, restricted_nacl):
             f'{resource_prefix}/{envname}/flowlogs',
             log_group_name=f'{resource_prefix}/{envname}/flowlogs',
             removal_policy=RemovalPolicy.DESTROY,
-            retention=getattr(logs.RetentionDays,  self.log_retention_duration)
+            retention=getattr(logs.RetentionDays, self.log_retention_duration),
         )
         iam_policy = iam.PolicyDocument(
             assign_sids=True,