| | |
| | """ |
| | Part of the astor library for Python AST manipulation. |
| | |
| | License: 3-clause BSD |
| | |
| | Copyright (c) 2015 Patrick Maupin |
| | |
| | This module provides data and functions for mapping |
| | AST nodes to symbols and precedences. |
| | |
| | """ |
| |
|
| | import ast |
| |
|
| | op_data = """ |
| | GeneratorExp 1 |
| | |
| | Assign 1 |
| | AnnAssign 1 |
| | AugAssign 0 |
| | Expr 0 |
| | Yield 1 |
| | YieldFrom 0 |
| | If 1 |
| | For 0 |
| | AsyncFor 0 |
| | While 0 |
| | Return 1 |
| | |
| | Slice 1 |
| | Subscript 0 |
| | Index 1 |
| | ExtSlice 1 |
| | comprehension_target 1 |
| | Tuple 0 |
| | FormattedValue 0 |
| | |
| | Comma 1 |
| | NamedExpr 1 |
| | Assert 0 |
| | Raise 0 |
| | call_one_arg 1 |
| | |
| | Lambda 1 |
| | IfExp 0 |
| | |
| | comprehension 1 |
| | Or or 1 |
| | And and 1 |
| | Not not 1 |
| | |
| | Eq == 1 |
| | Gt > 0 |
| | GtE >= 0 |
| | In in 0 |
| | Is is 0 |
| | NotEq != 0 |
| | Lt < 0 |
| | LtE <= 0 |
| | NotIn not in 0 |
| | IsNot is not 0 |
| | |
| | BitOr | 1 |
| | BitXor ^ 1 |
| | BitAnd & 1 |
| | LShift << 1 |
| | RShift >> 0 |
| | Add + 1 |
| | Sub - 0 |
| | Mult * 1 |
| | Div / 0 |
| | Mod % 0 |
| | FloorDiv // 0 |
| | MatMult @ 0 |
| | PowRHS 1 |
| | Invert ~ 1 |
| | UAdd + 0 |
| | USub - 0 |
| | Pow ** 1 |
| | Await 1 |
| | Num 1 |
| | Constant 1 |
| | """ |
| |
|
| | op_data = [x.split() for x in op_data.splitlines()] |
| | op_data = [[x[0], ' '.join(x[1:-1]), int(x[-1])] for x in op_data if x] |
| | for index in range(1, len(op_data)): |
| | op_data[index][2] *= 2 |
| | op_data[index][2] += op_data[index - 1][2] |
| |
|
| | precedence_data = dict((getattr(ast, x, None), z) for x, y, z in op_data) |
| | symbol_data = dict((getattr(ast, x, None), y) for x, y, z in op_data) |
| |
|
| |
|
| | def get_op_symbol(obj, fmt='%s', symbol_data=symbol_data, type=type): |
| | """Given an AST node object, returns a string containing the symbol. |
| | """ |
| | return fmt % symbol_data[type(obj)] |
| |
|
| |
|
| | def get_op_precedence(obj, precedence_data=precedence_data, type=type): |
| | """Given an AST node object, returns the precedence. |
| | """ |
| | return precedence_data[type(obj)] |
| |
|
| |
|
| | class Precedence(object): |
| | vars().update((x, z) for x, y, z in op_data) |
| | highest = max(z for x, y, z in op_data) + 2 |
| |
|
