minification.py: Fixed the bug reported in issue #34 where multiple d…

…ecorators on the same function were not being indented properly.
liftoff · Feb 6, 2015 · 92bdf14 · 92bdf14
1 parent 63240c3
commit 92bdf14
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/pyminifier/minification.py b/pyminifier/minification.py
@@ -223,6 +223,10 @@ def foo(foo,bar,blah):
                 # mixed strings using both single quotes and double quotes.
                 out += "'''" + new_string + "'''"
                 joining_strings = False
+            if token_string == '@': # Decorators need special handling
+                if prev_tok[0] == tokenize.NEWLINE:
+                    # Ensure it gets indented properly
+                    out += (" " * (start_col - last_col))
         if not joining_strings:
             out += token_string
         last_col = end_col