carrotquest/django-clickhouse
1
1
Fork 0
mirror of https://github.com/carrotquest/django-clickhouse.git synced 2025-06-28 00:43:07 +03:00
Code Issues Packages Projects Releases Wiki Activity

Made get_database for ClickHouseModel a class method

Browse Source
This commit is contained in:
M1ha 2018-11-23 11:58:44 +05:00
parent 68914a20f0
commit a5bbc9ba00
1 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
src/django_clickhouse/clickhouse_models.py
View File

@ -51,7 +51,8 @@ class ClickHouseModel(with_metaclass(ClickHouseModelMeta, InfiModel)):
sync_database_alias = None sync_database_alias = None
sync_lock_timeout = None sync_lock_timeout = None
def get_database(self, for_write=False): @classmethod
def get_database(cls, for_write=False):
# type: (bool) -> Database # type: (bool) -> Database
""" """
Gets database for read or write purposes Gets database for read or write purposes
@ -60,9 +61,9 @@ class ClickHouseModel(with_metaclass(ClickHouseModelMeta, InfiModel)):
""" """
db_router = lazy_class_import(config.DATABASE_ROUTER)() db_router = lazy_class_import(config.DATABASE_ROUTER)()
if for_write: if for_write:
return db_router.db_for_write(self.__class__, instance=self) return db_router.db_for_write(cls)
else: else:
return db_router.db_for_read(self.__class__, instance=self) return db_router.db_for_read(self)
@classmethod @classmethod
def get_django_model_serializer(cls, writable=False): # type: (bool) -> Django2ClickHouseModelSerializer def get_django_model_serializer(cls, writable=False): # type: (bool) -> Django2ClickHouseModelSerializer