a ¢º”h‘ã@sDddlmZmZdgZGdd„dƒZd eeeddedœdd „Zd S) é)Ú GlobGroupÚ GlobPatternÚ Directoryc@sreZdZdZeedœdd„Zeeddœdd„Zedœd d „Z eed œd d „Z dd„Z deeeedœdd„Z dS)rzËA 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`.)ÚnameÚis_dircCs||_||_i|_dS)N)rrÚchildren)Úselfrr©r úY/var/www/auris/lib/python3.9/site-packages/torch/package/file_structure_representation.pyÚ__init__szDirectory.__init__)ÚdirsÚreturncCsJt|ƒdkr|S|d}||jvr2t|dƒ|j|<|j| |dd…¡S)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. éTrN)ÚlenrrÚ_get_dir)rr Údir_namer r r rs  zDirectory._get_dir)Ú file_pathcCs.| d¡^}}| |¡}t|dƒ|j|<dS)zÇAdds 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)rrr ÚfileÚdirr r r Ú _add_file$s zDirectory._add_file)Úfilenamer cCs\|jddd}|d}t|ƒdkr*|dnd}||j ¡vrX|durHdS|j| |¡SdS)zÜChecks 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ÚkeysÚhas_file)rrZlineageÚchildZ grandchildrenr r r r/szDirectory.has_filecCsg}| |¡d |¡S)NÚ)Ú_stringify_treeÚjoin)rÚstr_listr r r Ú__str__As zDirectory.__str__rõ ─── )r ÚpreambleÚdir_ptrcCs$d}d}d}d}| |›|›|j›d¡||kr<||}n||}g}g} |j ¡D]$\} } | jrp|  | ¡qV| | ¡qVtt| ƒƒD]N\} } | t| ƒdkrÂt|ƒdkrÂ|j|  |||¡qˆ|j|  |||¡qˆtt|ƒƒD]:\} } | t|ƒdkr|n|}| |›|›| ›d¡qädS) zCRecursive method to generate print-friendly version of a Directory.z u│ u ├── u └── Ú rrN) ÚappendrrÚitemsrÚ enumerateÚsortedrr)rr r#r$ÚspaceÚbranchÚteeÚlastZ file_keysZdir_keysÚkeyÚvalÚindexrÚpointerr r r rFs*   zDirectory._stringify_treeN)rr") Ú__name__Ú __module__Ú __qualname__Ú__doc__ÚstrÚboolr Úlistrrrr!rr r r r r s üüú**r r)rÚ file_listÚincludeÚexcluder cCs:t||dd}t|dƒ}|D]}| |¡r| |¡q|S)aâReturn 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)rrÚmatchesr)rr:r;r<Z glob_patternZtop_dirrr r r Ú _create_directory_from_file_listms    r?N)r9r )Z glob_grouprrÚ__all__rr6r8r?r r r r Úsgüû