Wtite TO only for value, default and multi

lelit · Feb 3, 2025 · e683d4e · e683d4e · lelit · Feb 4, 2025
1 parent 04e6bfc
commit e683d4e
Showing 1 changed file with 10 additions and 10 deletions.
diff --git a/pglast/printers/ddl.py b/pglast/printers/ddl.py
@@ -3363,16 +3363,16 @@ def variable_set_stmt(node, output):
             output.print_list(node.args, ',')
         else:
             output.print_name(node.name.split('.'))
-        if node.kind == vsk.VAR_SET_CURRENT:
-            output.write('FROM CURRENT')
-        else:
-            output.write(' TO ')
-        if node.kind == vsk.VAR_SET_VALUE:
-            output.print_list(node.args)
-        elif node.kind == vsk.VAR_SET_DEFAULT:
-            output.write('DEFAULT')
-        elif node.kind == vsk.VAR_SET_MULTI:
-            pass
+            if node.kind == vsk.VAR_SET_CURRENT:
+                output.write('FROM CURRENT')
+            else:
+                output.write(' TO ')
+            if node.kind == vsk.VAR_SET_VALUE:
+                output.print_list(node.args)
+            elif node.kind == vsk.VAR_SET_DEFAULT:
+                output.write('DEFAULT')
+            elif node.kind == vsk.VAR_SET_MULTI:
+                pass
 
 
 @node_printer(ast.VariableShowStmt)