o wi @sbUddlZddlZddlZddlmZddlmZddlmZdZ eej ddZ e ed<e ejej dd dd d 7Z ej e rIe e g7Z e fd eeeefd eeeefddfddZded eeddfddZdeddfddZe fd eeeefddfddZde ededede efddZe fdeded eeeefddfddZdS)N)Sequence)pprint)Unionzrequirements.txt requirementsz*.txtREQUIREMENT_FILES_ALLz**T) recursivepackages req_filesreturncCs8t|tr|g}t|tr|g}|D]}t||qdS)2Remove some packages from given requirement files.N) isinstancestr_prune_packages)rr reqra/home/ubuntu/sommelier/.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)r Ncsg|] }|s|qSr) startswith).0lnpkgrr 'sz#_prune_packages..w)open readlinesr r r writelines)rrfplinesrrrrs     "rfnamecCsrt|}|dd}Wdn1swYt|d}||WddS1s2wYdS)Nz>=z==r)rreadreplacewrite)r fopenrfwrrr _replace_min.s   "r&cCs&t|tr|g}|D]}t|q dS)z-Replace the min package version by fixed one.N)r r r&)r r rrrreplace_oldest_version5s  r' old_package new_packagecCs6t|D]\}}tdt|d||||<q|S)a Replace one package by another with same version in given requirement file. >>> _replace_package_name(["torch>=1.0 # comment", "torchvision>=0.2", "torchtext <0.3"], "torch", "pytorch") ['pytorch>=1.0 # comment', 'torchvision>=0.2', 'torchtext <0.3'] ^z (?=[ <=>#]|$)) enumerateresubescape)rr(r)irrrr_replace_package_name=s"r0c Cst|tr|g}|D]:}t| }|}Wdn1swYt|||}t|d }||Wdn1s?wYq dS)zLReplace one package by another with same version in given requirement files.Nr)r r rrr0r)r(r)r r r$reqsr%rrrreplace_package_in_requirementsIs      r2)globos.pathosr,collections.abcrrtypingrREQUIREMENT_ROOTpathjoinrlist__annotations__isfiler rrr&r'r0r2rrrrsB        ""