let black do it's magic

core/data/schema/user.py

@@ -17,72 +17,81 @@ aime_user = Table(
     Column("username", String(25), unique=True),
     Column("email", String(255), unique=True),
     Column("password", String(255)),
-    Column("permissions", Integer),    
+    Column("permissions", Integer),
     Column("created_date", TIMESTAMP, server_default=func.now()),
     Column("last_login_date", TIMESTAMP, onupdate=func.now()),
     Column("suspend_expire_time", TIMESTAMP),
-    mysql_charset='utf8mb4'
+    mysql_charset="utf8mb4",
 )
 
+
 class PermissionBits(Enum):
     PermUser = 1
     PermMod = 2
     PermSysAdmin = 4
 
+
 class UserData(BaseData):
-    def create_user(self, id: int = None, username: str = None, email: str = None, password: str = None, permission: int = 1) -> Optional[int]:
+    def create_user(
+        self,
+        id: int = None,
+        username: str = None,
+        email: str = None,
+        password: str = None,
+        permission: int = 1,
+    ) -> Optional[int]:
         if id is None:
             sql = insert(aime_user).values(
-                username=username, 
-                email=email, 
-                password=password, 
-                permissions=permission
+                username=username,
+                email=email,
+                password=password,
+                permissions=permission,
             )
         else:
             sql = insert(aime_user).values(
-                id=id, 
-                username=username, 
-                email=email, 
-                password=password, 
-                permissions=permission
+                id=id,
+                username=username,
+                email=email,
+                password=password,
+                permissions=permission,
             )
 
         conflict = sql.on_duplicate_key_update(
-            username=username, 
-            email=email, 
-            password=password, 
-            permissions=permission
+            username=username, email=email, password=password, permissions=permission
         )
-        
+
         result = self.execute(conflict)
-        if result is None: return None
+        if result is None:
+            return None
         return result.lastrowid
-    
+
     def get_user(self, user_id: int) -> Optional[Row]:
         sql = select(aime_user).where(aime_user.c.id == user_id)
         result = self.execute(sql)
-        if result is None: return False
+        if result is None:
+            return False
         return result.fetchone()
-    
+
     def check_password(self, user_id: int, passwd: bytes = None) -> bool:
         usr = self.get_user(user_id)
-        if usr is None: return False
-
-        if usr['password'] is None:
+        if usr is None:
             return False
 
-        return bcrypt.checkpw(passwd, usr['password'].encode())
+        if usr["password"] is None:
+            return False
+
+        return bcrypt.checkpw(passwd, usr["password"].encode())
 
     def reset_autoincrement(self, ai_value: int) -> None:
         # ALTER TABLE isn't in sqlalchemy so we do this the ugly way
         sql = f"ALTER TABLE aime_user AUTO_INCREMENT={ai_value}"
         self.execute(sql)
-    
+
     def delete_user(self, user_id: int) -> None:
         sql = aime_user.delete(aime_user.c.id == user_id)
 
         result = self.execute(sql)
-        if result is None: 
+        if result is None:
             self.logger.error(f"Failed to delete user with id {user_id}")
 
     def get_unregistered_users(self) -> List[Row]:
@@ -92,6 +101,6 @@ class UserData(BaseData):
         sql = select(aime_user).where(aime_user.c.password == None)
 
         result = self.execute(sql)
-        if result is None: 
+        if result is None:
             return None
-        return result.fetchall()
+        return result.fetchall()