Admin panel bookkeeping (#43)

* Keep track of all the functions we decorate

---------

Co-authored-by: Brett Beutell <brett@fiberplane.com>

Author: brettimus

src/autometrics/__init__.py

@@ -1 +1,3 @@
 from .decorator import *
+
+from .admin_panel import *

src/autometrics/admin_panel/__init__.py

@@ -0,0 +1 @@
+from .function_registry import register_function_info, get_decorated_functions_list

src/autometrics/admin_panel/function_registry.py

@@ -0,0 +1,22 @@
+# This module will do some bookkeeping on information of any functions that have been wrapped by autometrics decorators
+
+from typing import List, TypedDict
+
+
+class FunctionInfo(TypedDict):
+    name: str
+    module: str
+
+
+FUNCTION_REGISTRY: List[FunctionInfo] = []
+
+
+def register_function_info(func_name: str, module_name: str):
+    global FUNCTION_REGISTRY
+    function_info: FunctionInfo = {"name": func_name, "module": module_name}
+    FUNCTION_REGISTRY.append(function_info)
+
+
+def get_decorated_functions_list():
+    global FUNCTION_REGISTRY
+    return FUNCTION_REGISTRY

src/autometrics/decorator.py

@@ -7,6 +7,7 @@
 from typing_extensions import ParamSpec
 from .objectives import Objective
 from .tracker import get_tracker, Result
+from .admin_panel import register_function_info
 from .utils import get_module_name, get_caller_function, append_docs_to_docstring
 
 
@@ -68,6 +69,7 @@ def sync_decorator(func: Callable[P, T]) -> Callable[P, T]:
 
         module_name = get_module_name(func)
         func_name = func.__name__
+        register_function_info(func_name, module_name)
 
         @wraps(func)
         def sync_wrapper(*args: P.args, **kwds: P.kwargs) -> T:
@@ -100,6 +102,7 @@ def async_decorator(func: Callable[P, Awaitable[T]]) -> Callable[P, Awaitable[T]
 
         module_name = get_module_name(func)
         func_name = func.__name__
+        register_function_info(func_name, module_name)
 
         @wraps(func)
         async def async_wrapper(*args: P.args, **kwds: P.kwargs) -> T: