TelegramAktienBot/telegram_bot/bot_updates.py

"""
script for regularly sending updates on shares and news based on user interval
"""
__author__ = "Florian Kellermann, Linus Eickhoff"
__date__ = "05.04.2022"
__version__ = "1.0.1" 
__license__ = "None"

from calendar import month
from symtable import Symbol
from shares.share_fetcher import get_share_price
import time
import datetime
from bot import bot
import sys
from multiprocessing import Process
from apscheduler.schedulers.background import BackgroundScheduler
from api_handling.api_handler import API_Handler


'''
* * * * * code
┬ ┬ ┬ ┬ ┬
│ │ │ │ │
│ │ │ │ └──── weekday (0->Monday, 7->Sunday)
│ │ │ └────── Month (1-12)
│ │ └──────── Day (1-31)
│ └────────── Hour (0-23)
└──────────── Minute (0-59)

example 0 8 * * * -> daily update at 8am
'''
user_ids = []
user_crontab = []

def main_loop():
    """ main loop for regularly sending updates
    :raises: none

    :rtype: none
    """
    
    current_time_datetime = datetime.datetime.now()
    my_handler = API_Handler("https://gruppe1.testsites.info/api", "bot@example.com", "bot")
    
    
    # update_for_user(5270256395, my_handler) # Debug (running test update for kevins shares)
    
    
    update_crontab(current_time_datetime, my_handler)
    
    
def update_crontab(pCurrent_Time, p_my_handler):
    """ Updating crontab lists every hour
    :type pCurrent_Time: time when starting crontab update
    :param pCurrent_Time: datetime

    :raises: none

    :rtype: none
    """
    
    global user_crontab
    global user_ids
    
    #p_my_handler.set_cron_interval(user_id = 1770205310, cron_interval = "23 08 * * *")
    
    all_users = p_my_handler.get_all_users()
    
    user_ids = []
    user_crontab = []
    
    for element in all_users:
        if element["cron"] != '' and element["telegram_user_id"] != '':
            user_ids.append(int(element["telegram_user_id"]))
            user_crontab.append(str(element["cron"]))
    
    print(user_ids)
    
    update_based_on_crontab(user_ids, user_crontab, p_my_handler)
    
    update_crontab(datetime.datetime.now(), p_my_handler)
    
    
def update_based_on_crontab(p_user_ids, p_user_crontab, p_my_handler):
    
    """ Check all the crontab codes and add jobs to start in time
    :type p_user_ids: array
    :param p_user_ids: user id array of all users

    :type p_user_crontab: array
    :param p_user_crontab: crontabs for all users equivalent to the user array

    :type p_my_handler: Api_Handler
    :param p_my_handler: get database stuff

    :raises: none

    :rtype: none
    """
    
    my_scheduler = BackgroundScheduler()
    
    print(len(user_ids)) #Debug
    
    for i in range(len(p_user_ids)):
        cron_split = p_user_crontab[i].split(" ")
        print(cron_split[4], cron_split[1], cron_split[0], cron_split[3], cron_split[2])
        my_scheduler.add_job(update_for_user, 'cron', day_of_week = cron_split[4] , hour= cron_split[1] , minute = cron_split[0], month= cron_split[3] , day=cron_split[2], args=(p_user_ids[i], p_my_handler ))
    
    my_scheduler.start()
    
    time.sleep( 600 )
    my_scheduler.shutdown()
                    
def update_for_user(p_user_id, p_my_handler):
    
    """ Pull shares and send updates for specific user id
    :type p_user_id: integer
    :param p_user_id: user id of user that shall receive update

    :type p_my_handler: Api_Handler
    :param p_my_handler: handle the api and pull from database

    :raises: none

    :rtype: none
    """
    share_symbols = []
    share_amounts = []
    share_courses = []
    
    my_portfolio = p_my_handler.get_user_portfolio(p_user_id)
    
    for element in my_portfolio:
        if element["count"] != '' and element["symbol"]!= '':
            print(element["count"], element["symbol"])
            share_symbols.append(element["symbol"])
            share_amounts.append(element["count"])
            share_courses.append(element["current_price"])

    my_user = p_my_handler.get_user(p_user_id)
    send_to_user("Hello %s this is your share update:"%str(my_user["username"]), pUser_id=p_user_id)
    
    if len(share_symbols) != 0:
        for i in range(len(share_symbols)):
            my_update_message = f'Symbol: {share_symbols[i]}\nCurrent Price per Share: {share_courses[i]}\nAmount owned: {share_amounts[i]}\nTotal Investment: {float(share_courses[i]) * float(share_amounts[i])}'
            send_to_user(my_update_message, pUser_id=p_user_id)
    else:
        send_to_user("No shares found for your account. Check https://gruppe1.testsites.info/api to change your settings and add shares.", pUser_id=p_user_id)
         
    
def send_to_user(pText, pUser_id = 1770205310):
    
    """ Send message to user
    :type pText: string
    :param pText: Text to send to user

    :type pUser_id: int
    :param pUser_id: user to send to. per default me (Florian Kellermann)

    :raises: none

    :rtype: none
    """
    bot.send_message(chat_id=pUser_id, text=pText)


if __name__ == "__main__":
    
    print('bot_updates.py starting.')
    try:
        main_loop()
        sys.exit(-1)
    except KeyboardInterrupt:
        print("Ending")
        sys.exit(-1)