set default database for non model migrations

This commit is contained in:
amirhgh 2022-01-04 13:53:49 +03:30
parent 06f5929c92
commit 5bcd9cc0b1

View File

@ -1,4 +1,5 @@
from infi import clickhouse_orm
from .configuration import config
class HintParamMixin:
@ -7,6 +8,12 @@ class HintParamMixin:
hints = dict()
self.hints = hints
super(HintParamMixin, self).__init__(*args, **kwargs)
if not any([
isinstance(self, clickhouse_orm.ModelOperation),
'model' not in hints.keys(),
'force_migrate_on_databases' not in hints.keys()
]):
self.hints['force_migrate_on_databases'] = (config.DEFAULT_DB_ALIAS,)
class CreateTable(HintParamMixin, clickhouse_orm.CreateTable):