o h\ @s`ddlmZmZddlmZmZdgZGdddZ  dded eed d d d d ef ddZ dS))DictList) GlobGroup GlobPattern Directoryc@s~eZdZdZdedefddZdeeddfdd Zd efd d Z d edefddZ ddZ ddeededefddZ 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 a/var/www/html/ai/venv/lib/python3.10/site-packages/torch/package/file_structure_representation.py__init__ s zDirectory.__init__dirsreturncCsJt|dkr|S|d}||jvrt|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_filefilenamecCs\|jddd}|d}t|dkr|dnd}||jvr,|dur$dS|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 rlineagechild grandchildrenr r r r.szDirectory.has_filecCsg}||d|S)N)_stringify_treejoin)r str_listr r r __str__@s  zDirectory.__str__r! ─── r$preambledir_ptrcCs"d}d}d}d}||||jd||kr||}n||}g}g} |jD]\} } | jr8| | q+|| q+tt| D]'\} } | t| dkrat|dkra|j| |||qD|j| |||qDtt|D]\} } | t|dkr|n|}|||| dqrdS) zCRecursive method to generate print-friendly version of a Directory.z u│ u ├── u └──  rrN) appendrr itemsr enumeratesortedrr")r r$r'r(spacebranchteelast file_keysdir_keyskeyvalindexrpointerr r r r"Es,   zDirectory._stringify_treeN)r!r&) __name__ __module__ __qualname____doc__strboolrrrrrr%r"r r r r rs **r r file_listincluderexcludercCs: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)rA separatorT)rrmatchesr)rr?r@rA glob_patterntop_dirrr r r _create_directory_from_file_listis   rFN)r>r ) typingrr glob_grouprr__all__rr<rFr r r r s"d