i4 UdZddlmZddlZddlmZmZmZmZm Z m Z m Z ddl m Z ddlmZddlmZdd lmZmZmZdd lmZdd lmZdd lmZmZdd lmZerddlmZdZ dZ!Gdde Z"GddZ#ee$e%gdfZ&de'd< d!dZ(e e(Z)Gdde#Z*ee$e%gdfZ+de'd< d"dZ,e e,Z-ej\dZ/ d# d$dZ0ejbejdejfd Z4y)%z& Utility module to manipulate queries ) annotationsN) TYPE_CHECKINGAnyCallableMappingMatch NamedTupleSequence) lru_cache)errors)pq)BufferParamsQuery) Composable)PyFormat) TypeAlias TypeGuard) conn_encoding) Transformeri2c,eZdZUded<ded<ded<y) QueryPartbytesprez int | stritemrformatN)__name__ __module__ __qualname____annotations__p/var/www/html/hwPaymentPortal-be-dev/hw-payment-portal-api/venv/lib/python3.12/site-packages/psycopg/_queries.pyrrs J O r$rc|eZdZdZdj Zd dZd dZd dZe d dZ e d dZ y) PostgresQueryzO Helper to convert a Python query and parameters into Postgres format. z^ query params types formats _tx _want_formats _parts _encoding _order c||_d|_d|_d|_d|_t |j |_|d|_d|_ y)Nr#r$) _txparamstypes _want_formatsformatsr connection _encodingquery_order)self transformers r%__init__zPostgresQuery.__init__,sK6: &( 5937 &{'='=>  (, r$ct|tr|j|j}n.t|tr|j |j }n|}|^t|tkrt|tkrt}nt}|||j\|_ |_ |_|_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) isinstancestrencoder/ras_bytesr)lenMAX_CACHED_STATEMENT_LENGTHMAX_CACHED_STATEMENT_PARAMS _query2pg_query2pg_nocacher0r,r1_partsdumpr2r0varsbqueryfs r%convertzPostgresQuery.convert<s eS !\\$..1F z *^^DHH-FF  F ::I!<<(%IJJ 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@r1r,r) dump_sequencer*r+r-r2rCr*s r%rAzPostgresQuery.dumpas  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 ) typelisttupledictr7r rr8r TypeErrorr)rCtsequences r%is_params_sequencez PostgresQuery.is_params_sequencers J 9U H$YHh ' 4%0NHg &H  T ++,. 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| yw)Nr#).0irCs r% z.s(Qq1D=(Qs ) r'rTr;eProgrammingErrorr7rintrQr8KeyErrorjoinsorted)partsrCorderrs ` r%rIz)PostgresQuery.validate_and_reorder_paramss)  + +D 14yCJN*(($SZ!^$45D {"9;JuQx}}c: TUUKE Qz%(1+s/KS 3894DJ99: (( &(QEKR(Q"QRSU sC CCCAD"N)r3rr0rrC Params | NonereturnNonerCrcrdre)rCrrdzTypeGuard[Sequence[Any]])r`list[QueryPart]rCrrazlist[str] | Nonerdz Sequence[Any]) rr r!__doc__split __slots__r4rFrA staticmethodrTrIr#r$r%r'r'"sr  EG - #J "$  &, 5E    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_queryr7rr\appendrrr8r;rZr[r^) r0encodingr`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$eZdZdZdZddZddZy)PostgresClientQueryzI PostgresQuery subclass merging query and arguments client-side. )templatect|tr|j|j}n.t|tr|j |j }n|}|Xt|tkrt|tkrt}nt}|||j\|_ |_ |_n||_d|_ |j!|yr6)r7r8r9r/rr:r)r;r<r=_query2pg_client_query2pg_client_nocacheryr1r@r0rArBs r%rFzPostgresClientQuery.converts eS !\\$..1F z *^^DHH-FF  F ::I!<<%5,89&$..8Q 5T]DKDJDK $r$c|_jj|j}tfd|D_j jz_yd_y)rHNc3\K|]#}|jj|nd%yw)NsNULL)r) as_literal)rWpr2s r%rYz+PostgresClientQuery.dump..s. IJ!-##A&WD s),)rIr@r1rOr*ryr0rKs` r%rAzPostgresClientQuery.dumps]  55dkk4UF NT DK4DJDKr$Nrbrf)rr r!rhrjrFrAr#r$r%rxrxsI< r$rx/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_percentNrros%sr$) rpr7rr\rqrr8rr^)r0rrr`rarsrtrurvs r%r|r|#sb % 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 '%%'rossbtz8only '%s', '%b', '%t' are allowed as placeholders, got 'z1positional and named placeholders cannot be mixed)_re_placeholderfinditerspanrqr;rrAUTOgrouprZr[ridecoderM _ph_to_fmt)r0rrrr`curmrrvrXphtypervpre1m1rrs r%rprpZs68E 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$)stb)r0rrrr8rdrl)r0rrrr8rdr)asciiT)r0rrrr8rboolrdrg)5rh __future__rretypingrrrrrr r functoolsr r rZrabcrrrsqlr_enumsr_compatrr _encodingsrrr<r=rr'rr8rmr"r?r>rxrr|r{compilerrprTEXTBINARYrr#r$r%rsl # UUU&&)% "   DDP  CLSS 9 /3 /3/3D/3n ' ( 2-2j& CLCC "* "*"*4"*J56"**    LPF FFDHFFT -- -- // r$