39 lines
1.2 KiB
Python
39 lines
1.2 KiB
Python
import os
|
||
import certifi
|
||
|
||
from beanie import init_beanie
|
||
|
||
from app.config import settings
|
||
from app.models.user import User
|
||
from app.models.report import Report
|
||
from app.models.analysis import Analysis
|
||
from app.models.chat import ChatSession
|
||
from app.models.checkup import CheckupPlan
|
||
|
||
|
||
async def init_db():
|
||
use_mock = os.environ.get("USE_MONGOMOCK", "").lower() in ("1", "true", "yes")
|
||
|
||
if use_mock:
|
||
# 使用 mongomock 内存数据库（无需安装 MongoDB）
|
||
from mongomock_motor import AsyncMongoMockClient
|
||
client = AsyncMongoMockClient()
|
||
else:
|
||
from motor.motor_asyncio import AsyncIOMotorClient
|
||
# Only use TLS for Atlas (mongodb+srv://) connections
|
||
if settings.mongodb_url.startswith("mongodb+srv://") or "mongodb.net" in settings.mongodb_url:
|
||
client = AsyncIOMotorClient(settings.mongodb_url, tlsCAFile=certifi.where())
|
||
else:
|
||
client = AsyncIOMotorClient(settings.mongodb_url)
|
||
|
||
await init_beanie(
|
||
database=client[settings.mongodb_db_name],
|
||
document_models=[
|
||
User,
|
||
Report,
|
||
Analysis,
|
||
ChatSession,
|
||
CheckupPlan,
|
||
],
|
||
)
|