fix: code inspect issues

Author: heitorlessa

aws_lambda_powertools/utilities/parser/envelopes/base.py

@@ -1,6 +1,6 @@
 import logging
 from abc import ABC, abstractmethod
-from typing import Any, Dict
+from typing import Any, Dict, Union
 
 from pydantic import BaseModel, ValidationError
 
@@ -11,7 +11,7 @@
 
 class BaseEnvelope(ABC):
  @staticmethod
- def _parse(event: Dict[str, Any], schema: BaseModel) -> Any:
+ def _parse(event: Union[Dict[str, Any], str], schema: BaseModel) -> Any:
  if event is None:
  logger.debug("Skipping parsing as event is None")
  return event

aws_lambda_powertools/utilities/parser/envelopes/dynamodb.py

@@ -51,4 +51,5 @@ def parse(self, event: Dict[str, Any], schema: BaseModel) -> List[Dict[Literal["
  "OldImage": self._parse(record.dynamodb.OldImage, schema),
  }
  )
+ # noinspection PyTypeChecker
  return output

aws_lambda_powertools/utilities/parser/envelopes/sqs.py

@@ -44,5 +44,7 @@ def parse(self, event: Dict[str, Any], schema: Union[BaseModel, str]) -> List[Un
  parsed_envelope = SqsSchema(**event)
  except (ValidationError, TypeError) as e:
  raise SchemaValidationError("SQS input doesn't conform with schema") from e
- output = [self._parse(record.body, schema) for record in parsed_envelope.Records]
+ output = []
+ for record in parsed_envelope.Records:
+ output.append(self._parse(record.body, schema))
  return output

aws_lambda_powertools/utilities/parser/parser.py

@@ -19,42 +19,43 @@ def parser(
  schema: BaseModel,
  envelope: Optional[BaseEnvelope] = None,
 ) -> Any:
+ # noinspection SpellCheckingInspection,SpellCheckingInspection
  """Decorator to conduct advanced parsing & validation for lambda handlers events
 
- As Lambda follows (event, context) signature we can remove some of the boilerplate
- and also capture any exception any Lambda function throws as metadata.
- event will be the parsed and passed as a BaseModel pydantic class of the input type "schema"
- to the lambda handler.
- event will be extracted from the envelope in case envelope is not None.
- In case envelope is None, the complete event is parsed to match the schema parameter BaseModel definition.
- In case envelope is not None, first the event is parsed as the envelope's schema definition, and the user
- message is extracted and parsed again as the schema parameter's definition.
+  As Lambda follows (event, context) signature we can remove some of the boilerplate
+  and also capture any exception any Lambda function throws as metadata.
+  event will be the parsed and passed as a BaseModel pydantic class of the input type "schema"
+  to the lambda handler.
+  event will be extracted from the envelope in case envelope is not None.
+  In case envelope is None, the complete event is parsed to match the schema parameter BaseModel definition.
+  In case envelope is not None, first the event is parsed as the envelope's schema definition, and the user
+  message is extracted and parsed again as the schema parameter's definition.
 
- Example
- -------
- **Lambda function using validation decorator**
+  Example
+  -------
+  **Lambda function using validation decorator**
 
- @parser(schema=MyBusiness, envelope=envelopes.EVENTBRIDGE)
- def handler(event: MyBusiness , context: LambdaContext):
- ...
+  @parser(schema=MyBusiness, envelope=envelopes.EVENTBRIDGE)
+  def handler(event: MyBusiness , context: LambdaContext):
+  ...
 
- Parameters
- ----------
- handler: input for lambda_handler_decorator, wraps the handler lambda
- event: AWS event dictionary
- context: AWS lambda context
- schema: pydantic BaseModel class. This is the user data schema that will replace the event.
- event parameter will be parsed and a new schema object will be created from it.
- envelope: what envelope to extract the schema from, can be any AWS service that is currently
- supported in the envelopes module. Can be None.
+  Parameters
+  ----------
+  handler: input for lambda_handler_decorator, wraps the handler lambda
+  event: AWS event dictionary
+  context: AWS lambda context
+  schema: pydantic BaseModel class. This is the user data schema that will replace the event.
+  event parameter will be parsed and a new schema object will be created from it.
+  envelope: what envelope to extract the schema from, can be any AWS service that is currently
+  supported in the envelopes module. Can be None.
 
- Raises
- ------
- SchemaValidationError
- When input event doesn't conform with schema provided
- InvalidSchemaTypeError
- When schema given does not implement BaseModel
- """
+  Raises
+  ------
+  SchemaValidationError
+  When input event doesn't conform with schema provided
+  InvalidSchemaTypeError
+  When schema given does not implement BaseModel
+  """
  if envelope is None:
  try:
  logger.debug("Parsing and validating event schema; no envelope used")