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

Small changes in FSM filter.

Browse source
This commit is contained in:
Alex Root Junior 2018-06-25 17:13:42 +03:00
parent 21ba9288d0
commit 74b56259e6
1 changed files with 6 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -141,8 +141,7 @@ class StateFilter(BaseFilter):
return {'state': Dispatcher.current().current_state()}
try:
if self.ctx_state.get() in self.state:
return {'state': Dispatcher.current().current_state(), 'raw_state': self.state}
state = self.ctx_state.get()
except LookupError:
chat, user = self.get_target(obj)
@ -150,7 +149,11 @@ class StateFilter(BaseFilter):
state = await self.dispatcher.storage.get_state(chat=chat, user=user)
self.ctx_state.set(state)
if state in self.state:
return {'state': Dispatcher.current().current_state(), 'raw_state': self.state}
return {'state': Dispatcher.current().current_state(), 'raw_state': state}
else:
if state in self.state:
return {'state': Dispatcher.current().current_state(), 'raw_state': state}
return False