110 lines
3.4 KiB
Python
110 lines
3.4 KiB
Python
#!/usr/bin/env python3.6
|
|
import os
|
|
from flask import Flask
|
|
from flask import render_template, url_for, send_from_directory, redirect, request, send_file
|
|
from flask_sqlalchemy import SQLAlchemy
|
|
import pymysql
|
|
from werkzeug.security import generate_password_hash, check_password_hash
|
|
import hashlib
|
|
|
|
import dualisauth
|
|
from fetchRAPLA import *
|
|
from get_mysql import get_mysql
|
|
|
|
app = Flask(__name__)
|
|
|
|
db = SQLAlchemy()
|
|
dbpw = get_mysql()
|
|
app.config['SECRET_KEY'] = 'ASDF)uhdsklvbkuezafdpo12i34rewfgvoukzgp3zerfpg8owiu2301394trilfkj'
|
|
app.config['SQLALCHEMY_DATABASE_URI'] = 'mysql+pymysql://paulmrtn:' + dbpw + '@localhost/paulmrtn_DUALHUB'
|
|
db.init_app(app)
|
|
|
|
|
|
class User(db.Model):
|
|
id = db.Column(db.Integer, primary_key=True) # primary keys are required by SQLAlchemy
|
|
email = db.Column(db.String(100), unique=True)
|
|
password = db.Column(db.String(100))
|
|
name = db.Column(db.String(1000))
|
|
|
|
|
|
@app.route("/")
|
|
def index():
|
|
return render_template('index.html', headermessage='Header', message='DualHub')
|
|
|
|
|
|
@app.route("/backendpoc/error<int:ecode>")
|
|
def error(ecode):
|
|
if ecode == 900:
|
|
msg = "Ungültige RAPLA-URL! Sicher, dass der Link zum DHBW-Rapla führt?"
|
|
elif ecode == 899:
|
|
msg = "Der Kalender wurde nicht gefunden! Sicher, dass der Link korrekt ist?"
|
|
else:
|
|
msg = "Unbekannter Fehler!"
|
|
return render_template('index.html', message=msg, headermessage="DualHub")
|
|
|
|
|
|
@app.route("/backendpoc/rapla")
|
|
def chooseRaplas():
|
|
r = getRaplas()
|
|
return render_template("rapla.html", raplas=r)
|
|
|
|
|
|
@app.route("/backendpoc/plan", methods=["POST"])
|
|
def getRapla():
|
|
file = str(request.form.get("file"))
|
|
url = str(request.form.get("url"))
|
|
if file == url == "None":
|
|
return redirect(url_for("chooseRaplas"))
|
|
if file != "None":
|
|
return send_file("calendars/" + file)
|
|
elif url != "None":
|
|
file = getNewRapla(url)
|
|
if type(file) is not int:
|
|
return send_file("calendars/" + file)
|
|
else:
|
|
return redirect(url_for("error", ecode=file + 900))
|
|
return render_template("index.html")
|
|
|
|
|
|
@app.route("/backendpoc/log-in")
|
|
def login(ecode: int = None):
|
|
if ecode:
|
|
print(ecode)
|
|
return render_template("login.html")
|
|
|
|
|
|
@app.route("/backendpoc/log-in", methods=["POST"])
|
|
def login_post():
|
|
email = request.form.get("email")
|
|
password = request.form.get("password")
|
|
|
|
user = User.query.filter_by(email=email).first()
|
|
|
|
if user:
|
|
if check_password_hash(user.password, password):
|
|
return redirect(url_for("index"))
|
|
else:
|
|
t = dualisauth.checkUser(email, password)
|
|
if t == -2:
|
|
return redirect(url_for("login", ecode=-2))
|
|
else:
|
|
user.password = generate_password_hash(password, method="pbkdf2:sha256")
|
|
db.session.commit()
|
|
return redirect(url_for("index"))
|
|
|
|
t = dualisauth.checkUser(email, password)
|
|
if t == -2:
|
|
return redirect(url_for("login", ecode=-2))
|
|
|
|
hashid = int(hashlib.sha1(email.encode("utf-8")).hexdigest(), 16) % (10 ** 12)
|
|
hashpw = generate_password_hash(password, method="pbkdf2:sha256")
|
|
new_user = User(id=hashid, email=email, password=hashpw)
|
|
db.session.add(new_user)
|
|
db.session.commit()
|
|
|
|
return redirect(url_for("index"))
|
|
|
|
|
|
if __name__ == "__main__":
|
|
app.run(host='0.0.0.0', port=2024, debug=True)
|