Fixed a bug with singup due to changing the way the User class works.

This commit is contained in:
Alexander Huddleston 2020-03-07 20:25:46 -06:00
parent a83507bcf9
commit 4f4b87e6ca
3 changed files with 5 additions and 40 deletions

View file

@ -1,8 +1,7 @@
from flask import Flask, Response
from flask_login import LoginManager, login_required
from flask import Flask
from os import urandom
app = Flask(__name__)
app.secret_key = urandom(12).hex()
from . import views
from . import uwume

View file

@ -1,34 +0,0 @@
from flask import Response
from flask_login import LoginManager, login_required
try:
from .lib.classUser import User
except:
from lib.classUser import User
@login_manager.request_loader
def load_user(request):
token = request.headers.get('Authorization')
if(token is None):
token = request.args.get('token')
if(token is not None):
username, password = token.split(':') # naive token
user_entry = User.get(username)
if(user_entry is not None):
user = User(user_entry[0], user_entry[1])
if(user.password == password):
return user
return None
@app.route("/", methods=["GET"])
def index():
return Response(response="Hello World!", status=200)
@app.route("/protected/", methods=["GET"])
@login_required
def protected():
return Response(response="Hello Protected World!", status=200)

View file

@ -1,9 +1,11 @@
try:
from .lib.helpers import get_static_paths, get_content_text
from .lib.classUser import User
from .lib.classUwuCounter import UwuCounter
except:
from lib.classUser import User
from lib.helpers import get_static_paths, get_content_text
from lib.classUwuCounter import UwuCounter
from . import app
from flask import Flask, render_template, redirect, url_for, request, flash
from flask_login import LoginManager, UserMixin, login_required, login_user, current_user, logout_user
@ -90,9 +92,7 @@ def signup():
username = request.form['username']
password = request.form['password']
if(not username in User.user_database.keys()):
userClass = User(username, password)
User.user_database[username] = (username, password)
User.user_database[username] = (username, password, UwuCounter(0))
return redirect('/')
else:
flash('It looks like a user already exists with that name.')