get source_file of wrapped functions (#657)

mitmproxy · Jan 17, 2024 · 8750631 · 8750631
1 parent 8c15000
commit 8750631
Show file tree

Hide file tree

Showing 3 changed files with 14 additions and 0 deletions.
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -4,6 +4,8 @@
 
 ## Unreleased: pdoc next
 
+ - pdoc now correctly extracts the source_file of wrapped functions.
+   ([#657](https://github.com/mitmproxy/pdoc/pull/657), @tmeyier)
 
 ## 2023-12-22: pdoc 14.3.0
 

diff --git a/pdoc/doc.py b/pdoc/doc.py
@@ -887,6 +887,8 @@ def __init__(
             unwrapped = func.__func__  # type: ignore
         elif isinstance(func, singledispatchmethod):
             unwrapped = func.func  # type: ignore
+        elif hasattr(func, "__wrapped__"):
+            unwrapped = func.__wrapped__
         else:
             unwrapped = func
         super().__init__(modulename, qualname, unwrapped, taken_from)

diff --git a/test/test_doc.py b/test/test_doc.py
@@ -175,3 +175,13 @@ def test_default_value_masks_env_vars(monkeypatch):
         assert v2.default_value_str == "'42.0.1'"
     finally:
         _environ_lookup.cache_clear()
+
+
+def test_source_file_method():
+    mod = extract.load_module(extract.parse_spec(here / "testdata" / "demo_long.py"))
+
+    m = Module(mod)
+
+    assert m.members["Foo"].members["a_cached_function"].source_file == (
+        here / "testdata" / "demo_long.py"
+    )