i UddlZddlZddlZddlZddlmZddlmZddlm Z dZ ejejjddZ eed<e ejejjdd dd z Z e ejejjd d z Z ejj!e re e gz Z e fd e eeefde eeefddfdZded eeddfdZdeddfdZddeddfdZe fde eeefddfdZdedededdfdZdedededdfdZe fdedede eeefddfdZy)N)Sequence)pprint)Unionzrequirements.txt requirementsz*.txtREQUIREMENT_FILES_ALLz**T) recursivepyproject.tomlpackages req_filesreturncvt|tr|g}t|tr|g}|D]}t||y)zRemove one or more packages from the specified requirement files. Args: packages: A package name or list of package names to remove. req_files: A path or list of paths to requirement files to process. N) isinstancestr_prune_packages)r r reqs y/Volumes/fast/ai/experiments/voice-extract-mac/.venv/lib/python3.12/site-packages/lightning_utilities/cli/dependencies.pyprune_packages_in_requirementsrs:(C :)S!K X&req_filec^t|5}|j}dddt|tr|g}|D]#}Dcgc]}|j |r|}}%t t|d5}|j |dddy#1swYvxYwcc}w#1swYyxYw)zRemove all occurrences of the given packages (by line prefix) from a requirements file. Args: req_file: Path to a requirements file. packages: Package names to remove. Lines starting with any of these names will be dropped. Nw)open readlinesrr startswithr writelines)rr fplinespkglns rrr's h2  (C :#>e2==+=e> 5M h  e    ?  s#BBB7B#B#B,ct|5}|jjdd}dddt|d5}|jdddy#1swY0xYw#1swYyxYw)zReplace all occurrences of '>=' with '==' in a plain text requirements file. Args: req_file: Path to the requirements.txt-like file to update. >===Nr)rreadreplacewrite)rfopenrfws r_replace_min_req_in_txtr(<sX h5jjl""4.  h        s!AA(A%(A1 proj_filecddl}t|d5}|j}ddd|j}|j dij d}|syt |D]\}}|j dd||<t|d d5}|j|j|dddy#1swYxYw#1swYyxYw) zReplace all '>=' with '==' in the [project.dependencies] section of a standard pyproject.toml. Preserves formatting and comments using tomlkit. Args: proj_file: Path to the pyproject.toml file. rNutf-8encodingproject dependenciesr!r"r) tomlkitrr#parseget enumerater$r%dumps)r)r0fcontentdocdepsirs r"_replace_min_req_in_pyproject_tomlr:Is i' *a&&( + -- C 779b ! % %n 5D D/3++dD)Q" iw /1  c"# 0 / + * 0 /sC!C C  Cct|tr|g}|D]m}|jdr t| tj j |dk(r t|Ntjd|dtdoy)a#Convert minimal version specifiers (>=) to pinned ones (==) in the given requirement files. Supports plain *.txt requirements and pyproject.toml files. Unsupported file types trigger a warning. Args: req_files: A path or list of paths to requirement files to process. .txtr ^Only *.txt with plain list of requirements or standard pyproject.toml are supported.Provided '' is not supported. stacklevelN) rrendswithr(ospathbasenamer:warningswarn UserWarning)r fnames rreplace_oldest_versionrJhsw)S!K  >>& ! #E * WW  e $(8 8 .u 5 MM"G#68   r old_package new_packagecTt|5}|j}dddtD]8\}}tjdtj |zdz||||<:t|d5}|j |dddy#1swYvxYw#1swYyxYw)a Rename a package in a plain text requirements file, preserving version specifiers and markers. Args: req_file: Path to the requirements.txt-like file to update. old_package: The original package name to replace. new_package: The new package name to use. N^z (?=[ <=>#]|$)r)rrr3resubescaper)rrKrLr&rr9rr's r_replace_package_name_in_txtrRs h5(  L)3&& +(>!>AQ!QS^`cd Q* h  l#     sB7BBB'cddl}t|d5}|j}ddd|j}|j dij d}|syt |D]8\}} t jdt j|zdz|| ||<:t|d d5}|j|j|dddy#1swYxYw#1swYyxYw) aRename a package in the [project.dependencies] section of a standard pyproject.toml, preserving constraints. Args: proj_file: Path to the pyproject.toml file. old_package: The original package name to replace. new_package: The new package name to use. rNr+r,r.r/rNz (?=[ <=>]|$)r) r0rr#r1r2r3rOrPrQr%r4) r)rKrLr0r5r6r7r8r9rs r'_replace_package_name_in_pyproject_tomlrTs i' *a&&( + -- C 779b ! % %n 5D D/3&& + 66H+WZ[Q" iw /1  c"# 0 / + * 0 /sC4!C*C'*C3ct|tr|g}|D]q}|jdrt|||"tj j |dk(rt|||Rtjd|dtdsy)aRename a package across multiple requirement files while keeping version constraints intact. Supports plain *.txt requirements and pyproject.toml files. Unsupported file types trigger a warning. Args: old_package: The original package name to replace. new_package: The new package name to use. req_files: A path or list of paths to requirement files to process. r<r r=r>r?r@N) rrrBrRrCrDrErTrFrGrH)rKrLr rIs rreplace_package_in_requirementsrVs)S!K  >>& ! ( [ I WW  e $(8 8 3E; T MM"G#68   r)r )globos.pathrCrOrFcollections.abcrrtypingrREQUIREMENT_ROOTrDjoinrlist__annotations__isfilerrrr(r:rJrRrTrVrrras   $%'dii ^W(MNtN277<<g#NZ^__277<<6F#GHH77>>"#.//Qf'C#&''49#x}:L4M' '$cXc]t* c d $#$T$>CXeC#,>&?\`2$3$S$s$W[$($s$$[^$cg$@Pe#&38hsm9K3L r