o DDiv=@sUdZddlmZddlZddlmZmZmZmZm Z ddl m Z ddl m Z mZmZddlmZdd lmZmZdd lmZmZmZmZdd lmZdd lmZdd lmZer_ddlm Z dZ!dZ"GdddeZ#GdddZ$e e%e&gdfZ'de(d<d0ddZ)e e)Z*Gdd d e$Z+e e%e&gd!fZ,de(d"<d1d#d$Z-e e-Z.e/d%Z0 'd2d3d+d,Z1ej2ej3ej4d-Z5Gd.d/d/e$Z6dS)4z& Utility module to manipulate queries ) annotationsN) TYPE_CHECKINGAny NamedTuple TypeAlias TypeGuard) lru_cache)CallableMappingSequence)errors)pqsql)BufferParamsQueryQueryNoTemplate)PyFormat)Template)TemplateProcessor) Transformeri2c@s&eZdZUded<ded<ded<dS) QueryPartbytesprez int | stritemrformatN)__name__ __module__ __qualname____annotations__r"r"I/var/www/Datamplify/venv/lib/python3.10/site-packages/psycopg/_queries.pyrs  rc@sfeZdZdZdZd$ddZd%d dZd&ddZe d'ddZ e d(ddZ d)ddZ d*d!d"Z d#S)+ PostgresQueryzO Helper to convert a Python query and parameters into Postgres format. z^ query params types formats _tx _want_formats _parts _encoding _order transformerrcCs2||_d|_d|_d|_d|_|d|_d|_dS)Nr")_txparamstypes _want_formatsformatsquery_order)selfr%r"r"r#__init__,s zPostgresQuery.__init__r,rvars Params | NonereturnNonecCst|tr |||S||}|dur5t|tkr#t|tkr#t}nt}|||j 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) isinstancer_convert_template _ensure_byteslenMAX_CACHED_STATEMENT_LENGTHMAX_CACHED_STATEMENT_PARAMS _query2pg_query2pg_nocacher'encodingr,r*r-_partsdumpr.r,r0fr"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.r0r(r"r"r#r?^s 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 ) typelisttupledictr5r rstrr TypeErrorr )r0tsequencer"r"r#is_params_sequenceos$   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").0rr0r"r# sz=PostgresQuery.validate_and_reorder_params..r"zquery parameter missing: z, c3s|] }|vr|VqdSNr")rVirWr"r# sz.) r$rPr8eProgrammingErrorr5rintrMrLKeyErrorjoinsorted)rQr0rSr"rWr#rDs2 " z)PostgresQuery.validate_and_reorder_paramsrrcCs4t|tr ||jjSt|tjr||jS|SrY)r5rLencoder'r=r Composableas_bytes)r.r,r"r"r#r7s   zPostgresQuery._ensure_bytesrcCs||durtdt|d|jd}||j|_|jr3|j|j|j|_|jjp*d|_|jj|_dSd|_d|_d|_dS)NA'execute()' with string template query doesn't support parametersT server_paramstxr") rMrr'processr,r(rEr+r)r.r,r0tpr"r"r#r6s zPostgresQuery._convert_templateN)r%rr,rr0r1r2r3r0r1r2r3)r0rr2rG)rQrRr0rrSrTr2rU)r,rr2rr,rr0r1r2r3)rrr __doc__split __slots__r/rBr? staticmethodrPrDr7r6r"r"r"r#r$"s   #   "r$?tuple[bytes, list[PyFormat], list[str] | None, list[QueryPart]]r _Query2Pgr,rr=rLr2c 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_queryr5rr^appendrrrLr8r\r]r`) r,r=rQrSchunksr+partseenphr"r"r#r<s<       r<c@s2eZdZdZdZdd d Zdd d ZdddZdS)PostgresClientQueryzI PostgresQuery subclass merging query and arguments client-side. )templater,rr0r1r2r3cCst|tr |||S||}|dur3t|tkr#t|tkr#t}nt}|||j j \|_ |_ |_ n||_d|_ ||dSr4)r5rr6r7r8r9r:_query2pg_client_query2pg_client_nocacher'r=r}r-r>r,r?r@r"r"r#rBs     zPostgresClientQuery.convertcsN|dur"j|j}tfdd|D_jj_dSd_dS)rCNc3s(|]}|durj|ndVqdS)NsNULL)r' as_literal)rVpr.r"r#r[+s z+PostgresClientQuery.dump..)rDr>r-rJr(r}r,rFr"rr#r?#s   zPostgresClientQuery.dumprcCs<|durtdt|d|jd}||j|_|j|_dS)NreFrf)rMrr'rir,r(rjr"r"r#r62s z%PostgresClientQuery._convert_templateNrlrmrn)rrr rorqrBr?r6r"r"r"r#r|s   r|/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_percentNrrus%sr&) rvr5rr^rwrrLrr`)r,r=rQrSrxryrzr{r"r"r#rCs0      rs"(?x) % # a literal % (?: (?: \( ([^)]+) \) # or a name in (braces) . # followed by a format ) | (?:.) # or any char, really ) asciiTrboolrRcCsg}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 '%%'russbtz8only '%s', '%b', '%t' are allowed as placeholders, got 'z1positional and named placeholders cannot be mixed)_re_placeholderfinditerspanrwr8rrAUTOgroupr\r]rpdecoderH _ph_to_fmt)r,r=rrQcurmrrvrZphtyper{pre1m1rrr"r"r#rvzsl  +"   0rv)stbc@s*eZdZdddZdd d Zdd d ZdS)PostgresRawQueryr,rr0r1r2r3cCs<t|tr |||S|||_d|_|_||dSrY)r5rr6r7r,r*r-r?r.r,r0r"r"r#rBs    zPostgresRawQuery.convertcCsp|dur-t|s tdtjgt||_|j||j|_ |jj p$d|_ |jj |_ dSd|_ d|_ d|_ dS)Nz,raw queries require a sequence of parametersr") r$rPrMrrr8r*r'rEr(r)r+)r.r0r"r"r#r?s  zPostgresRawQuery.dumprcCstt|jd)Nz! doesn't support template strings)r\NotSupportedErrorrHrrr"r"r#r6sz"PostgresRawQuery._convert_templateNrlrmrn)rrr rBr?r6r"r"r"r#rs  r)r,rr=rLr2rs)r,rr=rLr2r)rT)r,rr=rLrrr2rR)7ro __future__rretypingrrrrr functoolsrcollections.abcr r r r r\rrabcrrrr_enumsr_compatr _tstringsrrr9r:rr$rrLrtr!r<r;r|rrr~compilerrvrTEXTBINARYrrr"r"r"r#sP         7>  % J