diff --git a/debugbot.py b/debugbot.py new file mode 100644 index 0000000..a17ab9b --- /dev/null +++ b/debugbot.py @@ -0,0 +1,92 @@ +# 5108535940:AAF5FpPHNV96WxGCDt8aMrGGKke1VILYib4 + +import os + +import telebot +import time +import sys +import logging +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_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('5108535940:AAF5FpPHNV96WxGCDt8aMrGGKke1VILYib4') + +@bot.message_handler(commands=['start']) # /start -> saving as new user and sending welcome +def send_start(message): + new_user = User(int(message.from_user.id), message.from_user.first_name, int(message.chat.id)) + existing_already = False + for known_user in user_list: + if known_user.user_id == new_user.user_id: + existing_already = True + if existing_already == False: + user_list.append(new_user) + + bot.reply_to(message, "Welcome to this share bot project. Type /help to get information on what this bot can do") + + +@bot.message_handler(commands=['help']) # /help -> sending all functions +def send_welcome(message): + bot.reply_to(message, "/id or /auth for authentication. /update to get updates on your shares. /users to see all users. For further details see aktienbot.flokaiser.com") + +@bot.message_handler(commands=['users']) +def send_all_users(message): + print('Debug: users command') + user_id = int(message.from_user.id) + answer = 'Current number of users: ' + str(len(user_list)) + bot.send_message(chat_id = user_id, text=answer) + for known_user in user_list: + answer = str(known_user.user_id) + ' : ' + known_user.user_name + bot.send_message(chat_id=user_id, text=answer) + + +@bot.message_handler(commands=['id', 'auth']) # /auth or /id -> Authentication with user_id over web tool +def send_id(message): + answer = 'Your ID/Authentication Code is: [' + str(message.from_user.id) + ']. Enter this code in the settings on aktienbot.flokaiser.com to get updates on your shares.' + bot.reply_to(message, answer) + + +@bot.message_handler(commands=['update']) # /update -> send static update via user_id to this user, later fetch from database +def send_update(message): + user_id = int(message.from_user.id) + #Get Information for user with this id + bot.send_message(chat_id=user_id, text='This is your update') + + +@bot.message_handler(func=lambda message: True) # Returning that command is unkown for any other statement +def echo_all(message): + answer = 'Do not know this command or text: ' + message.text + bot.reply_to(message, answer) + + +telebot.logger.setLevel(logging.DEBUG) + + +@bot.inline_handler(lambda query: query.query == 'text') # inline prints for debugging +def query_text(inline_query): + try: + r = types.InlineQueryResultArticle('1', 'Result1', types.InputTextMessageContent('hi')) + r2 = types.InlineQueryResultArticle('2', 'Result2', types.InputTextMessageContent('hi')) + bot.answer_inline_query(inline_query.id, [r, r2]) + except Exception as e: + print(e) + + +def main_loop(): + bot.infinity_polling() + while 1: + time.sleep(3) + + +if __name__ == '__main__': + try: + main_loop() + except KeyboardInterrupt: + print('\nExiting by user request.\n') + sys.exit(0) diff --git a/telegram_bot/bot.py b/telegram_bot/bot.py index 5540439..82dd180 100644 --- a/telegram_bot/bot.py +++ b/telegram_bot/bot.py @@ -22,20 +22,30 @@ bot = telebot.TeleBot(os.getenv('BOT_API_KEY')) @bot.message_handler(commands=['start']) # /start -> saving as new user and sending welcome def send_start(message): - new_user = User(int(message.from_user.id), int(message.chat.id)) + new_user = User(int(message.from_user.id), message.from_user.first_name, int(message.chat.id)) existing_already = False for known_user in user_list: if known_user.user_id == new_user.user_id: existing_already = True if existing_already == False: - user_list.add(new_user) + user_list.append(new_user) bot.reply_to(message, "Welcome to this share bot project. Type /help to get information on what this bot can do") @bot.message_handler(commands=['help']) # /help -> sending all functions def send_welcome(message): - bot.reply_to(message, "/id or /auth for authentication. /update to get updates on your shares. For further details see aktienbot.flokaiser.com") + bot.reply_to(message, "/id or /auth for authentication. /update to get updates on your shares. /users to see all users. For further details see aktienbot.flokaiser.com") + +@bot.message_handler(commands=['users']) +def send_all_users(message): + print('Debug: users command') + user_id = int(message.from_user.id) + answer = 'Current number of users: ' + str(len(user_list)) + bot.send_message(chat_id = user_id, text=answer) + for known_user in user_list: + answer = str(known_user.user_id) + ' : ' + known_user.user_name + bot.send_message(chat_id=user_id, text=answer) @bot.message_handler(commands=['id', 'auth']) # /auth or /id -> Authentication with user_id over web tool @@ -56,6 +66,7 @@ def echo_all(message): answer = 'Do not know this command or text: ' + message.text bot.reply_to(message, answer) + telebot.logger.setLevel(logging.DEBUG)