Cleanup in User code

app/services/user_service.py

@@ -62,7 +62,7 @@ class UserService:
 			print(f"Error: {e}")
 			return {"Failed": "Failed to insert into database. Username or email are likely in use already"}, 500
 
-		send_mail("register", email)
+		UserService.__send_email("register", email=email)
 
 		return {"Success": "User created successfully"}, 200
 
@@ -98,7 +98,7 @@ class UserService:
 
 				token = create_access_token(identity=user_id, expires_delta=expire)
 
-				send_mail("login", email)
+				UserService.__send_email("login", email=email)
 
 				return {"token": token}, 200
 
@@ -106,7 +106,7 @@ class UserService:
 			return {"Failed": f"Failed to login. Error: {e}"}, 500
 
 	@staticmethod
-	def logout(jti, exp, user_id) -> Tuple[Union[dict, str], int]:
+	def logout(jwt_token, user_id) -> Tuple[Union[dict, str], int]:
 		"""
 		Logs out a user by invalidating the provided JWT.
 
@@ -118,6 +118,9 @@ class UserService:
 		:rtype: Tuple[Union[dict, str], int]
 		"""
 
+		jti = jwt['jti']
+		exp = jwt['exp']
+		
 		UserService.__invalidate_token(jti, exp)
 		UserService.__send_email("logout", id=user_id)
 
@@ -234,7 +237,7 @@ class UserService:
 					cursor.execute("select email from user where username = %s", (username,))
 					result = cursor.fetchone()
 					email = result['email']
-					send_mail("logout", email)
+					send_mail(message, email)
 
 			except Error as e:
 				return {"Failed": f"Failed to fetch some data. Error: {e}"}, 500
@@ -246,7 +249,7 @@ class UserService:
 					cursor.execute("select email from user where id = %s", (id,))
 					result = cursor.fetchone()
 					email = result['email']
-					send_mail("logout", email)
+					send_mail(message, email)
 
 			except Error as e:
 				return {"Failed": f"Failed to fetch some data. Error: {e}"}, 500