changing column types can work in reverse

piccolo-orm · Feb 13, 2021 · 3859bfa · 3859bfa
1 parent f7d8147
commit 3859bfa
Showing 1 changed file with 15 additions and 2 deletions.
diff --git a/piccolo/apps/migrations/auto/migration_manager.py b/piccolo/apps/migrations/auto/migration_manager.py
@@ -340,9 +340,20 @@ async def _run_alter_columns(self, backwards=False):
                 )
                 column_name = alter_column.column_name
 
+                ###############################################################
+
                 # Change the column type if possible
-                column_class = alter_column.column_class
-                old_column_class = alter_column.old_column_class
+                column_class = (
+                    alter_column.old_column_class
+                    if backwards
+                    else alter_column.column_class
+                )
+                old_column_class = (
+                    alter_column.column_class
+                    if backwards
+                    else alter_column.old_column_class
+                )
+
                 if (old_column_class is not None) and (
                     column_class is not None
                 ):
@@ -361,6 +372,8 @@ async def _run_alter_columns(self, backwards=False):
                             old_column=old_column, new_column=new_column
                         )
 
+                ###############################################################
+
                 null = params.get("null")
                 if null is not None:
                     await _Table.alter().set_null(