ValroseDocs.py

import os
from json import JSONDecodeError

import requests
import yaml
from flask import Flask, jsonify, Response
from flask_cors import CORS
from requests import Response
from dotenv import load_dotenv

load_dotenv()
app = Flask(__name__)
CORS(app)

if os.environ.get('FLASK_ENV') != 'development':
    import flask_monitoringdashboard as dashboard

    dashboard.bind(app)

config = None

with open("static/config.yml", "r") as stream:
    try:
        config = yaml.safe_load(stream)
    except yaml.YAMLError as exc:
        print(exc)


# Router for the API

# / - Returns the API version and the API name
@app.route('/')
def index() -> Response:
    return jsonify({
        "name": "ValroseDocs",
        "version": "1.0.0"
    })


# /api/get_data - Returns the data from the API
@app.route('/api/get_data', methods=['GET'])
def api_get_data() -> tuple[Response, int] | Response:
    """
    Returns the data from the API
    :return Response: The response from the API
    """

    # First, we need to login to the API
    # We get the login data from the .env file as we don't want to expose it on GitHub
    json_data = {
        'email': os.environ.get('EMAIL'),
        'password': os.environ.get('PASSWORD'),
    }

    loginReq = requests.post('https://www.mathcha.io/api/up/login', json=json_data)

    # If the request was unsuccessful, return the error
    if loginReq.status_code != 200:
        return jsonify({'error': 'Login failed'}), 500

    # Otherwise, return get the data from the API
    req_url: str = f"https://www.mathcha.io/api/init2?sharedLink="  # The URL to request

    # headers to send with the request
    headers_list = {
        "Accept": "application/json",
        "Content-Type": "application/json"
    }

    # Make the request
    response: Response = requests.request("GET", req_url, headers=headers_list, cookies=loginReq.cookies)

    # response's data encoded as a json object
    try:
        data = response.json()["tree"]["directories"]
    except JSONDecodeError:
        return jsonify({
            "error": "Could not decode JSON"
        }), 500

    valrose = [f for f in data if f['id'] == config['base_folder']['id']][0]
    resp_data = valrose["directories"]

    c = config["conversion"]
    ti = config["tag_image"]

    periods = []
    p_i = 0
    for p in resp_data:
        if p["id"] not in c:
            continue
        periods.append({})
        periods[p_i]["id"] = p["id"]
        if "name" in c[p["id"]]:
            periods[p_i]["name"] = c[p["id"]]["name"]
        if "tag" in c[p["id"]]:
            periods[p_i]["tag"] = c[p["id"]]["tag"]
        if "image" in c[p["id"]]:
            periods[p_i]["image"] = c[p["id"]]["image"]
        units = []
        u_i = 0
        for u in p["directories"]:
            units.append({})
            units[u_i]["id"] = u["id"]
            units[u_i]["name"] = c[u["id"]]["name"] if "name" in c[u["id"]] else u["name"]
            if "tag" in c[u["id"]]:
                units[u_i]["tag"] = c[u["id"]]["tag"]
            units[u_i]["image"] = c[u["id"]]["image"] if "image" in c[u["id"]] else "file.svg"
            lessons = []
            l_i = 0
            for l in u["documents"]:
                if "shared" not in l or l["shared"] is not True:
                    continue
                lessons.append({})
                lessons[l_i]["id"] = l["id"]
                lessons[l_i]["name"] = l["name"]
                # if l["name"] starts with TD, Amphi, TP, etc.
                if l["name"].startswith(tuple(ti.keys())):
                    lessons[l_i]["tag"] = l["name"].split(" ")[0]
                    lessons[l_i]["image"] = ti[lessons[l_i]["tag"]]
                else:
                    lessons[l_i]["image"] = "file.svg"
                lessons[l_i]["url"] = l["generatedLink"] + "?embedded=true"
                l_i += 1
            units[u_i]["children"] = lessons
            u_i += 1
        periods[p_i]["children"] = units
        p_i += 1

    response = jsonify(periods)
    return response