<reponame>simonasvaitkus/B.LambdaLayerCommon<filename>b_lambda_layer_common/util/pynamodb_encoder.py
from typing import Any

from pynamodb.attributes import MapAttribute

from b_lambda_layer_common.util.dynamodb_encoder import DynamoDBEncoder


# TODO move to pynamo_db directory.
class PynamoDBEncoder(DynamoDBEncoder):
    def default(self, o: Any):
        if isinstance(o, MapAttribute):
            return o.as_dict()
        return super(PynamoDBEncoder, self).default(o)
