o h6@sddlZddlZddlZddlmZddlmZddlmZm Z m Z m Z m Z m Z mZmZmZmZmZddlmZddlmZmZmZddlmZmZddlmZdd lmZej d krqzddl!Z!Wne"ypesnddl#Z!Ynwddl#Z!dd l$m%Z%dd l&m'Z'dd l(m)Z)ddl*m+Z+erddl,Z,e dFdee-dee-deee-ffddZ. dFdee-dfdee-dee-fddZ/eddde-de e-e ffddZ0de e-e fdee e'fddZ1d e-dee e'fd!d"Z2d e-dee e'fd#d$Z3d%edefd&d'Z4edefd(d)Z5ed*edefd+d,Z6 dFd-ed*ed.ee+dee-fd/d0Z7d1e-d*ed2e eefde8fd3d4Z9d5e-d6eee-de8fd7d8Z:d9e ed*ed:ee-d;ee-de8d?e8de efd@dAZ;dBej>sz$find_project_root..cSsg|] }tt|qSr!)rcwdresolve)r#srcr!r!r& Bsz%find_project_root..cSs(g|]}t|j|r|gngqSr!)listparentsis_dir)r#pathr!r!r&r+Fscs|]}t|VqdSN)set)r#r-r!r!r&r'KcSs|jSr1)parts)r/r!r!r&Lsz#find_project_root..)keyz.gitz.git directoryz.hgz .hg directorypyproject.tomlzfile system root) tuplestrrr(r)maxr2 intersectionr-existsr.is_file)rr path_srcs src_parents common_base directoryr!r%r&find_project_root-s*      rBpath_search_start.c Cs~t||\}}|d}|rt|Szt}|r t|WSdWSttfy>}ztd|WYd}~dSd}~ww)z;Find the absolute filepath to a pyproject.toml if it existsr7Nz-Ignoring user configuration directory due to )rBr=r9find_user_pyproject_tomlPermissionError RuntimeErrorr)rCrpath_project_root_path_pyproject_tomlpath_user_pyproject_tomler!r!r&find_pyproject_toml\s rLT) patchable path_configcCst|d }t|}Wdn1swY|didi}dd|D}d|vrBt|}|durBdd |D|d<|S) zParse a pyproject toml file, pulling out relevant parts for Black. If parsing fails, will raise a tomllib.TOMLDecodeError. rbNtoolblackcSs&i|]\}}|dddd|qS)z--r rH)replace)r#kvr!r!r& {s&z(parse_pyproject_toml..target_versioncSsg|]}|jqSr!)namelowerr#rUr!r!r&r+sz(parse_pyproject_toml..)opentomllibloadgetitemsinfer_target_version)rNfpyproject_tomlconfiginferred_target_versionr!r!r&parse_pyproject_tomlrs  rerbc Csf|di}|dd}|dur1zt|WStyYnwzt|WSttfy0YdSwdS)a#Infer Black's target version from the project metadata in pyproject.toml. Supports the PyPA standard format (PEP 621): https://packaging.python.org/en/latest/specifications/declaring-project-metadata/#requires-python If the target version cannot be inferred, returns None. projectzrequires-pythonN)r^parse_req_python_versionrparse_req_python_specifierr)rbproject_metadatarequires_pythonr!r!r&r`s    r`rjc CsFt|}|jddkr dSz t|jdgWSttfy"YdSw)zParse a version string (i.e. ``"3.7"``) to a list of TargetVersion. If parsing fails, will raise a packaging.version.InvalidVersion error. If the parsed version cannot be mapped to a valid TargetVersion, returns None. rrN)rreleaser IndexError ValueError)rjversionr!r!r&rgsrgcsJtt|}|s dSddtDt|}|r#fdd|DSdS)zParse a specifier string (i.e. ``">=3.7,<3.10"``) to a list of TargetVersion. If parsing fails, will raise a packaging.specifiers.InvalidSpecifier error. If the parsed specifier cannot be mapped to a valid TargetVersion, returns None. NcSsi|] }d|j|qS)z3.)valuerZr!r!r&rVsz.parse_req_python_specifier..csg|]}|qSr!r!rZtarget_version_mapr!r&r+sz.parse_req_python_specifier..)strip_specifier_setrrr,filter)rj specifier_setcompatible_versionsr!rqr&rhs rhrucCsg}|D]T}dt|vr||q|jdvr0t|j}t|j|jd|j}||q|jdkrSt|j}t|j dkrMtd|jd|j}||q||qt d dd |DS) zStrip minor versions for some specifiers in the specifier set. For background on version specifiers, see PEP 440: https://peps.python.org/pep-0440/#version-specifiers *)z~=z==>=z===.>rx,csr0r1)r9r"r!r!r&r'r3z&strip_specifier_set..) r9appendoperatorrrormajorminorlenrlrjoin)ru specifiersr$rostrippedr!r!r&rss         rscCsDtjdkrtd}|Stjdd}t|d}|S)a:Return the path to the top-level user configuration for black. This looks for ~\.black on Windows and ~/.config/black on Linux and other Unix systems. May raise: - RuntimeError: if the current user has no homedir - PermissionError: if the current process cannot access the user's homedir win32z.blackXDG_CONFIG_HOMEz ~/.configrQ) sysplatformrhomeosenvironr^ expanduserr))user_config_path config_rootr!r!r&rDs rDrootc Cs|d}g}|r$|jdd }|}Wdn1swYztd|WStyB}z td|d|d}~ww)z8Return a PathSpec matching gitignore content if present.z .gitignorezutf-8)encodingN gitwildmatchzCould not parse z: )r=r[ readlinesr from_linesrr)r gitignorelinesgfrKr!r!r& get_gitignores rr/reportc Csz2|r|nt|}|}z ||}WW|Sty2|r.||d|YWdSwtyP}z|rE||d|WYd}~dSd}~ww)zrNormalize `path`. May return `None` if `path` was ignored. `report` is where "path ignored" output goes. z'is a symbolic link that points outside Nzcannot be read because ) is_absoluterr(r) relative_toas_posixrn path_ignoredOSError)r/rrabspathnormalized_pathroot_relative_pathrKr!r!r&normalize_path_maybe_ignores&   rrgitignore_dictc CsX||}|D]!\}}z ||}Wn ty YdSw||r)dSqdS)NTF)r_rrrn match_file)rrrr/gitignore_pathpattern relative_pathr!r!r&_path_is_ignoreds  rrrcCs$|r||nd}t|o|dS)Nr)searchboolgroup)rrmatchr!r!r&path_is_excluded)srpathsincludeexcludeextend_exclude force_excludeverbosequietc csb|s Jd||D]} | |} |r(t| ||r(|| dqd| } | r4| d7} t| |r@|| dqt| |rL|| dqt| |rX|| dqt| ||} | durcq| r|durxi||| t | i} nd} t | ||||||| || d EdHq| r| j d krt|p| d sq|r|| nd }|r| VqdS) a7Generate all files under `path` whose paths are not excluded by the `exclude_regex`, `extend_exclude`, or `force_exclude` regexes, but are included by the `include` regex. Symbolic links pointing outside of the `root` directory are ignored. `report` is where output about exclusions goes. z/INTERNAL ERROR: `root` must be absolute but is z!matches a .gitignore file content/z(matches the --exclude regular expressionz/matches the --extend-exclude regular expressionz.matches the --force-exclude regular expressionN)rrz.ipynb)warnT)rabsoluterrrrr.rrrgen_python_filesiterdirr=suffixrr)rrrrrrrrrrchildrrnew_gitignore_dict include_matchr!r!r&r1sp         rrazcolorama.AnsiToWin32cCs:zddlm}Wn ty|YSw||dddddS)at Wrap stream with colorama's wrap_stream so colors are shown on Windows. If `colorama` is unavailable, the original stream is returned unmodified. Otherwise, the `wrap_stream()` function determines whether the stream needs to be wrapped for a Windows environment and will accordingly either return an `AnsiToWin32` wrapper or the original stream. r) wrap_streamNFT)convertstrip autoresetwrap)colorama.initialiser ImportError)rarr!r!r&wrap_stream_for_windowss   rr1)>iorr functoolsrpathlibrtypingrrrrrr r r r r rmypy_extensionsrpackaging.specifiersrrrpackaging.versionrrpathspecrpathspec.patterns.gitwildmatchr version_infor\rtomliblack.handle_ipynb_magicsr black.moder black.outputr black.reportrcoloramar9rBrLrer`rgrhrsrDrrrrrr TextIOWrapperrr!r!r!r&s  4            /               Y