Skip to content

Commit 3d18c2e

Browse files
author
App Generator
committed
Fixes
1 parent 0467abd commit 3d18c2e

File tree

2 files changed

+24
-5
lines changed

2 files changed

+24
-5
lines changed

api/models.py

Lines changed: 15 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -5,12 +5,14 @@
55

66
from datetime import datetime
77

8+
import json
9+
from json import JSONEncoder
10+
811
from werkzeug.security import generate_password_hash, check_password_hash
912
from flask_sqlalchemy import SQLAlchemy
1013

1114
db = SQLAlchemy()
1215

13-
1416
class Users(db.Model):
1517
id = db.Column(db.Integer(), primary_key=True)
1618
username = db.Column(db.String(32), nullable=False)
@@ -45,6 +47,18 @@ def get_by_id(cls, id):
4547
def get_by_email(cls, email):
4648
return cls.query.filter_by(email=email).first()
4749

50+
def toDICT(self):
51+
52+
cls_dict = {}
53+
cls_dict['id'] = self.id
54+
cls_dict['username'] = self.username
55+
cls_dict['email'] = self.email
56+
57+
return cls_dict
58+
59+
def toJSON(self):
60+
61+
return json.dumps( self.toDICT() )
4862

4963
class JWTTokenBlocklist(db.Model):
5064
id = db.Column(db.Integer(), primary_key=True)

api/routes.py

Lines changed: 9 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -5,6 +5,8 @@
55

66
from datetime import datetime, timezone
77

8+
import json
9+
810
from flask import request
911
from flask_restx import Api, Resource, fields, abort
1012
from flask_jwt_extended import JWTManager, jwt_required, get_jwt, create_access_token, get_jwt_identity
@@ -48,7 +50,7 @@ def check_if_token_revoked(jwt_header, jwt_payload):
4850
"""
4951

5052

51-
@rest_api.route('/api/users/signup')
53+
@rest_api.route('/api/users/register')
5254
class Register(Resource):
5355
"""
5456
Creates a new user by taking 'signup_model' input
@@ -66,15 +68,16 @@ def post(self):
6668
user_exists = Users.get_by_email(_email)
6769
if user_exists:
6870
return {"success": False,
69-
"msg": "Sorry. This email already exists."}, 400
71+
"msg": "Email already exists"}, 400
7072

7173
new_user = Users(username=_username, email=_email)
7274

7375
new_user.set_password(_password)
7476
new_user.save()
7577

7678
return {"success": True,
77-
"msg": "User with (%s, %s) created successfully!" % (_username, _email)}, 201
79+
"userID" : new_user.id,
80+
"msg": "The user was succesfully registered"}, 200
7881

7982

8083
@rest_api.route('/api/users/login')
@@ -104,7 +107,9 @@ def post(self):
104107
# create access token uwing JWT
105108
access_token = create_access_token(identity=_email)
106109

107-
return {"access_token": access_token}, 200
110+
return {"success": True,
111+
"token": access_token,
112+
"user" : user_exists.toJSON() }, 200
108113

109114

110115
@rest_api.route('/api/users/edit')

0 commit comments

Comments
 (0)