U Mh@sTddlmZmZddlmZmZdgZGdddZd eeedded d d Z d S))DictList) GlobGroup GlobPattern Directoryc@sreZdZdZeedddZeeddddZedd d Z eed d d Z ddZ deeeedddZ dS)rzA file structure representation. Organized as Directory nodes that have lists of their Directory children. Directories for a package are created by calling :meth:`PackageImporter.file_structure`.)nameis_dircCs||_||_i|_dS)N)rr children)selfrr r ]/var/www/html/venv/lib/python3.8/site-packages/torch/package/file_structure_representation.py__init__szDirectory.__init__)dirsreturncCsJt|dkr|S|d}||jkr2t|d|j|<|j||ddS)a$Builds path of Directories if not yet built and returns last directory in list. Args: dirs (List[str]): List of directory names that are treated like a path. Returns: :class:`Directory`: The last Directory specified in the dirs list. rTrN)lenr r_get_dir)r rdir_namer r r rs  zDirectory._get_dir) file_pathcCs.|d^}}||}t|d|j|<dS)zAdds a file to a Directory. Args: file_path (str): Path of file to add. Last element is added as a file while other paths items are added as directories. /FN)splitrrr )r rrfiledirr r r _add_file$s zDirectory._add_file)filenamercCs\|jddd}|d}t|dkr*|dnd}||jkrX|dkrHdS|j||SdS)zChecks if a file is present in a :class:`Directory`. Args: filename (str): Path of file to search for. Returns: bool: If a :class:`Directory` contains the specified file. rr)maxsplitrNTF)rrr keyshas_file)r rZlineagechildZ grandchildrenr r r r/szDirectory.has_filecCsg}||d|S)N)_stringify_treejoin)r str_listr r r __str__As zDirectory.__str__r ─── )r"preambledir_ptrcCs$d}d}d}d}||||jd||kr<||}n||}g}g} |jD]$\} } | jrp| | qV|| qVtt| D]N\} } | t| dkrt|dkr|j| |||q|j| |||qtt|D]:\} } | t|dkr|n|}|||| dqdS) zCRecursive method to generate print-friendly version of a Directory.z u│ u ├── u └──  rrN) appendrr itemsr enumeratesortedrr )r r"r%r&spacebranchteelastZ file_keysZdir_keyskeyvalindexrZpointerr r r r Fs*   zDirectory._stringify_treeN)rr$) __name__ __module__ __qualname____doc__strboolrrrrrr#r r r r r r s **r r)r file_listincludeexcludercCs:t||dd}t|d}|D]}||r||q|S)aReturn a :class:`Directory` file structure representation created from a list of files. Args: filename (str): The name given to the top-level directory that will be the relative root for all file paths found in the file_list. file_list (List[str]): List of files to add to the top-level directory. include (Union[List[str], str]): An optional pattern that limits what is included from the file_list to files whose name matches the pattern. exclude (Union[List[str], str]): An optional pattern that excludes files whose name match the pattern. Returns: :class:`Directory`: a :class:`Directory` file structure representation created from a list of files. r)r< separatorT)rrmatchesr)rr:r;r<Z glob_patternZtop_dirrr r r _create_directory_from_file_listms    r?N)r9r ) typingrrZ glob_grouprr__all__rr7r?r r r r sg