From 0ee9cc02a6a0ef532930e986aff1fe7fb9ea9335 Mon Sep 17 00:00:00 2001 From: Fenicu Date: Fri, 10 Sep 2021 01:39:01 +0300 Subject: [PATCH] Shortened the code --- examples/custom_filter_example.py | 8 +++----- 1 file changed, 3 insertions(+), 5 deletions(-) diff --git a/examples/custom_filter_example.py b/examples/custom_filter_example.py index bf6c5b24..b6e1f744 100644 --- a/examples/custom_filter_example.py +++ b/examples/custom_filter_example.py @@ -30,12 +30,10 @@ class UserMiddleware(LifetimeControllerMiddleware): update: Union[types.Message, types.CallbackQuery], data: dict, ): - if update.from_user.id in fake_db: - data["user_in_db"] = fake_db[update.from_user.id] - - else: + if update.from_user.id not in fake_db: fake_db[update.from_user.id] = {"ban": False} - data["user_in_db"] = fake_db[update.from_user.id] + + data["user_in_db"] = fake_db[update.from_user.id] class BanUserFilter(BoundFilter):