o ޥi@sddlZddlZddlmZddlmZmZmZddlm Z dedee deefdd Z dd ee efd ee ee fdefddZ    ddee efde de de de ddf ddZ    dd ee efd ee ee fde de de de defddZdS)N)Path)IterableOptionalUnion) load_dotenvpath indicatorsreturncCsF|D]}t||}t|dkr|Sq|j|krdSt|j|S)axRecursively search for files from the `indicators` list, starting from given path. Args: path (Path): Starting folder path. indicators (Iterable[str]): List of filenames to search for. Raises: FileNotFoundError: If root is not found. Returns: Optional[Path]: Path to folder containing at list one of the files from the list. rN)listgloblenparent_pyrootutils_recursive_search)rrfilefoundrK/home/ubuntu/.local/lib/python3.10/site-packages/pyrootutils/pyrootutils.pyr s    r.z .project-rootz setup.cfgzsetup.pyz.gitzpyproject.toml search_from indicatorcCst|ttfs tdt|}t|tr|g}|s!tdt|dr/tdd|Ds3tdt ||}|r>|sEtd||S)aRecursively searches for project root indicator(s), starting from given path. Args: search_from (str): Path to folder or file to start search from. indicator (Union[str, Iterable[str]], optional): List of filenames to search for. Finding at least one on these files indicates the project root. Raises: TypeError: If any input type is incorrect. FileNotFoundError: If root is not found. Returns: Path: Path to project root. z6search_from must be either a string or pathlib object.z search_from path does not exist.__iter__css|]}t|tVqdS)N) isinstancestr).0irrr Cszfind_root..z.indicator must be a string or list of strings.z.Project root directory not found. Indicators: ) rrr TypeErrorresolveexistsFileNotFoundErrorhasattrallr)rrrrrr find_root!s    r#TFproject_root_env_vardotenv pythonpathcwdcCsnt|}tj|std||r|tjd<|r#ttj|d|r,tj d||r5t |dSdS)aSet given path as a project root. Args: path (Union[str, Path]): Project root path. project_root_env_var (bool, optional): Whether to set PROJECT_ROOT environment variable. dotenv (bool, optional): Whether to load `.env` file from project root. pythonpath (bool, optional): Whether to add project root to pythonpath. cwd (bool, optional): Whether to set current working directory to project root. Raises: FileNotFoundError: If root path does not exist. Returns: None z"Project root path does not exist: PROJECT_ROOTz.envrN) rosrrr environrjoinsysinsertchdirrr$r%r&r'rrrset_rootNs  r0cCs t||}t|||||d|S)aCombines `get_root()` and `set_root()` into one method. Recursively searches for files from the `indicators` list, starting from given path. Args: search_from (str): Path to file or folder to start search from. indicator (Union[str, Iterable[str]], optional): List of filenames to search for. Finding at least one on these files indicates the project root. project_root_env_var (bool, optional): Whether to set PROJECT_ROOT environment variable. dotenv (bool, optional): Whether to load `.env` file from project root. pythonpath (bool, optional): Whether to add project root to pythonpath. cwd (bool, optional): Whether to set current working directory to project root. Raises: TypeError: If any input type is incorrect. FileNotFoundError: If root is not found. Returns: Path: Path to project root. r/)r#r0)rrr$r%r&r'rrrr setup_rootvs !r1)rr)TTFF)rTTFF)r)r,pathlibrtypingrrrr%rrrr#boolr0r1rrrrsh    /  *