qj&UdZddlmZddlmZmZddlmZmZm Z ddl m Z ddl m Z ddlmZdd lmZmZdd lmZe r dd lmZdd lmZd Zded<e j6j8Ze j6j:ZGddeZGddZy)z! Support for prepared statements ) annotations)IntEnumauto)AnySequence TYPE_CHECKING) OrderedDict)pq)PQGen)Deque TypeAlias) PostgresQuery)PGresult)BaseConnectionztuple[bytes, tuple[int, ...]]rKeyc6eZdZeZeZeZy)PrepareN)__name__ __module__ __qualname__rNOYESSHOULDS/var/www/html/kelly-backend/venv/lib/python3.12/site-packages/psycopg/_preparing.pyrrs B &C VFrrceZdZUdZded<dZded<ddZeddZ d dd Z dd Z edd Z dd Z ddZ ddZddZddZy )PrepareManagerz int | Noneprepare_thresholddint prepared_maxcrt|_t|_d|_t d|_y)Nrz bytes | None)r _counts_names _prepared_idxr _to_flushselfs r__init__zPrepareManager.__init__)s/.9m 0;} ~.0rc2|j|jfSN)querytypes)r/s rkeyzPrepareManager.key5s U[[))rNc|dus |jtjdfS|j|}|jj |}|rtj |fS|jj |d}||jk\s|rDd|jj}|xjdz c_tj|fStjdfS)zP Check if a query is prepared, tell back whether to prepare it. Frr_pg3_r ) r!rrr1r'getrr&r(encoder)r+r/preparer1namecounts rr4zPrepareManager.get9s e t55=::s? "hhuo{{s# ;;$ $   a( D** *g4--./668D   ! # >>4' '::s? "rc|js|tjk(rR|D]M}|jtk7r|j }|s&|j ds|dk(s=|jcSy)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+prepresultsresultcmdstats r_should_discardzPrepareManager._should_discardSsb ;;$'..0!==J. // 2 28 <;@V::<' " rclt|dk7ry|dj}t|cxk7r tk7ryyy)zCReturn False if 'results' are invalid for prepared statement cache.r FrT)lenr:r; TUPLES_OK)r@r:s r_check_resultszPrepareManager._check_resultsas< w<1 ""  ,9 , -rc:t|j|jkDr|jjdt|j|jkDr;|jjdd}|j j |yy)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)rEr&r$popitemr'r)append)r+r7s r_rotatezPrepareManager._rotateosz t|| t00 0 LL e , t{{ d// /;;&&E&215D NN ! !$ ' 0rc|jy|j|}||jvrb|tjur|j|=||j |<y|j|xxdz cc<|jj |y||j vr|j j |y|tjur||j |<|Sd|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!r1r&rrr' move_to_end)r+r/r?r7r1s rmaybe_add_to_cachez!PrepareManager.maybe_add_to_cache|s  ! ! )hhuo $,, w~~%LL%#' C  S!Q&! ((- DKK  KK # #C (w~~%#' C J%& S!Jrc|j||ry|j|s9|jj|d|jj|dy|j y)zValidate cached entry with 'key' by checking query 'results'. Possibly record a command to perform maintenance on database side. N)rCrGr'popr&rL)r+r1r?r7r@s rvalidatezPrepareManager.validatesT   g . ""7+ KKOOC & LL  S$ ' LLNrc|jj|jrP|jj|jj|jj dyy)zClear the cache of the maintenance commands. Clear the internal state and prepare a command to clear the state of the server. NTF)r&r>r'r)rKr*s rr>zPrepareManager.clearsS  ;; KK    NN " NN ! !$ 'rc#K|jrA|jj}|j|Ed{|jr@yy7w)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+connr7s r maintain_genzPrepareManager.maintain_gensAnn>>))+D''- - -nn -s;AAAA)returnNone)r/rrYrr.)r/rr6z bool | NonerYztuple[Prepare, bytes])r?rr@Sequence[PGresult]rYbool)r@r[rYr\)r/rr?rr7bytesrYz Key | None) r1rr?rr7r]r@r[rYrZ)rYr\)rWzBaseConnection[Any]rYz PQGen[None])rrrr!__annotations__r$r, staticmethodr1r4rCrGrLrOrRr>rXrrrrr"s$%z%L# 1**<@#"#-8# #4    ("*19> B   $   (  .rrN) __doc__ __future__renumrrtypingrrr collectionsr r abcr _compatr r_queriesrpq.abcr_connection_baserrr^ ExecStatusr;rFrrrrrrlsp #//#%# 00Y0 ]] % % MM # # g g.g.r