chore: change DB to SQLite for Hugging Face Space deployment

Dockerfile

@@ -15,8 +15,8 @@ COPY . .
 # 暴露端口
 EXPOSE 8000
 
-# 设置环境变量，使用容器内的数据库地址
-ENV DATABASE_URL="postgresql+asyncpg://postgres:postgres@db:5432/customs_data"
+# 设置环境变量，使用 SQLite
+ENV DATABASE_URL="sqlite+aiosqlite:///./customs_data.db"
 
 # 启动命令：先运行 mock 脚本初始化库和数据，再运行巴西的测试数据，最后启动 API 服务
 CMD ["bash", "-c", "python apps/worker/run_mock.py && python apps/worker/run_brazil.py && uvicorn apps.api.main:app --host 0.0.0.0 --port 8000"]

packages/core/database.py

@@ -3,12 +3,17 @@ from sqlalchemy.orm import declarative_base
 from packages.core.config import settings
 
 # 创建异步数据库引擎
+engine_kwargs = {
+    "echo": False,
+    "future": True,
+}
+if not settings.DATABASE_URL.startswith("sqlite"):
+    engine_kwargs["pool_size"] = 20
+    engine_kwargs["max_overflow"] = 10
+
 engine = create_async_engine(
     settings.DATABASE_URL,
-    echo=False,
-    future=True,
-    pool_size=20,
-    max_overflow=10
+    **engine_kwargs
 )
 
 # 创建异步 Session 工厂

requirements.txt

@@ -3,6 +3,7 @@ uvicorn==0.27.1
 sqlalchemy==2.0.28
 asyncpg==0.29.0
 psycopg2-binary==2.9.9
+aiosqlite==0.20.0
 pydantic==2.6.3
 pydantic-settings==2.2.1
 httpx==0.27.0