From 68458e0a1a77b23446966a7d621308d0ddc16314 Mon Sep 17 00:00:00 2001
From: Sergey <iceman9831@gmail.com>
Date: Tue, 23 Feb 2021 21:29:00 +0300
Subject: [PATCH] =?UTF-8?q?=D0=9E=D0=B3=D1=80=D0=B0=D0=BD=D0=B8=D1=87?=
 =?UTF-8?q?=D0=B5=D0=BD=D0=B8=D0=B5=20=D0=BF=D0=BE=20Telegram=20user=5Fid?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 config.py   | 2 ++
 telegram.py | 4 ++++
 2 files changed, 6 insertions(+)

diff --git a/config.py b/config.py
index 4745060..35187b0 100644
--- a/config.py
+++ b/config.py
@@ -1,5 +1,7 @@
 import os
 
+ALLOWED_USER_IDS = os.environ.get('ALLOWED_USER_IDS', '')
+
 DATABASE_USER = os.environ.get('POSTGRES_USER', 'postgres')
 DATABASE_PASSWORD = os.environ.get('POSTGRES_PASSWORD', 'postgres')
 DATABASE_HOST = os.environ.get('DATABASE_HOST', 'db')
diff --git a/telegram.py b/telegram.py
index 05baedf..e35b145 100644
--- a/telegram.py
+++ b/telegram.py
@@ -560,6 +560,10 @@ async def search_callback(call: types.CallbackQuery):
 
 @dp.message_handler(commands=['start'])
 async def send_welcome(msg: types.Message):
+    if ALLOWED_USER_IDS:
+        if str(msg.from_user.id) not in ALLOWED_USER_IDS.replace(' ','').split(','):
+            await msg.reply('⛔️ Бот недоступен для Вашего аккаунта.')
+            return
     user, created = await update_user_info(msg.from_user)
     tgchat, tgchat_created = await update_chat_info(msg.chat)
     if not tgchat: