@(iSrSSKJr SSKrSSKrSSKJr SSKJr SSK r SSK r SSK J r \(a SSK Jr SS KJr \R""\5rS rS r\"S S 55rSSjr"SS5rg)zf Cache suitable for shards, not allowed to change because they are named after their own sha256 hash. ) annotationsN) dataclass) TYPE_CHECKING)unlink_or_rename_to_trash)Path) ShardDictzrepodata_shards.dbic>\rSrSr%S SjrS\S'S\S'S\S'Srg ) AnnotatedRawShardurlstrpackagecompressed_shardbytescHSU;deSU;deXlX lX0lg)Nz://)r rr)selfr rrs Z/data/cameron/miniconda/lib/python3.13/site-packages/conda_libmamba_solver/shards_cache.py__init__AnnotatedRawShard.__init__!s,||G### 0)rrr N)r rrrrr)__name__ __module__ __qualname____firstlineno__r__annotations____static_attributes__rrr r s1 H Lrr c|[R"USS9n[RUlUR S5 U$)ze Get database connection. dburi: uri-style sqlite database filename; accepts certain ?= parameters. T)urizPRAGMA foreign_keys = ON)sqlite3connectRow row_factoryexecute)dburiconns rr"r"/s2 ??5d +D{{DLL+, Krc`\rSrSrSrS SSjjrSrS SjrSSjrSSjr SSjr S r S r S r g ) ShardCache;zA Handle caching for individual shards (not the index of shards). c0XlUR5 g)z0 base: directory and filename prefix for cache. N)baser")rr,creates rrShardCache.__init__@s  rc*[URSS9$)z5 Copy cache with new connection. Useful for threads. F)r-)r)r,rs rcopyShardCache.copyGs$))E22rcUR[- R5n[U5UlU(dgURR S5 g)z8 Args: create: if True, create table if not exists. NzCREATE TABLE IF NOT EXISTS shards (url TEXT PRIMARY KEY, package TEXT, shard BLOB, timestamp TIMESTAMP DEFAULT CURRENT_TIMESTAMP))r,SHARD_CACHE_NAMEas_urir"r'r%)rr-r&s rr"ShardCache.connectMsF --557EN    = rcURnURSURURUR45 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_shardcs rinsertShardCache.insert^s@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 r9rows rretrieveShardCache.retrievemsc 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 r=r>N) rCZstdDecompressorjoinlenr'r%rArBrDrE)rurlsdctxqueryr9rFresults rretrieve_multipleShardCache.retrieve_multiplexs 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%)rr9s r clear_cacheShardCache.clear_caches$YY! II* +YYs( 6cURR5 [[UR[ - 55 g)z$ Remove the sharded cache database. N)r'closerrr,r4r0s r remove_cacheShardCache.remove_caches) !#dii2B&B"CDr)r,r'N)T)r,r)r8r )returnzShardDict | None)rOz list[str]r\zdict[str, ShardDict | None])rrrr__doc__rr1r"r:rGrSrVrZrrrrr)r);s03  "  2,Err))zcache.db)r] __future__rloggingr! dataclassesrtypingrrArCconda.gateways.disk.deleterpathlibr shards_typingr getLoggerrlogr4rEr r"r)rrrrgsu #! @( !'       dEdEr