o ix@s&UddlmZmZmZmZddlZddlZddlZddlZddl m Z m Z m Z e eZgZee ed<eZdaee dfed<de d dfd d Zde d dfd d Zd ee dffddZd efddZdddZded efddZded eefddZded efddZde d efddZ dS) )AnyOptionalTupleListN) CacheProvider CacheContext CacheValue _providers._providers_snapshotproviderreturncCst4|tvrtd|jjd WddSt|ttat d|jjWddS1s9wYdS)zPRegister an external cache provider. Providers are called in registration order. Provider z already registeredNzRegistered cache provider: ) _providers_lockr _loggerwarning __class____name__appendtupler debugr r r ?z!_canonicalize..cStj|ddSNT) sort_keysjsondumpsxr r r@z_canonicalize..)key__set__cSr%r r&r(r r rr+Dr,cSr-r.r0r3r r rr5Er6 __tuple__cSr%r r&r(r r rr+Hr,cSr%r r&r(r r rr+Jr,__dict__cSs g|] \}}t|t|gqSr r&r)kvr r rr+Ms cSr-r.r0r3r r rr5Nr6 __bytes__zCannot canonicalize type: ) isinstance frozensetsortedsetrlistdictitemsintfloatstrrtyperbyteshexrr#r r rr'8s0         r' cache_keyc Csfzt|}tj|ddd}t|dWSty2}zt d|WYd}~dSd}~ww)NT),:)r/ separatorszutf-8zFailed to serialize cache key: ) r'r1r2hashlibsha256encode hexdigest Exceptionrr)rM canonicaljson_strer r r_serialize_cache_keyXsrYcCsz ||ksWdSWn tyYdSwt|ttttfr&tdd|DSt|tr6tdd|DSt |dr@t |j SdS)NTcss|]}t|VqdSr_contains_self_unequalr(r r r msz)_contains_self_unequal..css$|] \}}t|p t|VqdSrrZr;r r rr\os"valueF) rUr?r@rrCrBanyrDrEhasattrr[r]rLr r rr[ds    r[r]csNzddlWn tyYdSwdfdd|jD]}|qS)Nrcspt|jr||7dSt|tr$|D]}|qdSt|ttfr4|D]}|q-dSdSr)r?Tensornumel element_sizerDvaluesrCr)r#r=r*estimatetorchtotalr rre}s     z&_estimate_value_size..estimate)rf ImportErroroutputs)r]outputr rdr_estimate_value_sizeus   rk)r N)!typingrrrrrQr1logging threadingcomfy_api.latest._cachingrrr getLoggerrrr __annotations__Lockrr rrrrr r"r'rHrYr[rFrkr r r rs&