| 123456789101112131415161718192021222324252627282930313233343536373839404142434445 |
- import dis
- import array
- import collections
- import six
- OpArg = collections.namedtuple('OpArg', 'opcode arg')
- class Bytecode_compat(object):
- 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)
|
