mirror of
https://github.com/carrotquest/django-clickhouse.git
synced 2025-02-12 00:40:34 +03:00
50 lines
1.6 KiB
Python
50 lines
1.6 KiB
Python
from infi.clickhouse_orm.database import Database as InfiDatabase, ServerError
|
|
from infi.clickhouse_orm.migrations import MigrationHistory
|
|
|
|
from .configuration import config
|
|
from .exceptions import DBAliasError
|
|
|
|
|
|
class Database(InfiDatabase):
|
|
def __init__(self, **kwargs):
|
|
infi_kwargs = {
|
|
k: kwargs[k]
|
|
for k in ('db_name', 'db_url', 'username', 'password', 'readonly', 'autocreate')
|
|
if k in kwargs
|
|
}
|
|
super(Database, self).__init__(**infi_kwargs)
|
|
|
|
def drop_database(self):
|
|
# BUG fix https://github.com/Infinidat/infi.clickhouse_orm/issues/89
|
|
super(Database, self).drop_database()
|
|
self.db_exists = False
|
|
|
|
def migrate(self, migrations_package_name, up_to=9999):
|
|
raise NotImplementedError('This method is not supported by django-clickhouse.'
|
|
' Use django_clickhouse.migrations module instead.')
|
|
|
|
def _get_applied_migrations(self, migrations_package_name):
|
|
raise NotImplementedError("This method is not supported by django_clickhouse.")
|
|
|
|
|
|
class ConnectionProxy:
|
|
_connections = {}
|
|
|
|
def get_connection(self, alias):
|
|
if alias is None:
|
|
alias = config.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()
|