Implement caller inference to allow files() to be called without any parameter.

Author: jaraco

importlib_resources/_common.py

@@ -5,6 +5,8 @@
 import contextlib
 import types
 import importlib
+import inspect
+import itertools
 
 from typing import Union, Optional, cast
 from .abc import ResourceReader, Traversable
@@ -15,7 +17,7 @@
 Anchor = Package
 
 
-def files(package: Anchor) -> Traversable:
+def files(package: Optional[Anchor] = None) -> Traversable:
     """
     Get a Traversable resource for an anchor.
     """
@@ -39,7 +41,7 @@ def get_resource_reader(package: types.ModuleType) -> Optional[ResourceReader]:
 
 
 @functools.singledispatch
-def resolve(cand: Anchor) -> types.ModuleType:
+def resolve(cand: Optional[Anchor]) -> types.ModuleType:
     return cast(types.ModuleType, cand)
 
 
@@ -48,6 +50,28 @@ def _(cand: str) -> types.ModuleType:
     return importlib.import_module(cand)
 
 
+@resolve.register
+def _(cand: None) -> types.ModuleType:
+    return resolve(_infer_caller().f_globals['__name__'])
+
+
+def _infer_caller():
+    """
+    Walk the stack and find the frame of the first caller not in this module.
+    """
+
+    def is_this_file(frame_info):
+        return frame_info.filename == __file__
+
+    def is_wrapper(frame_info):
+        return frame_info.function == 'wrapper'
+
+    not_this_file = itertools.filterfalse(is_this_file, inspect.stack())
+    # also exclude 'wrapper' due to singledispatch in the call stack
+    callers = itertools.filterfalse(is_wrapper, not_this_file)
+    return next(callers).frame
+
+
 def from_package(package: types.ModuleType):
     """
     Return a Traversable object for the given package.

importlib_resources/tests/test_files.py

@@ -73,7 +73,6 @@ def test_module_resources(self):
 
 
 class ImplicitContextFilesTests(SiteDir, unittest.TestCase):
-    @__import__('pytest').mark.xfail(reason="work in progress")
     def test_implicit_files(self):
         """
         Without any parameter, files() will infer the location as the caller.