fix: persist avatar data and uploads

2026-08-19 14:44:58 +08:00
parent c5bfa47a23
commit 68ea87e1b2
4 changed files with 14 additions and 3 deletions
@@ -5,10 +5,11 @@ from sqlalchemy.orm import sessionmaker, declarative_base, Session

 BASE_DIR = os.path.dirname(os.path.abspath(__file__))
 DB_FILE = os.path.join(BASE_DIR, "avatar.db")
+DATABASE_URL = os.getenv("DATABASE_URL", f"sqlite:///{DB_FILE}")

 engine = create_engine(
-    f"sqlite:///{DB_FILE}",
-    connect_args={"check_same_thread": False},
+    DATABASE_URL,
+    connect_args={"check_same_thread": False} if DATABASE_URL.startswith("sqlite:") else {},
 )
 SessionLocal = sessionmaker(bind=engine, autoflush=False, expire_on_commit=False)
 Base = declarative_base()