jinx/python-uncompyle6
1
Fork 0
mirror of https://github.com/rocky/python-uncompyle6.git synced 2025-08-04 01:09:52 +08:00
Code Issues Packages Projects Releases Wiki Activity

Go over docstring handling

Browse Source
This commit is contained in:
rocky
2020-02-02 05:37:07 -05:00
parent de1e7d423c
commit 38e2b8a10b
2 changed files with 9 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
uncompyle6/semantics/pysource.py
View File

@@ -884,7 +884,11 @@ class SourceWalker(GenericASTTraversal, object):
def n_docstring(self, node): def n_docstring(self, node):
indent = self.indent indent = self.indent
docstring = node[0].pattr doc_node = node[0]
if doc_node.attr:
docstring = doc_node.attr
else:
docstring = node[0].pattr
quote = '"""' quote = '"""'
if docstring.find(quote) >= 0: if docstring.find(quote) >= 0:

7
uncompyle6/semantics/transform.py
View File

@@ -26,7 +26,7 @@ from uncompyle6.semantics.consts import RETURN_NONE
def is_docstring(node): def is_docstring(node):
try: try:
return node[0].kind == "assign" and node[0][1][0].pattr == "__doc__" return node.kind == "assign" and node[1][0].pattr == "__doc__"
except: except:
return False return False
@@ -397,7 +397,7 @@ class TreeTransform(GenericASTTraversal, object):
# Disambiguate a string (expression) which appears as a "call_stmt" at # Disambiguate a string (expression) which appears as a "call_stmt" at
# the beginning of a function versus a docstring. Seems pretty academic, # the beginning of a function versus a docstring. Seems pretty academic,
# but this is Python. # but this is Python.
call_stmt = ast[0][0][0] call_stmt = ast[0][0]
if is_not_docstring(call_stmt): if is_not_docstring(call_stmt):
call_stmt.kind = "string_at_beginning" call_stmt.kind = "string_at_beginning"
call_stmt.transformed_by = "transform" call_stmt.transformed_by = "transform"
@@ -414,7 +414,8 @@ class TreeTransform(GenericASTTraversal, object):
"LOAD_STR", "LOAD_STR",
has_arg=True, has_arg=True,
offset=0, offset=0,
pattr=self.ast[i][0][0][0].pattr, attr=self.ast[i][0][0].attr,
pattr=self.ast[i][0][0].pattr,
) )
], ],
) )