black bytecode.py

Author: rocky

xdis/bytecode.py

@@ -234,7 +234,7 @@ def get_logical_instruction_at_offset(
     linestarts=None,
     line_offset=0,
     exception_entries=None,
-    labels = None
+    labels=None,
 ):
     """
     Return a single logical instruction for `bytecode` at offset `offset`.
@@ -306,11 +306,7 @@ def get_logical_instruction_at_offset(
                     + extended_arg
                 )
                 i += 2
-                extended_arg = (
-                    arg * 0x10000
-                    if opname == "EXTENDED_ARG"
-                    else 0
-                )
+                extended_arg = arg * 0x10000 if opname == "EXTENDED_ARG" else 0
 
             #  Set argval to the dereferenced value of the argument when
             #  available, and argrepr to the string representation of argval.
@@ -320,7 +316,9 @@ def get_logical_instruction_at_offset(
             argval = arg
 
             # create a localsplusnames table that resolves duplicates.
-            localsplusnames = (varnames or tuple()) + tuple(name for name in (cells or tuple()) if name not in varnames)
+            localsplusnames = (varnames or tuple()) + tuple(
+                name for name in (cells or tuple()) if name not in varnames
+            )
 
             if op in opc.CONST_OPS:
                 argval, argrepr = _get_const_info(arg, constants)
@@ -333,9 +331,7 @@ def get_logical_instruction_at_offset(
                     argval, argrepr = _get_name_info(arg >> 1, names)
                     if arg & 1:
                         argrepr = "NULL|self + " + argrepr
-                elif (
-                    opc.version_tuple >= (3, 12) and opname == "LOAD_SUPER_ATTR"
-                ):
+                elif opc.version_tuple >= (3, 12) and opname == "LOAD_SUPER_ATTR":
                     argval, argrepr = _get_name_info(arg >> 2, names)
                     if arg & 1:
                         argrepr = "NULL|self + " + argrepr
@@ -347,7 +343,13 @@ def get_logical_instruction_at_offset(
 
                 # check cache instructions for python 3.13
                 if opc.version_tuple >= (3, 13):
-                    if opc.opname[op] in ["POP_JUMP_IF_TRUE", "POP_JUMP_IF_FALSE", "POP_JUMP_IF_NONE", "POP_JUMP_IF_NOT_NONE", "JUMP_BACKWARD"]:
+                    if opc.opname[op] in [
+                        "POP_JUMP_IF_TRUE",
+                        "POP_JUMP_IF_FALSE",
+                        "POP_JUMP_IF_NONE",
+                        "POP_JUMP_IF_NOT_NONE",
+                        "JUMP_BACKWARD",
+                    ]:
                         argval += 2
 
                 # FOR_ITER has a cache instruction in 3.12
@@ -358,7 +360,11 @@ def get_logical_instruction_at_offset(
                 argval = get_jump_val(arg, opc.python_version)
                 argrepr = "to " + repr(argval)
             elif op in opc.LOCAL_OPS:
-                if opc.version_tuple >= (3, 13) and opname in ("LOAD_FAST_LOAD_FAST", "STORE_FAST_LOAD_FAST", "STORE_FAST_STORE_FAST"):
+                if opc.version_tuple >= (3, 13) and opname in (
+                    "LOAD_FAST_LOAD_FAST",
+                    "STORE_FAST_LOAD_FAST",
+                    "STORE_FAST_STORE_FAST",
+                ):
                     arg1 = arg >> 4
                     arg2 = arg & 15
                     argval1, argrepr1 = _get_name_info(arg1, localsplusnames)
@@ -377,11 +383,11 @@ def get_logical_instruction_at_offset(
             elif op in opc.COMPARE_OPS:
                 if opc.python_version >= (3, 13):
                     # The fifth-lowest bit of the oparg now indicates a forced conversion to bool.
-                    argval = (opc.cmp_op[arg >> 5])
+                    argval = opc.cmp_op[arg >> 5]
                 elif opc.python_version >= (3, 12):
-                    argval = (opc.cmp_op[arg >> 4])
+                    argval = opc.cmp_op[arg >> 4]
                 else:
-                    argval = (opc.cmp_op[arg])
+                    argval = opc.cmp_op[arg]
                 argrepr = argval
             elif op in opc.NARGS_OPS:
                 opname = opname
@@ -473,17 +479,18 @@ def get_instructions_bytes(
     offset = 0
 
     while offset < n:
-        instructions = list(get_logical_instruction_at_offset(
-            bytecode,
-            offset,
-            opc,
-            varnames=varnames,
-            names=names,
-            constants=constants,
-            cells=cells,
-            linestarts=linestarts,
-            line_offset=0,
-            exception_entries=exception_entries,
+        instructions = list(
+            get_logical_instruction_at_offset(
+                bytecode,
+                offset,
+                opc,
+                varnames=varnames,
+                names=names,
+                constants=constants,
+                cells=cells,
+                linestarts=linestarts,
+                line_offset=0,
+                exception_entries=exception_entries,
             )
         )
 
@@ -492,7 +499,6 @@ def get_instructions_bytes(
         offset = next_offset(instruction.opcode, opc, instruction.offset)
 
 
-
 class Bytecode:
     """Bytecode operations involving a Python code object.