fix: supports keyword only method spy

Author: zouyu.zzx

flight_profiler/common/bytecode_transformer.py

@@ -93,6 +93,10 @@ def _execute_bytecode_transform_intern(
     copy_fn = types.FunctionType(
         fn.__code__, global_space, fn.__name__, fn.__defaults__, fn.__closure__
     )
+    copy_fn.__kwdefaults__ = fn.__kwdefaults__
+    copy_fn.__annotations__ = fn.__annotations__
+    copy_fn.__dict__.update(fn.__dict__)  # 别忘了同步装饰器可能加上的其他属性
+    copy_fn.__qualname__ = fn.__qualname__
 
     if sys.version_info >= (3, 11):
         # emit COPY_FREE_VARIABLE

flight_profiler/test/plugins/watch/watch_plugin_test.py

@@ -35,6 +35,35 @@ def test_watch_module_method(self):
             raise
         finally:
             integration.stop()
+    
+    def test_watch_keyword_only_module_method(self):
+        current_directory = os.path.dirname(os.path.abspath(__file__))
+        file = os.path.join(current_directory, "watch_server_script.py")
+        integration = ProfileIntegration()
+        integration.start(file, 15)
+        try:
+            integration.execute_profile_cmd(
+                "watch __main__ kwonly_func --expr {args,return_obj}"
+            )
+            process = integration.client_process
+            find = False
+            start = time.time()
+            while time.time() - start < 15:
+                output = process.stdout.readline()
+                print(output)
+                if output:
+                    line = str(output)
+                    if line.find("kwonly_func_called") >= 0:
+                        find = True
+                        break
+                else:
+                    break
+            
+            self.assertTrue(find)
+        except:
+            raise
+        finally:
+            integration.stop()
 
     def test_watch_module_class_nested_method(self):
         current_directory = os.path.dirname(os.path.abspath(__file__))
@@ -151,3 +180,5 @@ def test_watch_module_exp_method(self):
             raise
         finally:
             integration.stop()
+            
+    

flight_profiler/test/plugins/watch/watch_server_script.py

@@ -18,6 +18,10 @@ def nested_func_inner():
             return test_func("nested_method")
         return nested_func_inner()
 
+def kwonly_func(x, y, *, c=1):
+    print("kwonly_func called")
+    
+    
 def test_func(name):
     print("hello func")
     return name + " " + "watch_plugin"
@@ -35,5 +39,6 @@ def exp_func():
     A().hello()
     A().nested_method()
     A.cls_method()
+    kwonly_func("kwonly_func_called", 2)
     time.sleep(1)
     idx += 1

pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "flight_profiler"
-version = "1.0.3"
+version = "1.0.4"
 description = "A diagnostic toolkit for on-the-fly analysis of remote python processes."
 homepage = "https://github.com/alibaba/PyFlightProfiler"
 readme = "README.md"