格式化代码

Author: EITS403

app/apis/db_health.py

@@ -1,5 +1,6 @@
 from fastapi import APIRouter
 from sqlalchemy import text
+
 from app.database.database import engine
 
 router = APIRouter(prefix="/db", tags=["Database"])

app/core/sqlmap_core.py

@@ -1,7 +1,8 @@
-from app.database.database import AsyncSessionLocal
-from app.models.sqlmap_result import SqlmapScanPayload, SqlmapScanLog, SqlmapScanResult
 from sqlalchemy import select
 
+from app.database.database import AsyncSessionLocal
+from app.models.sqlmap_result import SqlmapScanPayload, SqlmapScanLog
+
 
 async def task_add(
     task_id: int,

app/database/celery_sync_database.py

@@ -1,9 +1,9 @@
 import os
 from urllib.parse import quote_plus
 
+import dotenv
 from sqlalchemy import create_engine
 from sqlalchemy.orm import sessionmaker, declarative_base
-import dotenv
 
 dotenv.load_dotenv()
 

app/database/database.py

@@ -1,12 +1,12 @@
 import os
 from urllib.parse import quote_plus
 
+import dotenv
 from sqlalchemy.ext.asyncio import (
     create_async_engine,
     AsyncSession,
 )
 from sqlalchemy.orm import sessionmaker, declarative_base
-import dotenv
 
 dotenv.load_dotenv()
 
@@ -16,7 +16,6 @@
 database_password = os.getenv("POSTGRES_PASSWORD")
 database_name = os.getenv("POSTGRES_DB")
 
-
 # 密码 URL 编码（非常重要）
 database_password = quote_plus(database_password)
 

app/middleware/celery_app.py

@@ -1,5 +1,6 @@
-from celery import Celery
 import os
+
+from celery import Celery
 from dotenv import load_dotenv
 
 load_dotenv()
@@ -15,7 +16,6 @@
 mq_dbname = os.getenv("MQ_DBNAME")
 mq_port = os.getenv("MQ_DBPORT")
 
-
 celery_app = Celery(
     "fastapi_sqlmap",
     broker=f"amqp://{mq_username}:{mq_password}@{mq_host}:{mq_port}/{mq_dbname}",

app/models/sqlmap_result.py

@@ -1,9 +1,9 @@
+import enum
 from datetime import datetime
 
 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()
 

app/schema/sqlmap.py

@@ -1,6 +1,7 @@
-from pydantic import BaseModel, HttpUrl, Field
 from typing import Optional, List
 
+from pydantic import BaseModel, HttpUrl, Field
+
 
 class SQLMapScanRequest(BaseModel):
     target: HttpUrl = Field(..., description="扫描目标 URL")

app/tasks/sqlmap_scheduler.py

@@ -1,4 +1,5 @@
 from celery import shared_task
+
 from app.database.celery_sync_database import SessionLocal
 from app.models.sqlmap_result import SqlmapScanPayload, ScanStatus
 from app.tasks.sqlmap_worker import poll_single_sqlmap_task