You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
224 lines
8.6 KiB
Python
224 lines
8.6 KiB
Python
![]()
2 months ago
|
"""Unix."""
|
||
|
from __future__ import annotations
|
||
|
|
||
|
import os
|
||
|
import sys
|
||
|
from configparser import ConfigParser
|
||
|
from pathlib import Path
|
||
|
|
||
|
from .api import PlatformDirsABC
|
||
|
|
||
|
if sys.platform == "win32":
|
||
|
|
||
|
def getuid() -> int:
|
||
|
msg = "should only be used on Unix"
|
||
|
raise RuntimeError(msg)
|
||
|
|
||
|
else:
|
||
|
from os import getuid
|
||
|
|
||
|
|
||
|
class Unix(PlatformDirsABC):
|
||
|
"""
|
||
|
On Unix/Linux, we follow the
|
||
|
`XDG Basedir Spec <https://specifications.freedesktop.org/basedir-spec/basedir-spec-latest.html>`_. The spec allows
|
||
|
overriding directories with environment variables. The examples show are the default values, alongside the name of
|
||
|
the environment variable that overrides them. Makes use of the
|
||
|
`appname <platformdirs.api.PlatformDirsABC.appname>`,
|
||
|
`version <platformdirs.api.PlatformDirsABC.version>`,
|
||
|
`multipath <platformdirs.api.PlatformDirsABC.multipath>`,
|
||
|
`opinion <platformdirs.api.PlatformDirsABC.opinion>`,
|
||
|
`ensure_exists <platformdirs.api.PlatformDirsABC.ensure_exists>`.
|
||
|
"""
|
||
|
|
||
|
@property
|
||
|
def user_data_dir(self) -> str:
|
||
|
"""
|
||
|
:return: data directory tied to the user, e.g. ``~/.local/share/$appname/$version`` or
|
||
|
``$XDG_DATA_HOME/$appname/$version``
|
||
|
"""
|
||
|
path = os.environ.get("XDG_DATA_HOME", "")
|
||
|
if not path.strip():
|
||
|
path = os.path.expanduser("~/.local/share") # noqa: PTH111
|
||
|
return self._append_app_name_and_version(path)
|
||
|
|
||
|
@property
|
||
|
def site_data_dir(self) -> str:
|
||
|
"""
|
||
|
:return: data directories shared by users (if `multipath <platformdirs.api.PlatformDirsABC.multipath>` is
|
||
|
enabled and ``XDG_DATA_DIR`` is set and a multi path the response is also a multi path separated by the OS
|
||
|
path separator), e.g. ``/usr/local/share/$appname/$version`` or ``/usr/share/$appname/$version``
|
||
|
"""
|
||
|
# XDG default for $XDG_DATA_DIRS; only first, if multipath is False
|
||
|
path = os.environ.get("XDG_DATA_DIRS", "")
|
||
|
if not path.strip():
|
||
|
path = f"/usr/local/share{os.pathsep}/usr/share"
|
||
|
return self._with_multi_path(path)
|
||
|
|
||
|
def _with_multi_path(self, path: str) -> str:
|
||
|
path_list = path.split(os.pathsep)
|
||
|
if not self.multipath:
|
||
|
path_list = path_list[0:1]
|
||
|
path_list = [self._append_app_name_and_version(os.path.expanduser(p)) for p in path_list] # noqa: PTH111
|
||
|
return os.pathsep.join(path_list)
|
||
|
|
||
|
@property
|
||
|
def user_config_dir(self) -> str:
|
||
|
"""
|
||
|
:return: config directory tied to the user, e.g. ``~/.config/$appname/$version`` or
|
||
|
``$XDG_CONFIG_HOME/$appname/$version``
|
||
|
"""
|
||
|
path = os.environ.get("XDG_CONFIG_HOME", "")
|
||
|
if not path.strip():
|
||
|
path = os.path.expanduser("~/.config") # noqa: PTH111
|
||
|
return self._append_app_name_and_version(path)
|
||
|
|
||
|
@property
|
||
|
def site_config_dir(self) -> str:
|
||
|
"""
|
||
|
:return: config directories shared by users (if `multipath <platformdirs.api.PlatformDirsABC.multipath>`
|
||
|
is enabled and ``XDG_DATA_DIR`` is set and a multi path the response is also a multi path separated by the OS
|
||
|
path separator), e.g. ``/etc/xdg/$appname/$version``
|
||
|
"""
|
||
|
# XDG default for $XDG_CONFIG_DIRS only first, if multipath is False
|
||
|
path = os.environ.get("XDG_CONFIG_DIRS", "")
|
||
|
if not path.strip():
|
||
|
path = "/etc/xdg"
|
||
|
return self._with_multi_path(path)
|
||
|
|
||
|
@property
|
||
|
def user_cache_dir(self) -> str:
|
||
|
"""
|
||
|
:return: cache directory tied to the user, e.g. ``~/.cache/$appname/$version`` or
|
||
|
``~/$XDG_CACHE_HOME/$appname/$version``
|
||
|
"""
|
||
|
path = os.environ.get("XDG_CACHE_HOME", "")
|
||
|
if not path.strip():
|
||
|
path = os.path.expanduser("~/.cache") # noqa: PTH111
|
||
|
return self._append_app_name_and_version(path)
|
||
|
|
||
|
@property
|
||
|
def site_cache_dir(self) -> str:
|
||
|
""":return: cache directory shared by users, e.g. ``/var/tmp/$appname/$version``"""
|
||
|
return self._append_app_name_and_version("/var/tmp") # noqa: S108
|
||
|
|
||
|
@property
|
||
|
def user_state_dir(self) -> str:
|
||
|
"""
|
||
|
:return: state directory tied to the user, e.g. ``~/.local/state/$appname/$version`` or
|
||
|
``$XDG_STATE_HOME/$appname/$version``
|
||
|
"""
|
||
|
path = os.environ.get("XDG_STATE_HOME", "")
|
||
|
if not path.strip():
|
||
|
path = os.path.expanduser("~/.local/state") # noqa: PTH111
|
||
|
return self._append_app_name_and_version(path)
|
||
|
|
||
|
@property
|
||
|
def user_log_dir(self) -> str:
|
||
|
""":return: log directory tied to the user, same as `user_state_dir` if not opinionated else ``log`` in it"""
|
||
|
path = self.user_state_dir
|
||
|
if self.opinion:
|
||
|
path = os.path.join(path, "log") # noqa: PTH118
|
||
|
return path
|
||
|
|
||
|
@property
|
||
|
def user_documents_dir(self) -> str:
|
||
|
""":return: documents directory tied to the user, e.g. ``~/Documents``"""
|
||
|
return _get_user_media_dir("XDG_DOCUMENTS_DIR", "~/Documents")
|
||
|
|
||
|
@property
|
||
|
def user_downloads_dir(self) -> str:
|
||
|
""":return: downloads directory tied to the user, e.g. ``~/Downloads``"""
|
||
|
return _get_user_media_dir("XDG_DOWNLOAD_DIR", "~/Downloads")
|
||
|
|
||
|
@property
|
||
|
def user_pictures_dir(self) -> str:
|
||
|
""":return: pictures directory tied to the user, e.g. ``~/Pictures``"""
|
||
|
return _get_user_media_dir("XDG_PICTURES_DIR", "~/Pictures")
|
||
|
|
||
|
@property
|
||
|
def user_videos_dir(self) -> str:
|
||
|
""":return: videos directory tied to the user, e.g. ``~/Videos``"""
|
||
|
return _get_user_media_dir("XDG_VIDEOS_DIR", "~/Videos")
|
||
|
|
||
|
@property
|
||
|
def user_music_dir(self) -> str:
|
||
|
""":return: music directory tied to the user, e.g. ``~/Music``"""
|
||
|
return _get_user_media_dir("XDG_MUSIC_DIR", "~/Music")
|
||
|
|
||
|
@property
|
||
|
def user_runtime_dir(self) -> str:
|
||
|
"""
|
||
|
:return: runtime directory tied to the user, e.g. ``/run/user/$(id -u)/$appname/$version`` or
|
||
|
``$XDG_RUNTIME_DIR/$appname/$version``.
|
||
|
|
||
|
For FreeBSD/OpenBSD/NetBSD, it would return ``/var/run/user/$(id -u)/$appname/$version`` if
|
||
|
exists, otherwise ``/tmp/runtime-$(id -u)/$appname/$version``, if``$XDG_RUNTIME_DIR``
|
||
|
is not set.
|
||
|
"""
|
||
|
path = os.environ.get("XDG_RUNTIME_DIR", "")
|
||
|
if not path.strip():
|
||
|
if sys.platform.startswith(("freebsd", "openbsd", "netbsd")):
|
||
|
path = f"/var/run/user/{getuid()}"
|
||
|
if not Path(path).exists():
|
||
|
path = f"/tmp/runtime-{getuid()}" # noqa: S108
|
||
|
else:
|
||
|
path = f"/run/user/{getuid()}"
|
||
|
return self._append_app_name_and_version(path)
|
||
|
|
||
|
@property
|
||
|
def site_data_path(self) -> Path:
|
||
|
""":return: data path shared by users. Only return first item, even if ``multipath`` is set to ``True``"""
|
||
|
return self._first_item_as_path_if_multipath(self.site_data_dir)
|
||
|
|
||
|
@property
|
||
|
def site_config_path(self) -> Path:
|
||
|
""":return: config path shared by the users. Only return first item, even if ``multipath`` is set to ``True``"""
|
||
|
return self._first_item_as_path_if_multipath(self.site_config_dir)
|
||
|
|
||
|
@property
|
||
|
def site_cache_path(self) -> Path:
|
||
|
""":return: cache path shared by users. Only return first item, even if ``multipath`` is set to ``True``"""
|
||
|
return self._first_item_as_path_if_multipath(self.site_cache_dir)
|
||
|
|
||
|
def _first_item_as_path_if_multipath(self, directory: str) -> Path:
|
||
|
if self.multipath:
|
||
|
# If multipath is True, the first path is returned.
|
||
|
directory = directory.split(os.pathsep)[0]
|
||
|
return Path(directory)
|
||
|
|
||
|
|
||
|
def _get_user_media_dir(env_var: str, fallback_tilde_path: str) -> str:
|
||
|
media_dir = _get_user_dirs_folder(env_var)
|
||
|
if media_dir is None:
|
||
|
media_dir = os.environ.get(env_var, "").strip()
|
||
|
if not media_dir:
|
||
|
media_dir = os.path.expanduser(fallback_tilde_path) # noqa: PTH111
|
||
|
|
||
|
return media_dir
|
||
|
|
||
|
|
||
|
def _get_user_dirs_folder(key: str) -> str | None:
|
||
|
"""Return directory from user-dirs.dirs config file. See https://freedesktop.org/wiki/Software/xdg-user-dirs/."""
|
||
|
user_dirs_config_path = Path(Unix().user_config_dir) / "user-dirs.dirs"
|
||
|
if user_dirs_config_path.exists():
|
||
|
parser = ConfigParser()
|
||
|
|
||
|
with user_dirs_config_path.open() as stream:
|
||
|
# Add fake section header, so ConfigParser doesn't complain
|
||
|
parser.read_string(f"[top]\n{stream.read()}")
|
||
|
|
||
|
if key not in parser["top"]:
|
||
|
return None
|
||
|
|
||
|
path = parser["top"][key].strip('"')
|
||
|
# Handle relative home paths
|
||
|
return path.replace("$HOME", os.path.expanduser("~")) # noqa: PTH111
|
||
|
|
||
|
return None
|
||
|
|
||
|
|
||
|
__all__ = [
|
||
|
"Unix",
|
||
|
]
|