diff --git a/src/django_clickhouse/migrations.py b/src/django_clickhouse/migrations.py index edc4f03..47a52a2 100644 --- a/src/django_clickhouse/migrations.py +++ b/src/django_clickhouse/migrations.py @@ -8,8 +8,6 @@ from django.db import DEFAULT_DB_ALIAS as DJANGO_DEFAULT_DB_ALIAS from django.db.models.signals import post_migrate from django.dispatch import receiver -# In order to support all operations import here -from infi.clickhouse_orm.migrations import * # noqa F401, F403 from infi.clickhouse_orm.database import ServerError, DatabaseException from infi.clickhouse_orm.fields import StringField, DateField @@ -20,6 +18,7 @@ from .clickhouse_models import ClickHouseModel from .configuration import config from .database import connections, Database from .utils import lazy_class_import, module_exists +from .migration_operators import * class Migration: diff --git a/tests/clickhouse_migrations/0001_initial.py b/tests/clickhouse_migrations/0001_initial.py index d7bf69a..d9c0277 100644 --- a/tests/clickhouse_migrations/0001_initial.py +++ b/tests/clickhouse_migrations/0001_initial.py @@ -1,10 +1,10 @@ -from django_clickhouse import migrations, migration_operators +from django_clickhouse import migrations from tests.clickhouse_models import ClickHouseTestModel, ClickHouseCollapseTestModel class Migration(migrations.Migration): operations = [ - migration_operators.CreateTable(ClickHouseTestModel), - migration_operators.CreateTable(ClickHouseCollapseTestModel) + migrations.CreateTable(ClickHouseTestModel), + migrations.CreateTable(ClickHouseCollapseTestModel) ]