iUdZddlmZddlZddlmZmZmZmZddlm Z m Z m Z m Z m Z ddlmZddlmZdd lmZdd lmZmZdd lmZerdd lmZdd lmZddlmZddlm Z ejBjDZ"ejBjFZ#ejBjHZ$ejBjJZ%eddZ&edddZ'Gdde e'Z(Gdde e'Z)Gdde e'Z*Gdde e'Z+e edfZ,de-d< ee.efZ/de-d < d-d!Z0d.d"Z1d/d#Z2ejfd$d0d%Z4d1d&Z5d2d'Z6d2d(Z7d3d)Z8d4d*Z9d5d+Z:d6d,Z;y)7z psycopg row factories ) annotationsN) TYPE_CHECKINGAnyCallableDict) NamedTupleNoReturnProtocolSequenceTuple) namedtuple)errors)pq) TypeAliasTypeVar)_as_python_identifier)PGresult)Cursor) BaseCursor) AsyncCursorTT) covariantRowTupleRow)rdefaultceZdZdZddZy)RowMakera Callable protocol taking a sequence of value and returning an object. The sequence of value is what is returned from a database query, already adapted to the right Python types. The return value is the object that your program would like to receive: by default (`tuple_row()`) it is a simple tuple, but it may be any type of object. Typically, `!RowMaker` functions are returned by `RowFactory`. cyN)self_RowMaker__valuess l/var/www/html/hwPaymentPortal-be-dev/hw-payment-portal-api/venv/lib/python3.12/site-packages/psycopg/rows.py__call__zRowMaker.__call__2N)r# Sequence[Any]returnr__name__ __module__ __qualname____doc__r%r!r'r$rr&s  .dict_row_ysE6*+ +r')rGr(r)zdict[str, Any] _get_names no_result)r@rIrHs @r$dict_rowrMrs* F##0 ,r'c|jxrAtx}4t|jgfdt |D}|j St S)zRow factory to represent rows as `~collections.namedtuple`. The field names are taken from the column names of the returned columns, with some mangling to deal with invalid names. c3@K|]}j|ywr )fname).0iress r$ z!namedtuple_row..s)O1#))A,)Os)pgresult _get_nfields_make_nt _encodingrange_makerL)r@nfieldsntrSs @r$namedtuple_rowr]sQ  S0A%AW$N f&& P)Og)O Pxxr'icDtfd|D}td|S)Nc3RK|]}t|j ywr )rdecode)rQnencs r$rTz_make_nt..sGA(#7Gs$'r)r?r )rbrHsnamess` r$rWrWs GG GF eV $$r'cdfd }|S)aQGenerate a row factory to represent rows as instances of the class `!cls`. The class must support every output column name as a keyword parameter. :param cls: The class to return for each row. It must support the fields returned by the query as keyword arguments. :rtype: `!Callable[[Cursor],` `RowMaker`\[~T]] c>t|x dfd }|StS)Nc :ditt|SNr!rD)rGclsrHs r$ class_row__z2class_row..class_row_..class_row__s6T#eV"4566r'rGr(r)rrJ)r@rirHrhs @r$ class_row_zclass_row..class_row_s(' 'E 4 7  r')r@r<r) RowMaker[T]r!)rhrks` r$ class_rowrms  r'cdfd }|S)zGenerate a row factory calling `!func` with positional parameters for every row. :param func: The function to call for each row. It must support the fields returned by the query as positional arguments. cdfd }|S)Nc|Sr r!)rGfuncs r$ args_row__z/args_row..args_row_..args_row__s = r'rjr!)currrrqs r$ args_row_zargs_row..args_row_s !r')rsBaseCursor[Any, T]r)rlr!)rqrts` r$args_rowrvs r'cdfd }|S)zGenerate a row factory calling `!func` with keyword parameters for every row. :param func: The function to call for each row. It must support the fields returned by the query as keyword arguments. c>t|x dfd }|StS)Nc :ditt|SrgrD)rGrqrHs r$ kwargs_row__z5kwargs_row..kwargs_row_..kwargs_row__s7d3uf#5677r'rjrJ)r@rzrHrqs @r$ kwargs_row_zkwargs_row..kwargs_row_s(' 'E 4 8  r')r@rur)rlr!)rqr{s` r$ kwargs_rowr|s  r'c|jx}r-t|x} |dkrtjddd}|StS)zR Generate a row factory returning the first column as a scalar value. rzat least one column expectedc |dS)Nrr!rGs r$ scalar_row_zscalar_row..scalar_row_s !9 r')rGr(r)r)rUrVeProgrammingErrorrL)r@rSr[rs r$ scalar_rowrsP  S0A%AW$N Q;$$%CD D r'c,tjd)zA `RowMaker` that always fail. It can be used as return value for a `RowFactory` called with no result. Note that the `!RowFactory` *will* be called with no result, but the resulting `!RowMaker` never should. z the cursor doesn't have a result)rInterfaceErrorrs r$rLrLs  = >>r'c|jx}rPt|x}C|j}t|Dcgc]"}|j |j |$c}Sycc}wr )rUrVrXrYrPr`)r@rSr[rbrRs r$rKrKsd S0A%AW$N.3Gn )*CIIaL   $    s'A c|j}|jtk(s;|jtk(s(|jtk(s|jt k(r|r|Sy)z Return the number of columns in a result, if it returns tuples else None Take into account the special case of results with zero columns. N)r[status TUPLES_OK SINGLE_TUPLE TUPLES_CHUNK COMMAND_OK)rSr[s r$rVrVsI kkG i :: % :: % JJ* $r')r@r<r)zRowMaker[TupleRow])r@r<r)zRowMaker[DictRow])r@r<r)zRowMaker[NamedTuple])rbstrrHbytesr)ztype[NamedTuple])rhztype[T]r)BaseRowFactory[T])rqzCallable[..., T]r)r)r@r<r)z RowMaker[Any])rGr(r)r )r@r<r)zlist[str] | None)rSrr)z int | None)rs_ #55BB"'-'() ]] % % MM # # }})) }}))  C4 etZ8