o wvXjMK@sddlZddlZddlZddlmZddlmZddlmZm Z ddl m Z ddl m Z ddlmZmZmZdd lmZdd lmZmZdd lmZeeeejfZed Zd dZGdddZdS)N) OrderedDict)settings)DEFAULT_TIMEOUT get_key_func)ImproperlyConfigured) import_string)ConnectionError ResponseError TimeoutError)pool)CompressorErrorConnectionInterrupted)CacheKeyz([*?[])cCs td|S)Nz[\1]) special_resub)srX/var/www/html/myproject/venv/lib/python3.10/site-packages/django_redis/client/default.py glob_escapes rc@sbeZdZddZddZdFddZdGd d ZdHd dZeddd d fddZ dIddZ eddfddZ dJddZ dKddZ dKddZ      dLddZdJd d!Z dMd"d#ZdKd$d%ZdNd&d'Zd(d)Zd*d+ZdKd,d-Zeddfd.d/ZdOd0d1ZdOd2d3ZdId4d5ZdKd6d7ZdKd8d9ZdJd:d;ZdKdd?ZdKd@dAZ dBdCZ!eddfdDdEZ"dS)P DefaultClientcCs||_||_||_t|dpd|_|jstdt|jtt t fs*|j d|_dgt |j|_ |di|_|jdd|_|jdd }t|}|jd d }t|}||jd |_||jd |_tj|jd |_dS) NREVERSE_KEY_FUNCTIONz%django_redis.util.default_reverse_keyzMissing connections string,OPTIONSSLAVE_READ_ONLYT SERIALIZERz0django_redis.serializers.pickle.PickleSerializer COMPRESSORz4django_redis.compressors.identity.IdentityCompressor)options)_backend_server_paramsrget reverse_keyr isinstancelisttuplesetsplitlen_clients_options_slave_read_onlyr _serializer _compressorr get_connection_factoryconnection_factory)selfserverparamsbackendserializer_pathserializer_clscompressor_pathcompressor_clsrrr__init__s2 zDefaultClient.__init__cCs ||SN)has_key)r0keyrrr __contains__>s zDefaultClient.__contains__Trcsjrtt|jkrfddtdt|jD}t|S|s(t|jdkr*dStdt|jdS)z Return a next index for read client. This function implements a default behavior for get a next read client for master-slave setup. Overwrite this function if you want a specific behavior. csg|]}|vr|qSrr).0itriedrr Ksz7DefaultClient.get_next_client_index..r)r(rrangerandomchoicerandint)r0writer@ not_triedrr?rget_next_client_indexAs   z#DefaultClient.get_next_client_indexFcCsL|j||pgd}|j|dur|||j|<|r!|j||fS|j|S)z Method used for obtain a raw redis client. This function is used by almost all cache backend operations for obtain a native redis client/connection instance. )rGr@N)rIr)connect)r0rGr@ show_indexindexrrr get_clientSs  zDefaultClient.get_clientrcCs|j|j|S)z Given a connection index, returns a new raw redis client/connection instance. Index is used for master/slave setups and indicates that connection string should be used. In normal setups, index is 0. )r/rJr)r0rLrrrrJeszDefaultClient.connectNc Cs |j||d}||} |tur|jj}|} g} z>|dur(|jd| dd\}} |durKt|d}|dkrK|rB|j|||d WS|j|||dWSt |j || |||d WSt y} z!| sz|j szt | t |jkrz| | d}WYd} ~ qt|d | d} ~ ww) z Persist a value to the cache, and set an optional expiration time. Also supports optional nx parameter. If set to True - will use redis setnx instead of set. versionTN)rGr@rKrrOclient)rRrO)nxpxxx connection)make_keyencoderrdefault_timeoutrMintr:deleteboolr&_main_exceptionsr+r(rappendr)r0r;valuetimeoutrOrRrSrUnkeynvalueoriginal_clientr@rLerrrr&ms>      zDefaultClient.setrBc Cs|dur |jdd}|dur|jj}|||}|j|||d}z||}Wnty9}zt|d|d}~ww|durDtd|t |t rU|j| ||d} n |j|||d} |j | |||d|j ||d ||S) zh Adds delta to the cache version for the supplied key. Returns the new version. NTrGrQrVKey '%s' not foundrN)rarR)rR)rMrrOrXr!ttlr^r ValueErrorr#r original_keyr&r\) r0r;deltarOrRold_keyr`rhrenew_keyrrr incr_versions(     zDefaultClient.incr_versioncCs|j|||||ddS)z Add a value to the cache, failing if the key already exists. Returns ``True`` if the object was added, ``False`` if not. T)rOrRrS)r&)r0r;r`rarOrRrrraddszDefaultClient.addc Csl|dur |jdd}|j||d}z||}Wnty*}zt|d|d}~ww|dur1|S||S)zv Retrieve a value from the cache. Returns decoded value if key is found, the default if not. NFrfrNrV)rMrXr!r^rdecode)r0r;defaultrOrRr`rerrrr!s   zDefaultClient.getcCs>|dur |jdd}|j||d}||r||dSdSNTrfrN)rMrXexistspersist)r0r;rOrRrrrrts   zDefaultClient.persistcCs@|dur |jdd}|j||d}||r|||dSdSrr)rMrXrsexpirer0r;rarOrRrrrrus   zDefaultClient.expire皙?cCs6|dur |jdd}|j||d}|j|||||dS)NTrfrN)rasleepblocking_timeout thread_local)rMrXlock)r0r;rOrarxryrRrzrrrr{s zDefaultClient.lockc CsR|dur |jdd}z ||j|||dWSty(}zt|d|d}~ww)z. Remove a key from the cache. NTrfrOprefixrV)rMr\rXr^r)r0r;rOr}rRrerrrr\ s  zDefaultClient.deletec Cs|dur |jdd}|j|||d}d|i}|r||d<zd}|jd i|D] }|||d7}q'|WStyG} zt|d | d} ~ ww) z3 Remove all keys matching pattern. NTrfr|matchcountrrBrVr)rM make_pattern scan_iterr\r^r) r0patternrOr}rRitersizekwargsrr;rerrrdelete_patterns     zDefaultClient.delete_patternc sb|dur jdd}fdd|D}|sdSz|j|WSty0}zt|d|d}~ww)z/ Remove multiple keys at once. NTrfcsg|] }j|dqS)rNrXr=kr0rOrrrA:z-DefaultClient.delete_many..rV)rMr\r^r)r0keysrOrRrerrr delete_many2s   zDefaultClient.delete_manyc CsH|dur |jdd}z|WdSty#}zt|d|d}~ww)z' Flush all cache keys. NTrfrV)rMflushdbr^r)r0rRrerrrclearDs  zDefaultClient.clearc CsXzt|}W|Sttfy+z|j|}Wn ty!Ynw|j|}Y|Sw)z) Decode the given value. )r[ri TypeErrorr- decompressr r,loadsr0r`rrrrpQs  zDefaultClient.decodecCs4t|ts t|ts|j|}|j|}|S|S)z) Encode the given value. )r#r]r[r,dumpsr-compressrrrrrY`s   zDefaultClient.encodec s|dur jdd}|siSt}tfdd|D}z|j|}Wnty6}zt|d|d}~wwt||D]\}} | durEq<| |||<q<|S)z% Retrieve many keys. NFrfc3s"|] }j|d|fVqdS)rNNrrrrr ys z)DefaultClient.get_many..rV)rMrmgetr^rziprp) r0rrOrRrecovered_datamap_keysresultsrer;r`rrrget_manyls"  zDefaultClient.get_manyc Csv|dur |jdd}z|}|D]\}}|j|||||dq|WdSty:}zt|d|d}~ww)a" Set a bunch of values in the cache at once from a dict of key/value pairs. This is much more efficient than calling set() multiple times. If timeout is given, that timeout will be used for the key; otherwise the default cache timeout will be used. NTrfrQrV)rMpipelineitemsr&executer^r) r0datararOrRrr;r`rerrrset_manys  zDefaultClient.set_manyc Cs|dur |jdd}|j||d}zLz|sd}nd}||d||}|dur,td|WW|Sty]||}|dkrEtd||j|||d |}|j|||||d YW|Swtyo} zt |d | d} ~ ww) NTrfrNz local exists = redis.call('EXISTS', KEYS[1]) if (exists == 1) then return redis.call('INCRBY', KEYS[1], ARGV[1]) else return false end zW return redis.call('INCRBY', KEYS[1], ARGV[1]) rBrgrQ)rOrarRrV) rMrXevalrir rhr!r&r^r) r0r;rkrOrRignore_key_checkluar`rarerrr_incrs2     zDefaultClient._incrcCs|j|||||dS)z Add delta to value in the cache. If the key does not exist, raise a ValueError exception. if ignore_key_check=True then the key will be created and set to the delta value by default. )r;rkrOrRrr)r0r;rkrOrRrrrrincrszDefaultClient.incrcCs|j|| ||dS)zx Decreace delta to value in the cache. If the key does not exist, raise a ValueError exception. )r;rkrOrRr)r0r;rkrOrRrrrdecrszDefaultClient.decrcCsb|dur |jdd}|j||d}||sdS||}|dkr#|S|dkr)dS|dkr/dSdS)z Executes TTL redis command and return the "time-to-live" of specified key. If key is a non volatile key, it returns None. NFrfrNrr)rMrXrsrh)r0r;rOrRtrrrrhs   zDefaultClient.ttlc CsX|dur |jdd}|j||d}z||dkWSty+}zt|d|d}~ww)z% Test if key exists. NFrfrNrBrV)rMrXrsr^r)r0r;rOrRrerrrr:s  zDefaultClient.has_keyccsL|dur |jdd}|j||d}|j||dD] }||VqdS)zo Same as keys, but uses redis >= 2.8 cursors for make memory efficient keys iteration. NFrfrN)r~r)rMrrr"rp)r0searchrrRrOritemrrr iter_keyss zDefaultClient.iter_keysc sb|dur jdd}j||d}z fdd||DWSty0}zt|d|d}~ww)z Execute KEYS command and return matched results. Warning: this can return huge number of results, in this case, it strongly recommended use iter_keys for it. NFrfrNcsg|] }|qSr)r"rprr0rrrArz&DefaultClient.keys..rV)rMrrr^r)r0rrOrRrrerrrrs  zDefaultClient.keyscCsBt|tr|S|dur|jj}|dur|jj}t|j|||Sr9)r#rr key_prefixrOkey_func)r0r;rOr}rrrrXs zDefaultClient.make_keycCsVt|tr|S|dur|jj}t|}|dur|jj}tt|}t|j|||Sr9)r#rrrrrOstrr)r0rrOr}rrrr+s  zDefaultClient.make_patterncKsNttddr#tt|jD]}|j|jjD]}|qd|j|<q dSdS)NDJANGO_REDIS_CLOSE_CONNECTIONF)getattrrrCr(r)connection_pool_available_connections disconnect)r0rr>crrrclose9s   zDefaultClient.closecCsd|tur|jj}|dur|jdd}|j||d}|dur$t||St|d}t|||S)z2 Sets a new expiration for a key. NTrfrNrP) rrrZrMrXr]rtr[pexpirervrrrtouch@s  zDefaultClient.touch)Tr)TrF)r)rBNN)NNN)NN)NNrwNNT)NNNNr9)rBNNF)#__name__ __module__ __qualname__r8r<rIrMrJrr&rnror!rtrur{r\rrrrprYrrrrrrhr:rrrXrrrrrrrrsX#     ;           -      r)rDresocket collectionsr django.confrdjango.core.cache.backends.baserrdjango.core.exceptionsrdjango.utils.module_loadingrredis.exceptionsrr r r exceptionsr rutilrrar^compilerrrrrrrs