File skola_online.py

import requests
from bs4 import BeautifulSoup
from lxml import etree
from universal import GetRequestException


class SkolaOnlineLoginException(Exception):
    """! @brief Vyjímka pro chybné přístupové údaje.
    """

    def __init__(self, message=""):
        """! @brief Inicializuje vyjímku."""
        self.messagemessage = f"SkolaOnline Login Exception: {message}"
        super().__init__(self.messagemessage)


class LastMark:
    """! @brief Uchovává informace o známkách."""

    # date of the last mark
    date: str
    # subject of the last mark
    subject: str
    # mark of the last mark
    mark: int

    def __init__(self, date: str, subject: str, mark: str):
        """! @brief Konstruktor pro vytvoření instance třídy LastMark.

        @param date datum známky
        @param subject předmět známky
        @param mark známka
        """
        self.datedate = date
        self.subjectsubject = subject
        self.markmark = mark

    def __str__(self):
        """! @brief Vrací řetězec s informacemi o známkách."""
        return f"{self.date} {self.subject} {self.mark}"


class SkolaOnline:
    """! @brief Získává data ze Školy Online"""

    def __init__(self, name: str, pwd: str):
        """! @brief Konstruktor pro vytvoření instance třídy SkolaOnline.
        @param name uživatelské jméno
        @param pwd heslo
        """
        self.namename = name
        self.pwdpwd = pwd
        self.urlBaseurlBase = "https://aplikace.skolaonline.cz/SOL/App/"

        self.webSessionwebSession = requests.Session()
        self.__logInSol__logInSol()

    def __logInSol(self):
        """! @brief Přihlásí se do Školy Online."""
        url = 'https://aplikace.skolaonline.cz/SOL/Prihlaseni.aspx'
        data = {"__EVENTTARGET": "dnn$ctr994$SOLLogin$btnODeslat", "__EVENTARGUMENT": "", "__VIEWSTATE": "",
                "__VIEWSTATEGENERATOR": "", "__VIEWSTATEENCRYPTED": "", "__PREVIOUSPAGE": "", "__EVENTVALIDATION": "",
                "dnn$dnnSearch$txtSearch": "", "JmenoUzivatele": self.namename, "HesloUzivatele": self.pwdpwd, "ScrollTop": "",
                "__dnnVariable": "", "__RequestVerificationToken": "", }

        postLogin = self.webSessionwebSession.post(url, data=data)
        if postLogin.status_code != 200:
            raise SkolaOnlineLoginException(f"status_code = {postLogin.status_code}")

    def __getRequest(self, urlBase, params=None):
        """! @brief Vrací obsah stránky.
        @param urlBase adresa stránky
        @param params parametry stránky
        @return obsah stránky
        @throws GetRequestException
        """
        getMarks = self.webSessionwebSession.get(urlBase + params)
        if getMarks.status_code == 200:
            return getMarks.text
        else:
            raise GetRequestException()

    # def getTimeTable(self):
    #   html = self.__getRequest("https://aplikace.skolaonline.cz/SOL/App", "/Kalendar/KZK001_KalendarTyden.aspx")
    #   soup = BeautifulSoup(html, 'html.parser')
    #
    #   # htmlTable = soup.find(id="CCADynamicCalendarTable")
    #   htmlTables = soup.findAll("table", attrs={'class': 'DctInnerTableType10'})
    #
    #   for table in htmlTables:
    #       sub = table.contents[1].contents[1].contents[0].text
    #       xclass = table.contents[1].contents[1].contents[1].contents[0]
    #       room = table.contents[1].contents[1].contents[1].contents[2]
    #       print(f"{sub} -> {room}")
    #
    #   domTable = etree.HTML(str(soup))
    #   # domTable = etree.HTML(html)
    #   table = domTable.xpath('//*[@id="CCADynamicCalendarTable"]')
    #   name = domTable.xpath('//*[@id="LBLStudent"]')
    #   test = domTable.xpath('//*[@id="C34353962#1#2#2"]/span[1]')
    #
    #   return domTable

    # def getMarks(self):
    #   rawMarks = self.__getRequest("https://aplikace.skolaonline.cz/SOL/App", "/Hodnoceni/KZH001_HodnVypisStud.aspx")
    #   return rawMarks

    def getLastMarks(self):
        """! @brief Vrací pole objektů LastMark.

        @return pole objektů LastMark
        """

        html = self.__getRequest__getRequest("https://aplikace.skolaonline.cz/SOL/App", "/Spolecne/KZZ010_RychlyPrehled.aspx")
        domTable = etree.HTML(str(html))
        table = domTable.xpath('//*[@id="ctl00_bodyFooter_boxHodnVypisStud_ctl00_Content"]/table')

        if not table:
            raise SkolaOnlineLoginException("Bad username or password")

        lastMarks = []
        for tableRow in table[0]:
            if tableRow.tag == "tr":
                if tableRow[2][0][0].text:
                    mark = tableRow[2][0][0].text
                else:
                    mark = tableRow[2][0][0][0].text
            else:
                mark = "Err"
            lastMarks.append(LastMark(tableRow[0].text, tableRow[1].text, mark))
        return lastMarks