o fiD @sdZddlZddlmZmZddlmZddlmZm Z m Z ddl m Z ddl mZde eeefd eefd d ZGd d d e Zde eeefd eefddZddededed eefddZdS)zUtilities to parse and adjust Python requirements files. This module parses requirement lines while preserving inline comments and pip arguments and supports relaxing version pins based on a chosen unfreeze strategy: "none", "major", or "all". N)IterableIterator)Path)AnyOptionalUnion) Requirement)Versionstrsreturnccs>t|tr |}|D]}|}|r|ds|Vq dS)z{Yield non-empty, non-comment lines from a string or iterable of strings. Adapted from pkg_resources.yield_lines. #N) isinstancestr splitlinesstrip startswith)r linerf/home/ubuntu/SoloSpeech/.venv/lib/python3.10/site-packages/lightning_utilities/install/requirements.py _yield_liness rc sXeZdZdZdZddddededeed ed df fd d Zd ed efddZ Z S)_RequirementWithCommentaRequirement subclass that preserves an inline comment and optional pip argument. Attributes: comment: The trailing comment captured from the requirement line (including the leading '# ...'). pip_argument: A preceding pip argument line (e.g., ``"--extra-index-url ..."``) associated with this requirement, or ``None`` if not provided. strict: Whether the special marker ``"# strict"`` appears in ``comment`` (case-insensitive), in which case upper bound adjustments are disabled. z# strictNcomment pip_argumentargsrrkwargsr csLtj|i|||_|dus|std|||_|j|v|_dS)Nzwrong pip argument: )super__init__r RuntimeErrorr strict_stringlowerstrict)selfrrrr __class__rrr0s  z _RequirementWithComment.__init__unfreezecCst|}|jr|d|jS|dkr<|jD]#}|jdvr9t|jj}||j|jdt |ddSq|S|dkru|jD]/}|jdvrr|j|j}||dd d|d }||vrl||d }| SqC|S|d krt d |d |S) a`Adjust version specifiers according to the selected unfreeze strategy. The special marker ``"# strict"`` in the captured comment disables any relaxation of upper bounds. >>> _RequirementWithComment("arrow<=1.2.2,>=1.2.0", comment="# anything").adjust("none") 'arrow<=1.2.2,>=1.2.0' >>> _RequirementWithComment("arrow<=1.2.2,>=1.2.0", comment="# strict").adjust("none") 'arrow<=1.2.2,>=1.2.0 # strict' >>> _RequirementWithComment("arrow<=1.2.2,>=1.2.0", comment="# my name").adjust("all") 'arrow>=1.2.0' >>> _RequirementWithComment("arrow>=1.2.0, <=1.2.2", comment="# strict").adjust("all") 'arrow<=1.2.2,>=1.2.0 # strict' >>> _RequirementWithComment("arrow").adjust("all") 'arrow' >>> _RequirementWithComment("arrow>=1.2.0, <=1.2.2", comment="# cool").adjust("major") 'arrow<2.0,>=1.2.0' >>> _RequirementWithComment("arrow>=1.2.0, <=1.2.2", comment="# strict").adjust("major") 'arrow<=1.2.2,>=1.2.0 # strict' >>> _RequirementWithComment("arrow>=1.2.0").adjust("major") 'arrow>=1.2.0' >>> _RequirementWithComment("arrow").adjust("major") 'arrow' Args: unfreeze: One of: - ``"none"``: Keep all version specifiers unchanged. - ``"major"``: Relax the upper bound to the next major version (e.g., ``<2.0``). - ``"all"``: Drop any upper bound constraint entirely. Returns: The adjusted requirement string. If strict, the original string is returned with the strict marker appended. z major)>> txt = ['# ignored', '', 'this # is an', '--piparg', 'example', 'foo # strict', 'thing', '-r different/file.txt'] >>> [r.adjust('none') for r in _parse_requirements(txt)] ['this', 'example', 'foo # strict', 'thing'] >>> txt = '\\n'.join(txt) >>> [r.adjust('none') for r in _parse_requirements(txt)] ['this', 'example', 'foo # strict', 'thing'] Args: strs: Either an iterable of requirement lines or a single multi-line string. Yields: _RequirementWithComment: Parsed requirement objects with preserved comment and pip argument. Nz #r\z--z-r @z https?://r) rfindendswithrnext StopIterationrresearchr)r linesrr comment_posrrrr_parse_requirementsqs2     rHbase.txtr*path_dir file_namer&cs^dvr tddt||}|s td|||f|}fddt|DS)aMLoad, parse, and optionally relax requirement specifiers from a file. >>> import os >>> from lightning_utilities import _PROJECT_ROOT >>> path_req = os.path.join(_PROJECT_ROOT, "requirements") >>> load_requirements(path_req, "docs.txt", unfreeze="major") # doctest: +ELLIPSIS +NORMALIZE_WHITESPACE ['sphinx<6.0,>=4.0', ...] Args: path_dir: Directory containing the requirements file. file_name: The requirements filename inside ``path_dir``. unfreeze: Unfreeze strategy: ``"none"``, ``"major"``, or ``"all"`` (see ``_RequirementWithComment.adjust``). Returns: A list of requirement strings adjusted according to ``unfreeze``. Raises: ValueError: If ``unfreeze`` is not one of the supported options. FileNotFoundError: If the composed path does not exist. >r*r,r'zunsupported option of ""zmissing file for csg|]}|qSr)r7).0reqr&rr sz%load_requirements..)r2rexistsFileNotFoundError read_textrH)rJrKr&pathtextrrOrload_requirementss rV)rIr*)r;rDcollections.abcrrpathlibrtypingrrrpackaging.requirementsrpackaging.versionr rrrrHlistrVrrrrs   ""O$3