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