"""DB connection helper - shared across modules"""
import os
from motor.motor_asyncio import AsyncIOMotorClient

_MONGO_URL = os.environ["MONGO_URL"]
_DB_NAME = os.environ["DB_NAME"]

_client = AsyncIOMotorClient(_MONGO_URL)
_db = _client[_DB_NAME]


def get_db():
    return _db


async def close_db():
    _client.close()