aswi2021jmsd-gitlab/src/utils/file_anchor.py @ 5b396f6f

import os


class FileAnchor:
    def __init__(self, root_dir, tail, searching=False):
        self.__root_dir = root_dir
        self.__tail = tail
        if searching:
            self.shortest_relative_path = self.__srp_with_tree_search
        else:
            self.shortest_relative_path = self.__srp

    def __srp_with_tree_search(self, src=None):
        """
        Shortest relative path handling the situation where src is not a descendant of root_dir.
        Searches the subtree of src until it finds root_dir.
        :param src: source node or cwd
        :return: path from source to the anchored file
        """
        raise NotImplementedError("Not implemented yet.")

    def __srp(self, src=None):
        """
        Shortest relative path from src to the anchored file. If src is not a descendant of root_dir, returns None.
        :param src: source node or cwd
        :return: path from source to the anchored file or None
        """
        if src is None:
            src = os.getcwd()

        ret = ""
        h = src
        while len(h) > 0 and h != "/":
            h, t = os.path.split(h)
            if t != self.__root_dir:
                ret = os.path.join(ret, "..")
            else:
                return os.path.join(ret, self.__tail)

        return None

    def __str__(self):
        return self.shortest_relative_path(None)