From 1f1f2d4e22707650aec063f1cbb7d8f0d470c380 Mon Sep 17 00:00:00 2001
From: Kohaku-Blueleaf <59680068+KohakuBlueleaf@users.noreply.github.com>
Date: Thu, 16 Oct 2025 01:24:58 +0800
Subject: [PATCH] Formatting

---
 scripts/db_migrations/006_invitation_multi_use.py | 10 ++++++++--
 src/kohakuhub/api/admin.py                        |  8 ++++++--
 src/kohakuhub/api/invitation/router.py            |  7 ++++++-
 src/kohakuhub/config.py                           |  4 +++-
 src/kohakuhub/db_operations.py                    |  5 ++++-
 5 files changed, 27 insertions(+), 7 deletions(-)

diff --git a/scripts/db_migrations/006_invitation_multi_use.py b/scripts/db_migrations/006_invitation_multi_use.py
index b93f25a..e4b2890 100644
--- a/scripts/db_migrations/006_invitation_multi_use.py
+++ b/scripts/db_migrations/006_invitation_multi_use.py
@@ -70,7 +70,10 @@ def migrate_sqlite():
 
     # Invitation multi-use fields
     for column, sql in [
-        ("max_usage", "ALTER TABLE invitation ADD COLUMN max_usage INTEGER DEFAULT NULL"),
+        (
+            "max_usage",
+            "ALTER TABLE invitation ADD COLUMN max_usage INTEGER DEFAULT NULL",
+        ),
         (
             "usage_count",
             "ALTER TABLE invitation ADD COLUMN usage_count INTEGER DEFAULT 0",
@@ -109,7 +112,10 @@ def migrate_postgres():
 
     # Invitation multi-use fields
     for column, sql in [
-        ("max_usage", "ALTER TABLE invitation ADD COLUMN max_usage INTEGER DEFAULT NULL"),
+        (
+            "max_usage",
+            "ALTER TABLE invitation ADD COLUMN max_usage INTEGER DEFAULT NULL",
+        ),
         (
             "usage_count",
             "ALTER TABLE invitation ADD COLUMN usage_count INTEGER DEFAULT 0",
diff --git a/src/kohakuhub/api/admin.py b/src/kohakuhub/api/admin.py
index b566412..41ce5e2 100644
--- a/src/kohakuhub/api/admin.py
+++ b/src/kohakuhub/api/admin.py
@@ -407,7 +407,9 @@ async def create_register_invitation_admin(
         # Verify organization exists
         org = Organization.get_or_none(Organization.id == request.org_id)
         if not org:
-            raise HTTPException(404, detail={"error": f"Organization not found: {request.org_id}"})
+            raise HTTPException(
+                404, detail={"error": f"Organization not found: {request.org_id}"}
+            )
 
         org_name = org.name
     else:
@@ -544,7 +546,9 @@ async def delete_invitation_admin(
 
     delete_invitation(invitation)
 
-    logger.info(f"Admin deleted invitation: {token[:8]}... (action={invitation.action})")
+    logger.info(
+        f"Admin deleted invitation: {token[:8]}... (action={invitation.action})"
+    )
 
     return {"success": True, "message": "Invitation deleted successfully"}
 
diff --git a/src/kohakuhub/api/invitation/router.py b/src/kohakuhub/api/invitation/router.py
index 3b66ca2..ac029b4 100644
--- a/src/kohakuhub/api/invitation/router.py
+++ b/src/kohakuhub/api/invitation/router.py
@@ -127,7 +127,12 @@ async def create_org_invitation(
     # Send email if SMTP enabled and email provided (async)
     if req.email:
         await asyncio.to_thread(
-            send_org_invitation_email, req.email, org.name, user.username, token, req.role
+            send_org_invitation_email,
+            req.email,
+            org.name,
+            user.username,
+            token,
+            req.role,
         )
 
     # Determine invitation type for logging
diff --git a/src/kohakuhub/config.py b/src/kohakuhub/config.py
index b810688..a9fea99 100644
--- a/src/kohakuhub/config.py
+++ b/src/kohakuhub/config.py
@@ -178,7 +178,9 @@ def load_config(path: str = None) -> Config:
                 "KOHAKU_HUB_REQUIRE_EMAIL_VERIFICATION", "false"
             ).lower()
             == "true",
-            invitation_only=os.environ.get("KOHAKU_HUB_INVITATION_ONLY", "false").lower()
+            invitation_only=os.environ.get(
+                "KOHAKU_HUB_INVITATION_ONLY", "false"
+            ).lower()
             == "true",
             session_secret=os.environ.get(
                 "KOHAKU_HUB_SESSION_SECRET", "change-me-in-production"
diff --git a/src/kohakuhub/db_operations.py b/src/kohakuhub/db_operations.py
index 2fc4b4a..84993dd 100644
--- a/src/kohakuhub/db_operations.py
+++ b/src/kohakuhub/db_operations.py
@@ -597,7 +597,10 @@ def check_invitation_available(invitation: Invitation) -> tuple[bool, str | None
     else:
         # Limited use (check count)
         if invitation.usage_count >= invitation.max_usage:
-            return False, f"Invitation has reached maximum usage limit ({invitation.max_usage})"
+            return (
+                False,
+                f"Invitation has reached maximum usage limit ({invitation.max_usage})",
+            )
 
     return True, None