Add audeer.script_dir()

audeer/__init__.py

@@ -14,6 +14,7 @@
 from audeer.core.io import move_file
 from audeer.core.io import replace_file_extension
 from audeer.core.io import rmdir
+from audeer.core.io import script_dir
 from audeer.core.io import touch
 from audeer.core.path import path
 from audeer.core.path import safe_path

audeer/core/io.py

@@ -1,6 +1,7 @@
 import errno
 import fnmatch
 import hashlib
+import inspect
 import itertools
 import os
 import platform
@@ -1007,6 +1008,30 @@ def rmdir(
         shutil.rmtree(path)
 
 
+def script_dir() -> str:
+    r"""Folder in which caller of this function is located.
+
+    When called from a file,
+    it returns the directory,
+    in which the file is stored.
+    When called in an interactive session,
+    it returns the current working directory
+    of the interactive session.
+
+    Returns:
+        current directory of caller
+
+    Examples:
+        >>> os.path.basename(script_dir())  # folder of docstring test
+        'audeer_core_io_script_dir0'
+
+    """
+    # See https://stackoverflow.com/a/37792573
+    caller = inspect.stack()[1].filename
+    # See https://stackoverflow.com/a/5137509
+    return os.path.dirname(os.path.realpath(caller))
+
+
 def touch(
     path: typing.Union[str, bytes],
     *paths: typing.Sequence[typing.Union[str, bytes]],

docs/api-src/audeer.rst

@@ -39,6 +39,7 @@ audeer
     rmdir
     run_tasks
     safe_path
+    script_dir
     sort_versions
     StrictVersion
     to_list

tests/test_io.py

@@ -1543,6 +1543,23 @@ def test_rmdir(tmpdir):
     assert not os.path.exists(path)
 
 
+def test_script_dir(tmpdir):
+    r"""Test estimation of current directory of caller.
+
+    See https://stackoverflow.com/a/5137509.
+
+    Args:
+        tmpdir: tmpdir fixture
+
+    """
+    expected_script_dir = os.path.dirname(os.path.realpath(__file__))
+    assert audeer.script_dir() == expected_script_dir
+    current_dir = os.getcwd()
+    os.chdir(tmpdir)
+    assert audeer.script_dir() == expected_script_dir
+    os.chdir(current_dir)
+
+
 def test_touch(tmpdir):
     path = audeer.mkdir(tmpdir, "folder1")
     path = os.path.join(path, "file")