use setuptools to install/build statswrapper

Author: guenp

build.py

@@ -0,0 +1,16 @@
+from __future__ import annotations
+
+from hatchling.builders.hooks.plugin.interface import BuildHookInterface
+from setuptools import Extension
+from setuptools.command.build_ext import build_ext
+
+
+class CustomBuildHook(BuildHookInterface):
+    def initialize(self, version, build_data):
+        build_data['ext_modules'] = [
+            Extension(
+                "api_tracer._stats_wrapper",
+                ["src/api_tracer/_stats_wrapper.c"],
+            ),
+        ]
+        build_data['cmdclass'] = {"build_ext": build_ext} 

examples/test_scipy.py

@@ -1,4 +1,13 @@
-from scipy import stats
+# from opentelemetry.instrumentation.auto_instrumentation import initialize
+from api_tracer import install
+
+# from api_tracer.console import setup_console
+
+install(["scipy.stats._correlation", "scipy.stats._distn_infrastructure"])
+# initialize()
+# setup_console()
+
+from scipy import stats  # noqa: E402
 
 stats.norm.pdf(x=1, loc=1, scale=0.01)
 stats.norm(loc=1, scale=0.02).pdf(1)
@@ -8,3 +17,11 @@
 # Y = stats.exp((X + 1)*0.01)
 # from scipy import test
 # test()
+
+# How many calls (errors, and how often something was odd with the args)
+print("counts: ", stats.norm.pdf._get_counts())
+
+# Detailed statics for each parameter.  How often was it passed
+# and how often were "a" and 3 passed (or something equal to them,
+# I do not consider types right now, although one could)?
+print("param stats: ", stats.norm.pdf._get_param_stats())

examples/test_statswrapper.py

@@ -0,0 +1,20 @@
+from __future__ import annotations
+
+from api_tracer.statswrapper import stats_deco
+
+
+@stats_deco(None, b=("a", 3), c=None)
+def test_func(a, b=None, c=None):
+    return a, b, c
+
+
+# Do things with test_func
+test_func(1, 3)
+
+# How many calls (errors, and how often something was odd with the args)
+print("counts: ", test_func._get_counts())
+
+# Detailed statics for each parameter.  How often was it passed
+# and how often were "a" and 3 passed (or something equal to them,
+# I do not consider types right now, although one could)?
+print("param stats: ", test_func._get_param_stats())

pyproject.toml

@@ -1,12 +1,12 @@
 [build-system]
-requires = ["hatchling>=1.26", "hatch-vcs"]
-build-backend = "hatchling.build"
+requires = ["setuptools>=77", "setuptools_scm[toml]>=7"]
+build-backend = "setuptools.build_meta"
 
 
 [project]
 name = "api-tracer"
 authors = [
-  { name = "Guen Prawiroatmodjo", email = "guenp@hey.com" },
+  { name = "Scientific Python", email = "guenp@hey.com" },
   { name = "Sebastian Berg", email = "sebastian@sipsolutions.net" },
 ]
 description = "API tracer to gather usage telemetry for Python libraries"
@@ -56,14 +56,14 @@ docs = [
 ]
 
 
-[tool.hatch]
-version.source = "vcs"
-build.hooks.vcs.version-file = "src/api_tracer/_version.py"
+[tool.setuptools_scm]
+write_to = "src/api_tracer/_version.py"
 
-[tool.hatch.envs.default]
-# duplicate for now since hatch doesn't support groups yet
-dependencies = []
-scripts.test = "pytest {args}"
+
+[tool.setuptools]
+ext-modules = [
+  {name = "api_tracer.statswrapper._stats_wrapper", sources=["src/api_tracer/statswrapper/_stats_wrapper.c"]}
+]
 
 
 [tool.pytest.ini_options]

src/api_tracer/__init__.py

@@ -1,5 +1,5 @@
 """
-Copyright (c) 2025 Guen Prawiroatmodjo. All rights reserved.
+Copyright (c) 2025 Scientific Python. All rights reserved.
 
 api-tracer: API tracer to gather usage telemetry for Python libraries
 """
@@ -8,11 +8,14 @@
 
 from api_tracer.path_finder import install
 from api_tracer.span import span
+from api_tracer.statswrapper import stats_deco, stats_deco_auto
 
 from ._version import version as __version__
 
 __all__ = [
     "__version__",
     "install",
     "span",
+    "stats_deco",
+    "stats_deco_auto",
 ]

src/api_tracer/path_finder.py

@@ -1,10 +1,12 @@
+from __future__ import annotations
+
 import inspect
 import sys
 from importlib.abc import MetaPathFinder
 from importlib.machinery import SourceFileLoader
 from importlib.util import spec_from_loader
 
-from api_tracer.span import span
+from api_tracer.statswrapper import stats_deco_auto
 
 __all__ = ["install"]
 
@@ -21,7 +23,7 @@ def __init__(self, module_names, *args, **kwargs):
         super().__init__(*args, **kwargs)
 
     def find_spec(self, fullname, path, target=None):
-        if any([name in fullname for name in self._module_names]):
+        if any(name in fullname for name in self._module_names):
             for finder in sys.meta_path:
                 if finder != self:
                     spec = finder.find_spec(fullname, path, target)
@@ -32,8 +34,7 @@ def find_spec(self, fullname, path, target=None):
                                 loader=TelemetrySpanSourceFileLoader(spec.name, spec.origin),
                                 origin=spec.origin,
                             )
-                        else:
-                            return spec
+                        return spec
 
         return None
 
@@ -48,14 +49,13 @@ def exec_module(self, module):
         for name, _function in functions:
             _module = inspect.getmodule(_function)
             if module == _module:
-                setattr(_module, name, span(_function))
+                setattr(_module, name, stats_deco_auto(_function))
 
         # Add telemetry to methods
         for _, _class in classes:
             for name, method in inspect.getmembers(_class, predicate=inspect.isfunction):
-                if inspect.getmodule(_class) == module:
-                    if not name.startswith("_"):
-                        setattr(_class, name, span(method))
+                if inspect.getmodule(_class) == module and not name.startswith("_"):
+                        setattr(_class, name, stats_deco_auto(method))
 
 
 def install(module_names: list[str]):