application.py

from flask import Flask, request, jsonify
from flask_cors import CORS
from uims_api import SessionUIMS
from uims_api.exceptions import IncorrectCredentialsError, UIMSInternalError

app = Flask(__name__)
CORS(app)


@app.route("/")
def home():
    return "OK!"


@app.route("/api/signin", methods=["POST"])
def signin():

    if not request.form.get("uid"):
        return jsonify({"error": "UID not provided"})
    if not request.form.get("password"):
        return jsonify({"error": "Password not provided"})

    try:
        SessionUIMS(request.form.get("uid"), request.form.get("password"))
    except Exception as e:
        if e.__class__ == IncorrectCredentialsError:
            return jsonify({"error": "Invalid credentials"})
        else:
            return jsonify({"error": "Looks like this Module is inactive on UIMS"})
    else:
        return jsonify({"success": True})


@app.route("/api/attendance", methods=["POST"])
def get_minimal_attendance():
    if not request.form.get("uid"):
        return jsonify({"error": "UID not provided"})
    if not request.form.get("password"):
        return jsonify({"error": "Password not provided"})
    try:
        my_acc = SessionUIMS(request.form.get("uid"), request.form.get("password"))
    except Exception as e:
        if e.__class__ == IncorrectCredentialsError:
            return jsonify({"error": "Invalid credentials"})
        else:
            return jsonify({"error": "Looks like this Module is inactive on UIMS"})
    try:
        subjects = my_acc.attendance
    except Exception as e:
        if e.__class__ == UIMSInternalError:
            return jsonify({"error": "UIMS Internal Failure"})
        else:
            return jsonify({"error": "Looks like this Module is inactive on UIMS"})
    else:
        return jsonify(subjects)


@app.route("/api/fullattendance", methods=["POST"])
def get_full_attendance():
    if not request.form.get("uid"):
        return jsonify({"error": "UID not provided"})
    if not request.form.get("password"):
        return jsonify({"error": "Password not provided"})
    try:
        my_acc = SessionUIMS(request.form.get("uid"), request.form.get("password"))
    except Exception as e:
        if e.__class__ == IncorrectCredentialsError:
            return jsonify({"error": "Invalid credentials"})
        else:
            return jsonify({"error": "Looks like this Module is inactive on UIMS"})
    try:
        subjects = my_acc.full_attendance
    except Exception as e:
        if e.__class__ == UIMSInternalError:
            return jsonify({"error": "UIMS Internal Failure"})
        else:
            return jsonify({"error": "Looks like this Module is inactive on UIMS"})
    else:
        return jsonify(subjects)


@app.route("/api/timetable", methods=["POST"])
def get_timetable():
    if not request.form.get("uid"):
        return jsonify({"error": "UID not provided"})
    if not request.form.get("password"):
        return jsonify({"error": "Password not provided"})

    try:
        my_acc = SessionUIMS(request.form.get("uid"), request.form.get("password"))
    except Exception as e:
        if e.__class__ == IncorrectCredentialsError:
            return jsonify({"error": "Invalid credentials"})
        else:
            return jsonify({"error": "Looks like this Module is inactive on UIMS"})
    try:
        timetable = my_acc.timetable
    except Exception as e:
        if e.__class__ == UIMSInternalError:
            return jsonify({"error": "UIMS Internal Failure"})
        else:
            return jsonify({"error": "Looks like this Module is inactive on UIMS"})
    else:
        return jsonify(timetable)


if __name__ == "__main__":
    # This is used when running locally. Gunicorn is used to run the
    # application on Google App Engine. See entrypoint in app.yaml.
    app.run(host="127.0.0.1", port=5000, debug=True)
# [END gae_flex_python_static_files]