get-kurs
This commit is contained in:
132
app.py
132
app.py
@ -1,30 +1,57 @@
|
||||
#!/usr/bin/env python3.6
|
||||
import os
|
||||
import requests
|
||||
from flask import Flask
|
||||
from flask import render_template, url_for, send_from_directory, redirect, request, send_file
|
||||
from flask_login import login_user, login_required, current_user, LoginManager, UserMixin, logout_user
|
||||
from flask_sqlalchemy import SQLAlchemy
|
||||
import pymysql
|
||||
from werkzeug.security import generate_password_hash, check_password_hash
|
||||
import hashlib
|
||||
|
||||
import dualisauth
|
||||
import requesthelpers
|
||||
from fetchRAPLA import *
|
||||
from get_mysql import get_mysql
|
||||
import time
|
||||
|
||||
|
||||
def create():
|
||||
app = Flask(__name__)
|
||||
dbpw = get_mysql()[1]
|
||||
dbun = get_mysql()[0]
|
||||
|
||||
app.config['SECRET_KEY'] = 'SECRET_KEY_GOES_HERE'
|
||||
app.config['SQLALCHEMY_DATABASE_URI'] = 'mysql+pymysql://' + dbun + ':' + dbpw + '@localhost/paulmrtn_DUALHUB'
|
||||
db.init_app(app)
|
||||
|
||||
login_manager = LoginManager()
|
||||
login_manager.init_app(app)
|
||||
login_manager.login_view = "login"
|
||||
|
||||
@login_manager.user_loader
|
||||
def load_user(uid: int):
|
||||
return User.query.filter_by(id=uid).first()
|
||||
|
||||
return app
|
||||
|
||||
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)
|
||||
app = create()
|
||||
|
||||
|
||||
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))
|
||||
class User(UserMixin, db.Model):
|
||||
id = db.Column(db.Integer, primary_key=True)
|
||||
email = db.Column(db.String(255), unique=True)
|
||||
password = db.Column(db.String(255))
|
||||
name = db.Column(db.String(255))
|
||||
kurs = db.Column(db.String(15))
|
||||
|
||||
|
||||
class Dualis(db.Model):
|
||||
token = db.Column(db.String(255), unique=True)
|
||||
uid = db.Column(db.Integer, primary_key=True)
|
||||
token_created = db.Column(db.Integer)
|
||||
result_lists = db.Column(db.String(255))
|
||||
cookie = db.Column(db.String(255))
|
||||
|
||||
|
||||
@app.route("/")
|
||||
@ -32,6 +59,20 @@ def index():
|
||||
return render_template('index.html', headermessage='Header', message='DualHub')
|
||||
|
||||
|
||||
@app.route("/welcome")
|
||||
@login_required
|
||||
def welcome():
|
||||
d = Dualis.query.filter_by(uid=current_user.id).first()
|
||||
if not d.kurs:
|
||||
kurs = dualisauth.getKurs(d.token, d.cookie)
|
||||
current_user.kurs = kurs
|
||||
db.session.commit()
|
||||
else:
|
||||
kurs = d.kurs
|
||||
name = current_user.name
|
||||
return render_template('index.html', headermessage='DualHub', message="Hallo, "
|
||||
+ name + " (" + kurs + ")")
|
||||
|
||||
@app.route("/backendpoc/error<int:ecode>")
|
||||
def error(ecode):
|
||||
if ecode == 900:
|
||||
@ -44,11 +85,13 @@ def error(ecode):
|
||||
|
||||
|
||||
@app.route("/backendpoc/rapla")
|
||||
@login_required
|
||||
def chooseRaplas():
|
||||
r = getRaplas()
|
||||
return render_template("rapla.html", raplas=r)
|
||||
|
||||
|
||||
@login_required
|
||||
@app.route("/backendpoc/plan", methods=["POST"])
|
||||
def getRapla():
|
||||
file = str(request.form.get("file"))
|
||||
@ -56,10 +99,14 @@ def getRapla():
|
||||
if file == url == "None":
|
||||
return redirect(url_for("chooseRaplas"))
|
||||
if file != "None":
|
||||
User.query.filter_by(id=current_user.id).first().kurs = file[5:-5]
|
||||
db.session.commit()
|
||||
return send_file("calendars/" + file)
|
||||
elif url != "None":
|
||||
file = getNewRapla(url)
|
||||
if type(file) is not int:
|
||||
User.query.filter_by(id=current_user.id).first().kurs = file[5:-5]
|
||||
db.session.commit()
|
||||
return send_file("calendars/" + file)
|
||||
else:
|
||||
return redirect(url_for("error", ecode=file + 900))
|
||||
@ -67,9 +114,9 @@ def getRapla():
|
||||
|
||||
|
||||
@app.route("/backendpoc/log-in")
|
||||
def login(ecode: int = None):
|
||||
if ecode:
|
||||
print(ecode)
|
||||
def login(code: int = None):
|
||||
if code:
|
||||
print(code)
|
||||
return render_template("login.html")
|
||||
|
||||
|
||||
@ -77,32 +124,65 @@ def login(ecode: int = None):
|
||||
def login_post():
|
||||
email = request.form.get("email")
|
||||
password = request.form.get("password")
|
||||
n = request.args.get("next")
|
||||
if n:
|
||||
success = redirect(n)
|
||||
else:
|
||||
success = redirect(url_for("welcome"))
|
||||
|
||||
user = User.query.filter_by(email=email).first()
|
||||
|
||||
if user:
|
||||
dualis = Dualis.query.filter_by(uid=user.id).first()
|
||||
if check_password_hash(user.password, password):
|
||||
return redirect(url_for("index"))
|
||||
if not dualisauth.checkLifetime(dualis.token_created):
|
||||
new_token = dualisauth.checkUser(email, password)
|
||||
dualis.token = new_token[0]
|
||||
dualis.cookie = requesthelpers.getCookie(new_token[1].cookies)
|
||||
dualis.token_created = time.time()
|
||||
db.session.commit()
|
||||
login_user(user)
|
||||
return success
|
||||
else:
|
||||
t = dualisauth.checkUser(email, password)
|
||||
if t == -2:
|
||||
return redirect(url_for("login", ecode=-2))
|
||||
if t[0] == -2:
|
||||
return redirect(url_for("login", code=-2))
|
||||
else:
|
||||
user.password = generate_password_hash(password, method="pbkdf2:sha256")
|
||||
dualis.token = t[0]
|
||||
dualis.cookie = requesthelpers.getCookie(t[1].cookies)
|
||||
dualis.token_created = time.time()
|
||||
db.session.commit()
|
||||
return redirect(url_for("index"))
|
||||
login_user(user)
|
||||
return success
|
||||
|
||||
t = dualisauth.checkUser(email, password)
|
||||
if t == -2:
|
||||
return redirect(url_for("login", ecode=-2))
|
||||
if t[0] == -2:
|
||||
return redirect(url_for("login", code=-2))
|
||||
|
||||
hashid = int(hashlib.sha1(email.encode("utf-8")).hexdigest(), 16) % (10 ** 12)
|
||||
hashid = int(hashlib.sha1(email.encode("utf-8")).hexdigest(), 16) % (10 ** 8)
|
||||
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()
|
||||
pname = email.find(".") + 1
|
||||
ename = min(email[pname:].find("."), email[pname:].find("@"))
|
||||
name = email[pname:pname + ename].capitalize()
|
||||
|
||||
return redirect(url_for("index"))
|
||||
new_user = User(email=email, password=hashpw, name=name, id=hashid)
|
||||
db.session.add(new_user)
|
||||
|
||||
cookie = requesthelpers.getCookie(t[1].cookies)
|
||||
|
||||
new_dualis = Dualis(uid=hashid, token=t[0], token_created=int(time.time()), cookie=cookie)
|
||||
|
||||
db.session.add(new_dualis)
|
||||
db.session.commit()
|
||||
login_user(new_user)
|
||||
return success
|
||||
|
||||
|
||||
@app.route("/backendpoc/log-out")
|
||||
def logout():
|
||||
logout_user()
|
||||
return redirect(url_for("login", code=1))
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
|
||||
Reference in New Issue
Block a user