You've already forked python-uncompyle6
mirror of
https://github.com/rocky/python-uncompyle6.git
synced 2025-08-03 00:45:53 +08:00
80 lines
3.0 KiB
Python
80 lines
3.0 KiB
Python
# Copyright (c) 2017-2018 Rocky Bernstein
|
|
|
|
# This program is free software: you can redistribute it and/or modify
|
|
# it under the terms of the GNU General Public License as published by
|
|
# the Free Software Foundation, either version 3 of the License, or
|
|
# (at your option) any later version.
|
|
#
|
|
# This program is distributed in the hope that it will be useful,
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
# GNU General Public License for more details.
|
|
#
|
|
# You should have received a copy of the GNU General Public License
|
|
# along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
"""
|
|
spark grammar differences over Python 3.3 for Python 3.4
|
|
"""
|
|
|
|
from uncompyle6.parser import PythonParserSingle
|
|
from spark_parser import DEFAULT_DEBUG as PARSER_DEFAULT_DEBUG
|
|
from uncompyle6.parsers.parse33 import Python33Parser
|
|
|
|
class Python34Parser(Python33Parser):
|
|
|
|
def __init__(self, debug_parser=PARSER_DEFAULT_DEBUG):
|
|
super(Python34Parser, self).__init__(debug_parser)
|
|
self.customized = {}
|
|
|
|
def p_misc34(self, args):
|
|
"""
|
|
expr ::= LOAD_ASSERT
|
|
|
|
|
|
while1stmt ::= SETUP_LOOP l_stmts
|
|
COME_FROM JUMP_BACK POP_BLOCK COME_FROM_LOOP
|
|
|
|
# FIXME the below masks a bug in not detecting COME_FROM_LOOP
|
|
# grammar rules with COME_FROM -> COME_FROM_LOOP already exist
|
|
whileelsestmt ::= SETUP_LOOP testexpr l_stmts_opt JUMP_BACK POP_BLOCK
|
|
else_suitel COME_FROM
|
|
|
|
# Python 3.4+ optimizes the trailing two JUMPS away
|
|
|
|
# Is this 3.4 only?
|
|
yield_from ::= expr GET_ITER LOAD_CONST YIELD_FROM
|
|
"""
|
|
|
|
def customize_grammar_rules(self, tokens, customize):
|
|
self.remove_rules("""
|
|
while1stmt ::= SETUP_LOOP l_stmts COME_FROM JUMP_BACK COME_FROM_LOOP
|
|
""")
|
|
super(Python34Parser, self).customize_grammar_rules(tokens, customize)
|
|
return
|
|
|
|
class Python34ParserSingle(Python34Parser, PythonParserSingle):
|
|
pass
|
|
|
|
|
|
if __name__ == '__main__':
|
|
# Check grammar
|
|
p = Python34Parser()
|
|
p.check_grammar()
|
|
from uncompyle6 import PYTHON_VERSION, IS_PYPY
|
|
if PYTHON_VERSION == 3.4:
|
|
lhs, rhs, tokens, right_recursive = p.check_sets()
|
|
from uncompyle6.scanner import get_scanner
|
|
s = get_scanner(PYTHON_VERSION, IS_PYPY)
|
|
opcode_set = set(s.opc.opname).union(set(
|
|
"""JUMP_BACK CONTINUE RETURN_END_IF COME_FROM
|
|
LOAD_GENEXPR LOAD_ASSERT LOAD_SETCOMP LOAD_DICTCOMP LOAD_CLASSNAME
|
|
LAMBDA_MARKER RETURN_LAST
|
|
""".split()))
|
|
remain_tokens = set(tokens) - opcode_set
|
|
import re
|
|
remain_tokens = set([re.sub('_\d+$', '', t) for t in remain_tokens])
|
|
remain_tokens = set([re.sub('_CONT$', '', t) for t in remain_tokens])
|
|
remain_tokens = set(remain_tokens) - opcode_set
|
|
print(remain_tokens)
|
|
# print(sorted(p.rule2name.items()))
|