diff --git a/telegram_bot/bot.py b/telegram_bot/bot.py index 82dd180..7c3c1dd 100644 --- a/telegram_bot/bot.py +++ b/telegram_bot/bot.py @@ -14,9 +14,10 @@ from telebot import types user_list = [] class User: # Currently saving users in this class to test functionality -> later database - def __init__(self, p_user_id, p_chat_id): + def __init__(self, p_user_id, p_user_name, p_chat_id): self.user_id = int(p_user_id) self.chat_id = int(p_chat_id) + self.user_name = str(p_user_name) bot = telebot.TeleBot(os.getenv('BOT_API_KEY'))