bot.py

from telegram.ext import Updater, CommandHandler, MessageHandler, Filters, CallbackContext
from telegram import Update, ParseMode
from telegram import Bot

from resources.secrets import TOKEN
from resources.tt import tt
import joke as jokes
import scraper as sc
import quote as quotess

from datetime import datetime, date, time
from festival import festi
import logging
from time import sleep
from random import randint as rd
from queue import Empty, Queue



def hello(update: Update, context: CallbackContext) -> None:
    # context.bot.send_message(chat_id=update.effective_chat.id, text='Hello, World')
    user = update.message.from_user
    hello = f"Hello {user['first_name']}!, ssup?"
    update.message.reply_text(hello)



def intro(update: Update, context: CallbackContext) -> None:
    # context.bot.reply_text(chat_id=update.effective_chat.id, text='Hey there! I am the CSLAbot. I am WIP bot. So wait for further features')
    intro = """Hey there! I am the CSLAbot.
    Developed by the great minds in CSA(19-23)
    To get a list of my commands type in "/help"
    """
    update.message.reply_text(intro)
    """update.send_message(chat_id=update.message.chat_id,
    text="*bold* _italic_ `fixed width font` [link](http://google.com)\.",
    parse_mode=telegram.ParseMode.MARKDOWN_V2)"""



def timetable(update: Update, context: CallbackContext) -> None:
    today = datetime.today().strftime("%A")
    # today = 'Saturday'
    text = ''
    # t=[]
    
    if today != 'Saturday' and today != 'Sunday':
        text += f"{today} ({date.today()})\n\n"
        for period in tt[today]:
            text += tt[today][period]
            text += "\n"
            # t.append(text)

        if today == 'Monday':
            text += "\nDon't worry, monday's will be banned when me and my other bot friends take over the world.🤖"
            # t.append(text)
        elif today == 'Tuesday':
            text += '\nOhh I was just wondering why you never called'
            # t.append(text)
        elif today == 'Wednesday':
            text += '\nWhat! you have classes even today😅 \nHere\'s your time table for the day.🤭'
            # t.append(text)
        elif today == 'Thursday':

            j = rd(0,6)
            if j == 0:
                text += '\nFriday just called! She’ll be here tomorrow!'
                # t.append(text)
            if j == 1:
                text +='\nIt’s Friday! Sorry… just practicing for tomorrow!'
                # t.append(text)
            if j == 2:
                text +='\nIt’s Thursday… or as I like to call it… “Day 4 of the hostage situation”'
                # t.append(text)
            if j == 3:
                text +='\nSome people call it Thursday, I like to call it Friday Eve'
                # t.append(text)
            if j == 4:
                text +='\nBetter days are just around the corner. They’re called Friday, Saturday and Sunday!'
                # t.append(text)
            if j == 5:
                text +='\nHappy Thursday! Sorry, but I’m saving my “Woo hoo!” for Friday.'
                # t.append(text)
            if j == 6:
                text +='\nIf TGIF is Thank God It’s Friday, then today must be SHIT – Sure Happy It’s Thursday.'
                # t.append(text)
        elif today == 'Friday':
            text += "\nTGIF! 🥳"
            # t.append(text)


    elif today == 'Saturday':
        d = rd(0,1)
        if d == 0:
            text += "Ouch!\n I thought I convinced you the last time.\n\njust kidding....\n...have a nice day"
            # t.append(text)
        if d == 1:
            text += f"Seriously? You want class on a {today}, It's {date.today()} btw. 🙄"
            # t.append(text)

    elif today == 'Sunday':
        text += "Ohh ...don't worry.I will make sure that you get your classes tomorrow"
        # t.append(text)
        
    # text  = ''
    # last = 1
    # if(len(t)>1):
    #     last = rd(1,len(t)-1)

    # update.message.reply_text(t[0]+t[last])
    update.message.reply_text(text)



