commit
0028c40825
3 changed files with 68 additions and 41 deletions
3
app.py
3
app.py
|
@ -10,7 +10,8 @@ from flask_cors import CORS
|
|||
|
||||
def create_app():
|
||||
app = Flask(__name__)
|
||||
CORS(app, origins="http://localhost:3004")
|
||||
CORS(app, resources={r"/api/*": {"origins": "http://localhost:3000"},
|
||||
r"/auth/*": {"origins": "http://localhost:3000"}})
|
||||
app.config['CORS_HEADERS'] = 'Content-Type'
|
||||
app.config.from_object(DevelopmentConfig)
|
||||
db.init_app(app)
|
||||
|
|
77
auth/auth.py
77
auth/auth.py
|
@ -1,5 +1,4 @@
|
|||
from flask import Blueprint, request, jsonify, session
|
||||
|
||||
from database import db
|
||||
from models.User import User
|
||||
|
||||
|
@ -10,33 +9,36 @@ def auth_signup():
|
|||
data = request.get_json()
|
||||
user = User.query.filter_by(email=data.get('email')).first()
|
||||
if not user:
|
||||
try:
|
||||
print('getting here 1')
|
||||
user = User(
|
||||
email = data['email'],
|
||||
password = data['password'],
|
||||
)
|
||||
print('getting here 2')
|
||||
db.session.add(user)
|
||||
print('wtf')
|
||||
db.session.commit()
|
||||
print('user')
|
||||
auth_token = user.encode_auth_token(user.id)
|
||||
print('getting here 4')
|
||||
response = {
|
||||
'status': 'success',
|
||||
'message': 'Succesfully registered.',
|
||||
'auth_token': auth_token.decode()
|
||||
}
|
||||
return jsonify(response), 201
|
||||
except Exception as e:
|
||||
print(e.__dict__)
|
||||
user = User.query.filter_by(username=data.get('username')).first()
|
||||
if not user:
|
||||
try:
|
||||
user = User(
|
||||
username = data['username'],
|
||||
email = data['email'],
|
||||
password = data['password'],
|
||||
)
|
||||
db.session.add(user)
|
||||
db.session.commit()
|
||||
auth_token = user.encode_auth_token(user.id)
|
||||
response = {
|
||||
'status': 'success',
|
||||
'message': 'Succesfully registered.',
|
||||
'auth_token': auth_token.decode()
|
||||
}
|
||||
return jsonify(response), 201
|
||||
except Exception as e:
|
||||
print(e.__dict__)
|
||||
response = {
|
||||
'status': 'fail',
|
||||
'message': 'There was an error. Please try again.'
|
||||
}
|
||||
return jsonify(response), 401
|
||||
else: # username is taken
|
||||
response = {
|
||||
'status': 'fail',
|
||||
'message': 'There was an error. Please try again.'
|
||||
'message': 'User already exists. Please login.'
|
||||
}
|
||||
return jsonify(response), 401
|
||||
else:
|
||||
else: # email is taken
|
||||
response = {
|
||||
'status': 'fail',
|
||||
'message': 'User already exists. Please login.'
|
||||
|
@ -45,5 +47,26 @@ def auth_signup():
|
|||
|
||||
@auth.route('/login', methods=['POST'])
|
||||
def auth_login():
|
||||
response = {"message": "login post"}
|
||||
return jsonify(response)
|
||||
# get the post data
|
||||
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
|
|
@ -1,9 +1,12 @@
|
|||
from database import db, ma
|
||||
from marshmallow import fields
|
||||
from app import bcrypt
|
||||
from configuration import config
|
||||
import datetime
|
||||
import enum
|
||||
import json
|
||||
import jwt
|
||||
import os
|
||||
|
||||
class Ranks(enum.Enum): # with minimal Elo rating
|
||||
D7 = "Seven Dan" # Elo 2700+
|
||||
|
@ -58,7 +61,9 @@ class User(db.Model):
|
|||
elo = db.Column(db.Integer)
|
||||
rank_certainty = db.Column(db.Boolean, nullable=False, default=False)
|
||||
|
||||
def __init__(self, username, email, password, rank='RU', admin=False):
|
||||
def __init__(self, username, email, password, rank=Ranks.K1, admin=False):
|
||||
print(rank)
|
||||
print(Ranks)
|
||||
self.username = username
|
||||
self.email = email
|
||||
self.password = bcrypt.generate_password_hash(
|
||||
|
@ -81,7 +86,7 @@ class User(db.Model):
|
|||
}
|
||||
return jwt.encode(
|
||||
payload,
|
||||
app.config.get('SECRET_KEY'),
|
||||
os.environ.get('SECRET_KEY'),
|
||||
algorithm='HS256'
|
||||
)
|
||||
except Exception as e:
|
||||
|
@ -95,7 +100,7 @@ class User(db.Model):
|
|||
:return: integer|string
|
||||
"""
|
||||
try:
|
||||
payload = jwt.decode(auth_token, app.config.get('SECRET_KEY'))
|
||||
payload = jwt.decode(auth_token, os.environ.get('SECRET_KEY'))
|
||||
return payload['sub']
|
||||
except jwt.ExpiredSignatureError:
|
||||
return 'Signature expired. Please log in again.'
|
||||
|
@ -103,16 +108,14 @@ class User(db.Model):
|
|||
return 'Invalid token. Please log in again.'
|
||||
|
||||
class UserSchema(ma.ModelSchema):
|
||||
class Meta:
|
||||
fields = (
|
||||
'id',
|
||||
'username',
|
||||
'email',
|
||||
'registered_on',
|
||||
'rank',
|
||||
'rank_certainty',
|
||||
'elo'
|
||||
)
|
||||
id = fields.Int()
|
||||
username = fields.Str()
|
||||
email = fields.Str()
|
||||
registered_on = fields.Date()
|
||||
rank = fields.Str()
|
||||
rank_certainty = fields.Bool()
|
||||
elo = fields.Int()
|
||||
|
||||
|
||||
user_schema = UserSchema()
|
||||
users_schema = UserSchema(many=True)
|
Loading…
Reference in a new issue