added unique id

ChitChat/common/utils.py

@@ -20,7 +20,7 @@ def getChatHistory(user):
 
 def saveChatHistory(user, chat_history_ids):
     location = current_app.config.get('SAVE_FOLDER')
-    file_name = location + str(user.id) + '.pt'
+    file_name = location + user.user_id + '.pt'
     if chat_history_ids.shape[-1] > 100:
         user.history = None 
         db.session.commit()

ChitChat/models.py

@@ -1,8 +1,10 @@
 from ChitChat import db 
 from datetime import datetime 
+import uuid
 
 class ChatHistory(db.Model):
     id = db.Column(db.Integer, primary_key = True)
+    user_id = db.Column(db.String(), unique = True, nullable = False, default = str(uuid.uuid4()))
     username  = db.Column(db.String(10), unique = True, nullable = False)
     email = db.Column(db.String(100), nullable = False)
     password = db.Column(db.String(60), nullable = False)

ChitChat/resources/routes.py

@@ -16,11 +16,11 @@ class UserLogin(Resource):
         user = ChatHistory.query.filter_by(email = userInfo['email']).first()
 
         if user is None:
-            return {'status' : "Account doesn't exist", 'user_id' : -1}
+            return {'status' : "Account doesn't exist", 'user_id' : None}
         elif bcrypt.check_password_hash(pw_hash = user.password, password = userInfo['password']):
-            return {'status' : "login successful", 'user_id' : user.id}
+            return {'status' : "login successful", 'user_id' : user.user_id}
         else:
-            return {'status' : "Invalid password", 'user_id' : -1}
+            return {'status' : "Invalid password", 'user_id' : None}
 api.add_resource(UserLogin, '/login/')
 
 class RegisterUser(Resource):
@@ -43,7 +43,7 @@ api.add_resource(RegisterUser, '/register/')
 
 class ChatBot(Resource):
     def post(self, user_id):
-        user = ChatHistory.query.filter_by(id = user_id).first()
+        user = ChatHistory.query.filter_by(user_id = user_id).first()
         userInput = request.json['user']
         if user is None:
             return {'error' : "User doesn't exist"}, 400