21 lines
626 B
Python
21 lines
626 B
Python
import os
|
|
|
|
from beanie import init_beanie
|
|
from motor.motor_asyncio import AsyncIOMotorClient
|
|
|
|
from hub.user import User
|
|
from hub.auth import AccessToken
|
|
from hub.firm import Firm
|
|
|
|
|
|
MONGO_USERNAME = os.getenv("MONGO_INITDB_ROOT_USERNAME")
|
|
MONGO_PASSWORD = os.getenv("MONGO_INITDB_ROOT_PASSWORD")
|
|
|
|
DATABASE_URL = f"mongodb://{MONGO_USERNAME}:{MONGO_PASSWORD}@mongo:27017"
|
|
|
|
async def init_db():
|
|
client = AsyncIOMotorClient(DATABASE_URL, uuidRepresentation="standard")
|
|
|
|
await init_beanie(database=client.hub,
|
|
document_models=[User, AccessToken, Firm],
|
|
allow_index_dropping=True) |