Don't include doc string of class in its code child

da8db8ca · Jeremy Hylton · 8aea0cc9 · da8db8ca · da8db8ca
Kaydet (Commit) da8db8ca authored Agu 29, 2001 tarafından Jeremy Hylton
Show whitespace changes
Inline Side-by-side

Showing with 10 additions and 0 deletions

transformer.py Lib/compiler/transformer.py +5 -0

transformer.py Tools/compiler/compiler/transformer.py +5 -0

No files found.
--- a/Lib/compiler/transformer.py
+++ b/Lib/compiler/transformer.py
@@ -222,6 +222,11 @@ class Transformer:
        # code for class
        code = self.com_node(nodelist[-1])

+        if doc is not None:
+            assert isinstance(code, Stmt)
+            assert isinstance(code.nodes[0], Discard)
+            del code.nodes[0]
+
        n = Class(name, bases, doc, code)
        n.lineno = nodelist[1][2]
        return n

--- a/Tools/compiler/compiler/transformer.py
+++ b/Tools/compiler/compiler/transformer.py
@@ -222,6 +222,11 @@ class Transformer:
        # code for class
        code = self.com_node(nodelist[-1])

+        if doc is not None:
+            assert isinstance(code, Stmt)
+            assert isinstance(code.nodes[0], Discard)
+            del code.nodes[0]
+
        n = Class(name, bases, doc, code)
        n.lineno = nodelist[1][2]
        return n