a 紫j@sddlmZddlZddlZddlZddlmZmZddlmZddl m Z ddl m Z m Z m Z mZmZmZddlmZmZdd lmZmZmZmZdd lmZdd lmZmZdd lmZm Z m!Z!m"Z"dd l#m$Z$e rddlm%Z%m&Z&ddl'm(Z(ddl)m*Z*ddlm+Z+e,e-Z.gdZ/ddddZ0edZ1edZ2GdddeZ3ede3dZ4e5dZ6ddd d!d"Z7d#d#d$d%d&Z8d#d'd'd(d)d*Z9d+d,d-d.d/Z:d0d1d#d2d3d4d5Z;d0d1d#dd3d6d7Zd?d@Z>d0d1ddAdBZ?d0d1d\z_toml_value..) isinstancerrrrHr)rJrOr0r0r1 _toml_valueXs rVzlist[tuple[str, Any]]zdict[str, Any])datar-cCsdd|DS)NcSs(i|] \}}|durt|t||qSr.)rNrV)rQrJrOr0r0r1 asz&_toml_dict_factory..r0)rWr0r0r1_toml_dict_factory`srYr6ztype[_T]z _T | None)r8 expected_typerJr-cCsX||}durdSt||r2|turTt|trTtdt|jd|jd|d|S)zBGet a value from the dictionary and verify it's the expected type.NUnexpected type (expected )context)getrUintrBr*typer<r8rZrJrOr0r0r1_geths rdcCs t|||}durt||S)zKGet a required value from the dictionary and verify it's the expected type.N)rd_PylockRequiredKeyErrorrcr0r0r1 _get_requiredxsrfzSequence[_T] | None)r8expected_item_typerJr-cCst|t|}durdSt|ttfr@tdt|jd|dt|D]@\}}t||sHtdt|jd|jd|d|ddqH|S) zMGet a list value from the dictionary and verify it's the expected items type.Nr[z (expected Sequence)r^r\r][]) rdrrUrHbytesr*rbr< enumerate)r8rgrJrOiitemr0r0r1 _get_sequences  rnzCallable[[_T], _T2]z _T2 | None)r8rZ target_typerJr-c CsZt|||}durdSz ||WStyT}zt||d|WYd}~n d}~00dS)zGet a value from the dictionary, verify it's the expected type, and convert to the target type. This assumes the target_type constructor accepts the value. Nr^)rd Exceptionr*)r8rZrorJrOer0r0r1_get_ass  rrcCs"t||||}durt||S)zfGet a required value from the dict, verify it's the expected type, and convert to the target type.N)rrre)r8rZrorJrOr0r0r1_get_required_assrszlist[_T2] | None)r8rgtarget_item_typerJr-c Cst|||}durdSg}z |D]}||}||q"WnBty~}z*t||dt|dd|WYd}~n d}~00|S)z>Get list value from dictionary and verify expected items type.Nrhrir^)rnappendrpr*len) r8rgrtrJrOresultrm typed_itemrqr0r0r1_get_sequence_ass4ryztype[_FromMappingProtocolT]z_FromMappingProtocolT | None)r8rorJr-c Cs\t|t|}durdSz ||WStyV}zt||d|WYd}~n d}~00dS)zIGet a dictionary value from the dictionary and convert it to a dataclass.Nr^)rdrr;rpr*)r8rorJrOrqr0r0r1 _get_objects  rzz"list[_FromMappingProtocolT] | None)r8rtrJr-c Cst|t|}durdSg}z"|D]}||}||q"WnBty}z*t||dt|dd|WYd}~n d}~00|S)zJGet a list value from the dictionary and convert its items to a dataclass.Nrhrir^)rnrr;rurpr*rv)r8rtrJrOrwrmrxrqr0r0r1_get_sequence_of_objectss 4r{zSequence[_FromMappingProtocolT]cCs t|||}durt||S)zWGet a required list value from the dictionary and convert its items to a dataclass.N)r{re)r8rtrJrwr0r0r1!_get_required_sequence_of_objectssr|r)rDr-cCs t|std|dt|S)z+Validate that a string is a NormalizedName.zName z is not normalized)rr*r)rDr0r0r1_validate_normalized_namesr} str | NoneNone)rCurlr-cCs|s|stddS)Nzpath or url must be provided)r*)rCrr0r0r1_validate_path_urlsrcCs@|sdSd|vr |dddSd|vr8|dddS|SdS)N/r\)rsplitrGr0r0r1 _path_namesr)rr-cCs&|sdSt|j}t|dddS)Nrrr)rrCr r)rZurl_pathr0r0r1 _url_names r)hashesr-cCs.|s tdtdd|Ds*td|S)Nz"At least one hash must be providedcss|]}t|tVqdSr.)rUrH)rQZhash_valr0r0r1 rTz#_validate_hashes..zHash values must be strings)r*allvalues)rr0r0r1_validate_hashess rc@sLeZdZUdZdZded<ded<ddddd d d d Zdd ddZdS)r*z2Raised when when input data is not spec-compliant.Nr~r_rHmessager^zstr | Exceptionr)causer_r-cCsRt|tr>|jr.|r$|d|jn|j|_n||_|j|_n||_t||_dS)N.)rUr*r_rrH)selfrr_r0r0r1__init__s  zPylockValidationError.__init__r,cCs|jr|jd|jS|jS)Nz in )r_rrr0r0r1__str__/szPylockValidationError.__str__)r<r=r>__doc__r___annotations__rrr0r0r0r1r*s  r*cs$eZdZdddfdd ZZS)rerHrrIcstjd|ddS)NzMissing required valuer^)superr)rrJ __class__r0r1r6sz _PylockRequiredKeyError.__init__)r<r=r>r __classcell__r0r0rr1re5srec@seZdZdZdS)r)z7Raised when encountering an unsupported `lock_version`.Nr<r=r>rr0r0r0r1r):sr)c@seZdZdZdS)r(z[Base exception for errors raised by :meth:`Pylock.select`. .. versionadded:: 26.1 Nrr0r0r0r1r(>sr(TF)frozeninitc @seZdZUded<dZded<dZded<dZded<ded<dZded <ddddd ddddddd d d dZe dddddZ dS)r%rHrbNr~rrCrequested_revision commit_id subdirectory)rrCrrr)rbrrCrrrr-cCsXt|d|t|d|t|d|t|d|t|d|t|d|dS)NrbrrCrrrobject __setattr__)rrbrrCrrrr0r0r1rNs zPackageVcs.__init__r6rr7c CsV|t|tdt|tdt|tdt|tdt|tdt|tdd}t|j|j|S)NrbrrCzrequested-revisionz commit-idr)rbrrCrrr)rfrHrdrrCr)r:r8Z package_vcsr0r0r1r;`s      zPackageVcs._from_dict) r<r=r>rrrCrrrr?r;r0r0r0r1r%Es     r%c@s^eZdZUded<dZded<dZded<ddddddd d d d Zed ddddZdS)r#rHrCNz bool | Noneeditabler~r)rrr)rCrrr-cCs.t|d|t|d|t|d|dS)NrCrrr)rrCrrr0r0r1rtszPackageDirectory.__init__r6rr7cCs&|t|tdt|tdt|tddS)NrCrr)rCrr)rfrHrdrBr9r0r0r1r;s    zPackageDirectory._from_dict) r<r=r>rrrrr?r;r0r0r0r1r#ns    r#c @seZdZUdZded<dZded<dZded<dZded<d ed <dZded <dddddd ddddd dd dddZ e dddddZ dS)r"Nr~rrC int | Nonesizedatetime | None upload_timeMapping[str, str]rr)rrCrrrr)rrCrrrrr-cCsXt|d|t|d|t|d|t|d|t|d|t|d|dS)NrrCrrrrr)rrrCrrrrr0r0r1rs zPackageArchive.__init__r6rr7c CsX|t|tdt|tdt|tdt|tdt|ttdt|tdd}t|j|j |S)NrrCr upload-timerr)rrCrrrr) rdrHrarrsrrrrCr)r:r8Zpackage_archiver0r0r1r;s      zPackageArchive._from_dict) r<r=r>rrrCrrrrr?r;r0r0r0r1r"s      r"c @seZdZUdZded<dZded<dZded<dZded<dZded <d ed <dddddd dddddd d dddZ e dddddZ e ddddZ dS)r$Nr~rDrrrrCrrrrrDrrrCrrrDrrrCrrr-cCsXt|d|t|d|t|d|t|d|t|d|t|d|dSNrDrrrCrrrrrDrrrCrrr0r0r1rs zPackageSdist.__init__r6rr7c CsX|t|tdt|tdt|tdt|tdt|tdt|ttdd}t|j|j |SNrDrrrCrr)rDrrrCrr rdrHrrarsrrrrCr)r:r8Z package_sdistr0r0r1r;s      zPackageSdist._from_dictrHr,cCs*|jpt|jpt|j}|s&td|S)zGGet the filename of the sdist. .. versionadded:: 26.1 zCannot determine sdist filenamerDrrCrrr*rfilenamer0r0r1rszPackageSdist.filenamer<r=r>rDrrrrCrrr?r;propertyrr0r0r0r1r$s       r$c @seZdZUdZded<dZded<dZded<dZded<dZded <d ed <dddddd dddddd d dddZ e dddddZ e ddddZ dS)r&Nr~rDrrrrCrrrrrrrcCsXt|d|t|d|t|d|t|d|t|d|t|d|dSrrrr0r0r1rs zPackageWheel.__init__r6rr7c CsX|t|tdt|tdt|tdt|tdt|tdt|ttdd}t|j|j |Srr)r:r8Z package_wheelr0r0r1r;s      zPackageWheel._from_dictrHr,cCs*|jpt|jpt|j}|s&td|S)zGet the filename of the wheel.zCannot determine wheel filenamerrr0r0r1rszPackageWheel.filenamerr0r0r0r1r&s       r&c@seZdZUded<dZded<dZded<dZded <dZd ed <dZd ed <dZ ded<dZ ded<dZ ded<dZ ded<dZ ded<dZd ed<dZded<ddddddddddddd ddddd d dddddd dddddZed d!d"d#d$Zed%d&d'd(ZdS))r!rrDNzVersion | Noneversionz Marker | NonemarkerSpecifierSet | Nonerequires_pythonz"Sequence[Mapping[str, Any]] | None dependencieszPackageVcs | NonevcszPackageDirectory | None directoryzPackageArchive | Nonearchiver~indexzPackageSdist | NonesdistzSequence[PackageWheel] | Nonewheelsattestation_identitiesMapping[str, Any] | Nonetool) rrrrrrrrrrrrr)rDrrrrrrrrrrrrr-c Cst|d|t|d|t|d|t|d|t|d|t|d|t|d|t|d|t|d | t|d | t|d | t|d | t|d | dS)NrDrrrrrrrrrrrrr)rrDrrrrrrrrrrrrr0r0r1r'szPackage.__init__r6rr7c Cs|t|ttdt|ttdt|ttdt|tdt|ttdt |t dt |t dt |t dt |td t |td t|td t|td t |td d }t|jt|jpg}t|jt|jt|j}|dkr|dkrtd|dkr|dkrtdt|jpgD]\}}zt|j\}}} } WnDtyj} z*td|jd|dd| WYd} ~ n d} ~ 00||jkrtd|jd|jd|dd|jr||jkrtd|jdt|jd|ddq|jrzt|jj\}}Wn>ty6} z$td|jjd d| WYd} ~ n d} ~ 00||jkrbtd|jjd|jd d|jr||jkrtd|jjdt|jd dz,t|j pgD]\}} t!| tdqWn<ty} z"t| d|dd| WYd} ~ n d} ~ 00|S)NrDrrequires-pythonrrrrrrrrzattestation-identitiesr) rDrrrrrrrrrrrrrzFNone of vcs, directory, archive must be set if sdist or wheels are setrzRExactly one of vcs, directory, archive must be set if sdist and wheels are not setzInvalid wheel filename zwheels[rir^zName in z% is not consistent with package name z Version in z( is not consistent with package version zInvalid sdist filename kindzattestation-identities[)"rsrHr}rrrrrnrrrzr%r#r"rdr$r{r&rBrrvrrrrr*rkrrrprDrrrrf) r:r8package distributionsZ direct_urlsrlwheelrDrrKrqZattestation_identityr0r0r1r;Gs                          zPackage._from_dictrBr,cCs|jp |j Sr.)rrrr0r0r1 is_directszPackage.is_direct)r<r=r>rrrrrrrrrrrrrrr?r;rrr0r0r0r1r!s<             , Pr!c @s eZdZUdZded<dZded<dZded<dZd ed <dZd ed <dZ d ed <ded<ded<dZ ded<ddddddddddd d d ddddd ddZ e dddddZ e dddddZdd d!d"Zdd d#d$Zdddddd%d&d'd(d(d)d*d+d,d-ZdS).r'z#A class representing a pylock file.r lock_versionNzSequence[Marker] | NonerPrrzSequence[NormalizedName] | NoneextraszSequence[str] | Nonedependency_groupsdefault_groupsrH created_byzSequence[Package]packagesrr)rPrrrrrr) rrPrrrrrrrr-c Cst|d|t|d|t|d|t|d|t|d|t|d|t|d|t|d|t|d | dS) NrrPrrrrrrrr) rrrPrrrrrrrr0r0r1rszPylock.__init__r6rr7c Cs|t|ttdt|ttdt|ttdt|tdt|tdt|tdt|tt dt |t dt |t d d }td |jkrtd ksntd |jd|jtdkrtd|j|S)Nz lock-versionrPrzdependency-groupszdefault-groupsz created-byrrr) rrPrrrrrrr12zpylock version z is not supportedz1.0z(pylock minor version %s is not supported)rsrHrryrr}rnrfrrrr|r!rdrrr)_loggerwarning)r:r8Zpylockr0r0r1r;s(           zPylock._from_dictcCs ||S)zCreate and validate a Pylock instance from a TOML dictionary. Raises :class:`PylockValidationError` if the input data is not spec-compliant. )r;r9r0r0r1 from_dictszPylock.from_dictr,cCstj|tdS)z1Convert the Pylock instance to a TOML dictionary.) dict_factory) dataclassesasdictrYrr0r0r1to_dictszPylock.to_dictcCs||dS)zqValidate the Pylock instance against the specification. Raises :class:`PylockValidationError` otherwise.N)rrrr0r0r1validateszPylock.validate) environmenttagsrrprefer_sdist_predicatezEnvironment | NonezSequence[Tag] | NonezCollection[str] | Nonez'Callable[[NormalizedName], bool] | NonezfIterator[tuple[Package, PackageVcs | PackageDirectory | PackageArchive | PackageWheel | PackageSdist]])rrrrrr-c cst|dur|nt}tdt|p"it|p*gt|dur@|jpBgn|d}t|rZ|dntd}|jr|j |st d|dt |j|j r|j D] } | j td|pidd rqqt d i} t|jD]\} } | jr| jj |d d sq| jr6| j |s6t d|dt | jd | jd | d | j| vrjt d| jd| d| | jdd| | f| | j<q| D]\} } | jdur| | jfVn| jdur| | jfVn| jdur| | jfVn| jdur |dur || jr | | jfVn| jrrt|dd| jDd} | r>| | fVn2| jdurX| | jfVnt d| jd | dn| jdur| | jfVntqdS)aRSelect what to install from the lock file. The *environment* and *tags* parameters represent the environment being selected for. If unspecified, ``packaging.markers.default_environment()`` and ``packaging.tags.sys_tags()`` are used. The *extras* parameter represents the extras to install. The *dependency_groups* parameter represents the groups to install. If unspecified, the default groups are used. The *prefer_sdist_predicate* parameter is called for packages with a source distribution. If it returns ``True``, the source distribution is selected before attempting wheel compatibility. If no source distribution is available, wheel selection proceeds as usual without calling the predicate. This method must be used on valid Pylock instances (i.e. one obtained from :meth:`Pylock.from_dict` or if constructed manually, after calling :meth:`Pylock.validate`). .. versionadded:: 26.1 .. versionchanged:: 26.3 Added the *prefer_sdist_predicate* parameter. Nzdict[str, str | frozenset[str]])rrpython_full_versionzpython_full_version zI in provided environment does not satisfy the Python version requirement zdict[str, str] requirementr^zXProvided environment does not satisfy any of the environments specified in the lock fileZ lock_filez for package z at packages[riz Multiple packages with the name z are selected at packages[z] and packages[rcss |]}|t|jdfVqdS)rN)rr)rQrr0r0r1rsz Pylock.select..z6No wheel found matching the provided tags for package z'], and no sdist available as a fallback)rrr dict frozensetrrrrcontainsr(rHrPevaluaterkrrrDrrrrrrnextNotImplementedError)rrrrrrZcompatible_tags_selectorenvZenv_python_full_versionZ env_markerZselected_packages_by_nameZ package_indexrZ best_wheelr0r0r1selects+                  z Pylock.select)r<r=r>rrrPrrrrrrr?r;rrrrr0r0r0r1r's:       $ r')U __future__rrloggingrecollections.abcrrrrtypingrrr r r r urllib.parser rmarkersrrrr specifiersrrrrutilsrrrrrrrrpathlibrZtyping_extensionsrr getLoggerr<rr/r2r3r4r5r@compilerEr+rNrVrYrdrfrnrrrsryrzr{r|r}rrrrrpr*rer)r(r%r#r"r$r&r!r'r0r0r0r1s~                 (  ( 3 0