Skip to content

Commit aa33793

Browse files
committed
EM-1209 Add support for deserialising Enums
1 parent eab7a01 commit aa33793

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

nhs_aws_helpers/dynamodb_model_store/base_model_store.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -3,6 +3,7 @@
33
import itertools
44
from dataclasses import dataclass, fields, is_dataclass
55
from datetime import date, datetime
6+
from enum import Enum
67
from time import time
78
from typing import (
89
Any,
@@ -174,6 +175,9 @@ def deserialise_value(cls, value_type: type, value: Any, **kwargs) -> Any: # no
174175
if origin_type is set:
175176
return set(value)
176177

178+
if issubclass(value_type, Enum):
179+
return value_type(value)
180+
177181
return value
178182

179183
@classmethod

0 commit comments

Comments
 (0)