a d%@sdZddlZddlZddlZddlZddlZddlmZddlm Z ddl m Z ddl m Z dZ dZd Zd d Zd d ZddZe jfddZddZd"ddZddZddZddZddZd d!ZdS)#zyInterface to file resources. This module provides functions for interfacing with files: opening, writing, and querying. N) ConfigParser)detect_encoding)errors)style  z cCsg}tj|rt|tjrt|d:}|D]$}|r.|ds.||q.Wdn1sh0Yt dd|Drt d|S)z7Get a list of file patterns to ignore from .yapfignore.r#Ncss|]}|dVqdSz./N startswith.0erd/var/www/html/stable-diffusion-webui/venv/lib/python3.9/site-packages/yapf/yapflib/file_resources.py -z4_GetExcludePatternsFromYapfIgnore..z,path in .yapfignore should not start with ./) ospathisfileaccessR_OKopenstripr appendanyr YapfError)filenameignore_patternsfdlinerrr!_GetExcludePatternsFromYapfIgnore$s . r"cCsg}z ddl}Wnty,tdYn0tj|rt|tjrt |d}| |}Wdn1sr0Y| di di dg}t dd |Drtd |S) z:Get a list of file patterns to ignore from pyproject.toml.rNHtomli package is needed for using pyproject.toml as a configuration filerbtoolZ yapfignorercss|]}|dVqdSr r r rrrrCrz7_GetExcludePatternsFromPyprojectToml..z/path in pyproject.toml should not start with ./) tomli ImportErrorrrrrrrrrloadgetr)rrtomllibr pyproject_tomlrrr$_GetExcludePatternsFromPyprojectToml3s*    ( r,cCsTg}tj|d}tj|r*|t|7}tj|d}tj|rP|t|7}|S)a%Return patterns of files to exclude from ignorefile in a given directory. Looks for .yapfignore in the directory dirname. Arguments: dirname: (unicode) The name of the directory. Returns: A List of file patterns to exclude if ignore file is found, otherwise empty List. z .yapfignorezpyproject.toml)rrjoinexistsr"r,)dirnamerZyapfignore_fileZpyproject_toml_filerrrGetExcludePatternsForDirIs     r0c Cstj|}tj|tj}tj|r,|Stj|tj}z t|}Wnt yZYnR0|<t }| || dr|WdSWdn1s0Ytj|tj }zt|d}Wnt yYn0|vz ddl}WntytdYn0||}|didd}|durJ|WdSWdn1s`0Y|rtj|r|tjtjjkrqtj|}q tjtj} tj| r| S|S)aReturn default style name for a given directory. Looks for .style.yapf or setup.cfg or pyproject.toml in the parent directories. Arguments: dirname: (unicode) The name of the directory. default_style: The style to return if nothing is found. Defaults to the global default style ('pep8') unless otherwise specified. Returns: The filename if found, otherwise return the default style. ZyapfNr$rr#r%)rrabspathr-rZ LOCAL_STYLEr.Z SETUP_CONFIGrIOErrorr read_file has_sectionZPYPROJECT_TOMLr&r'rrr(r)basenamesepr/ expanduserZ GLOBAL_STYLE) r/Z default_styleZ style_file config_filer configr*r+Z style_dictZ global_filerrrGetDefaultStyleForDirasN      0     2r:cCs t|||S)z7Return the list of files specified on the command line.)_FindPythonFiles)Zcommand_line_file_list recursiveexcluderrrGetCommandLineFilessr>FcCsX|r@tj|d|d}||WdqT1s40Yntjj|ddS)axEmit the reformatted code. Write the reformatted code into the file, if in_place is True. Otherwise, write to stdout. Arguments: filename: (unicode) The name of the unformatted file. reformatted_code: (unicode) The reformatted code. encoding: (unicode) The encoding of the file. in_place: (bool) If True, then write the reformatted code to the file. wmodeencodingNzutf-8)codecsrwritesysstdoutbufferencode)rZreformatted_coderCZin_placer rrrWriteReformattedCodes*rJcCstdtdtdi}|D]V}|tr4|td7<q|trP|td7<q|tr|td7<qttttf|jdddS)z0Retrieve the line ending of the original source.rT)keyreverse)CRLFCRLFendswithsortedr))linesZendingsr!rrr LineEndings   rTc s||r tdd|Dr td|o0dd|D}g}|D]:}|dkrX|rXt||rXq:tj|r`|sxtd|g}t|D]\}dkr|rt|r|qntfdd|Drq|D]4}tj |}|rt||rqt |r||qЇfd dt t D} | D]6} tj | } t| |rN|| n | q$qq:tj |r:||q:|S) zFind all Python files.css|]}|dVqdSr r r rrrrrz#_FindPythonFiles..z,path in '--exclude' should not start with ./cSsg|]}|dtjjqS)/)rstriprrr6r rrr rz$_FindPythonFiles...z2directory specified without '--recursive' flag: %sc3s|]}|VqdSNr r )dirpathrrrrcsg|]}dqS)r)pop)ri)dirnamesrrrWr)rrr IsIgnoredrrisdirwalkrr- IsPythonFilerangelenr) filenamesr<r=Z python_filesrZ excluded_dirsfilelistffilepathZ dirnames_r/Zdir_r)r]rZrr;sD        r;csP|dur dStjjdtjjr:ddqtfdd|DS)z8Return True if filename matches any patterns in exclude.NFrXc3s$|]}t|tjjVqdSrY)fnmatchrVrrr6r rrrrrzIsIgnored..)lstriprrr6r r)rr=rrjrr^s r^c Cstj|ddkrdSztt|d}t|jd}Wdn1sH0Ytj|d|d}|Wdn1s0YWn,tyd }Ynt t fyYd S0z>tj|d|d}|d }Wdn1s0YWnt yYd S0t d |S) z)Return True if filename is a Python file.rKz.pyTr$rNrrAzlatin-1Fz^#!.*\bpython[23]?\b) rrsplitextrrreadlinerDreadUnicodeDecodeErrorr2 SyntaxErrorrematch)rr rC first_linerrrras" ,* ,racCs<t|d}t|jdWdS1s.0YdS)zReturn the file's encoding.r$rN)rrrn)rr rrr FileEncoding"s ru)r?F)__doc__rDrirrrrF configparserrtokenizerZ yapf.yapflibrrrOrPrNr"r,r0Z DEFAULT_STYLEr:r>rJrTr;r^rarurrrrs2    B  0