From 4f0a3c607f0a077ebdeeaa82e84b7abc3f4f3c39 Mon Sep 17 00:00:00 2001 From: Alex Root Junior Date: Thu, 28 Jun 2018 19:24:25 +0300 Subject: [PATCH] Allow to use '*' and `None` as state in State object. --- aiogram/dispatcher/filters/state.py | 10 +++++++++- 1 file changed, 9 insertions(+), 1 deletion(-) diff --git a/aiogram/dispatcher/filters/state.py b/aiogram/dispatcher/filters/state.py index 0861a7e0..f6785bbe 100644 --- a/aiogram/dispatcher/filters/state.py +++ b/aiogram/dispatcher/filters/state.py @@ -16,7 +16,11 @@ class State: @property def state(self): - if self._group_name is None and self._group: + if self._state is None: + return None + elif self._state == '*': + return self._state + elif self._group_name is None and self._group: group = self._group.__full_group_name__ elif self._group_name: group = self._group_name @@ -157,3 +161,7 @@ class StatesGroup(metaclass=MetaStatesGroup): await state.set_state(last_step_name) return last_step_name + + +default_state = State() +any_state = State(state='*')