o wZŽh‘ã @sPddlmZmZdgZGdd„dƒZ  ddedeedd d d d ef d d „ZdS)é)Ú GlobGroupÚ GlobPatternÚ Directoryc@s€eZdZdZdedefdd„Zdeeddfdd „Zd efd d „Z d edefdd„Z dd„Z  ddeededefdd„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 úZ/var/www/auris/lib/python3.10/site-packages/torch/package/file_structure_representation.pyÚ__init__s zDirectory.__init__ÚdirsÚreturncCsJt|ƒdkr|S|d}||jvrt|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ÚfilenamecCs\|jddd}|d}t|ƒdkr|dnd}||j ¡vr,|dur$dS|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]\} } | jr8|  | ¡q+| | ¡q+tt| ƒƒD]'\} } | t| ƒdkrat|ƒdkra|j|  |||¡qD|j|  |||¡qDtt|ƒƒD]\} } | t|ƒdkr€|n|}| |›|›| ›d¡qrdS) 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ÚindexrZpointerr r r rFs,   þzDirectory._stringify_treeN)rr") Ú__name__Ú __module__Ú __qualname__Ú__doc__ÚstrÚboolr Úlistrrrr!rr r r r r s  üþýüú**r rÚ file_listÚincluderÚ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)rr9r:r;Z glob_patternZtop_dirrr r r Ú _create_directory_from_file_listms   €r>N)r8r )Z glob_grouprrÚ__all__rr5r7r>r r r r Ús güÿþýüû