o TPf@sUdZddlmZddlZddlmZmZmZmZddlm Z m Z m Z ddl m Z ddlmZdd lmZdd lmZmZdd lmZe r]dd lmZdd lmZddlmZddlmZejj Z ejj!Z!ejj"Z"ejj#Z#edddZ$eddddZ%Gddde e%Z&Gddde e%Z'Gddde e%Z(Gddde e%Z)dZ*de+d< d Z,de+d!< dQd&d'Z-dRd)d*Z.dSd,d-Z/e0d.dTd4d5Z1dUd9d:Z2dVd=d>Z3dVd?d@Z4dWdBdCZ5dXdGdHZ6dYdJdKZ7dZdOdPZ8dS)[z psycopg row factories ) annotationsN)AnyCallable NamedTupleNoReturn) TYPE_CHECKINGProtocolSequence) 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!X/var/www/html/Testing_prj/Navya-Bakers/venv/lib/python3.10/site-packages/psycopg/rows.py__call__1zRowMaker.__call__N)rrrr__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]cCrr r!)r"r,r!r!r#r$Br%zRowFactory.__call__N)r,r-rr.r&r!r!r!r#r+4s r+c@r) AsyncRowFactoryz@ Like `RowFactory`, taking an async cursor as argument. _AsyncRowFactory__cursorAsyncCursor[Any]rr.cCrr r!)r"r0r!r!r#r$Jr%zAsyncRowFactory.__call__N)r0r1rr.r&r!r!r!r#r/Er/c@r) BaseRowFactoryzF Like `RowFactory`, taking either type of cursor as argument. _BaseRowFactory__cursorBaseCursor[Any, Any]rr.cCrr r!)r"r4r!r!r#r$Rr%zBaseRowFactory.__call__N)r4r5rr.r&r!r!r!r#r3Mr2r3ztuple[Any, ...]rdict[str, Any]DictRowcursorr5rRowMaker[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)r8r!r!r# tuple_rowds 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. Nvaluesrrr6cstt|Sr dictzipr=namesr!r# dict_row_zszdict_row..dict_row_)r=rrr6 _get_names no_result)r8rDr!rBr#dict_rowqs rHRowMaker[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..)pgresultrG _get_nfields_make_nt _encodingrange_make)r8nfieldsntr!rNr#namedtuple_rows$rYiencstrrCbytestype[NamedTuple]cs tfdd|D}td|S)Nc3s|] }t|VqdSr )rdecode)rLnrZr!r#rPsz_make_nt..r)r:r )rZrCsnamesr!r`r#rSs rSclstype[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]] r8r5r RowMaker[T]c(t|dur tSdfdd }|S)Nr=rrrcditt|SNr!r>rA)rbrCr!r# class_row__z2class_row..class_row_..class_row__r=rrrrE)r8rjrbrBr# class_row_ zclass_row..class_row_N)r8r5rrfr!)rbrnr!rmr# class_rows  rpfuncCallable[..., T]cre) 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]rrfcsdfdd }|S)Nr=rrrcs|Sr r!rArqr!r# args_row__z/args_row..args_row_..args_row__rlr!)rsrvrur!r# args_row_szargs_row..args_row_N)rsrtrrfr!)rqrxr!rur#args_rowsrycre) 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. r8rtrrfcrg)Nr=rrrcrhrir>rA)rqrCr!r# kwargs_row__rkz5kwargs_row..kwargs_row_..kwargs_row__rlrE)r8rzrurBr# kwargs_row_rozkwargs_row..kwargs_row_N)r8rtrrfr!)rqr{r!rur# 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=rrrcSs|dS)Nrr!rAr!r!r# scalar_row_rwzscalar_row..scalar_row_)r=rrr)rQrGrReProgrammingError)r8rOrWr~r!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)rInterfaceErrorrAr!r!r#rGs rGlist[str] | Nonecs@|jsdSt}|durdS|jfddt|DS)Ncsg|] }|qSr!)rJr^rKrZrOr!r# sz_get_names..)rQrRrTrU)r8rWr!rr#rFs rFrOr 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)rWstatus TUPLES_OK SINGLE_TUPLE TUPLES_CHUNK COMMAND_OK)rOrWr!r!r#rRs   rR)r8r5rr9)r8r5rr<)r8r5rrI)rZr[rCr\rr])rbrcrrd)rqrrrrd)r8r5rr})r=rrr)r8r5rr)rOrrr)9r* __future__r functoolstypingrrrrrrr collectionsr r r r_compatrr _encodingsrr8r _cursor_baser cursor_asyncrpsycopg.pq.abcr ExecStatusrrrrrrrr+r/r3r__annotations__r7r;rHrY lru_cacherSrpryr|rrGrFrRr!r!r!r#sR