Revise for streamline method calls

Author: rocky

pytest/test_std.py

@@ -155,6 +155,7 @@ def test_distb(traceback_fixture, stream_fixture):
     #     assert actual == expected
 
     def test_get_instructions():
+        # opc = get_opcode_module(PYTHON_VERSION_TRIPLE, PYTHON_IMPLEMENTATION)
         actual = list(dis.get_instructions(TEST_SOURCE_CODE))
         actual_len = len(actual)
         assert actual_len > 0

xdis/bytecode.py

@@ -814,7 +814,7 @@ def get_instructions(self, x):
         the disassembled code object.
         """
         co = get_code_object(x)
-        return get_instructions_bytes(co.co_code, self.opc)
+        return get_instructions_bytes(co, self.opc)
 
 
 def list2bytecode(

xdis/std.py

@@ -223,7 +223,7 @@ def disco(self, code, lasti=-1, file=None) -> None:
             python_implementation=self.python_version_tuple,
         )
 
-    def get_instructions(self, x, first_line=None):
+    def get_instructions(self, x):
         """Iterator for the opcodes in methods, functions or code
 
         Generates a series of Instruction named tuples giving the details of
@@ -234,7 +234,11 @@ def get_instructions(self, x, first_line=None):
         Otherwise, the source line information (if any) is taken directly from
         the disassembled code object.
         """
-        return self.Bytecode(x).get_instructions(x, first_line)
+        if isinstance(x, str):
+            code_obj = compile(x, f"<std.py {str}>", "exec")
+        else:
+            code_obj = x
+        return self.Bytecode(code_obj).get_instructions(code_obj)
 
     def findlinestarts(self, code):
         """Find the offsets in a byte code which are start of lines in the source.