From 42fd99b88b78119a9e8d5a618a4ae91ebea82458 Mon Sep 17 00:00:00 2001
From: amirhgh <amirhgholamian78@gmail.com>
Date: Fri, 31 Dec 2021 22:06:55 +0330
Subject: [PATCH] add hints to migration operators

---
 src/django_clickhouse/migration_operators.py | 41 ++++++++++++++++++++
 1 file changed, 41 insertions(+)
 create mode 100644 src/django_clickhouse/migration_operators.py

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