From 1349d5ffae74b3d71ea3e7d22d39874d67042486 Mon Sep 17 00:00:00 2001
From: Pavel Kirilin <win10@list.ru>
Date: Mon, 12 Apr 2021 15:11:37 +0400
Subject: [PATCH] Updated danger message decorator.
Signed-off-by: Pavel Kirilin <win10@list.ru>
---
src/actions/basic.py | 3 ++-
src/utils/responses.py | 6 ++----
2 files changed, 4 insertions(+), 5 deletions(-)
diff --git a/src/actions/basic.py b/src/actions/basic.py
index c3889ce..c9fe005 100644
--- a/src/actions/basic.py
+++ b/src/actions/basic.py
@@ -28,7 +28,8 @@ async def ci(event):
@config.telegram_client.on(events.NewMessage(pattern=r"^\.cid$"))
@mark_unread
async def get_chat_id(event):
- await event.respond(str(event.chat.id))
+ chat = await event.get_chat()
+ await event.respond(str(chat.id))
@config.telegram_client.on(events.NewMessage(pattern=r"^\.t$"))
diff --git a/src/utils/responses.py b/src/utils/responses.py
index ae9f31e..6a97f57 100644
--- a/src/utils/responses.py
+++ b/src/utils/responses.py
@@ -78,10 +78,8 @@ def mark_unread(f):
def danger_message(f):
@functools.wraps(f)
async def wrapper(event: events.NewMessage.Event):
-
- if hasattr(event.chat, "id") and str(event.chat.id) in set(
- config.excluded_chats
- ):
+ chat = await event.get_chat()
+ if str(chat.id) in set(config.excluded_chats):
logger.debug("Skipping sending message")
return
await f(event)
--
GitLab