black 格式化代码

Author: EITS403

app/apis/sqlmap.py

@@ -70,6 +70,7 @@ async def list_tasks():
         ],
     }
 
+
 @router.get("/tasks/{task_id}")
 async def task_status(task_id: str):
     r = requests.get(f"{SQLMAP_API}/scan/{task_id}/status", auth=AUTH)

app/core/sqlmap_task.py

@@ -2,6 +2,7 @@
 from app.models.sqlmap_result import SqlmapScanPayload
 from sqlalchemy import select
 
+
 async def task_add(
     task_id: int,
     scan_url: str,
@@ -20,9 +21,10 @@ async def task_add(
         session.add(task)
         await session.commit()
 
+
 async def list_tasks():
     async with AsyncSessionLocal() as session:
         result = await session.execute(
             select(SqlmapScanPayload).order_by(SqlmapScanPayload.created_at.desc())
         )
-        return result.scalars().all()
+        return result.scalars().all()

app/models/sqlmap_result.py

@@ -1,19 +1,13 @@
 from datetime import datetime
 
-from sqlalchemy import (
-    String,
-    Boolean,
-    DateTime,
-    Text,
-    Integer,
-    Enum
-)
+from sqlalchemy import String, Boolean, DateTime, Text, Integer, Enum
 from sqlalchemy.dialects.postgresql import JSONB
 from sqlalchemy.orm import Mapped, mapped_column, declarative_base
 import enum
 
 Base = declarative_base()
 
+
 class ScanStatus(str, enum.Enum):
     pending = "pending"
     running = "running"