1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859 |
- 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
|