python/aiogram
1
0
Fork 0
mirror of https://github.com/aiogram/aiogram.git synced 2026-04-08 16:37:47 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Fix unexpected behavior of sequences in StateFilter (#791)

Browse source 
* Fix sequence check behavior in StateFilter

* Add sequence cases to StateFilter tests

* Add the changelog
This commit is contained in:
Daniil Kovalenko 2021-12-29 08:39:28 +07:00 • committed by GitHub
parent 7c14a6b16b
commit 6e39f9fada
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
3 changed files with 9 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

9
aiogram/dispatcher/filters/state.py
View file

@ -37,11 +37,12 @@ class StateFilter(BaseFilter):
allowed_states = cast(Sequence[StateType], self.state)
for allowed_state in allowed_states:
if isinstance(allowed_state, str) or allowed_state is None:
if allowed_state == "*":
if allowed_state == "*" or raw_state == allowed_state:
return True
return raw_state == allowed_state
elif isinstance(allowed_state, (State, StatesGroup)):
return allowed_state(event=obj, raw_state=raw_state)
if allowed_state(event=obj, raw_state=raw_state):
return True
elif isclass(allowed_state) and issubclass(allowed_state, StatesGroup):
return allowed_state()(event=obj, raw_state=raw_state)
if allowed_state()(event=obj, raw_state=raw_state):
return True
return False