File size: 3,288 Bytes
bbae060 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 | try:
import zoneinfo
reset_tzpath = zoneinfo.reset_tzpath
InvalidTZPathWarning = zoneinfo.InvalidTZPathWarning
def __getattr__(name):
if name == "TZPATH":
return zoneinfo.TZPATH
else:
raise AttributeError(
"Module %s has no attribute %s" % (repr(__name__), repr(name))
)
except ImportError:
import os
import sys
import warnings
import six
class InvalidTZPathWarning(RuntimeWarning):
"""Warning raised if an invalid path is specified in PYTHONTZPATH."""
def _parse_python_tzpath(env_var, stacklevel):
if not env_var:
return ()
raw_tzpath = env_var.split(os.pathsep)
new_tzpath = tuple(filter(os.path.isabs, raw_tzpath))
# If anything has been filtered out, we will warn about it
if len(new_tzpath) != len(raw_tzpath):
msg = _get_invalid_paths_message(raw_tzpath)
warnings.warn(
"Invalid paths specified in PYTHONTZPATH environment variable. "
+ msg,
InvalidTZPathWarning,
stacklevel=stacklevel,
)
return new_tzpath
def _get_invalid_paths_message(tzpaths):
invalid_paths = (path for path in tzpaths if not os.path.isabs(path))
prefix = "\n "
indented_str = prefix + prefix.join(invalid_paths)
return (
"Paths should be absolute but found the following relative paths:"
+ indented_str
)
def _reset_tzpath(to=None, stacklevel=4):
global TZPATH
tzpaths = to
if tzpaths is not None:
if isinstance(tzpaths, six.string_types + (bytes,)):
raise TypeError(
"tzpaths must be a sequence, not %s: %s"
% (
type(tzpaths),
repr(tzpaths),
)
)
# Path-like objects not supported by os.path operations before 3.6
if sys.version_info < (3, 6):
tzpaths = tuple(map(str, tzpaths))
if not all(map(os.path.isabs, tzpaths)):
raise ValueError(_get_invalid_paths_message(tzpaths))
base_tzpath = tzpaths
else:
env_var = os.environ.get("PYTHONTZPATH", None)
if env_var is not None:
base_tzpath = _parse_python_tzpath(env_var, stacklevel)
elif sys.platform != "win32":
base_tzpath = (
"/usr/share/zoneinfo",
"/usr/lib/zoneinfo",
"/usr/share/lib/zoneinfo",
"/etc/zoneinfo",
)
else:
base_tzpath = ()
TZPATH = tuple(base_tzpath)
if TZPATH_CALLBACKS:
for callback in TZPATH_CALLBACKS:
callback(TZPATH)
def reset_tzpath(to=None):
# As in CPython, the warning about PYTHONTZPATH is raised from a
# helper that is used both at import time and by this function, so
# that each can pass the stacklevel that points at its caller.
_reset_tzpath(to)
TZPATH = ()
TZPATH_CALLBACKS = []
_reset_tzpath(stacklevel=5)
|