o "4g@sUdZddlmZddlZddlmZmZmZmZddlm Z m Z m Z ddlm Z m Z ddlmZdd lmZdd lmZdd lmZmZdd lmZe redd lmZddlmZddlmZddlm Z ej!j"Z"ej!j#Z#ej!j$Z$ej!j%Z%edddZ&eddddZ'Gddde e'Z(Gddde e'Z)Gddde e'Z*Gddde e'Z+e edfZ,d e-d< e e.efZ/d e-d!< dQd&d'Z0dRd)d*Z1dSd,d-Z2e3d.dTd4d5Z4dUd9d:Z5dVd=d>Z6dVd?d@Z7dWdBdCZ8dXdGdHZ9dYdJdKZ:dZdOdPZ;dS)[z psycopg row factories ) annotationsN)AnyCallable NamedTupleNoReturn) TYPE_CHECKINGProtocolSequence)DictTuple) namedtuple)pq)errors) TypeAliasTypeVar)_as_python_identifier)Cursor) BaseCursor) AsyncCursor)PGresultTT) covariantRowTupleRow)rdefaultc@eZdZdZd ddZdS) 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`. _RowMaker__values Sequence[Any]returnrcCdSN)selfrr#r#U/var/www/html/authentication-server/venv/lib/python3.10/site-packages/psycopg/rows.py__call__2zRowMaker.__call__N)rrr r__name__ __module__ __qualname____doc__r&r#r#r#r%r&s rc@r) RowFactorya Callable protocol taking a `~psycopg.Cursor` and returning a `RowMaker`. A `!RowFactory` is typically called when a `!Cursor` receives a result. This way it can inspect the cursor state (for instance the `~psycopg.Cursor.description` attribute) and help a `!RowMaker` to create a complete object. For instance the `dict_row()` `!RowFactory` uses the names of the column to define the dictionary key and returns a `!RowMaker` function which would use the values to create a dictionary for each record. _RowFactory__cursor Cursor[Any]r RowMaker[Row]cCr!r"r#)r$r.r#r#r%r&Cr'zRowFactory.__call__N)r.r/r r0r(r#r#r#r%r-5s r-c@r) AsyncRowFactoryz@ Like `RowFactory`, taking an async cursor as argument. _AsyncRowFactory__cursorAsyncCursor[Any]r r0cCr!r"r#)r$r2r#r#r%r&Kr'zAsyncRowFactory.__call__N)r2r3r r0r(r#r#r#r%r1Fr1c@r) BaseRowFactoryzF Like `RowFactory`, taking either type of cursor as argument. _BaseRowFactory__cursorBaseCursor[Any, Any]r r0cCr!r"r#)r$r6r#r#r%r&Sr'zBaseRowFactory.__call__N)r6r7r r0r(r#r#r#r%r5Nr4r5.rDictRowcursorr7r RowMaker[TupleRow]cCstS)zRow factory to represent rows as simple tuples. This is the default factory, used when `~psycopg.Connection.connect()` or `~psycopg.Connection.cursor()` are called without a `!row_factory` parameter. )tuple)r9r#r#r% tuple_rowes r<RowMaker[DictRow]cs&t|dur tSdfdd }|S) zRow factory to represent rows as dictionaries. The dictionary keys are taken from the column names of the returned columns. Nvaluesrr dict[str, Any]cstt|Sr"dictzipr>namesr#r% dict_row_{szdict_row..dict_row_)r>rr r? _get_names no_result)r9rFr#rDr%dict_rowrs rJRowMaker[NamedTuple]csL|jstSt}|durtSt|jgfddt|DR}|jS)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. Nc3s|]}|VqdSr")fname.0iresr#r% sz!namedtuple_row..)pgresultrI _get_nfields_make_nt _encodingrange_make)r9nfieldsntr#rPr%namedtuple_rows$r[iencstrrEbytestype[NamedTuple]cs tfdd|D}td|S)Nc3s|] }t|VqdSr")rdecode)rNnr\r#r%rRsz_make_nt..r)r;r )r\rEsnamesr#rbr%rUs rUclstype[T]BaseRowFactory[T]cdfdd }|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]] r9r7r RowMaker[T]c(t|dur tSdfdd }|S)Nr>rr rcditt|SNr#r@rC)rdrEr#r% class_row__z2class_row..class_row_..class_row__r>rr rrG)r9rlrdrDr% class_row_ zclass_row..class_row_N)r9r7r rhr#)rdrpr#ror% class_rows  rrfuncCallable[..., T]crg) 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. curBaseCursor[Any, T]r rhcsdfdd }|S)Nr>rr rcs|Sr"r#rCrsr#r% args_row__z/args_row..args_row_..args_row__rnr#)rurxrwr#r% args_row_szargs_row..args_row_N)rurvr rhr#)rsrzr#rwr%args_rowsr{crg) 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. r9rvr rhcri)Nr>rr rcrjrkr@rC)rsrEr#r% kwargs_row__rmz5kwargs_row..kwargs_row_..kwargs_row__rnrG)r9r|rwrDr% kwargs_row_rqzkwargs_row..kwargs_row_N)r9rvr rhr#)rsr}r#rwr% kwargs_rows r~ RowMaker[Any]cCsB|j}|stSt|}|durtS|dkrtdd dd }|S) zR Generate a row factory returning the first column as a scalar value. Nr zat least one column expectedr>rr rcSs|dS)Nrr#rCr#r#r% scalar_row_ryzscalar_row..scalar_row_)r>rr r)rSrIrTeProgrammingError)r9rQrYrr#r#r% scalar_rows  rr>rrcCs td)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)rInterfaceErrorrCr#r#r%rIs rIlist[str] | Nonecs@|jsdSt}|durdS|jfddt|DS)Ncsg|] }|qSr#)rLr`rMr\rQr#r% sz_get_names..)rSrTrVrW)r9rYr#rr%rHs rHrQr int | NonecCs:|j}|jtks|jtks|jtks|jtkr|r|SdS)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)rYstatus TUPLES_OK SINGLE_TUPLE TUPLES_CHUNK COMMAND_OK)rQrYr#r#r%rTs   rT)r9r7r r:)r9r7r r=)r9r7r rK)r\r]rEr^r r_)rdrer rf)rsrtr rf)r9r7r r)r>rr r)r9r7r r)rQrr r)sT