import motor.motor_asyncio

from beanie import init_beanie

from .user import User, AccessToken

DB_PASSWORD = "IBO3eber0mdw2R9pnInLdtFykQFY2f06"
DATABASE_URL = f"mongodb://root:{DB_PASSWORD}@mongo:27017/"

client = motor.motor_asyncio.AsyncIOMotorClient(
    DATABASE_URL, uuidRepresentation="standard"
)

async def init_db():
    await init_beanie(database=client.core,
                      document_models=[User, AccessToken, ], # Entity, ContractTemplate, ProvisionTemplate, ContractDraft, Contract,
                      allow_index_dropping=True)


async def stop_db():
    client.close()

def get_db_client():
    yield client