| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181 | from __future__ import annotationsimport osimport sysfrom configparser import ConfigParserfrom pathlib import Pathfrom .api import PlatformDirsABCif sys.platform.startswith("linux"):  # pragma: no branch # no op check, only to please the type checker    from os import getuidelse:    def getuid() -> int:        raise RuntimeError("should only be used on Linux")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>`.    """    @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")        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]        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")        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")        return self._append_app_name_and_version(path)    @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")        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")        return path    @property    def user_documents_dir(self) -> str:        """        :return: documents directory tied to the user, e.g. ``~/Documents``        """        documents_dir = _get_user_dirs_folder("XDG_DOCUMENTS_DIR")        if documents_dir is None:            documents_dir = os.environ.get("XDG_DOCUMENTS_DIR", "").strip()            if not documents_dir:                documents_dir = os.path.expanduser("~/Documents")        return documents_dir    @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``        """        path = os.environ.get("XDG_RUNTIME_DIR", "")        if not path.strip():            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)    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_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 = os.path.join(Unix().user_config_dir, "user-dirs.dirs")    if os.path.exists(user_dirs_config_path):        parser = ConfigParser()        with open(user_dirs_config_path) 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        path = path.replace("$HOME", os.path.expanduser("~"))        return path    return None__all__ = [    "Unix",]
 |