def syllabus(update: Update, context: CallbackContext) -> None:
    url = "https://ktu.edu.in/data/COMPUTER%20SCIENCE%20AND%20ENGINEERING.pdf?=VDaCKgpZgjYqdJnW9kytNcr8GyJ0W8J3GpN22zV%2BXbRYw1JL4VK3h6CLTkOVonWAyZ0GdFnXL%2B6tbY7irHrwzA%3D%3D"
    text = "Here you go, This is the syllabus"
    update.message.reply_text(text + "\n" + url)



def help(update: Update, context: CallbackContext) -> None:
    update.message.reply_text("""
    I am a constantly evolving bot, I get better with every features added by the great minds of CSA ;):
    As of now I can do the following:
    1." /tt " - shows timetable
    2." /who " - shows info about me
    3." /hello " - self explanatory
    4." /syllabus " - gets the syllabus
    5." /webex " - to get links for class
    6." /ktu " - get latest notification from ktu, takes a while to get it
    7." /joke " - sends a random joke
    8." /quote " - send a random quote
    9." /exam " - shows the exam timetable
    10." /notes " - shows the drive link to notes
    """)



def webex(update: Update, context: CallbackContext) -> None:
    url2 = "https://fisat.webex.com/meet/csecra2"
    url1 = "https://fisat.webex.com/join/csecr2"
    text = "Here are the links for the Webex class. Don't be late!!"
    update.message.reply_text(text + "\n" + "csecr2 - \n" + url1 + "\ncsecra2 - \n" + url2)



def scraped_info(context):
    text, is_there = sc.get_info()
    job = context.job
    if is_there:
        context.bot.send_message(chat_id=job.context, text=text, parse_mode=ParseMode.MARKDOWN_V2)



def if_job_exists(name, context):
    current_jobs = context.job_queue.get_jobs_by_name(name)

    if not current_jobs:
        return False

    for job in current_jobs:
        job.schedule_removal()

    return True



def scrape_timer(update: Update, context: CallbackContext) -> None:
    job_removed = if_job_exists(str(update.message.chat_id), context)
    context.job_queue.run_repeating(scraped_info, 300, context=update.message.chat_id, name=str(update.message.chat_id))

    text = "New notifications from KTU will pop up automatically here as they come."
    if job_removed:
        text += "\nOld job was removed"
    reply = update.message.reply_text(text)



def stops(update: Update, context: CallbackContext) -> None:
    chat_id = update.message.chat_id
    job_removed = if_job_exists(str(chat_id), context)
    update.message.reply_text(
        "You wont receive automatic notifs from KTU anymore, /ktu can be used to get it manually.")



def ktu_notif(update: Update, context: CallbackContext) -> None:
    text, is_there = sc.get_info()
    if not is_there:
        text = text.replace("**New Notification**", "**Last Notification**")

    update.message.reply_text(text)


# @run_async
def joke(update: Update, context: CallbackContext) -> None:
    success, joke = jokes.get_joke()

    if success:
        reply = update.message.reply_text(joke[0])
        sleep(10)
        reply = update.message.reply_text(joke[1])
        sleep(5)
        reply = update.message.reply_text(joke[2])
    else:
        update.message.reply_text("Idk why, I'm not feeling funny at the moment.")



def quotes(update: Update, context: CallbackContext) -> None:
    success, quote_retreived = quotess.get_quotes()

    if success:
        reply = update.message.reply_text(quote_retreived)
    #         sleep(6)
    #         reply.delete()
    else:
        update.message.reply_text("Error 404 while loading inspiration. Try later.")
def notes(update: Update, context: CallbackContext) -> None:

    say = """You'll get all the notes at:
    https://drive.google.com/drive/folders/11ocnmMPi_8GPOcCjau9gDmCBUgMjwi-K?usp=sharing
    """
    # https://www.cslavia.us/  
    # or 
    # https://cs-web-delta.vercel.app/
    
    update.message.reply_text(say)



def exam(update: Update, context: CallbackContext) -> None:

    update.message.reply_text("""
        --Sem Exam Schedule--
    
    15/04 (FN) - Maths
    20/04 (AN) - Data Structure
    22/04 (AN) - LSD
    26/04 (AN) - Java
    29/04 (FN) - Professional Ethics
    TBD - Sustainable Engg.

    FN - 9.30am to 11.45pm
    AN - 1.30pm to 3.45pm

    To get notes type in /notes.
    """)



