Ytj,UdZddlmZddlmZmZddlmZmZm Z ddl m Z m Z ddl mZddlmZdd lmZdd lmZer dd lmZdd lmZeeeed ffZded<ejjZejj Z GddeZ!GddZ"dS)z! Support for prepared statements ) annotations)IntEnumauto) TYPE_CHECKINGAny TypeAlias) OrderedDictdeque)Sequence)pq)PQGen) PostgresQuery)PGresult)BaseConnection.rKeycJeZdZeZeZeZdS)PrepareN)__name__ __module__ __qualname__rNOYESSHOULDP/var/www/html/CA-Chatbot/venv/lib/python3.11/site-packages/psycopg/_preparing.pyrrs/ B $&&C TVVFFFrrceZdZUdZded<dZded<d&d Zed'd Z d(d)dZ d*dZ ed+dZ d&dZ d,dZ d-d Zd.d!Zd/d%ZdS)0PrepareManagerz int | Noneprepare_thresholddint prepared_maxreturnNonect|_t|_d|_t t dz|_dS)Nr)r _counts_names _prepared_idxr bytes _to_flushselfs r__init__zPrepareManager.__init__)s>.9mm 0;}} ut|,..rqueryrrc|j|jfSN)r0types)r0s rkeyzPrepareManager.key5s U[))rNprepare bool | Nonetuple[Prepare, bytes]c|dus|jtjdfS|j||x}x}rtj|fS|j|d}||jks|r:d|j }|xjdz c_tj |fStjdfS)zP Check if a query is prepared, tell back whether to prepare it. FNrr_pg3_r ) r!rrr)getr4rr(r*encoder)r.r0r5r4namecounts rr:zPrepareManager.get9s e  t5=:s? ";??$((5//#93:: :4 %;$ $   a(( D* * *g */4-//6688D   ! #  >4' ':s? "rpreprresultsSequence[PGresult]boolc|js|tjkrP|D]M}|jtkr|j}|r1|ds|dkr|cSNdS)zCheck if we need to discard our entire state: it should happen on rollback or on dropping objects, because the same object may get recreated and postgres would fail internal lookups. sDROP sROLLBACKF)r)rrstatus COMMAND_OKcommand_status startswithclear)r.r>r?resultcmdstats r_should_discardzPrepareManager._should_discardQs ; ($'.00! ( (=J.. /( 2 28 < <(;@V@V::<<'''urc~t|dkrdSt|djcxkr tkrnndSdS)zCReturn False if 'results' are invalid for prepared statement cache.r FrT)lenrDrC TUPLES_OK)r?s r_check_resultszPrepareManager._check_results_sM w<<1  5 * 7 7 7 7i 7 7 7 7 75trc*t|j|jkr|jdt|j|jkr=|jdd}|j|dSdS)zEvict an old value from the cache. If it was prepared, deallocate it. Do it only once: if the cache was resized, deallocate gradually. F)lastr N)rLr(r$popitemr)r,append)r.r<s r_rotatezPrepareManager._rotatels t|  t0 0 0 L e , , , t{  d/ / /;&&E&2215D N ! !$ ' ' ' ' ' 0 /rr<r+ Key | Nonec|jdS||x}|jvrR|tjur|j|=||j|<n/|j|xxdz cc<|j|dS||jvr|j|dS|tjur ||j|<n d|j|<|S)zHandle 'query' for possible addition to the cache. If a new entry has been added, return its key. Return None otherwise (meaning the query is already in cache or cache is not enabled). Nr )r!r4r(rrr) move_to_end)r.r0r>r<r4s rmaybe_add_to_cachez!PrepareManager.maybe_add_to_cacheys  ! )488E?? "Ct| 3 3w~%%L%#' C   S!!!Q&!!! ((---4 DK   K # #C ( ( (4w~%%#' C  $% S!Jrr4c|||rdS||s8|j|d|j|ddS|dS)zValidate cached entry with 'key' by checking query 'results'. Possibly record a command to perform maintenance on database side. N)rJrNr)popr(rS)r.r4r>r<r?s rvalidatezPrepareManager.validates|   g . .  F""7++  KOOC & & & L  S$ ' ' ' ' ' LLNNNNNrc|j|jrN|j|j|jddSdS)zClear the cache of the maintenance commands. Clear the internal state and prepare a command to clear the state of the server. NTF)r(rGr)r,rRr-s rrGzPrepareManager.clearsj  ;  K      N " " " N ! !$ ' ' '45rconnBaseConnection[Any] PQGen[None]c#K|jr=|j}||Ed{V|j;dSdS)z Generator to send the commands to perform periodic maintenance Deallocate unneeded command in the server, or flush the prepared statements server state entirely if necessary. N)r,popleft _deallocate)r.r\r<s r maintain_genzPrepareManager.maintain_genskn .>))++D''-- - - - - - - -n . . . . .r)r%r&)r0rr%rr2)r0rr5r6r%r7)r>rr?r@r%rA)r?r@r%rA)r0rr>rr<r+r%rT) r4rr>rr<r+r?r@r%r&)r%rA)r\r]r%r^)rrrr!__annotations__r$r/ staticmethodr4r:rJrNrSrWrZrGrbrrrrr"s'$%%%%%L / / / /***\*<@#####0       \  ( ( ( (@     . . . . . .rrN)#__doc__ __future__renumrrtypingrrr collectionsr r collections.abcr r abcr_queriesrpq.abcr_connection_basertupler+r#rrc ExecStatusrDrMrrrrrrrs #"""""0000000000********$$$$$$######1      000000ueCHo-..... ] % M # g _._._._._._._._._._.r