ɯei]ddlmZddlZddlZddlZddlZddlZddlZddlZddl Z ddl m Z ddlm Z m Z ddlmZddlmZmZmZddlmZmZddlmZmZd d lmZd d lmZejj:Zejj<Zed Zej@e!Z"ejFed jIdk(Z%GddeeeZ&edZ'edZ(ddZ)Gddee'e(fZ*Gdde*Z+y)) annotationsN)Iterator)makedirspath)Lock)GenericNoReturnTypeVar)FileLockTimeout) NamedTupleSelf) constants)ENV_VAR_TEST_MODETtruec"eZdZUded<ded<y) CacheEntrydatetime.datetimeexpiryrentryN)__name__ __module__ __qualname____annotations__[/var/www/html/glpi_dashboard/venv/lib/python3.12/site-packages/snowflake/connector/cache.pyrr s  HrrKVct|k\SN)now)ds r is_expiredr')s 5A:rceZdZdZej f d dZd!dZe d"dZ dd d#dZ e Z d$dZ d%d Z d$d Zd&d Zd'd Zd(d Zd)dZ d* d+dZd,dZ d-dZ d-dZ d.dZ d/ d0dZ d/ d0dZ d1dZd,dZd,dZd,dZd2dZd2dZd2dZ d3dZ!d,dZ"y)4 SFDictCachezA generic in-memory cache that acts somewhat like a dictionary. Unlike normal dictionaries keys(), values() and items() return list materialized at call time, unlike normal dictionaries that return a view object. ctj||_i|_t |_|j y)z"Inits a SFDictCache with lifetime.)secondsN)datetime timedelta_entry_lifetime_cacher_lock_reset_telemetry)selfentry_lifetimes r __init__zSFDictCache.__init__4s3 (11.I.0 V  rcp|j5t|jcdddS#1swYyxYwr$)r0lenr/r2s r __len__zSFDictCache.__len__>s* ZZ $t{{# $ $ $s,5c 8|di|}|j||S)zCreate an dictionary cache from an already existing dictionary. Note that the same references will be stored in the cache than in the dictionary provided. r)update)cls_dictkwcaches r from_dictzSFDictCache.from_dictBs  b  U rTshould_record_hitsctr!|jjsJd |j|\}}t |r|j||r|j||S#t$r|j |wxYw)zNon-locking version of __getitem__. This should only be used by internal functions when already holding self._lock. 4The mutex self._lock should be locked by this thread) test_moder0lockedr/KeyError_missr'_expire_hit)r2krAtvs r _getitemzSFDictCache._getitemQs  !!# FE F# ;;q>DAq a= LLO  IIaL  JJqM  s A,,Bctr!|jjsJdtt |j z||j |<|jy)zNon-locking version of __setitem__. This should only be used by internal functions when already holding self._lock. rCrrN)rDr0rErr%r.r/_add_or_remover2rJrLs r _setitemzSFDictCache._setitemns\  !!# FE F##54///  A rcl|j5|j|dcdddS#1swYyxYw)AReturns an element if it hasn't expired yet in a thread-safe way.Tr@N)r0rMr2rJs r __getitem__zSFDictCache.__getitem__~s0 ZZ ===t=< = = =*3cj|j5|j||dddy#1swYyxYw)z(Inserts an element in a thread-safe way.N)r0rRrQs r __setitem__zSFDictCache.__setitem__s. ZZ MM!Q    s)2c4t|jSr$)iterkeysr7s r __iter__zSFDictCache.__iter__sDIIK  rcT|jDcgc]\}}| c}}Scc}}wr$items)r2rJ_s r r\zSFDictCache.keys "jjl+da+++ $c |j5g}t|jjD]'} |j ||j |df) ddd|S#t $rY@wxYw#1swYSxYw)NFr@)r0listr/r\appendrMrF)r2valuesrJs r r`zSFDictCache.itemss ZZ (*F$++**,- MM1dmmA%m&P"QR       s.)A7$A(A7( A41A73A44A77BcT|jDcgc]\}}| c}}Scc}}wr$r_)r2rarLs r rgzSFDictCache.valuesrbrcNc0 ||S#t$r|cYSwxYwr$)rF)r2rJdefaults r getzSFDictCache.gets$  7N N s  c|j5|jj|jdddy#1swYyxYwr$)r0r/clearr1r7s r rmzSFDictCache.clears: ZZ $ KK     ! ! # $ $ $s +AA ctr!|jjsJd|j|=|j y)zNon-locking version of __delitem__. This should only be used by internal functions when already holding self._lock. rCN)rDr0rEr/rPr2keys r _delitemzSFDictCache._delitemsA  !!# FE F# KK  rch|j5|j|dddy#1swYyxYwr$)r0rqros r __delitem__zSFDictCache.__delitem__s,ZZ  MM#    s(1c|j5 |j|d dddy#t$r YdddywxYw#1swYyxYw)NTr@F)r0rMrFros r __contains__zSFDictCache.__contains__s[ZZ   cd ;          s$A, AAAAA c Ntr!|jjsJd|jt |t t fr}t|jz}t |t r t|}n)t |t rt|j}Dcic]\}}|t||}}}nt |tr|jt |jj}|Dcic]>\}}||jvs(|r&|j|j|jkr||@}}}nt |jj#||r|j%t'|dkDScc}}wcc}}w)zNon-locking version of update. This should only be used by internal functions when already holding self._lock and other._lock. rCrOr)rDr0rE_clear_expired_entries isinstanceredictr%r.r[r`rr)clear_expired_entriesr/r TypeErrorr:rPr6) r2otherupdate_newer_onlyrgrJrL to_insert others_itemss r _updatezSFDictCache._updatesh  !!# FE F# ##% edD\ *UT111F%&KE4('KLM41aJfA>>MIM { +  ' ' ) 2 2 45L ) AqT[[(-{{1~,,qxx71 I O 9%     !9~!!1N s /FAF!cj|j5|j||cdddS#1swYyxYw)aInsert multiple values at the same time, if self could learn from the other. If this function is given a dictionary, or list expiration timestamps will be all the same a self._entry_lifetime form now. If it's given another SFDictCache then the timestamps will be taken from the other cache. Returns a boolean. It describes whether self learnt anything from other. Note that clear_expired_entries will be called on both caches. To prevent deadlocks this is done without acquiring other._lock. The intended behavior is to use this function with an unpickled/unused cache. If live caches are being merged then use .items() on them first and merge those into the other caches. Nr0r)r2r|r}s r r:zSFDictCache.updates/(ZZ :<<'89 : : :s)2cl|j5|j|dcdddS#1swYyxYw)zAThis function is like update, but it only updates newer elements.Tr}Nr)r2r|s r update_newerzSFDictCache.update_newers7 ZZ <<"&    rWctr!|jjsJdd}t|jj D]} |j |d|r|jyy#t$rd}Y:wxYw)NrCFr@T) rDr0rErer/r\rMrFrP)r2 cache_updatedrJs r rwz"SFDictCache._clear_expired_entries&s  !!# FE F# dkk&&() %A % aE : %     !   %!%  %sA:: BBcf|j5|jdddy#1swYyxYw)z&Remove expired entries from the cache.N)r0rwr7s r rzz!SFDictCache.clear_expired_entries6s* ZZ *  ' ' ) * * *s'0cddddd|_y)z(Re)set telemetry fields. This function will be called by the initalizer and other functions that should reset telemtry entries. r)hitmiss expirationsizeN telemetryr7s r r1zSFDictCache._reset_telemetry<s  rc2|jdxxdz cc<y)aThis function gets called when a hit occurs. Functions that hit every entry (like values) is not going to count. Note that while this function does not interact with lock, but it's only called from contexts where the lock is already held. rrNrrUs r rIzSFDictCache._hitIs u"rc2|jdxxdz cc<y)zThis function gets called when a miss occurs. Note that while this function does not interact with lock, but it's only called from contexts where the lock is already held. rrNrrUs r rGzSFDictCache._missSs v!#rc2|jdxxdz cc<y)zThis function gets called when an expiration occurs. Note that while this function does not interact with lock, but it's only called from contexts where the lock is already held. rrNrrUs r _expirationzSFDictCache._expiration[s |$)$rcR|j||j|t)z7Helper function to call _expiration and delete an item.)rrqrFrUs r rHzSFDictCache._expirecs!  arcHt|j|jd<y)This function gets called when an element is added, or removed. Note that while this function does not interact with lock, but it's only called from contexts where the lock is already held. rN)r6r/rr7s r rPzSFDictCache._add_or_removeis "%T[[!1vr)r3intreturnNone)rr)r<z dict[K, V]rrrJr!rAboolrr"rJr!rLr"rrrJr!rr")rz Iterator[K])rzlist[K])rzlist[tuple[K, V]])rzlist[V]r$)rJr!rjV | Nonerrrr)rpr!rr)rpr!rr)F)r|2dict[K, V] | list[tuple[K, V]] | SFDictCache[K, V]r}rrr)r|rrr)rJr!rr)rJr!rr )#rrr__doc__rDAY_IN_SECONDSr4r8 classmethodr?rM_getitem_non_lockingrRrVrYr]r\r`rgrkrmrqrsrurr:rrwrzr1rIrGrrHrPrrr r)r)-s(66    $     $$(  !  6$ = = =       !,, !   $           #(."A." ."  ."f#(:A: :  :. A   " *  #$* 2rr)ceZdZdZdZej ddf dfd Zdd ddZddZ dfd Z dd Z dd Z d Z ed ZdddZdd dZddZddZddZddZd!fd Zd"dZd#dZd!fd ZxZS)$SFDictFileCache )r.r/r file_path file_timeout_file_lock_path last_loadedrTc@t||t|tr%tj j ||_nwtjj}|||vr(tt|j|_n'tj j |||_djdtdD}tj j!|j}t j"|s t%|d t-j.| \} } t7| d 5} | j9|ddd t7| 5} | j;|k7r t'd dddtj j#| r5tj j=| rt j>|  ||_ |jd|_!tE|jB|j@|_#d|_$tj j#|jr/|r-|jJ5|jMdddd|_'yd|_'y#t&$r"}t(j+d||Yd}~d}~wwxYw#t0$r} t3| j4d |d} ~ wwxYw#1swYxYw#1swYixYw#t0$r} t3| j4d | d} ~ wwxYw#tj j#| r6tj j=| rt j>| wwwxYw#1swYd|_'yxYw)aInits an SFDictFileCache with path, lifetime. File path can be a dictionary that contains different paths for different OSes, possible keys are: 'darwin', 'linux' and 'windows'. If a current platform cannot be determined, or is not in the dictionary we'll use the first value. Once we select a location based on file path, we write and read a random temporary file to check for read/write permissions. If this fails OSError might be thrown. )r3Nrc3bK|]'}tjtj)ywr$)randomchoicestring ascii_letters).0ras r z+SFDictFileCache.__init__..sV f.B.B CVs-/i)modez/cannot create a cache directory: [%s], err=[%s])dirCache folder is not writeablewz-Temporary file just written has wrong contentzCache file is not readablez.locktimeoutF)(superr4rxstrosr expanduserrplatformsystemlowernextr[rgjoinrangedirnameexistsr ExceptionloggerdebugtempfilemkstempOSErrorPermissionErrorerrnoopenwritereadisfileunlinkrrr _file_lockrr0_load_cache_modified)r2rr3rload_if_file_existscurrent_platform random_string cache_folderextmp_file tmp_file_patho_errw_filer_file __class__s r r4zSFDictFileCache.__init__s" )   i %WW// :DN'0668 '+;9+L!%d9+;+;+=&>!?!#!3!3I>N4O!PVUSTXVV wwt~~6 {{<( E2 &.&6&6 ' #Hm )h$ , ]+ , -(SF{{} 5!"QRSww~~m, 1N -(("&..!17"4#7#7ARARS59 77>>$.. ).A    %uW  E   ! /   , ,SS %KK0! ww~~m, 1N -(2O,  %s J,K  L3K1$L3- L 8K>L 2N  K$KK K.K))K.1K;6L3>LL L0L++L00L33AN  Nr@ct||jvr7|j}|r||jvr|j|t|j|\}}t |rD|j}d}|r|j|\}}t |}|r|j ||j ||S)a[Non-locking version of __getitem__ of SFDictFileCache. This should only be used by internal functions when already holding self._lock. Note that we do not overwrite _getitem because _getitem is used by self._load to clear in-memory expired caches. Overwriting would cause infinite recursive call. T)r/_load_if_shouldrGrFr'rHrI)r2rJrAloadedrKrL expire_items r rz$SFDictFileCache._getitem_non_lockings DKK ))+Fq 3 1 {{1~1 a=))+FK{{1~1(m  Q ! rch|j5|j|cdddS#1swYyxYw)rTN)r0rrUs r rVzSFDictFileCache.__getitem__s, ZZ 0,,Q/ 0 0 0(1cFt||||jyr$)rrR_save_if_should)r2rJrLrs r rRzSFDictFileCache._setitems A rc, t|jd5}|j|}ddd|jj j |d}|jj |j|jc|_|_|jd|jd<||_ t|_ y#1swYxYw#ttf$rt$r }tj!d|Yd}~yd}~wwxYw)zFLoad cache from disk if possible, returns whether it was able to load.rbNTrrz-Fail to read cache from disk due to error: %sF)rr _deserializer0releaser:acquirer/rrr%rAssertionError RuntimeErrorrrr)r2rr|cache_file_learntes r rzSFDictFileCache._loads dnnd+ Cv)-):):6)B C JJ   % "&!-!  JJ   (- dkk %DK%*__V%.>$ /+Hmh-8 T__%678JJ}dnn='/'8'8'F'F 0(D$,1D( ww~~m4 9V -0S) 1) 188!*") 7ww~~m4 9V -0ww~~m4 9V -0:W4Q) 1de) 1d  LLT1122PQ   -   L LLG K K Ls K?J8&F): FA&F)AJ8KF& "F)) G>3G G> GG>GG>IG>"G99G>>IAJ8AJ,,J8/K8K=KK,L)2L) L$$L)ch|j5|j|cdddS#1swYyxYw)zSave cache to disk if possible, returns whether it was able to save. This is the public version of _save, it makes sure that all the necessary locks are acquired. N)r0r)r2rs r savezSFDictFileCache.savees, ZZ *::j) * * *rcD|jr|jSy)zSaves file to disk if necessary and returns whether it saved. Uses self._should_save to decide whether to save. F) _should_saverr7s r rzSFDictFileCache._save_if_shouldn    ::< rcD|jr|jSy)zLoad file to disk if necessary and returns whether it loaded. Uses self._should_load to decide whether to load. F) _should_loadrr7s r rzSFDictFileCache._load_if_shouldwr rcHtjd|jdk(S)zDecide whether we should save. This is a simple random number generator to randomize writes across processes that are possibly saving the same values in this cache. r)rrandint MAX_RAND_INTr7s r r zSFDictFileCache._should_saves ~~a!2!23q88rc@tjj|jrutjj |jrL|j yt j jt|j|j kDSy)z{Decide whether we should load. We should load if the file on disk has changed since we have last read it. TF) rrrrrrr,rrr7s r rzSFDictFileCache._should_loadst 77>>$.. )bggnnT^^.L'!!//T^^,""#  rct||j5tjj |j rHtjj|j rtj|j ddd|j5|jdddddy#1swY2xYw#1swYyxYw)NFT)rr) rrmrrrrrrrr0rr2rs r rmzSFDictFileCache.clears   __ *ww~~dnn-"''..2P $..) *ZZ ; JJ%TJ : ; ;  * * ; ;sA2B?"C ?C Cc|jjDcic]\}}|tjvr||c}}Scc}}wr$)__dict__r`r_ATTRIBUTES_TO_PICKLErQs r __getstate__zSFDictFileCache.__getstate__sG ++- 1O999 qD   sAc|jj|d|_t|_t |j |j|_y)NFr) rr:rrr0r rrr)r2states r __setstate__zSFDictFileCache.__setstate__s? U#$V "4#7#7ARARSrc0t|d|_y)rTN)rrPrrs r rPzSFDictFileCache._add_or_removes  #r) rzstr | dict[str, str]r3rrrrrrrrrr)rr)TF)rrrrrr)T)rrrrr)rry)rryrr)rrrrrrrr4rrVrRrrrrrrr rrr rrmrrrP __classcell__)rs@r rrrsL(66$( N%'N%N% N% " N%  N%h$(  !  B0 4 "##@D*9 ; T $$rr)r&rrr), __future__rr,loggingrrrrrrcollections.abcrrr threadingrtypingrr r filelockr r typing_extensionsr rrrrr%rr getLoggerrrgetenvrrDrr!r"r'r)rrrr r(s"  $--&.( 77   CL   8 $ BII' , 2 2 4 >  WQZ  CL CLB2'!Q$-B2J F$kF$r