Fix

loon/mqtt/handlers/change_password_request.py

@@ -1,5 +1,6 @@
 import json
 
+from sqlalchemy.exc import IntegrityError
 from sqlmodel import Session, select
 
 from loon.web.db import engine
@@ -23,11 +24,18 @@ async def change_password_handler(client, userdata, msg, data):
                            json.dumps({"success": False, "error": "The user is not registered!"}))
             return
 
-    with Session(engine) as session:
         user = session.exec(
             select(User).where(User.uuid == uuid)
         ).first()
         user.password = hash_password(password)
+
+        try:
             session.commit()
+        except IntegrityError:
+            session.rollback()
+            client.publish(f"loon/register/{uuid}/response",
+                           json.dumps({"success": False, "error": "The user is not registered!"}))
+
+            return
 
     client.publish(f"loon/register/{uuid}/response", json.dumps({"success": True, "error": None}))

loon/mqtt/handlers/register_request.py

@@ -1,5 +1,6 @@
 import json
 
+from sqlalchemy.exc import IntegrityError
 from sqlmodel import Session, select
 
 from loon.web.db import engine
@@ -24,9 +25,15 @@ async def register_request_handler(client, userdata, msg, data):
                            json.dumps({"success": False, "error": "The user is already registered!"}))
             return
 
-    with Session(engine) as session:
         user = User(uuid=uuid, password=hash_password(password), internal_username=internal_username, admin=False)
         session.add(user)
+        try:
             session.commit()
+        except IntegrityError:
+            session.rollback()
+            client.publish(f"loon/register/{uuid}/response",
+                           json.dumps({"success": False, "error": "The user is already registered!"}))
+
+            return
 
     client.publish(f"loon/register/{uuid}/response", json.dumps({"success": True, "error": None}))