Use customized unparser instead of ast.unparse

yunline · Nov 30, 2024 · a5316ce · a5316ce
1 parent e05f397
commit a5316ce
Show file tree

Hide file tree

Showing 2 changed files with 520 additions and 1 deletion.
diff --git a/oneliner/__init__.py b/oneliner/__init__.py
@@ -1,4 +1,5 @@
 from oneliner.convert import OnelinerConvertor
+from oneliner.expr_unparse import unparse
 
 __all__ = ["OnelinerConvertor", "convert_code_string"]
 
@@ -10,4 +11,4 @@ def convert_code_string(code: str, filename="<string>"):
     ast_root = ast.parse(code, filename, "exec")
     symtable_root = symtable.symtable(code, filename, "exec")
     out = OnelinerConvertor().cvt(ast_root, symtable_root)
-    return ast.unparse(out).replace("\n", "")
+    return unparse(out)