Spaces:
Running
Running
import dis | |
import array | |
import collections | |
try: | |
import html | |
except ImportError: | |
html = None | |
from setuptools.extern import six | |
from setuptools.extern.six.moves import html_parser | |
__metaclass__ = type | |
OpArg = collections.namedtuple('OpArg', 'opcode arg') | |
class Bytecode_compat: | |
def __init__(self, code): | |
self.code = code | |
def __iter__(self): | |
"""Yield '(op,arg)' pair for each operation in code object 'code'""" | |
bytes = array.array('b', self.code.co_code) | |
eof = len(self.code.co_code) | |
ptr = 0 | |
extended_arg = 0 | |
while ptr < eof: | |
op = bytes[ptr] | |
if op >= dis.HAVE_ARGUMENT: | |
arg = bytes[ptr + 1] + bytes[ptr + 2] * 256 + extended_arg | |
ptr += 3 | |
if op == dis.EXTENDED_ARG: | |
long_type = six.integer_types[-1] | |
extended_arg = arg * long_type(65536) | |
continue | |
else: | |
arg = None | |
ptr += 1 | |
yield OpArg(op, arg) | |
Bytecode = getattr(dis, 'Bytecode', Bytecode_compat) | |
unescape = getattr(html, 'unescape', None) | |
if unescape is None: | |
# HTMLParser.unescape is deprecated since Python 3.4, and will be removed | |
# from 3.9. | |
unescape = html_parser.HTMLParser().unescape | |