2iv=&UdZddlmZddlZddlmZmZmZmZm Z ddl m Z ddl m Z mZmZddlmZdd lmZmZdd lmZmZmZmZdd lmZdd lmZdd lmZerddlm Z dZ!dZ"GddeZ#GddZ$e e%e&gdfZ'de(d< d#dZ)e e)Z*Gdde$Z+e e%e&gdfZ,de(d< d$dZ-e e-Z.ej^dZ0 d% d&dZ1ejdejfejhd Z5Gd!d"e$Z6y)'z& Utility module to manipulate queries ) annotationsN) TYPE_CHECKINGAny NamedTuple TypeAlias TypeGuard) lru_cache)CallableMappingSequence)errors)pqsql)BufferParamsQueryQueryNoTemplate)PyFormat)Template)TemplateProcessor) Transformeri2c,eZdZUded<ded<ded<y) QueryPartbytesprez int | stritemrformatN)__name__ __module__ __qualname____annotations__O/var/www/html/secretsanta/venv/lib/python3.12/site-packages/psycopg/_queries.pyrrs J O r%rceZdZdZdj Zd dZd dZd dZe ddZ e ddZ ddZ dd Z y ) PostgresQueryzO Helper to convert a Python query and parameters into Postgres format. z^ query params types formats _tx _want_formats _parts _encoding _order cj||_d|_d|_d|_d|_|d|_d|_y)Nr$r%)_txparamstypes _want_formatsformatsquery_order)self transformers r&__init__zPostgresQuery.__init__,s96: &( 5937   (, r%ct|tr|j||S|j|}|ht |t krt |t krt}nt}|||jj\|_ |_ |_ |_n||_ dx|_ |_ |j|yz Set up the query and parameters to convert. The results of this function can be obtained accessing the object attributes (`query`, `params`, `types`, `formats`). N) isinstancer_convert_template _ensure_byteslenMAX_CACHED_STATEMENT_LENGTHMAX_CACHED_STATEMENT_PARAMS _query2pg_query2pg_nocacher*encodingr/r-r0_partsdumpr1r/varsfs r&convertzPostgresQuery.convert;s eX &))%6 6""5)  E 99I!<<(%IJtxx((J FTZ+T[$+DJ/3 3D  $r%ch||j|j||j}|jJ|jj ||j|_|jjxsd|_|jj|_yd|_d|_d|_y) Process a new set of variables on the query processed by `convert()`. This method updates `params` and `types`. Nr$) validate_and_reorder_paramsr?r0r-r* dump_sequencer+r,r.r1rBr+s r&r@zPostgresQuery.dump^s  55dkk4UF%%1 11((009K9KLDK-2DJ88++DLDKDJDLr%ct|}|tus|turd}|S|turd}|St |t rt |t tfsd}|St |trd}|Stdt|j)NTFz8query parameters should be a sequence or a mapping, got ) typelisttupledictr6r rstrr TypeErrorr")rBtsequences r&is_params_sequencez PostgresQuery.is_params_sequenceos J 9U H$YHh ' 4%0NHg &H  T //02 r%c Jtjrwtt|dz k7r1tjdt|dz dtdr(t |dj ts tdSr/t|dkDr!t |ddts td |r|Dcgc]}| c}Sycc}w#t$r>tjd d jtfd |xsdDwxYw) zO Verify the compatibility between a query and a set of params. r zthe query has z placeholders but z parameters were passedrz2named placeholders require a mapping of parametersz=positional placeholders (%s) require a sequence of parametersr$zquery parameter missing: z, c3,K|] }|vs| ywNr$).0irBs r& z.s(Qq1D=(Qs ) r(rSr9eProgrammingErrorr6rintrPrOKeyErrorjoinsorted)partsrBorderrs ` r&rGz)PostgresQuery.validate_and_reorder_paramss)  + +D 14yCJN*(($SZ!^$45D {"9;JuQx}}c: TUUKE Qz%(1+s/KS 3894DJ99: (( &(QEKR(Q"QRSU sC CCCAD"ct|tr%|j|jjSt|t j r|j|jS|SrV)r6rOencoder*r>r Composableas_bytes)r1r/s r&r8zPostgresQuery._ensure_bytessK eS !<< 1 12 2 s~~ .>>$((+ +Lr%c| tdt|d|j}|j|j|_|j rp|jj |j |j|_|jjxsd|_|jj|_yd|_d|_d|_y)NA'execute()' with string template query doesn't support parametersT server_paramstxr$) rPrr*processr/r+rHr.r,r1r/rBtps r&r7zPostgresQuery._convert_templates  S uDTXX F XX 99((00BJJGDK-2DJ88++DLDKDJDLr%N)r2rr/rrB Params | NonereturnNonerBrorprq)rBrrpzTypeGuard[Sequence[Any]])r`list[QueryPart]rBrrazlist[str] | Nonerpz Sequence[Any])r/rrprr/rrBrorprq)r r!r"__doc__split __slots__r3rDr@ staticmethodrSrGr8r7r$r%r&r(r("s}  EG -!F "$  &, 5E    D r%r(?tuple[bytes, list[PyFormat], list[str] | None, list[QueryPart]]r _Query2PgcXt||}d}g}g}t|djtr|ddD]u}t|jtsJ|j |j |j d|jdzz|j |j wnJt|djtr,i}g}|ddD]}t|jtsJ|j |j |j|vrtdt|dzz}||j f||j<|j |j|j ||j |j ||jd|j k7r#tjd|jd|j ||jd"|j |dj dj||||fS) a Convert Python query and params into something Postgres understands. - Convert Python placeholders (``%s``, ``%(name)s``) into Postgres format (``$1``, ``$2``) - placeholders can be %s, %t, or %b (auto, text or binary) - return ``query`` (bytes), ``formats`` (list of formats) ``order`` (sequence of names used in the query, in the position they appear) ``parts`` (splits of queries and placeholders). Nrs$%dr z placeholder 'z' cannot have different formatsr%) _split_queryr6rr\appendrrrOr9rZr[r^) r/r>r`rachunksr.partseenphs r&r=r=s  )E"EFG%(--%#2J (Ddii- -- MM$(( # MM&DIIM2 3 NN4;; '  ( E!HMM3 '24#2J 2Ddii- -- MM$(( #yy$s4y1}-#%t{{"3TYY TYY' b!t{{+ ?1%4,,' {2QR d499oa01 2" MM%)-- 88F WeU 22r%c,eZdZdZdZddZddZd dZy) PostgresClientQueryzI PostgresQuery subclass merging query and arguments client-side. )templatecpt|tr|j||S|j|}|bt |t krt |t krt}nt}|||jj\|_ |_ |_ n||_d|_ |j|yr5)r6rr7r8r9r:r;_query2pg_client_query2pg_client_nocacher*r>rr0r?r/r@rAs r&rDzPostgresClientQuery.converts eX &))%6 6""5)  E 99I!<<%5,89%ARAR8S 5T]DKDJDK $r%c|_jj|j}tfd|D_j jz_yd_y)rFNc3\K|]#}|jj|nd%yw)NsNULL)r* as_literal)rWpr1s r&rYz+PostgresClientQuery.dump..+s. IJ!-##A&WD s),)rGr?r0rMr+rr/rIs` r&r@zPostgresClientQuery.dump#s]  55dkk4UF NT DK4DJDKr%c| tdt|d|j}|j|j|_|j |_y)NrgFrh)rPrr*rkr/r+rls r&r7z%PostgresClientQuery._convert_template2sK  S uEdhh G XX ii r%Nrnrrrt)r r!r"rurwrDr@r7r$r%r&rrsI8   r%r/tuple[bytes, list[str] | None, list[QueryPart]]_Query2PgClientcVt||d}d}g}t|djtrS|ddD]J}t|jtsJ|j |j |j dLnt|djt ri}g}|ddD]}t|jt sJ|j |j |j|vrJd}||jf||j<|j |j|j ||j ||jd|j |j|j |dj dj|||fS)zX Convert Python query and params into a template to perform client-side binding F)collapse_double_percentNrr|s%sr%) r}r6rr\r~rrOrr^)r/r>r`rarrrrs r&rrCsb % HE"EF%(--%#2J !Ddii- -- MM$(( # MM%  ! E!HMM3 '24#2J (Ddii- -- MM$(( #yy$#%t{{"3TYY TYY' b! d499oa01 TYY' ( MM%)-- 88F UE ))r%s"(?x) % # a literal % (?: (?: \( ([^)]+) \) # or a name in (braces) . # followed by a format ) | (?:.) # or any char, really ) cg}d}d}tj|D]@}|||jdd}|j||f|jdd}B|r|j||ddfn|j|dfg}d}d} |t |kr||\}}|-|jt |dt j |S|jdx} dk(r#|rd} ||dz\} } || z| z| f||dz<||=| dk(rNtjd||jdddjdj|d| dk(rtjd | d dd vr7tjd |jdj|d|jdr |jdj|n|} | s t| } n"| t| urtjd t| d d}|jt || ||dz }|t |kr|S)Nrr s%%%s%(zincomplete placeholder: ''s% zfincomplete placeholder: '%'; if you want to use '%' as an operator you can double it up, i.e. use '%%'r|ssbtz8only '%s', '%b', '%t' are allowed as placeholders, got 'z1positional and named placeholders cannot be mixed)_re_placeholderfinditerspanr~r9rrAUTOgrouprZr[rvdecoderK _ph_to_fmt)r/r>rr`curmrrvrXphtyperpre1m1rrs r&r}r}zs9;E C A  % %e ,C!&&)A,' c1XffQil  eCDk4() eT]# B A F c%j.qQ 9 IIiQ 6 7 V IS''!* B &&QU|HD""HtOR0E!a%La  ;$$166!9Q<=)//1!4;;HEFaI 5[$$@ WF "$$WWQZ&&x014 /0ggajqwwqz  *a$ZF 4: %$$C BrsG$ )Cv./ Q] c%j.` Ir%)stbc$eZdZddZddZddZy)PostgresRawQueryct|tr|j||S|j||_dx|_|_|j|yrV)r6rr7r8r/r-r0r@r1r/rBs r&rDzPostgresRawQuery.convertsK eX &))%6 6''. +//T[ $r%c|tj|s tdtjgt |z|_|jj||j |_ |jjxsd|_ |jj|_ yd|_ d|_ d|_ y)Nz,raw queries require a sequence of parametersr$) r(rSrPrrr9r-r*rHr+r,r.)r1rBs r&r@zPostgresRawQuery.dumps   33D9 NOO"*--3t9!rOrpry)r/rr>rOrpr)asciiT)r/rr>rOrboolrprs)7ru __future__rretypingrrrrr functoolsr collections.abcr r r rrZrrabcrrrr_enumsr_compatr _tstringsrrr:r;rr(rrOrzr#r=r<rrrrcompilerr}rTEXTBINARYrrr$r%r&rsx # GG7777( "   [ [ ~  CLSS 9 /3 /3/3D/3n ' ( ; -; |& CLCC "* "*"*4"*J56"**    LPF FFDHFFT -- -- //  } r%