R hUdZddlmZddlZddlmZmZmZmZddlm Z m Z m Z m Z m Z ddlmZddlmZdd lmZdd lmZmZdd lmZerdd lmZdd lmZddlmZddlm Z ej!j"Z"ej!j#Z#ej!j$Z$ej!j%Z%eddZ&edddZ'Gdde e'Z(Gdde e'Z)Gdde e'Z*Gdde e'Z+e edfZ,de-d< ee.efZ/de-d < dEd%Z0dFd'Z1dGd)Z2ej3d*dHd0Z4dId4Z5dJd7Z6dJd8Z7dKd:Z8dLd>Z9dMd@Z:dNdDZ;dS)Oz psycopg row factories ) annotationsN) TYPE_CHECKINGAnyCallableDict) NamedTupleNoReturnProtocolSequenceTuple) namedtuple)errors)pq) TypeAliasTypeVar)_as_python_identifier)PGresult)Cursor) BaseCursor) AsyncCursorTT) covariantRowTupleRow)rdefaultceZdZdZddZdS) 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]returnrcdSN)selfrs W/var/www/html/01_SiteInternet/03_Maps/venv/lib/python3.11/site-packages/psycopg/rows.py__call__zRowMaker.__call__2N)rr r!r__name__ __module__ __qualname____doc__r'r$r)r&rr&s(  <;;;;;r)rceZdZdZddZdS) 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]cdSr#r$)r%r1s r&r'zRowFactory.__call__Cr(r)N)r1r2r!r3r*r$r)r&r0r05s)  DCCCCCr)r0ceZdZdZddZdS) AsyncRowFactoryz@ Like `RowFactory`, taking an async cursor as argument. _AsyncRowFactory__cursorAsyncCursor[Any]r!r3cdSr#r$)r%r7s r&r'zAsyncRowFactory.__call__Kr(r)N)r7r8r!r3r*r$r)r&r6r6Fs)IHHHHHr)r6ceZdZdZddZdS) BaseRowFactoryzF Like `RowFactory`, taking either type of cursor as argument. _BaseRowFactory__cursorBaseCursor[Any, Any]r!r3cdSr#r$)r%r<s r&r'zBaseRowFactory.__call__Sr(r)N)r<r=r!r3r*r$r)r&r;r;Ns)MLLLLLr)r;.rDictRowcursorr=r!RowMaker[TupleRow]ctS)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)r@s r& tuple_rowrDes  Lr)RowMaker[DictRow]cDt|tSdfd }|S)zRow factory to represent rows as dictionaries. The dictionary keys are taken from the column names of the returned columns. Nvaluesr r!dict[str, Any]c>tt|Sr#dictzip)rGnamess r& dict_row_zdict_row..dict_row_{sCv&&'''r))rGr r!rH _get_names no_result)r@rNrMs @r&dict_rowrRrsC v  E }(((((( r)RowMaker[NamedTuple]c|jstSt}|tSt|jgfdt |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. Nc3BK|]}|VdSr#)fname).0iress r& z!namedtuple_row..s-%K%Kqciill%K%K%K%K%K%Kr))pgresultrQ _get_nfields_make_nt _encodingrange_make)r@nfieldsntrYs @r&namedtuple_rowrcsl /C 3G &" L%K%K%K%KE'NN%K%K%K L L LB 8Or)iencstrrMbytestype[NamedTuple]cZtfd|D}td|S)Nc3\K|]&}t|V'dSr#)rdecode)rWnrds r&rZz_make_nt..s6GGA(#77GGGGGGr)r)rCr )rdrMsnamess` r&r]r]s5 GGGGGGG G GF eV $ $$r)clstype[T]BaseRowFactory[T]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]] r@r=r! RowMaker[T]cHt|tSdfd }|S)NrGr r!rc Jditt|SNr$rJ)rGrmrMs r& class_row__z2class_row..class_row_..class_row__s+322c%001122 2r)rGr r!rrO)r@rurMrms @r& class_row_zclass_row..class_row_sG6"" =  3 3 3 3 3 3 3r))r@r=r!rqr$)rmrws` r& class_rowrxs) r)funcCallable[..., T]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. curBaseCursor[Any, T]r!rqcdfd }|S)NrGr r!rc|Sr#r$)rGrys r& args_row__z/args_row..args_row_..args_row__s4= r)rvr$)r|rrys r& args_row_zargs_row..args_row_s) ! ! ! ! ! !r))r|r}r!rqr$)ryrs` r&args_rowrs) 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. r@r}r!rqcHt|tSdfd }|S)NrGr r!rc Jditt|SrtrJ)rGryrMs r& kwargs_row__z5kwargs_row..kwargs_row_..kwargs_row__s+433$s5&112233 3r)rvrO)r@rrMrys @r& kwargs_row_zkwargs_row..kwargs_row_sG6"" =  4 4 4 4 4 4 4r))r@r}r!rqr$)ryrs` r& kwargs_rowrs) r) RowMaker[Any]c|j}|stSt|}|tS|dkrtjdd d}|S) zR Generate a row factory returning the first column as a scalar value. Nrzat least one column expectedrGr r!rc|dS)Nrr$rGs r& scalar_row_zscalar_row..scalar_row_s ayr))rGr r!r)r[rQr\eProgrammingError)r@rYrars r& scalar_rowrsh /C 3G{{ !?@@@ r)rGr 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&rQrQs = > >>r)list[str] | Nonec|jsdSt}|dS|jfdt|DS)Nc`g|]*}|+Sr$)rVrj)rWrXrdrYs r& z_get_names..s>   %& ! C     r))r[r\r^r_)r@rardrYs @@r&rPrPsm /C t3Gt  C     */..   r)rYr int | Nonec|j}|jtks2|jtks"|jtks|jt kr|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)rastatus TUPLES_OK SINGLE_TUPLE TUPLES_CHUNK COMMAND_OK)rYras r&r\r\sP kG i : % % : % % J* $ $ $tr))r@r=r!rA)r@r=r!rE)r@r=r!rS)rdrerMrfr!rg)rmrnr!ro)ryrzr!ro)r@r=r!r)rGr r!r )r@r=r!r)rYrr!r)rs #"""""555555555555BBBBBBBBBBBBBB""""""''''''''------*''''''(((((()))))) ] % M # }) })  GC4   getZ888 < < < <