o fi @sUddlZddlZddlZddlZddlmZddlmZddlm Z dZ eej ddZ eed<e ejej dd dd d 7Z e eej d d 7Z ej e rYe e g7Z e fd e eeefde eeefddfddZded eeddfddZdeddfddZd$deddfddZe fde eeefddfddZdedededdfddZdedededdfd d!Ze fdedede eeefddfd"d#ZdS)%N)Sequence)pprint)Unionzrequirements.txt requirementsz*.txtREQUIREMENT_FILES_ALLz**T) recursivepyproject.tomlpackages req_filesreturncCs8t|tr|g}t|tr|g}|D]}t||qdS)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 reqrb/home/ubuntu/SoloSpeech/.venv/lib/python3.10/site-packages/lightning_utilities/cli/dependencies.pyprune_packages_in_requirementss   rreq_filecst| }|}Wdn1swYt|tr |g}|D] fdd|D}q"t|t|d}||WddS1sHwYdS)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. Ncsg|] }|s|qSr) startswith).0lnpkgrr 5sz#_prune_packages..w)open readlinesr r r writelines)rr fplinesrrrr's     "rcCsrt|}|dd}Wdn1swYt|d}||WddS1s2wYdS)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)rfopenrfwrrr_replace_min_req_in_txt<s   "r' proj_filecCsddl}t|dd }|}Wdn1swY||}|did}|s0dSt|D] \}}|dd||<q4t|d dd}|||WddS1s\wYdS) 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(r.fcontentdocdepsirrrr"_replace_min_req_in_pyproject_tomlIs   "r8cCsdt|tr|g}|D]%}|drt|q tj|dkr#t|q tj d|dt ddq dS)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) r r endswithr'ospathbasenamer8warningswarn UserWarning)r fnamerrrreplace_oldest_versionhs    rG old_package new_packagecCst| }|}Wdn1swYt|D]\}}tdt|d||||<qt|d}||WddS1sGwYdS)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)rrr1resubescaper)rrHrIr%rr7rr&rrr_replace_package_name_in_txts "  "rNc Csddl}t|dd }|}Wdn1swY||}|did}|s0dSt|D]\}} tdt|d|| ||<q4t|d dd}| | |WddS1sdwYdS) 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-rJz (?=[ <=>]|$)r) r.rr"r/r0r1rKrLrMr$r2) r(rHrIr.r3r4r5r6r7rrrr'_replace_package_name_in_pyproject_tomls   ""rOcCslt|tr|g}|D])}|drt|||q tj|dkr't|||q tj d|dt ddq dS)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. r9rr:r;r<r=N) r r r?rNr@rArBrOrCrDrE)rHrIr rFrrrreplace_package_in_requirementss  rP)r)globos.pathr@rKrCcollections.abcrrtypingrREQUIREMENT_ROOTrAjoinrlist__annotations__isfiler rrr'r8rGrNrOrPrrrrsJ        "