style: normalize python formatting gates

2026-06-21 06:52:19 +08:00
parent 0809a079c5
commit bfbaddbd9a
129 changed files with 3024 additions and 2485 deletions
@@ -1,7 +1,9 @@
 """
 PostgreSQL Workspace Repository 实现
 """
+
 from typing import Optional
+
 import psycopg2
 from psycopg2.extras import RealDictCursor

@@ -11,21 +13,23 @@ from packages.ports.workspace_repository import WorkspaceRepository

 class PostgresWorkspaceRepository(WorkspaceRepository):
    """Workspace 仓储 PostgreSQL 实现"""
-    
+
    def __init__(self, connection_string: str):
        self.connection_string = connection_string
-    
+
    def _get_connection(self):
        """获取数据库连接（使用连接池）"""
        from packages.adapters.postgres.connection_pool import PooledConnection
+
        return PooledConnection()
-    
+
    def save(self, workspace: Workspace) -> None:
        """保存工作空间"""
        conn = self._get_connection()
        try:
            with conn.cursor() as cur:
-                cur.execute("""
+                cur.execute(
+                    """
                    INSERT INTO workspaces (
                        id, name, owner_user_id, subscription_plan,
                        subscription_status, subscription_expires_at,
@@ -43,22 +47,24 @@ class PostgresWorkspaceRepository(WorkspaceRepository):
                        max_projects = EXCLUDED.max_projects,
                        max_storage_gb = EXCLUDED.max_storage_gb,
                        used_storage_gb = EXCLUDED.used_storage_gb
-                """, {
-                    "id": workspace.id,
-                    "name": workspace.name,
-                    "owner_user_id": workspace.owner_user_id,
-                    "subscription_plan": workspace.subscription_plan,
-                    "subscription_status": workspace.subscription_status,
-                    "subscription_expires_at": workspace.subscription_expires_at,
-                    "max_projects": workspace.max_projects,
-                    "max_storage_gb": workspace.max_storage_gb,
-                    "used_storage_gb": workspace.used_storage_gb,
-                    "created_at": workspace.created_at,
-                })
+                """,
+                    {
+                        "id": workspace.id,
+                        "name": workspace.name,
+                        "owner_user_id": workspace.owner_user_id,
+                        "subscription_plan": workspace.subscription_plan,
+                        "subscription_status": workspace.subscription_status,
+                        "subscription_expires_at": workspace.subscription_expires_at,
+                        "max_projects": workspace.max_projects,
+                        "max_storage_gb": workspace.max_storage_gb,
+                        "used_storage_gb": workspace.used_storage_gb,
+                        "created_at": workspace.created_at,
+                    },
+                )
            conn.commit()
        finally:
            conn.close()
-    
+
    def find_by_id(self, workspace_id: str) -> Optional[Workspace]:
        """根据 ID 查找工作空间"""
        conn = self._get_connection()
@@ -66,13 +72,13 @@ class PostgresWorkspaceRepository(WorkspaceRepository):
            with conn.cursor() as cur:
                cur.execute("SELECT * FROM workspaces WHERE id = %s", (workspace_id,))
                row = cur.fetchone()
-                
+
                if row:
                    return self._row_to_workspace(row)
                return None
        finally:
            conn.close()
-    
+
    def delete(self, workspace_id: str) -> bool:
        """删除工作空间"""
        conn = self._get_connection()
@@ -84,7 +90,7 @@ class PostgresWorkspaceRepository(WorkspaceRepository):
            return deleted
        finally:
            conn.close()
-    
+
    def _row_to_workspace(self, row: dict) -> Workspace:
        """将数据库行转换为 Workspace 对象"""
        return Workspace(