diff --git a/src/django_clickhouse/migration_operators.py b/src/django_clickhouse/migration_operators.py new file mode 100644 index 0000000..1964256 --- /dev/null +++ b/src/django_clickhouse/migration_operators.py @@ -0,0 +1,41 @@ +from infi import clickhouse_orm + + +class HintParamMixin: + def __init__(self, *args, hints=None, **kwargs): + if hints is None: + hints = dict() + self.hints = hints + super(HintParamMixin, self).__init__(*args, **kwargs) + + +class CreateTable(HintParamMixin, clickhouse_orm.CreateTable): + pass + + +class AlterTable(HintParamMixin, clickhouse_orm.AlterTable): + pass + + +class AlterTableWithBuffer(HintParamMixin, clickhouse_orm.AlterTableWithBuffer): + pass + + +class DropTable(HintParamMixin, clickhouse_orm.DropTable): + pass + + +class AlterConstraints(HintParamMixin, clickhouse_orm.AlterConstraints): + pass + + +class AlterIndexes(HintParamMixin, clickhouse_orm.AlterIndexes): + pass + + +class RunPython(HintParamMixin, clickhouse_orm.RunPython): + pass + + +class RunSQL(HintParamMixin, clickhouse_orm.RunSQL): + pass