o poi @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)2Remove some packages from given requirement files.N) isinstancestr_prune_packages)r r reqrX/home/ubuntu/.local/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)r Ncsg|] }|s|qSr) startswith).0lnpkgrr )sz#_prune_packages..w)open readlinesr rr writelines)rr fplinesrrrr!s     "rcCsrt|}|dd}Wdn1swYt|d}||WddS1s2wYdS)N>===r)rreadreplacewrite)rfopenrfwrrr_replace_min_req_in_txt0s   "r( proj_filecCsddl}t|dd }|}Wdn1swY||}|did}|s0dSt|D] \}}|dd||<q4t|d dd}|||WddS1s\wYdS) zYReplace all `>=` with `==` in the standard pyproject.toml file in [project.dependencies].rNutf-8encodingproject dependenciesr!r"r) tomlkitrr#parseget enumerater$r%dumps)r)r/fcontentdocdepsirrrr"_replace_min_req_in_pyproject_toml7s  "r9cCsdt|tr|g}|D]%}|drt|q tj|dkr#t|q tj d|dt ddq dS)z-Replace the min package version by fixed one..txtr^Only *.txt with plain list of requirements or standard pyproject.toml are supported.Provided '' is not supported. stacklevelN) r rendswithr(ospathbasenamer9warningswarn UserWarning)r fnamerrrreplace_oldest_versionOs    rH old_package new_packagecCst| }|}Wdn1swYt|D]\}}tdt|d||||<qt|d}||WddS1sGwYdS)zQReplace one package by another with the same version in a given requirement file.N^z (?=[ <=>#]|$)r)rrr2resubescaper)rrIrJr&rr8rr'rrr_replace_package_name_in_txtas  "  "rOc Csddl}t|dd }|}Wdn1swY||}|did}|s0dSt|D]\}} tdt|d|| ||<q4t|d dd}| | |WddS1sdwYdS) zYReplace one package by another with the same version in the standard pyproject.toml file.rNr*r+r-r.rKz (?=[ <=>]|$)r) r/rr#r0r1r2rLrMrNr%r3) r)rIrJr/r4r5r6r7r8rrrr'_replace_package_name_in_pyproject_tomlns  ""rPcCslt|tr|g}|D])}|drt|||q tj|dkr't|||q tj d|dt ddq dS)zLReplace one package by another with same version in given requirement files.r:rr;r<r=r>N) r rr@rOrArBrCrPrDrErF)rIrJr rGrrrreplace_package_in_requirementss  rQ)r)globos.pathrArLrDcollections.abcrrtypingrREQUIREMENT_ROOTrBjoinrlist__annotations__isfilerrrr(r9rHrOrPrQrrrrsJ        "