*JfvpUdZddlZddlmZmZmZmZmZmZm Z ddlm Z m Z m Z m Z mZddlmZddlmZddlmZdd lmZdd lmZdd lmZe rdd lmZmZdd lmZddl m!Z!ej"j#Z#ej"j$Z$ej"j%Z%eddZ&eddZ'Gddee'Z(Gddee'Z)Gddee'Z*Gddee'Z+e edfZ,ee-d< ee.efZ/ee-d< d6d"Z0d7d$Z1 d8d&Z2ej3d'd(e.d)e4d e efd*Z5d+e e&d e+e&fd,Z6d-ede&fd e+e&fd.Z7d-ede&fd e+e&fd/Z8d0e ed e fd1Z9ddd eee.fd2Z:d3d4d ee;fd5Ztt|Sr"dictzip)rDnamess r% dict_row_zdict_row..dict_row_wsCv&&'''r() _get_names no_resultr rrstr)r=rJrIs @r%dict_rowrNns] v  E }((3-(DcN(((((( 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()pgresultrL _get_nfields_make_nt _encodingrange_make)r=nfieldsntrUs @r%namedtuple_rowr_}sl /C 3G &" L%K%K%K%KE'NN%K%K%K L L LB 8Or(iencrIcZtfd|D}td|S)Nc3\K|]&}t|V'dSr")rdecode)rSnr`s r%rVz_make_nt..s6GGA(#77GGGGGGr(r)r@r)r`rIsnamess` r%rYrYs5 GGGGGGG G GF eV $ $$r(clscdfd }|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=r9r RowMaker[T]czt|tSdttdtffd }|S)NrDr c Jditt|SNr#rF)rDrfrIs r% class_row__z2class_row..class_row_..class_row__s+322c%001122 2r(rKrLr rr)r=rlrIrfs @r% class_row_zclass_row..class_row_sY6"" =  3  3! 3 3 3 3 3 3 3r()r=r9r rhr#)rfrns` r% class_rowros) r(funccdfd }|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 rhcFdttdtffd }|S)NrDr c|Sr"r#)rDrps r% args_row__z/args_row..args_row_..args_row__s4= r()r rr)rrrvrps r% args_row_zargs_row..args_row_s; !x} ! ! ! ! ! ! !r()rrrsr rhr#)rprws` r%args_rowrxs) 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=rsr rhczt|tSdttdtffd }|S)NrDr c Jditt|SrkrF)rDrprIs r% kwargs_row__z5kwargs_row..kwargs_row_..kwargs_row__s+433$s5&112233 3r(rm)r=r|rIrps @r% kwargs_row_zkwargs_row..kwargs_row_sY6"" =  4# 41 4 4 4 4 4 4 4r()r=rsr rhr#)rpr}s` r% kwargs_rowr~s) r(rDc*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)eInterfaceError)rDs r%rLrLs = > >>r(c|jsdSt}|dS|jfdt|DS)Nc`g|]*}|+Sr#)rRrc)rSrTr`rUs r% z_get_names..s>   %& ! C     r()rWrXrZr[)r=r]r`rUs @@r%rKrKsm /C t3Gt  C     */..   r(rUrc||j}|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)r]status TUPLES_OK SINGLE_TUPLE COMMAND_OK)rUr]s r%rXrXsC kG i : % % J* $ $ $tr()r=r9r r>)r=r9r rB)r=r9r rO)=r, functoolstypingrrrrrrr r r r r r collectionsrtyping_extensionsrrrr_compatr _encodingsrr=rr cursor_asyncrpsycopg.pq.abcr ExecStatusrrrrrrr.r3r7r;__annotations__rMr<rArNr_ lru_cachebytesrYrorxr~rLrKintrXr#r(r%rs) LLLLLLLLLLLLLLLLLL@@@@@@@@@@@@@@""""""''''''------(********))))))'''''' ] % M # })  GC4   get$$$ < < < <