ibSrSSKJr SSKrSSKrSSKJr SSKJr SSK r SSK r \(a SSK J r SSK Jr \R"\5rS qS r\"S S 55rSS jr"SS5rg)zf Cache suitable for shards, not allowed to change because they are named after their own sha256 hash. ) annotationsN) dataclass) TYPE_CHECKING)Path) ShardDictzrepodata_shards.dbic>\rSrSr%S SjrS\S'S\S'S\S'Srg ) AnnotatedRawShardurlstrpackagecompressed_shardbytescHSU;deSU;deXlX lX0lg)Nz://)r rr)selfr rrs \/home/wildlama/miniconda3/lib/python3.13/site-packages/conda_libmamba_solver/shards_cache.py__init__AnnotatedRawShard.__init__ s,||G### 0)rrr N)r r rr rr)__name__ __module__ __qualname____firstlineno__r__annotations____static_attributes__rrr r s1 H Lrr c[R"USS9n[RUlUnUR S5 SSS5 U$!,(df  U$=f)ze Get database connection. dburi: uri-style sqlite database filename; accepts certain ?= parameters. T)urizPRAGMA foreign_keys = ONN)sqlite3connectRow row_factoryexecute)dburiconncs rr!r!.sI ??5d +D{{D  ,-  K  Ks A  Acr\rSrSrSrSSSjjrSrSrSrSr SSjr SS jr SS jr SS jr S rS rSrg) ShardCache;zA Handle caching for individual shards (not the index of shards). c.XlURUS9 g)z0 base: directory and filename prefix for cache. createN)baser!)rr.r-s rrShardCache.__init__@s  F #rcU$Nrrs r __enter__ShardCache.__enter__Gs rc$UR5 gr1)close)rexc_typeexc_valexec_tbs r__exit__ShardCache.__exit__Js  rcjUR(a"URR5 SUlgg)zF Clean up connection. ShardCache can no longer be used after close(). N)r&r6r2s rr6ShardCache.closeMs% 99 IIOO DI rc*[URSS9$)z5 Copy cache with new connection. Useful for threads. Fr,)r)r.r2s rcopyShardCache.copyUs$))E22rchUR[- R5n[U5UlU(dgURnUR S5 SSS5 g!,(df  g=f![ Ran[US5nU(aU(aUR[ R:Xal[RSX55 UR5 O/![a"n[RSX55 SqSnAOSnAff=fURUSS9sSnA$eSnAff=f) zr Args: create: if True, create table if not exists. retry: remove cache, log warning, and retry on error. NzCREATE TABLE IF NOT EXISTS shards (url TEXT PRIMARY KEY, package TEXT, shard BLOB, timestamp TIMESTAMP DEFAULT CURRENT_TIMESTAMP)sqlite_errorcodez%s '%s'; remove and retry.z %s '%s'; use alternate filename.zrepodata_shards_1.dbF)r-retry)r.SHARD_CACHE_NAMEas_urir!r&r$r DatabaseErrorhasattrrB SQLITE_NOTADBlogwarning remove_cacheOSError)rr-rCr%r'e has_errorcodes rr!ShardCache.connect[s--557EN   a E $$  $A'9:M}!2D2DH]H]2] 8%C>%%'>KK BEM'=$> ||6|?? ! sl A3A"A3" A0,A30A33D1AD,C'&D,' D1D D,DD,%D1+D,,D1cURnURSURURUR45 SSS5 g!,(df  g=f)zd Args: url: of shard package: package name raw_shard: msgpack.zst compressed shard data zCINSERT OR IGNORE INTO SHARDS (url, package, shard) VALUES (?, ?, ?)N)r&r$r rr)r raw_shardr's rinsertShardCache.inserts@YY! IIU 1 193M3MN YYs 4A  AcURnURSU45R5nU(a0[R"[ R "US[S95OSsSSS5 $!,(df  g=f)Nz&SELECT shard FROM shards WHERE url = ?shardmax_output_size)r&r$fetchonemsgpackloads zstandard decompressZSTD_MAX_SHARD_SIZE)rr r'rows rretrieveShardCache.retrievesc YY!))DsfMVVXC  ((WGZ[  YYs AA22 Bc U(d0$[R"5nSSRS[U5-5S3nURnUR X15Vs0sH;nUSU(a+[ R"URUS[S95OS_M= nnUsSSS5 $s snf!,(df  g=f) zt Query database for cached shard urls. Return a dict of urls in cache mapping to the Shard or None if not present. z,SELECT url, shard FROM shards WHERE url IN (,)?z) ORDER BY urlr rUrVN) r[ZstdDecompressorjoinlenr&r$rYrZr\r])rurlsdctxqueryr'r^results rretrieve_multipleShardCache.retrieve_multiples I))+>sxxQTUYQZHZ?[>\\jk YY!99U1 3 2C E $MMOOCLBUOV  2 3Y3Ys B4 AB/"B4/B44 CcrURnURS5 SSS5 g!,(df  g=f)z8 Truncate the database by removing all rows from tables zDELETE FROM shardsN)r&r$)rr's r clear_cacheShardCache.clear_caches$YY! II* +YYs( 6cUR5 UR[- R5 g![a9 UR[- R UR[S3- 5 gf=f)z$ Remove the sharded cache database. z .conda_trashN)r6r.rDunlinkrLrenamer2s rrKShardCache.remove_cachese  a YY) ) 1 1 3 a YY) ) 1 1$))AQ@RR^>_2_ ` as!4AA76A7)r.r&N)T)r.r)TT)rQr )returnzShardDict | None)rgz list[str]rtzdict[str, ShardDict | None])rrrr__doc__rr3r:r6r?r!rRr_rkrnrKrrrrr)r);s@$3 %N  2, arr))zcache.db)ru __future__rloggingr dataclassesrtypingrrYr[pathlibr shards_typingr getLoggerrrIrDr]r r!r)rrrr}sr #! ( !'       JaJar