carrotquest/django-clickhouse
1
1
Fork 0
mirror of https://github.com/carrotquest/django-clickhouse.git synced 2024-11-22 00:56:37 +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_lock_timeout = None
def get_database(self, for_write=False):
@classmethod
def get_database(cls, for_write=False):
# type: (bool) -> Database
"""
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)()
if for_write:
return db_router.db_for_write(self.__class__, instance=self)
return db_router.db_for_write(cls)
else:
return db_router.db_for_read(self.__class__, instance=self)
return db_router.db_for_read(self)
@classmethod
def get_django_model_serializer(cls, writable=False): # type: (bool) -> Django2ClickHouseModelSerializer