| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110 |
- # -*- coding: utf-8 -*-
- """
- 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
|
