setup login

This commit is contained in:
Sorrel Bri 2019-10-06 15:39:22 -07:00
parent 5b17dedcad
commit 8dcb80e456
2 changed files with 26 additions and 11 deletions

View file

@ -1,5 +1,4 @@
from flask import Blueprint, request, jsonify, session from flask import Blueprint, request, jsonify, session
from database import db from database import db
from models.User import User from models.User import User
@ -13,20 +12,14 @@ def auth_signup():
user = User.query.filter_by(username=data.get('username')).first() user = User.query.filter_by(username=data.get('username')).first()
if not user: if not user:
try: try:
print('getting here 1')
print(data)
user = User( user = User(
username = data['username'], username = data['username'],
email = data['email'], email = data['email'],
password = data['password'], password = data['password'],
) )
print('getting here 2')
db.session.add(user) db.session.add(user)
print('wtf')
db.session.commit() db.session.commit()
print('user')
auth_token = user.encode_auth_token(user.id) auth_token = user.encode_auth_token(user.id)
print('getting here 4')
response = { response = {
'status': 'success', 'status': 'success',
'message': 'Succesfully registered.', 'message': 'Succesfully registered.',
@ -54,5 +47,26 @@ def auth_signup():
@auth.route('/login', methods=['POST']) @auth.route('/login', methods=['POST'])
def auth_login(): def auth_login():
response = {"message": "login post"} # get the post data
return jsonify(response) data = request.get_json()
try:
# fetch the user data
print('getting here')
user = User.query.filter_by(email=data['email']).first()
print(user.username)
auth_token = user.encode_auth_token(user.id)
print(auth_token)
if auth_token:
response = {
'status': 'success',
'message': 'Successfully logged in.',
'auth_token': auth_token.decode()
}
return jsonify(response), 200
except Exception as e:
print(e)
response = {
'status': 'fail',
'message': 'Try again'
}
return jsonify(response), 500

View file

@ -6,6 +6,7 @@ import datetime
import enum import enum
import json import json
import jwt import jwt
import os
class Ranks(enum.Enum): # with minimal Elo rating class Ranks(enum.Enum): # with minimal Elo rating
D7 = "Seven Dan" # Elo 2700+ D7 = "Seven Dan" # Elo 2700+
@ -85,7 +86,7 @@ class User(db.Model):
} }
return jwt.encode( return jwt.encode(
payload, payload,
app.config.get('SECRET_KEY'), os.environ.get('SECRET_KEY'),
algorithm='HS256' algorithm='HS256'
) )
except Exception as e: except Exception as e:
@ -99,7 +100,7 @@ class User(db.Model):
:return: integer|string :return: integer|string
""" """
try: try:
payload = jwt.decode(auth_token, app.config.get('SECRET_KEY')) payload = jwt.decode(auth_token, os.environ.get('SECRET_KEY'))
return payload['sub'] return payload['sub']
except jwt.ExpiredSignatureError: except jwt.ExpiredSignatureError:
return 'Signature expired. Please log in again.' return 'Signature expired. Please log in again.'