fix(user): record user.id instead of user_id

Normal-OJ · Nov 4, 2024 · 9994e6a · 9994e6a
1 parent 934d563
commit 9994e6a
Showing 1 changed file with 10 additions and 4 deletions.
diff --git a/mongo/user.py b/mongo/user.py
@@ -129,11 +129,17 @@ def login(cls, username, password, ip_addr):
         user_id = hash_id(user.username, password)
         if (compare_digest(user.user_id, user_id)
                 or compare_digest(user.user_id2, user_id)):
-            engine.LoginRecords(user_id=user_id, ip_addr=ip_addr,
-                                success=True).save(force_insert=True)
+            engine.LoginRecords(
+                user_id=user.id,
+                ip_addr=ip_addr,
+                success=True,
+            ).save(force_insert=True)
             return user
-        engine.LoginRecords(user_id=user_id,
-                            ip_addr=ip_addr).save(force_insert=True)
+        engine.LoginRecords(
+            user_id=user.id,
+            ip_addr=ip_addr,
+            success=False,
+        ).save(force_insert=True)
         raise engine.DoesNotExist
 
     @classmethod