Make relative imports more compact

luk3yx · dflook · commit e56c3836db5c · 2023-04-08T21:02:19.000+01:00
diff --git a/src/python_minifier/module_printer.py b/src/python_minifier/module_printer.py
@@ -268,13 +268,13 @@ def visit_Import(self, node):
     def visit_ImportFrom(self, node):
         assert isinstance(node, ast.ImportFrom)
 
-        if node.module is None:
-            self.code += 'from ' + ('.' * node.level) + ' '
-        else:
-            self.code += 'from ' + ('.' * node.level)
-            self.code += node.module
+        self.code += 'from' + ('.' * node.level)
+        if node.module is not None:
+            if node.level == 0:
+                self.code += ' '
+            self.code += node.module + ' '
 
-        self.code += ' import '
+        self.code += 'import '
         first = True
         for n in node.names:
             if first:
