From 1850000dc5f0d0b79e9ef4489295703af518ca35 Mon Sep 17 00:00:00 2001 From: root Date: Mon, 25 Apr 2022 11:56:05 +0000 Subject: [PATCH] Add multi admin mode --- bot.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/bot.py b/bot.py index fc665f4..6e861f7 100755 --- a/bot.py +++ b/bot.py @@ -19,7 +19,7 @@ logging.basicConfig( format='%(asctime)s - %(name)s - %(levelname)s - %(message)s') log = logging.getLogger(__name__) token = os.environ.get('TG_TOKEN') -admin = os.environ.get('TG_ADMIN') +admin = os.environ.get('TG_ADMIN').replace('"', '').replace(' ', '').split(',') if not token or not admin: log.error("Env var TG_TOKEN or TG_ADMIN aren't set.") sys.exit(1) @@ -34,7 +34,7 @@ def _help(update, context): def auth(handler): def wrapper(update, context): - if update.message.chat.username != admin: + if update.message.chat.username not in admin: update.message.reply_text( 'You are not allowed to do that.', parse_mode='HTML', disable_web_page_preview=True) return False