exposing if_not_exists argument in create_index

piccolo-orm · Feb 21, 2021 · f255f33 · f255f33
1 parent 31f4646
commit f255f33
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 3 deletions.
diff --git a/piccolo/apps/migrations/auto/migration_manager.py b/piccolo/apps/migrations/auto/migration_manager.py
@@ -415,7 +415,7 @@ async def _run_alter_columns(self, backwards=False):
                         column._meta._name = column_name
                         await _Table.drop_index([column]).run()
                         await _Table.create_index(
-                            [column], method=index_method
+                            [column], method=index_method, if_not_exists=True
                         ).run()
                 else:
                     # If the index value has changed, then we are either
@@ -427,7 +427,9 @@ async def _run_alter_columns(self, backwards=False):
                         kwargs = (
                             {"method": index_method} if index_method else {}
                         )
-                        await _Table.create_index([column], **kwargs).run()
+                        await _Table.create_index(
+                            [column], if_not_exists=True, **kwargs
+                        ).run()
                     else:
                         await _Table.drop_index([column]).run()
 

diff --git a/piccolo/table.py b/piccolo/table.py
@@ -623,14 +623,20 @@ def create_index(
         cls,
         columns: t.List[t.Union[Column, str]],
         method: IndexMethod = IndexMethod.btree,
+        if_not_exists: bool = False,
     ) -> CreateIndex:
         """
         Create a table index. If multiple columns are specified, this refers
         to a multicolumn index, rather than multiple single column indexes.
 
         await Band.create_index([Band.name]).run()
         """
-        return CreateIndex(table=cls, columns=columns, method=method)
+        return CreateIndex(
+            table=cls,
+            columns=columns,
+            method=method,
+            if_not_exists=if_not_exists,
+        )
 
     @classmethod
     def drop_index(