o "4g4@szUdZddlmZddlZddlmZmZmZmZm Z ddlm Z m Z ddl m Z ddlmZdd lmZdd lmZdd lmZmZmZdd lmZdd lmZmZddlmZe raddlmZdZ dZ!Gddde Z"GdddZ#ee$e%gdfZ&de'd<d/ddZ(e e(Z)Gd d!d!e#Z*ee$e%gd"fZ+de'd#<d0d$d%Z,e e,Z-e.d&Z/ (d1d2d,d-Z0ej1ej2ej3d.Z4dS)3z& Utility module to manipulate queries ) annotationsN)AnyCallableMappingMatch NamedTuple)Sequence TYPE_CHECKING) lru_cache)pq)errors) Composable)BufferQueryParams)PyFormat) TypeAlias TypeGuard) conn_encoding) Transformeri2c@s&eZdZUded<ded<ded<dS) QueryPartbytesprez int | stritemrformatN)__name__ __module__ __qualname____annotations__r!r!Y/var/www/html/authentication-server/venv/lib/python3.10/site-packages/psycopg/_queries.pyrs  rc@sReZdZdZdZdddZdd dZdddZe d ddZ e d!ddZ dS)" PostgresQueryzO Helper to convert a Python query and parameters into Postgres format. z^ query params types formats _tx _want_formats _parts _encoding _order transformerrcCs>||_d|_d|_d|_d|_t|j|_|d|_d|_ dS)Nr!) _txparamstypes _want_formatsformatsr connection _encodingquery_order)selfr$r!r!r"__init__-s  zPostgresQuery.__init__r-rvars Params | NonereturnNonecCst|tr ||j}nt|tr||j}n|}|dur>t|tkr-t|t kr-t }nt }|||j\|_ |_ |_|_n ||_ d|_ |_||dSz 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_nocacher-r)r._partsdumpr/r-r1bqueryfr!r!r"convert=s     zPostgresQuery.convertcCsl|dur+||j||j}|jdusJ|j||j|_|jjp"d|_|jj|_dSd|_d|_d|_dS) Process a new set of variables on the query processed by `convert()`. This method updates `params` and `types`. Nr!) validate_and_reorder_paramsr?r.r)r& dump_sequencer'r(r*r/r1r'r!r!r"r@bs zPostgresQuery.dumprTypeGuard[Sequence[Any]]cCsvt|}|tus |turd}|S|turd}|St|tr(t|ttfs(d}|St|tr1d}|St dt|j )NTFz8query parameters should be a sequence or a mapping, got ) typelisttupledictr6rrr7r TypeErrorr)r1tsequencer!r!r"is_params_sequencess$   z PostgresQuery.is_params_sequencepartslist[QueryPart]orderlist[str] | None Sequence[Any]c str1tt|dkr!tdt|ddtdr/t|djts/tdSrFt|dkrFt|ddt sFtdz|rSfdd |DWSWd St yrtd d t fd d|pjd Dw)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 parameterscsg|]}|qSr!r!).0rr1r!r" sz=PostgresQuery.validate_and_reorder_params..r!zquery parameter missing: z, c3s|] }|vr|VqdS)Nr!)rWirXr!r" sz.) r#rQr:eProgrammingErrorr6rintrNr7KeyErrorjoinsorted)rRr1rTr!rXr"rFs2 " z)PostgresQuery.validate_and_reorder_paramsN)r$rr-rr1r2r3r4r1r2r3r4)r1rr3rI)rRrSr1rrTrUr3rV) rrr__doc__split __slots__r0rDr@ staticmethodrQrFr!r!r!r"r##s   % r#?tuple[bytes, list[PyFormat], list[str] | None, list[QueryPart]]r _Query2Pgr-rencodingr7r3c Csht||}d}g}g}t|djtr;|ddD] }t|jts#J||j|d|jd||jqnht|djtri}g}|ddD]U}t|jtsWJ||j|j|vrdt|d}||jf||j<||j||||jqM||jd|jkrt d|jd|||jdqM||djd ||||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^appendrrr7r:r\r]r`) r-rjrRrTchunksr*partseenphr!r!r"r>s<       r>c@s(eZdZdZdZdd d Zdd d Zd S)PostgresClientQueryzI PostgresQuery subclass merging query and arguments client-side. )templater-rr1r2r3r4cCst|tr ||j}nt|tr||j}n|}|dur.)rFr?r.rLr'rsr-rHr!rxr"r@s   zPostgresClientQuery.dumpNrbrc)rrrrdrfrDr@r!r!r!r"rrs  rr/tuple[bytes, list[str] | None, list[QueryPart]]_Query2PgClientcCst||dd}d}g}t|djtr0|ddD]}t|jts#J||j|dqnNt|djtr~i}g}|ddD];}t|jtsLJ||j|j|vrmd}||jf||j<||j||qB|||jd||jqB||djd|||fS)zX Convert Python query and params into a template to perform client-side binding F)collapse_double_percentNrrks%sr%) rlr6rr^rmrr7rr`)r-rjrRrTrnrorprqr!r!r"ru$s0      rus"(?x) % # a literal % (?: (?: \( ([^)]+) \) # or a name in (braces) . # followed by a format ) | (?:.) # or any char, really ) asciiTr{boolrScCsg}d}d}t|D]}|||dd}|||f|dd}q |r5|||ddfn||dfg}d}d} |t|kr||\}}|dur_|t|dtj |S|d} | dkr|rld} ||d\} } || | | f||d<||=qB| dkrt d||ddd d |d| dkrt d | d dd vrt d |d |d|dr|d |n|} | st | } n | t | urt d t| d d}|t|| ||d7}|t|ksH|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 '%%'rkssbtz8only '%s', '%b', '%t' are allowed as placeholders, got 'z1positional and named placeholders cannot be mixed)_re_placeholderfinditerspanrmr:rrAUTOgroupr\r]redecoderJ _ph_to_fmt)r-rjr{rRcurmrrvrZphtyperqpre1m1rrr!r!r"rl[sn  , "   1rl)stb)r-rrjr7r3rh)r-rrjr7r3ry)r|T)r-rrjr7r{r}r3rS)5rd __future__rretypingrrrrrrr functoolsr r r r\sqlrabcrrr_enumsr_compatrr _encodingsrrr;r<rr#rr7rir r>r=rrrzrurtcompilerrlrTEXTBINARYrr!r!r!r"sP          75  % K