o DDi@sUdZddlmZddlZddlmZmZmZmZm Z m Z ddl m Z ddl mZmZddlmZdd lmZdd lmZdd lmZer]dd lmZdd lmZddlmZddlmZej 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,de-d< e.e/efZ0de-d < dPd%d&Z1dQd(d)Z2dRd+d,Z3e4d-dSd3d4Z5dTd8d9Z6dUdd?Z8dVdAdBZ9dWdFdGZ:dXdIdJZ;dYdNdOZ.dict_row_)r<rrr= _get_names no_result)r7rDr!rBr#dict_rowqsrHRowMaker[NamedTuple]csH|jr"t}dur"t|jgfddt|DR}|jStS)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..)pgresult _get_nfields_make_nt _encodingrange_makerG)r7nfieldsntr!rNr#namedtuple_rows$rYiencstrrCbytestype[NamedTuple]cs tfdd|D}td|S)Nc3s|] }t|VqdSr )rdecode)rLnrZr!r#rPsz_make_nt..r)r9r )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]] r7r5r RowMaker[T]c(t|durdfdd }|StS)Nr<rrrcditt|SNr!r>rA)rbrCr!r# class_row__z2class_row..class_row_..class_row__r<rrrrE)r7rjrbrBr# class_row_zclass_row..class_row_N)r7r5rrfr!)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. r7rtrrfcrg)Nr<rrrcrhrir>rA)rqrCr!r# kwargs_row__rkz5kwargs_row..kwargs_row_..kwargs_row__rlrE)r7rzrurBr# kwargs_row_rozkwargs_row..kwargs_row_N)r7rtrrfr!)rqr{r!rur# kwargs_rows r| RowMaker[Any]cCs>|j}rt|}dur|dkrtdd dd }|StS) 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)rQrReProgrammingErrorrG)r7rOrWr~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<|jrt}dur|jfddt|DSdS)Ncsg|] }|qSr!)rJr^rKrZrOr!r# sz_get_names..)rQrRrTrU)r7rWr!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)r7r5rr8)r7r5rr;)r7r5rrI)rZr[rCr\rr])rbrcrrd)rqrrrrd)r7r5rr})r<rrr)r7r5rr)rOrrr)=r* __future__r functoolstypingrrrrrr collectionsr collections.abcr r r rr_compatr _encodingsrr7rpq.abcr _cursor_baser cursor_asyncr ExecStatusrrrrrrrr+r/r3r9r__annotations__r?r[r6r:rHrY lru_cacherSrpryr|rrGrFrRr!r!r!r#sR