initial version of handling pandas method call

Author: asmirnov69

pandas_flavor/__version__.py

@@ -1,2 +1,2 @@
 """Version number."""
-__version__ = "0.3.0"
+__version__ = "0.4.0"

pandas_flavor/register.py

@@ -1,6 +1,8 @@
 from functools import wraps
 from pandas.api.extensions import register_series_accessor, register_dataframe_accessor
+from . import stack_counter
 
+handle_pandas_method_call = None
 
 def register_dataframe_method(method):
     """Register a function as a method attached to the Pandas DataFrame.
@@ -23,7 +25,15 @@ def __init__(self, pandas_obj):
 
             @wraps(method)
             def __call__(self, *args, **kwargs):
-                return method(self._obj, *args, **kwargs)
+                with stack_counter.global_scf.get_sc() as sc:
+                    #ipdb.set_trace()
+                    ret = method(self._obj, *args, **kwargs)
+                    if sc.scf.level == 1:
+                        #print("pf dataframe __call__")
+                        global handle_pandas_method_call
+                        if handle_pandas_method_call:
+                            handle_pandas_method_call(self._obj, method.__name__, args, kwargs, ret)
+                    return ret
 
         register_dataframe_accessor(method.__name__)(AccessorMethod)
 

pandas_flavor/stack_counter.py

@@ -0,0 +1,21 @@
+class StackCounter:
+    def __init__(self, scf):
+        self.scf = scf
+        
+    def __enter__(self):
+        #print("StackCounter:__enter__", id(self))
+        self.scf.level += 1
+        return self
+
+    def __exit__(self, type, value, traceback):
+        #print("StackCounter:__exit__", id(self))
+        self.scf.level -= 1
+        
+class SCF:
+    def __init__(self):
+        self.level = 0
+        
+    def get_sc(self):
+        return StackCounter(self)
+
+global_scf = SCF()