Better solution for attribute access on integer literals.

1 files changed, 8 insertions, 14 deletions

diff --git a/Demo/parser/unparse.py b/Demo/parser/unparse.py
index fa8f434..b37eb82 100644
--- a/Demo/parser/unparse.py
+++ b/Demo/parser/unparse.py
@@ -316,22 +316,11 @@ class Unparser:
         self.write("`")
 
     def _Num(self, t):
-        if isinstance(t.n, float):
-            # A float literal should be nonnegative, and not a nan.
-            # It could be an infinity, though; in that case we
-            # substitute an overflowing decimal value.
-            assert not math.isnan(t.n)
-            assert math.copysign(1.0, t.n) > 0.0
-            if math.isinf(t.n):
-                self.write("1e" + repr(sys.float_info.max_10_exp + 1))
-            else:
-                self.write(repr(t.n))
+        if isinstance(t.n, float) and math.isinf(t.n):
+            # Subsitute overflowing decimal literal for AST infinity
+            self.write("1e" + repr(sys.float_info.max_10_exp + 1))
         else:
-            # Parenthesize integer literals to avoid turning
-            # "3 .__abs__()" into "3.__abs__()".
-            self.write("(")
             self.write(repr(t.n))
-            self.write(")")
 
     def _List(self, t):
         self.write("[")
@@ -449,6 +438,11 @@ class Unparser:
 
     def _Attribute(self,t):
         self.dispatch(t.value)
+        # Special case: 3.__abs__() is a syntax error, so if t.value
+        # is an integer literal then we need to either parenthesize
+        # it or add an extra space to get 3 .__abs__().
+        if isinstance(t.value, ast.Num) and isinstance(t.value.n, int):
+            self.write(" ")
         self.write(".")
         self.write(t.attr)