Making mypy happy

Author: leandrodamascena

examples/idempotency/src/customize_persistence_layer.py

@@ -6,7 +6,7 @@
 )
 from aws_lambda_powertools.utilities.typing import LambdaContext
 
-table = os.getenv("IDEMPOTENCY_TABLE")
+table = os.getenv("IDEMPOTENCY_TABLE", "")
 persistence_layer = DynamoDBPersistenceLayer(
     table_name=table,
     key_attr="idempotency_key",

examples/idempotency/src/customize_persistence_layer_redis.py

@@ -8,7 +8,7 @@
 )
 from aws_lambda_powertools.utilities.typing import LambdaContext
 
-redis_endpoint = os.getenv("REDIS_CLUSTER_ENDPOINT")
+redis_endpoint = os.getenv("REDIS_CLUSTER_ENDPOINT", "localhost")
 persistence_layer = RedisCachePersistenceLayer(
     host=redis_endpoint,
     port=6379,

examples/idempotency/src/getting_started_with_idempotency.py

@@ -8,7 +8,7 @@
 )
 from aws_lambda_powertools.utilities.typing import LambdaContext
 
-table = os.getenv("IDEMPOTENCY_TABLE")
+table = os.getenv("IDEMPOTENCY_TABLE", "")
 persistence_layer = DynamoDBPersistenceLayer(table_name=table)
 
 

examples/idempotency/src/getting_started_with_idempotency_redis_client.py

@@ -12,7 +12,7 @@
 )
 from aws_lambda_powertools.utilities.typing import LambdaContext
 
-redis_endpoint = os.getenv("REDIS_CLUSTER_ENDPOINT")
+redis_endpoint = os.getenv("REDIS_CLUSTER_ENDPOINT", "localhost")
 client = Redis(
     host=redis_endpoint,
     port=6379,

examples/idempotency/src/getting_started_with_idempotency_redis_config.py

@@ -10,7 +10,7 @@
 )
 from aws_lambda_powertools.utilities.typing import LambdaContext
 
-redis_endpoint = os.getenv("REDIS_CLUSTER_ENDPOINT")
+redis_endpoint = os.getenv("REDIS_CLUSTER_ENDPOINT", "localhost")
 persistence_layer = RedisCachePersistenceLayer(host=redis_endpoint, port=6379)
 
 

examples/idempotency/src/integrate_idempotency_with_batch_processor.py

@@ -1,4 +1,5 @@
 import os
+from typing import Any, Dict
 
 from aws_lambda_powertools.utilities.batch import BatchProcessor, EventType, process_partial_response
 from aws_lambda_powertools.utilities.data_classes.sqs_event import SQSRecord
@@ -11,7 +12,7 @@
 
 processor = BatchProcessor(event_type=EventType.SQS)
 
-table = os.getenv("IDEMPOTENCY_TABLE")
+table = os.getenv("IDEMPOTENCY_TABLE", "")
 dynamodb = DynamoDBPersistenceLayer(table_name=table)
 config = IdempotencyConfig(event_key_jmespath="messageId")
 
@@ -21,7 +22,7 @@ def record_handler(record: SQSRecord):
     return {"message": record.body}
 
 
-def lambda_handler(event: SQSRecord, context: LambdaContext):
+def lambda_handler(event: Dict[str, Any], context: LambdaContext):
     config.register_lambda_context(context)  # see Lambda timeouts section
 
     return process_partial_response(

examples/idempotency/src/integrate_idempotency_with_validator.py

@@ -8,7 +8,7 @@
 from aws_lambda_powertools.utilities.typing import LambdaContext
 from aws_lambda_powertools.utilities.validation import envelopes, validator
 
-table = os.getenv("IDEMPOTENCY_TABLE")
+table = os.getenv("IDEMPOTENCY_TABLE", "")
 config = IdempotencyConfig(event_key_jmespath='["message", "username"]')
 persistence_layer = DynamoDBPersistenceLayer(table_name=table)
 

examples/idempotency/src/working_with_composite_key.py

@@ -6,7 +6,7 @@
 )
 from aws_lambda_powertools.utilities.typing import LambdaContext
 
-table = os.getenv("IDEMPOTENCY_TABLE")
+table = os.getenv("IDEMPOTENCY_TABLE", "")
 persistence_layer = DynamoDBPersistenceLayer(table_name=table, sort_key_attr="sort_key")
 
 

examples/idempotency/src/working_with_custom_config.py

@@ -12,7 +12,7 @@
 # See: https://botocore.amazonaws.com/v1/documentation/api/latest/reference/config.html#botocore-config
 boto_config = Config()
 
-table = os.getenv("IDEMPOTENCY_TABLE")
+table = os.getenv("IDEMPOTENCY_TABLE", "")
 persistence_layer = DynamoDBPersistenceLayer(table_name=table, boto_config=boto_config)
 
 config = IdempotencyConfig(event_key_jmespath="body")

examples/idempotency/src/working_with_custom_session.py

@@ -12,7 +12,7 @@
 # See: https://boto3.amazonaws.com/v1/documentation/api/latest/reference/core/session.html#module-boto3.session
 boto3_session = boto3.session.Session()
 
-table = os.getenv("IDEMPOTENCY_TABLE")
+table = os.getenv("IDEMPOTENCY_TABLE", "")
 persistence_layer = DynamoDBPersistenceLayer(table_name=table, boto3_session=boto3_session)
 
 config = IdempotencyConfig(event_key_jmespath="body")