Fixed State filter

This commit is contained in:
Alex Root Junior 2021-10-25 01:33:10 +03:00
parent 208134430f
commit 626b4a8205

View file

@ -23,7 +23,11 @@ class StateFilter(BaseFilter):
@validator("state")
@no_type_check # issubclass breaks things
def _validate_state(cls, v: Union[StateType, Sequence[StateType]]) -> Sequence[StateType]:
if isinstance(v, (str, State, StatesGroup)) or (isclass(v) and issubclass(v, StatesGroup)):
if (
isinstance(v, (str, State, StatesGroup))
or (isclass(v) and issubclass(v, StatesGroup))
or v is None
):
return [v]
return v