django-clickhouse/src/django_clickhouse/db_clients.py

from infi.clickhouse_orm.database import Database

from .exceptions import DBAliasError
from .configuration import config


DEFAULT_DB_ALIAS = 'default'


class ConnectionProxy:
    _connections = {}

    def get_connection(self, alias):
        if alias is None:
            alias = DEFAULT_DB_ALIAS

        if alias not in self._connections:
            if alias not in config.DATABASES:
                raise DBAliasError(alias)

            self._connections[alias] = Database(**config.DATABASES[alias])

        return self._connections[alias]

    def __getitem__(self, item):
        return self.get_connection(item)


connections = ConnectionProxy()