qjUdZddlmZddlZddlmZmZmZmZddlm Z m Z m Z ddlm Z m Z ddlmZdd lmZdd lmZdd lmZmZdd lmZe rdd lmZddlmZddlmZddlm Z ejBjDZ"ejBjFZ#ejBjHZ$ejBjJZ%eddZ&edddZ'Gdde e'Z(Gdde e'Z)Gdde e'Z*Gdde e'Z+e edfZ,d e-d< e e.efZ/d e-d!< d.d"Z0d/d#Z1d0d$Z2ejfd%d1d&Z4d2d'Z5d3d(Z6d3d)Z7d4d*Z8d5d+Z9d6d,Z:d7d-Z;y)8z psycopg row factories ) annotationsN)AnyCallable NamedTupleNoReturn) TYPE_CHECKINGProtocolSequence)DictTuple) namedtuple)pq)errors) TypeAliasTypeVar)_as_python_identifier)Cursor) BaseCursor) AsyncCursor)PGresultTT) 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 M/var/www/html/kelly-backend/venv/lib/python3.12/site-packages/psycopg/rows.py__call__zRowMaker.__call__2sN)r# Sequence[Any]returnr__name__ __module__ __qualname____doc__r%r!r&r$rr&s  .dict_row_{sCv&''r&)rFr'r(zdict[str, Any] _get_names no_result)r?rHrGs @r$dict_rowrLrs' v E }( r&c|jstSt}|tSt|jgfdt |D}|j 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%KNqciilNs)pgresultrK _get_nfields_make_nt _encodingrange_make)r?nfieldsntrRs @r$namedtuple_rowr\sT //C 3G &"" L%KE'N%K LB 88Or&icDtfd|D}td|S)Nc3RK|]}t|j ywr )rdecode)rPnencs r$rSz_make_nt..s GA(#7s$'r)r>r )rarGsnamess` r$rVrVs 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|tSdfd }|S)Nc :ditt|SNr!rC)rFclsrGs r$ class_row__z2class_row..class_row_..class_row__s2c%012 2r&rFr'r(rrI)r?rhrGrgs @r$ class_row_zclass_row..class_row_s%6" =  3r&)r?r;r( RowMaker[T]r!)rgrjs` r$ class_rowrls 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!)rFfuncs r$ args_row__z/args_row..args_row_..args_row__s = r&rir!)currqrps r$ args_row_zargs_row..args_row_s !r&)rrBaseCursor[Any, T]r(rkr!)rprss` r$args_rowrus 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|tSdfd }|S)Nc :ditt|SrfrC)rFrprGs r$ kwargs_row__z5kwargs_row..kwargs_row_..kwargs_row__s3$s5&123 3r&rirI)r?ryrGrps @r$ kwargs_row_zkwargs_row..kwargs_row_s%6" =  4r&)r?rtr(rkr!)rprzs` r$ kwargs_rowr{s r&c|j}|stSt|}|tS|dkrtjddd}|S)zR Generate a row factory returning the first column as a scalar value. rzat least one column expectedc |dS)Nrr!rFs r$ scalar_row_zscalar_row..scalar_row_s ayr&)rFr'r(r)rTrKrUeProgrammingError)r?rRrZrs r$ scalar_rowrsQ //C 3G{  !?@@ 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)rInterfaceErrorr~s r$rKrKs  = >>r&c|j}|syt|}|y|j}t|Dcgc]"}|j |j |$c}Scc}wr )rTrUrWrXrOr_)r?rRrZrarQs r$rJrJsf //C 3G   C*/. *8Q ! C .  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)rZstatus TUPLES_OK SINGLE_TUPLE TUPLES_CHUNK COMMAND_OK)rRrZs r$rUrUsI kkG i :: % :: % JJ* $r&)r?r;r(zRowMaker[TupleRow])r?r;r(zRowMaker[DictRow])r?r;r(zRowMaker[NamedTuple])rastrrGbytesr(ztype[NamedTuple])rgztype[T]r(BaseRowFactory[T])rpzCallable[..., T]r(r)r?r;r(z RowMaker[Any])rFr'r(r)r?r;r(zlist[str] | None)rRrr(z int | None)rsb #6644"'-()' ]] % % MM # # }})) }}))  C4 etZ8