iD dZddlZddlmZmZddlmZddlmZm Z m Z ddl m Z ddl mZde eeefd eefd ZGd d e Zde eeefd eefd Zddededed eefdZy)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)Versionstrsreturnc#Kt|tr|j}|D]+}|j}|s|j dr(|-yw)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 lines }/Volumes/fast/ai/experiments/voice-extract-mac/.venv/lib/python3.12/site-packages/lightning_utilities/install/requirements.py _yield_linesrsF $ zz| ,Js8AA Ac ZeZdZdZdZddddededeed ed df fd Zd ed efd Z xZ 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 ct||i|||_||std|||_|j |j v|_y)Nzwrong pip argument: )super__init__r RuntimeErrorr strict_stringlowerstrict)selfrrrr __class__s rr!z _RequirementWithComment.__init__0sW $)&) $ !5l^DE E(((GMMO; unfreezect|}|jr|d|jS|dk(r||jD]k}|jdvst |j j}|j|j|j dt|dzdcS|S|dk(r|jD]z}|jdvs|j|j }|j|dd jd|d }||vr|j|d }|jcS|S|d k7rtd |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))'  4XLHI I r() __name__ __module__ __qualname____doc__r#rrrr!r; __classcell__)r's@rrr"sV M24TX>> 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_posrs r_parse_requirementsrMqs*  EL 4<))D/K +.[\0B'DG == 9??$D U # ??4 L  ??5 !  $;"))K6 %dG,WW 3!  s+ACC*AC C  C C  Cpath_dir file_namer)c|dvrtd|dt||z }|jstd|||f|j }t |Dcgc]}|j |c}Scc}w)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.r0r+zunsupported option of ""zmissing file for )r6rexistsFileNotFoundError read_textrMr;)rNrOr)pathtextreqs rload_requirementsrXs,//28*A>?? >I %D ;;="3Xy$4O3P QRR >> D,?,E F,ESCJJx ,E FF Fs A;)zbase.txtr.)r?rIcollections.abcrrpathlibrtypingrrrpackaging.requirementsr packaging.versionr rrrrMlistrXr(rr`s .''.% uS(3-/0 Xc] LkL^0eC#$670HE\<]0fGGGCG\`ad\eGr(