Преобразование поля json в теле в перечисление с помощью flask-restx

У меня есть следующее определение API для flask-restx (хотя оно также должно работать с flask-restplus). Есть ли способ преобразовать поле enum в теле запроса в Enum MyEnum без особых накладных расходов или с использованием DAO?

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

python flask flask-restplus flask-restx
person Teharez    schedule 28.07.2021    source источник

Ответы (1)


arrow_upward
0
arrow_downward

Хорошо, я написал декоратор, который заменит значение перечисления перечислением.

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

person Teharez    schedule 28.07.2021