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

Refactoring

Browse Source
This commit is contained in:
M1ha 2018-12-25 12:53:23 +05:00
parent e8a15e4330
commit dbe54da5e3
1 changed files with 5 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
src/django_clickhouse/migrations.py
View File

@ -64,8 +64,9 @@ def migrate_app(app_label, db_alias, up_to=9999, database=None):
if module_exists(migrations_package): if module_exists(migrations_package):
database = database or connections[db_alias] database = database or connections[db_alias]
migration_history_model = lazy_class_import(config.MIGRATION_HISTORY_MODEL)
applied_migrations = MigrationHistory.get_applied_migrations(db_alias, migrations_package) applied_migrations = migration_history_model.get_applied_migrations(db_alias, migrations_package)
modules = import_submodules(migrations_package) modules = import_submodules(migrations_package)
unapplied_migrations = set(modules.keys()) - applied_migrations unapplied_migrations = set(modules.keys()) - applied_migrations
@ -74,7 +75,6 @@ def migrate_app(app_label, db_alias, up_to=9999, database=None):
migration = modules[name].Migration() migration = modules[name].Migration()
migration.apply(db_alias, database=database) migration.apply(db_alias, database=database)
migration_history_model = lazy_class_import(config.MIGRATION_HISTORY_MODEL)
migration_history_model.set_migration_applied(db_alias, migrations_package, name) migration_history_model.set_migration_applied(db_alias, migrations_package, name)
if int(name[:4]) >= up_to: if int(name[:4]) >= up_to:
@ -121,11 +121,10 @@ class MigrationHistory(ClickHouseModel):
""" """
# Ensure that table for migration storing is created # Ensure that table for migration storing is created
for db_alias in cls.migrate_non_replicated_db_aliases: for db_alias in cls.migrate_non_replicated_db_aliases:
connections[db_alias].create_table(MigrationHistory) connections[db_alias].create_table(cls)
cls.objects.bulk_create([ cls.objects.bulk_create([
MigrationHistory(db_alias=db_alias, package_name=migrations_package, module_name=name, cls(db_alias=db_alias, package_name=migrations_package, module_name=name, applied=datetime.date.today())
applied=datetime.date.today())
]) ])
@classmethod @classmethod
@ -136,7 +135,7 @@ class MigrationHistory(ClickHouseModel):
:param migrations_package: Package name to check :param migrations_package: Package name to check
:return: Set of migration names :return: Set of migration names
""" """
qs = MigrationHistory.objects.filter(package_name=migrations_package, db_alias=db_alias).only('module_name') qs = cls.objects.filter(package_name=migrations_package, db_alias=db_alias).only('module_name')
try: try:
return set(obj.module_name for obj in qs) return set(obj.module_name for obj in qs)
except ServerError as ex: except ServerError as ex: