main.py

import asyncio
import signal
import re
from time import sleep
from types import FrameType
from selenium.webdriver.common.by import By
from selenium import webdriver
from selenium.webdriver.common.keys import Keys
from selenium.webdriver import ActionChains
from selenium.common.exceptions import NoSuchElementException
from telegram.ext import ApplicationBuilder

from selenium.webdriver.firefox.options import Options


class SleepInterruptException(Exception):
    pass


def sigClose(sig: int, frame: FrameType | None):
    if frame is not None:
        status = frame.f_locals.get("status")
        driver = frame.f_locals.get("driver")
        if status is not None:
            print("Writing file")
            writeFile(".status", status)
        if driver is not None:
            print("Closing browser")
            driver.quit()
    raise SleepInterruptException()


async def sendMessage(notifType: str, materia: str, chatId: int, application):
    await application.bot.sendMessage(chat_id=chatId, text=f"{materia}: {notifType}")


def writeFile(fileName: str, status: dict[int, dict]):
    with open(fileName, "w") as f:
        for key in status.keys():
            s = status[key]
            f.write(f"{key},{s['contenido']},{s['mensajeria']},{s['forov2']}\n")


async def main():
    signal.signal(signalnum=signal.SIGINT, handler=sigClose)

    dni: str = ""
    password: str = ""
    token: str = ""
    chatId: int = 0

    with open(".token") as f:
        token = f.readline()
        token = token.strip()
        chatId = int(f.readline())

    with open(".credentials") as f:
        dni = f.readline()
        password = f.readline()

    if not re.search("[0-9]{3,3}?[0-9]{3,3}?[0-9]{1,2}", dni):
        print("DNI incorrecto")
        exit(1)

    if len(password) < 8:
        print("Contraseña incorrecta")
        exit(1)

    application = ApplicationBuilder().token(token).build()

    options: Options = Options()
    options.add_argument("--headless")
    driver: webdriver.Firefox = webdriver.Firefox(options=options)
    driver.get("https://miel.unlam.edu.ar")
    ActionChains(driver=driver).send_keys_to_element(
        driver.find_element(By.ID, "usuario"), dni
    ).send_keys_to_element(driver.find_element(By.ID, "clave"), password).send_keys(
        Keys.RETURN
    ).perform()

    sleep(1)

    driver.get("https://miel.unlam.edu.ar/principal/interno/")

    if driver.current_url != "https://miel.unlam.edu.ar/principal/interno/":
        print("No se pudo iniciar sesion")
        driver.close()
        exit(1)

    elementIds = [
        element.get_attribute("data-id")
        for element in driver.find_elements(
            By.CSS_SELECTOR, "body main div.curso-sortable div[data-id]"
        )
    ]

    status: dict[int, dict] = {}

    for i, id in enumerate(elementIds):
        if id is not None:
            idInt = int(id)
            status[idInt] = {
                "contenido": 0,
                "mensajeria": 0,
                "forov2": 0,
            }

    with open(".status", "+a") as f:
        f.seek(0)
        for i, line in enumerate(f.readlines()):
            parts: list[str] = line.split(",")
            if len(parts) == 4:
                currentID = int(parts[0])
                if currentID in status.keys():
                    status[currentID]["contenido"] = int(parts[1])
                    status[currentID]["mensajeria"] = int(parts[2])
                    status[currentID]["forov2"] = int(parts[3])
                else:
                    f.truncate(0)
                    break

    while True:
        print("Polling")
        driver.refresh()

        for materiaDiv in driver.find_elements(By.CSS_SELECTOR, "div.w3-light-grey"):
            name: str = materiaDiv.find_element(By.CSS_SELECTOR, ".materia-titulo").text

            id: str | None = materiaDiv.get_attribute("data-id")
            idInt: int = 0
            if id is not None:
                idInt = int(id)

            contenidosCount = 0
            mensajeriaCount = 0
            foroCount = 0

            categories = materiaDiv.find_elements(
                By.CSS_SELECTOR, ".materia-herramientas>div>a"
            )

            if len(categories) < 4:
                continue
            for i in [0, 2, 3]:
                try:
                    badge = categories[i].find_element(By.CSS_SELECTOR, "span")
                except NoSuchElementException:
                    continue
                if badge is not None:
                    count: int = int(badge.text)
                    if i == 0:
                        contenidosCount = count
                    elif i == 2:
                        mensajeriaCount = count
                    elif i == 3:
                        foroCount = count

            if status[idInt]["contenido"] < contenidosCount:
                async with application:
                    await sendMessage(
                        notifType="contenido",
                        materia=name,
                        chatId=chatId,
                        application=application,
                    )
            status[idInt]["contenido"] = contenidosCount

            if status[idInt]["mensajeria"] < mensajeriaCount:
                async with application:
                    await sendMessage(
                        notifType="mensajeria",
                        materia=name,
                        chatId=chatId,
                        application=application,
                    )
            status[idInt]["mensajeria"] = mensajeriaCount

            if status[idInt]["forov2"] < foroCount:
                async with application:
                    await sendMessage(
                        notifType="foro",
                        materia=name,
                        chatId=chatId,
                        application=application,
                    )
            status[idInt]["forov2"] = foroCount
            writeFile(".status", status)

        try:
            sleep(15)
            # sleep(60 * 15)
        except SleepInterruptException:
            exit(0)


if __name__ == "__main__":
    asyncio.run(main())