mongomantic.core.database

Complexity

Total Complexity

4

Size/Duplication

Total Lines	28
Duplicated Lines	0 %

Importance

Changes

0

2 Functions

Rating	Name	Duplication	Size	Complexity
A	connect()	0	11	3
A	disconnect()	0	2	1

# # Package # #
from pymongo import MongoClient
from pymongo.database import Database

__all__ = ["MongomanticClient"]


class MongomanticClient:
    client: MongoClient = None
    db: Database = None


def connect(uri: str, database: str, mock: bool = False) -> None:
    if mock:
        try:
            import mongomock
        except ImportError:
            raise RuntimeError("Mongomock needs to be installed for mocking a connection")
        MongomanticClient.client = mongomock.MongoClient(uri)
    else:
        MongomanticClient.client = MongoClient(uri)

    MongomanticClient.db = MongomanticClient.client.__getattr__(database)


def disconnect() -> None:
    MongomanticClient.client.close()