wBf; TdZddlmZddlmZddlmZddlmZddl m Z ddl m Z dd l mZmZdd lmZmZdd lmZdd lmZmZdd lmZddlmZddlmZmZmZm Z ddl!m"Z"ddl#m$Z$ddl%m&Z&m'Z'erddlm(Z(ee)Z*d'dZ+gZ,edddddddde jZdf d(dZ.dde jZdf d)dZ/d*dZ0d+dZ1d,d Z2 d- d.d!Z3gf d/d"Z4d0d#Z5d1d$Z6 d2 d3d%Z7 d4d&Z8y)5z%Tools for fetching the current index.) annotations)chain) getLogger) TYPE_CHECKING) IndexedSet)context)ThreadLimitedThreadPoolExecutor time_recorder)ChannelNotAllowed InvalidSpec)initialize_logging)Channelall_channel_urls) PackageType) MatchSpec) EMPTY_LINKPackageCacheRecord PackageRecord PrefixRecord)PackageCacheData) PrefixData) SubdirDatamake_feature_record)AnyctjrxttjdtjD|D]?}t |j }tfd|Dr-tt |yy)z Check if the given channel URLs are allowed by the context's allowlist. :param channel_urls: A list of channel URLs to check against the allowlist. :raises ChannelNotAllowed: If any URL is not in the allowlist. c3FK|]}t|jywN)r base_urls).0cs 0lib/python3.12/site-packages/conda/core/index.py z"check_allowlist..)s )* $$ s!c3&K|]}|v ywr)r!this_urlallowlist_channel_urlss r#r$z"check_allowlist../sUhx#99UsN) r allowlist_channelstupler from_iterablerr allr ) channel_urlsurl these_urlsr(s @r#check_allowlistr0 s|!!!&    .5.H.H  "    6C //JU*UU' 55 6 " get_indexr&TNFcNttjr|d}t||||}tj tj |t|t|||}|r t|||r t|tjr t||S)a" Return the index of packages available on the channels If prepend=False, only the channels passed in as arguments are used. If platform=None, then the current platform is used. If prefix is supplied, then the packages installed in that prefix are added. :param channel_urls: Channels to include in the index. :param prepend: If False, only the channels passed in are used. :param platform: Target platform for the index. :param use_local: Whether to use local channels. :param use_cache: Whether to use cached index information. :param unknown: Include unknown packages. :param prefix: Path to environment prefix to include in the index. :param repodata_fn: Filename of the repodata file. :return: A dictionary representing the package index. T) use_cache repodata_fn) rr offlinecalculate_channel_urlsLAST_CHANNEL_URLSclearextendr0 fetch_index_supplement_index_with_prefix_supplement_index_with_cachetrack_features_supplement_index_with_features) r-prependplatform use_localr5unknownprefixr6indexs r#r2r26s87?),9UL\*L!  { SE %eV4$U+'. Lr1ctjdt|zi}t5}fd}|j ||D](}|j d|j D* ddd|S#1swY|SxYw)a Fetch the package index from the specified channels. :param channel_urls: A list of channel URLs to fetch the index from. :param use_cache: Whether to use the cached index data. :param index: An optional pre-existing index to update. :param repodata_fn: The name of the repodata file. :return: A dictionary representing the fetched or updated package index. z channel_urls=c0tt|S)N)r6)rr)r.r6s r#zfetch_index..zs* CLk+ r1c3$K|]}||f ywrr&)r!recs r#r$zfetch_index..~s@#s@sN)logdebugreprr mapupdate iter_records)r-r5rFr6executorsubdir_instantiatorfs ` r#r<r<hsIIo\ 223 E ( *Ah 1<@ AA LL@q~~/?@ @ A A L A Ls AA==Bcrtj|tfd|jDS)z Check if a distribution string matches any package in the index. :param index: The package index. :param dist_str: The distribution string to match against the index. :return: True if there is a match; False otherwise. c3@K|]}j|ywr)match)r!prec match_specs r#r$z$dist_str_in_index..sA$z%As)r from_dist_stranyvalues)rFdist_strrYs @r#dist_str_in_indexr^s,((2J A%,,.A AAr1cR|sJt|jD]}||vr|||}|j|jk(r5|jdxst}t j |||||<Z|j}|j|_|` |||<|||<y)z Supplement the given index with information from the specified environment prefix. :param index: The package index to supplement. :param prefix: The path to the environment prefix. link)r`N) rrQchannelgetrr from_objectsr._Channel__canonical_name_PackageRecord__pkey)rFrE prefix_recordcurrent_recordr`prefix_channels r#r=r=s M6#F+88:1 E !"=1N%%)>)>>%((0>J'3'@'@"M(m$"/!6!6:H:L:L:N7!6'4m$$1E- =1r1ctjD]*}||vr||}tj||||<&|||<,y)zv Supplement the given index with packages from the cache. :param index: The package index to supplement. N)rget_all_extracted_entriesrrc)rFpcrecrgs r#r>r>sK";;=! E>"5\N-::>5QE%L E%L !r1c rttj||xsd|xsddtjdd| S)a8 Create a virtual package record. :param name: The name of the virtual package. :param version: The version of the virtual package, defaults to "0". :param build_string: The build string of the virtual package, defaults to "0". :return: A PackageRecord representing the virtual package. 0@ 12345678901234567890123456789012r) package_typenameversion build_stringrasubdirmd5 build_numberfn)rrVIRTUAL_SYSTEMr rt)rqrrrss r#_make_virtual_packagerys?  // 3!(S~~ .   r1cbttj|D]}t|}|||<y)z Supplement the given index with virtual feature records. :param index: The package index to supplement. :param features: A list of feature names to add to the index. N)rr r?r)rFfeaturesfeaturerKs r#r@r@s3//:!'*c r1ctjjD]5}td|j|j |j }|||<7y)z Loads and populates virtual package records from conda plugins and adds them to the provided index, unless there is a naming conflict. :param index: The package index to supplement. __N)r plugin_managerget_virtual_packagesryrqrrbuild)rFpackagerKs r#_supplement_index_with_systemrsL))>>@#b$7'--Xc r1cPddlm}m}tjj d\}}||vr|}n&|dk(ry|j r|t|}nytj}|tjk7r|Sddl }t|jjS)z Determine the architecture specification name for the current environment. :return: The architecture name if available, otherwise None. r) _arch_namesnon_x86_machines-zosNr) base.contextrrr rtsplitisdigitint_native_subdir archspec.cpustrcpuhost)rr target_plat target_archmachine native_subdirarchspecs r#get_archspec_namers =&~~33C8K&&      c+./**,M&8<<$$&''r1c|rdgt|z}|r|tjz }||dfntj}t ||S)a} Calculate the full list of channel URLs to use based on the given parameters. :param channel_urls: Initial list of channel URLs. :param prepend: Whether to prepend default channels to the list. :param platform: The target platform for the channels. :param use_local: Whether to include the local channel. :return: The calculated list of channel URLs. localnoarch)subdirs)listr channelsrr)r-rArBrCrs r#r8r8sMy4 #55 ((( &.&:x"G L' ::r1ct}ttttdfd dfd }|r&t|jD] }|| |D] }| srۉrhj } j | t | }tj||||} | D] } ||  |j| rhrij } j | t | }tj||||} | D] } ||  |j| rir؉r|D cic]} | | }} | t||tjs,dtjvr%tjjr t|t}|j!D],} |jg| j"| j$.|jtj"|D]}t'|} | || <t)||Scc} w)ae Generate a reduced package index based on the given specifications. This function is useful for optimizing the solver by reducing the amount of data it needs to consider. :param prefix: Path to an environment prefix to include installed packages. :param channels: A list of channel names to include in the index. :param subdirs: A list of subdirectories to consider for each channel. :param specs: A list of MatchSpec objects to filter the packages. :param repodata_fn: Filename of the repodata file to use. :return: A dictionary representing the reduced package index. c|jd}|r|vrj||jd}|r|D]}|vsj|yy)z Add a package name or track feature from a MatchSpec to the pending set. :param spec: The MatchSpec to process. rqr?N) get_raw_valueadd)specrqr?ftr_namecollected_namescollected_track_features pending_namespending_track_featuress r# push_specz$get_reduced_index..push_specJsl !!&) D/   d #++,<= * 9#;;*..x8 9 r1cd |j}t |j|D] }| |jr$|jD]}t |yy#t$r<}tjd|j |j dYd}~yd}~wwxYw)z Process a package record to collect its dependencies and features. :param record: The package record to process. z"Skipping %s due to InvalidSpec: %s invalid_specNr?) combined_dependsr rLwarning record_id_kwargsrrqr?)recordre_specrrs r# push_recordz&get_reduced_index..push_recordYs  %66  )FKK()% E e    "11 >)8<= > !  KK4  " .)    s A** B/32B**B/)rrr6rrD)rrreturnNone)rrrr)rsetrrQpoprrr query_allrPr=r r7_argparse_argsrDr>r\r?r{rr)rErrspecsr6recordsr prefix_recrrq new_recordsr feature_namerK reduced_indexknown_featuresftr_strrrrrrs @@@@@r#get_reduced_indexr0sJ(lGeO"uEM U 9 9>,$V,99; $J  # $$ 1 $$&D    %T?D$..xkK& $F# $ NN; '%1557L $ ( ( 6L9D$..xkK& $F# $ NN; '% 1.*11#S#X1M1 %mV<W+++0F0F0N0N %]3UN##%DB 2 2BS\\BCD'001!!!'* c!"-0 12s! I)r- list[str]rr)r- tuple[str]rAboolrB str | NonerCrr5rrDz bool | NonerErr6rrdict) r-rr5rrFz dict | Noner6rrr)rFdict[Any, Any]r]rrr)rFrrErrr)rFrrr)NN)rqrrrrrsrrr)rF"dict[PackageRecord, PackageRecord]r{rrr)rFrrr)rr)r&TNF) r-rrArrBrrCrrr) rErrrrrrzlist[MatchSpec]r6rrr)9__doc__ __future__r itertoolsrloggingrtypingrboltons.setutilsrrr common.ior r exceptionsr r gateways.loggingrmodels.channelrr models.enumsrmodels.match_specrmodels.recordsrrrrpackage_cache_datar prefix_datar subdir_datarrr__name__rLr0r9 repodata_fnsr2r<r^r=r>ryr@rrr8rr&r1r#rs(," '"F716&)XX0#86&{!++B/.. .. .  .  . .. ..f++B/     4 B'1T !"GK "9C2FH  - 9B    (< " ;; ;; ;  ;0t ttt  t  t  tr1