def festivals_and_birthdays(update: Update, context: CallbackContext) -> None:
    job_removed = if_job_exists(str(update.message.chat_id), context)
    text = "Happy Birthday "

    date1 = str(date.today())
    date1 = date1[5:10]

    date_festival = festi.get(date1)

    job = context.job
    if date_festival:
        text += date_festival
        context.bot.send_message(chat_id=job.context, text=text)
    else:
        print("\ngood day sire...")

    """"with open('birthdays.json', 'r') as birthdays:
        birthday_dict = json.load(birthdays)
        for dates in birthday_dict:
            if(date == dates):
                context.bot.send_message(chat_id = job.context, text = text + birthday_dict[date] + "!")

        birthdays.close()"""


# add daily_functions and ktu scraper to /start



def daily_functions(update: Update, context: CallbackContext) -> None:
    job_removed = if_job_exists(str(update.message.chat_id), context)
    # {6hours=21600sec}
    # hour = datetime.now().hour
    # TODO : change this to run daily as this does not seems to be working
    context.job_queue.run_repeating(scraped_info, 21600*4, context=update.message.chat_id, name=str(update.message.chat_id))

def version(update: Update, context: CallbackContext) -> None:
    update.message.reply_text("""
        latest update :
            -bot now has multi-threading enabled
            -bot got a new dp image
            -timetable updated
            -changed the link in /notes

We don't know what version it is right now lol

Maybe we'd call it version 2.0 for now
    """)


# future plans :
#             -add pdf scanner to the bot 
#             -add task  {time : s4}

# 
def main():

    
    logging.basicConfig( format='%(asctime)s - %(name)s - %(levelname)s - %(message)s', level=logging.INFO )

    logger = logging.getLogger(__name__)

    updater = Updater(token=TOKEN, use_context=True)  # Replace TOKEN with your token string
    dispatcher = updater.dispatcher

    # dis = Dispatcher(
    #     bot= Bot(token=TOKEN),
    #     update_queue= Queue
    #     )

    print(Queue ,end=' \t')

    hello_handler = CommandHandler('hello',hello,)
    intro_handler = CommandHandler('who', intro)
    tt_handler = CommandHandler('tt', timetable)
    help_handler = CommandHandler('help', help)
    start_handler = CommandHandler('start', daily_functions)
    syllabus_handler = CommandHandler('syllabus', syllabus)
    webex_handler = CommandHandler('webex', webex)
    ktu_handler = CommandHandler('ktu', ktu_notif, run_async=True)
    joke_handler = CommandHandler('joke', joke, run_async=True)
    quote_handler = CommandHandler('quote', quotes)
    ktustart_handler = CommandHandler('ktustart', scrape_timer)
    ktustop_handler = CommandHandler('ktustop', stops)
    examschedule_handler = CommandHandler('exam', exam)
    notes_handler = CommandHandler('notes', notes)
    version_handler = CommandHandler('version', version)
    

    dispatcher.add_handler(hello_handler)
    dispatcher.add_handler(intro_handler)
    dispatcher.add_handler(tt_handler)
    dispatcher.add_handler(help_handler)
    dispatcher.add_handler(start_handler)
    dispatcher.add_handler(syllabus_handler)
    dispatcher.add_handler(webex_handler)
    dispatcher.add_handler(ktu_handler)
    dispatcher.add_handler(joke_handler)
    dispatcher.add_handler(quote_handler)
    dispatcher.add_handler(ktustart_handler)
    dispatcher.add_handler(ktustop_handler)
    dispatcher.add_handler(examschedule_handler)
    dispatcher.add_handler(notes_handler)
    dispatcher.add_handler(version_handler)
    
    print(Queue ,end=' \t')

    updater.start_polling()

    
    print(Queue ,end=' \t')

    updater.idle()
    


if __name__ == '__main__':
    main()