1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
|
#############################################################################
##
## Copyright (C) 2021 The Qt Company Ltd.
## Contact: https://www.qt.io/licensing/
##
## This file is part of the Qt for Python project.
##
## $QT_BEGIN_LICENSE:COMM$
##
## Commercial License Usage
## Licensees holding valid commercial Qt licenses may use this file in
## accordance with the commercial license agreement provided with the
## Software or, alternatively, in accordance with the terms contained in
## a written agreement between you and The Qt Company. For licensing terms
## and conditions see https://www.qt.io/terms-conditions. For further
## information use the contact form at https://www.qt.io/contact-us.
##
## $QT_END_LICENSE$
##
#############################################################################
"""Helper to dump AST nodes for debugging"""
import ast
def to_string(node):
"""Helper to retrieve a string from the (Lists of )Name/Attribute
aggregated into some nodes"""
if isinstance(node, ast.Name):
return node.id
if isinstance(node, ast.Attribute):
return node.attr
return ''
def debug_format_node(node):
"""Format AST node for debugging"""
if isinstance(node, ast.alias):
return f'alias("{node.name}")'
if isinstance(node, ast.arg):
return f'arg({node.arg})'
if isinstance(node, ast.Attribute):
if isinstance(node.value, ast.Name):
nested_name = debug_format_node(node.value)
return f'Attribute("{node.attr}", {nested_name})'
return f'Attribute("{node.attr}")'
if isinstance(node, ast.Call):
return 'Call({}({}))'.format(to_string(node.func), len(node.args))
if isinstance(node, ast.ClassDef):
base_names = [to_string(base) for base in node.bases]
bases = ': ' + ','.join(base_names) if base_names else ''
return f'ClassDef({node.name}{bases})'
if isinstance(node, ast.ImportFrom):
return f'ImportFrom("{node.module}")'
if isinstance(node, ast.FunctionDef):
arg_names = [a.arg for a in node.args.args]
return 'FunctionDef({}({}))'.format(node.name, ', '.join(arg_names))
if isinstance(node, ast.Name):
return 'Name("{}", Ctx={})'.format(node.id, type(node.ctx).__name__)
if isinstance(node, ast.NameConstant):
return f'NameConstant({node.value})'
if isinstance(node, ast.Num):
return f'Num({node.n})'
if isinstance(node, ast.Str):
return f'Str("{node.s}")'
return type(node).__name__
|
