feat: support funix_class

backend/funix/__init__.py

@@ -30,6 +30,7 @@
 # ---- Exports ----
 # ---- Decorators ----
 funix = decorator.funix
+funix_class = decorator.funix_class
 # ---- Decorators ----
 
 Limiter = decorator.Limiter

backend/funix/decorator/__init__.py

@@ -8,7 +8,14 @@
 from enum import Enum, auto
 from functools import wraps
 from importlib import import_module
-from inspect import Parameter, Signature, getsource, isgeneratorfunction, signature
+from inspect import (
+    Parameter,
+    Signature,
+    getsource,
+    isgeneratorfunction,
+    signature,
+    ismethod,
+)
 from json import dumps, loads
 from secrets import token_hex
 from traceback import format_exc
@@ -1910,3 +1917,12 @@ def wrapped_function(**wrapped_function_kwargs):
         return function
 
     return decorator
+
+
+def funix_class(inited_class):
+    for class_function in dir(inited_class):
+        if not class_function.startswith("_"):
+            function = getattr(inited_class, class_function)
+            print(class_function, function)
+            if ismethod(function):
+                funix()(function)

examples/class.py

@@ -0,0 +1,15 @@
+from funix import funix_class
+
+
+class A:
+    def __init__(self):
+        self.a = 1
+
+    def abc(self, b: int, c: int) -> None:
+        self.a = b + c
+
+    def b(self) -> int:
+        return self.a
+
+
+funix_class(A())