DualHub/fetchDUALIS.py

import requests
import urllib.parse
import time
from bs4 import BeautifulSoup
from flask import redirect, url_for
from init import Dualis

headers = {
    'Cookie': 'cnsc=0',
    'Content-Type': 'application/x-www-form-urlencoded',
    'User-Agent': 'Mozilla/5.0 (Windows NT 6.1; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) '
                  'Chrome/58.0.3029.110 Safari/537.36',
    'Accept-Encoding': 'utf-8'
}

url = "https://dualis.dhbw.de/scripts/mgrqispi.dll"


def checkUser(email: str, password: str):
    """
    Erhält von Dualis den Token und Cookie für User.
    :param email:
    :param password:
    :return (Token, Session):
    """
    s = requests.Session()
    fpw = urllib.parse.quote(password, safe='', encoding=None, errors=None)
    fmail = urllib.parse.quote(email, safe='', encoding=None, errors=None)
    payload = 'usrname=' + fmail + '&pass=' + fpw + ('&ARGUMENTS=clino%2Cusrname%2Cpass%2Cmenuno%2Cmenu_type%2Cbrowser'
                                                     '%2Cplatform&APPNAME=CampusNet&PRGNAME=LOGINCHECK')
    response = s.post(url, headers=headers, data=payload)
    header = response.headers
    try:
        refresh = header["REFRESH"]
        arg = refresh.find("=-N") + 3
        komma = refresh[arg:].find(",")
    except KeyError:
        return -2, s
    token = refresh[arg:komma + arg]
    return token, s


def getKurs(token: int, cookie: str):
    """
    Bestimmt aus der ersten Prüfung den Kursbezeichner des Users.
    TODO: Umstellen auf Bezeichner INKL. Standort
    :param token:
    :param cookie:
    :return Kurs-Bezeichner:
    """
    try:
        headers["Cookie"] = "cnsc=" + cookie
        token = str(token)
        response = requests.request("GET", url +
                                    "?APPNAME=CampusNet&PRGNAME=COURSERESULTS&ARGUMENTS=-N" + token + ",-N000307,",
                                    headers=headers, data={})
        html = BeautifulSoup(response.text, 'lxml')
        link = html.body.find('a', attrs={'id': "Popup_details0001"})['href']
        response = requests.request("GET", url + link[21:], headers=headers, data={})
        html = BeautifulSoup(response.text, 'lxml')
        content = html.body.find('td', attrs={'class': 'level02'}).text
        start = content.find(" ") + 4
        end = start + (content[start:].find(" "))
        kurs = content[start:end]
    except AttributeError:
        kurs = 0
    return kurs


def logOut(token: int, cookie: str):
    """
    Invalidiert Token und Cookie bei Dualis.
    :param token:
    :param cookie:
    """
    headers["Cookie"] = "cnsc=" + cookie
    requests.request("GET", url + "?APPNAME=CampusNet&PRGNAME=LOGOUT&ARGUMENTS=-N" + str(token)
                     + ", -N001", headers=headers, data={})


def getSem(token: int, cookie: str):
    """
    Liefert die Liste aller auf Dualis verfügbaren Semester.
    :param token:
    :param cookie:
    :return Liste [[Semester, Semester-ID], ...]:
    """
    headers["Cookie"] = "cnsc=" + cookie
    token = str(token)
    response = requests.request("GET", url +
                                "?APPNAME=CampusNet&PRGNAME=COURSERESULTS&ARGUMENTS=-N" + token + ",-N000307,",
                                headers=headers, data={})
    html = BeautifulSoup(response.text, 'lxml')
    select = html.find('select')
    select = select.find_all(value=True)
    optlist = []
    for i in select:
        t = i.text.replace("Wi", "Winter").replace("So", "Sommer")
        t = t.replace("Se", "semester")
        optlist += [[t, i['value']]]
    return optlist


def getResults(token, cookie: str, resl: str):
    """
    Liefert die Liste aller Prüfungsergebnisse eines Semesters.
    :param token:
    :param cookie:
    :param resl:
    :return [[Name, Note, Credits], ...]:
    """
    headers["Cookie"] = "cnsc=" + cookie
    response = requests.request("GET", url + "?APPNAME=CampusNet&PRGNAME=COURSERESULTS&ARGUMENTS=-N" + token +
                                ",-N000307," + ",-N" + resl, headers=headers, data={})
    html = BeautifulSoup(response.content.decode("utf-8"), 'lxml')
    table = html.find('table', attrs={"class": "nb list"})
    body = table.find("tbody")
    vorl = body.find_all("tr")
    vorlist = []
    for row in vorl:
        cols = row.find_all("td")
        col = [e.text.strip() for e in cols]
        if len(col) != 0:
            if len(col[4]) == 0:
                col[2] = getPruefung(row.find("a")["href"])
            vorlist += [col[1:4]]
    return vorlist


def getPruefung(url):
    """
    Ermittelt Noten "geschachtelter" Prüfungen, die nicht auf der Hauptseite angezeigt werden.
    TODO: Namen der spezifischen Prüfungen auch zurückgeben, um Zusammensetzung zu spezifizieren.
    :param url:
    :return:
    """
    response = requests.request("GET", "https://dualis.dhbw.de" + url, headers=headers, data={})
    html = BeautifulSoup(response.content.decode("utf-8"), 'lxml')
    table = html.find('table')
    pruefung = table.find_all("tr")
    ret = "   "
    for row in pruefung:
        cols = row.find_all("td")
        col = [e.text.strip() for e in cols]
        if len(col) == 6:
            ret = col[3][:3]
    if ret[1] != ",":
        ret = "noch nicht gesetzt"
    return ret


def checkLifetime(timecode: float):
    """
    Dualis-Token laufen nach 30 Minuten ab.
    True, wenn Token noch gültig ist.
    False wenn ungültig.
    :param timecode:
    :return:
    """
    if time.time() - timecode > 1800:
        return False
    else:
        return True


def timeOut(dualis: Dualis, cookie: str, origin: str):
    """
    Checkt, ob Token und Cookie noch valide/vorhanden sind.
    False, falls alles richtig ist.
    Weiterleitung zum Login, falls nicht.
    :param dualis:
    :param cookie:
    :param origin:
    :return:
    """
    if not checkLifetime(dualis.token_created) or not cookie:
        return redirect(url_for("login", next=url_for(origin), code=10))
    else:
        return False