问题描述
我对flask-restx 有以下api 定义(尽管也应该与flask-restplus 一起使用)。
有没有办法在没有太多开销或使用 DAO 的情况下将请求正文中的 enum
字段转换为 Enum MyEnum
?
class MyEnum(Enum):
FOO = auto()
BAR = auto()
@dataclass(frozen=True)
class MyClass:
enum: MyEnum
api = Namespace('ns')
model = api.model('Model',{
'enum': fields.String(enum=[x.name for x in MyEnum]),})
@api.route('/')
class MyClass(Resource):
@api.expect(Model)
def post(self) -> None:
c = MyClass(**api.payload)
print(type(c.enum)) # <class 'str'> (but I want <enum 'MyEnum'>)
assert(type(c.enum) == MyEnum) # Fails
解决方法
好的,我已经写了一个装饰器,它将用枚举替换枚举值
def decode_enum(api: Namespace,enum_cls: Type[Enum],keys: List[str]):
def replace_item(obj: dict,keys_: List[str],new_value: Type[Enum]):
if not keys_:
return new_value
obj[keys_[0]] = replace_item(obj[keys_[0]],keys_[1:],new_value)
return obj
def decoder(f):
@wraps(f)
def wrapper(*args,**kwds):
value = api.payload
for k in keys:
value = value[k]
enum = enum_cls[value]
api.payload[keys[0]] = replace_item(api.payload[keys[0]],keys[1:],enum)
return f(*args,**kwds)
return wrapper
return decoder
用法是这样的
@decode_enum(api,MyEnum,['enum'])
@api.expect(Model)
def post(self) -> None:
c = MyClass(**api.payload)
print(type(c.enum)) # <enum 'MyEnum'>
replace_item
函数的灵感来自这个 SO 答案:https://stackoverflow.com/a/45335542/6900162