o 6 fv@sXUdZddlZddlmZmZmZmZmZmZm Z ddlm Z m Z m Z m Z mZddlmZddlmZddlmZdd lmZdd lmZdd lmZe radd lmZmZdd lmZddl m!Z!ej"j#Z#ej"j$Z$ej"j%Z%edddZ&edddZ'Gdddee'Z(Gdddee'Z)Gdddee'Z*Gdddee'Z+e edfZ,ee-d< ee.efZ/ee-d< d@d"d#Z0dAd%d&Z1   'dBd(d)Z2e3d*d+e.d,e4d e efd-d.Z5d/e e&d e+e&fd0d1Z6d2ede&fd e+e&fd3d4Z7d2ede&fd e+e&fd5d6Z8d7e ed e fd8d9Z9ddd eee.fd:d;Z:dd?Z.dict_row_) _get_names no_resultr rrstr)r6rBr!r@r#dict_rowns "rFRowMaker[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..)pgresultrD _get_nfields_make_nt _encodingrange_make)r6nfieldsntr!rLr#namedtuple_row}s$rWiencrAcs tfdd|D}td|S)Nc3s|] }t|VqdSr )rdecode)rJnrXr!r#rNsz_make_nt..r)r8r)rXrAsnamesr!r[r#rQs rQclscdfdd }|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]] r6r3r RowMaker[T]c4t|dur tSdttdtffdd }|S)Nr;rcditt|SNr!r<r?)r]rAr!r# class_row__z2class_row..class_row_..class_row__rCrDr rr)r6rcr]r@r# class_row_ zclass_row..class_row_N)r6r3rr_r!)r]rgr!rfr# class_rows  rifunccr^) 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]rr_csdttdtffdd }|S)Nr;rcs|Sr r!r?rjr!r# args_row__sz/args_row..args_row_..args_row__)r rr)rkrnrmr!r# args_row_szargs_row..args_row_N)rkrlrr_r!)rjror!rmr#args_rowsrpcr^) 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. r6rlrr_cr`)Nr;rcrarbr<r?)rjrAr!r# kwargs_row__rdz5kwargs_row..kwargs_row_..kwargs_row__re)r6rqrmr@r# kwargs_row_rhzkwargs_row..kwargs_row_N)r6rlrr_r!)rjrrr!rmr# kwargs_rows rsr;cCs 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)eInterfaceErrorr?r!r!r#rDs rDcs@|jsdSt}|durdS|jfddt|DS)Ncsg|] }|qSr!)rHrYrIrXrMr!r# sz_get_names..)rOrPrRrS)r6rUr!rvr#rCs rCrMrcCs0|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)rUstatus TUPLES_OK SINGLE_TUPLE COMMAND_OK)rMrUr!r!r#rPs   rP)r6r3rr7)r6r3rr:)r6r3rrG)=r) functoolstypingrrrrrrrr r r r r collectionsrtyping_extensionsrrrrt_compatr _encodingsrr6rr cursor_asyncrpsycopg.pq.abcr ExecStatusr{ryrzrrrr+r.r1r4__annotations__rEr5r9rFrW lru_cachebytesrQrirprsrDrCintrPr!r!r!r#sT$