Use explicit stacklevel argument

python · serhiy-storchaka · Feb 6, 2024 · Jun 29, 2023 · Aug 28, 2023 · Feb 5, 2024
commit 4ce91237038839715e6764a4ebfd6754396a3418
diff --git a/Lib/zoneinfo/_tzpath.py b/Lib/zoneinfo/_tzpath.py
@@ -2,7 +2,7 @@
 import sysconfig
 
 
-def reset_tzpath(to=None):
+def _reset_tzpath(to=None, stacklevel=4):
     global TZPATH
 
     tzpaths = to
@@ -19,16 +19,25 @@ def reset_tzpath(to=None):
     else:
         env_var = os.environ.get("PYTHONTZPATH", None)
         if env_var is not None:
-            base_tzpath = _parse_python_tzpath(env_var)
+            base_tzpath = _parse_python_tzpath(env_var, stacklevel)
         else:
             base_tzpath = _parse_python_tzpath(
-                sysconfig.get_config_var("TZPATH")
+                sysconfig.get_config_var("TZPATH"),
+                stacklevel,
             )
 
     TZPATH = tuple(base_tzpath)
 
 
-def _parse_python_tzpath(env_var):
+def reset_tzpath(to=None):
+    """Reset global TZPATH."""
+    # We need `_reset_tzpath` helper function because it produces a warning,
+    # it is used as both a module-level call and a public API.
+    # This is how we equalize the stacklevel for both calls.
+    _reset_tzpath(to)
+
+
+def _parse_python_tzpath(env_var, stacklevel):
     if not env_var:
         return ()
 
@@ -45,7 +54,7 @@ def _parse_python_tzpath(env_var):
             "Invalid paths specified in PYTHONTZPATH environment variable. "
             + msg,
             InvalidTZPathWarning,
-            stacklevel=5,
+            stacklevel=stacklevel,
         )
 
     return new_tzpath
@@ -173,4 +182,4 @@ class InvalidTZPathWarning(RuntimeWarning):
 
 
 TZPATH = ()
-reset_tzpath()
+_reset_tzpath(stacklevel=5)