S?NjUdZddlmZddlZddlmZmZmZmZm Z m Z ddl m Z ddl mZmZddlmZdd lmZdd lmZdd lmZerdd lmZdd lmZddlmZddlmZej@jBZ!ej@jDZ"ej@jFZ#ej@jHZ$eddZ%edddZ&Gdde e&Z'Gdde e&Z(Gdde e&Z)Gdde e&Z*e+edfZ,de-d< e.e/efZ0de-d < d-d!Z1d.d"Z2d/d#Z3ejhd$d0d%Z5d1d&Z6d2d'Z7d2d(Z8d3d)Z9d4d*Z:d5d+Z;d6d,Zds  Lr$c:t|xdfd }|StS)zRow factory to represent rows as dictionaries. The dictionary keys are taken from the column names of the returned columns. c.tt|Srdictzip)valuesnamess r" dict_row_zdict_row..dict_row_xsE6*+ +r$)rDr%r&zdict[str, Any] _get_names no_result)r=rFrEs @r"dict_rowrJqs* F##0 ,r$c|jxrAtx}4t|jgfdt |D}|j St S)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. c3@K|]}j|ywr)fname).0iress r" z!namedtuple_row..s)O1#))A,s)pgresult _get_nfields_make_nt _encodingrange_makerI)r=nfieldsntrPs @r"namedtuple_rowrZsQ  S0A%AW$N f&& P)Og)O Pxxr$icDtfd|D}td|S)Nc3RK|]}t|j ywr)rdecode)rNnencs r"rQz_make_nt..s GA(#7s$'r)r<r )r_rEsnamess` r"rTrTs GG GF eV $$r$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]] c>t|x dfd }|StS)Nc :ditt|SNrrA)rDclsrEs r" class_row__z2class_row..class_row_..class_row__s6T#eV"4566r$rDr%r&rrG)r=rfrEres @r" class_row_zclass_row..class_row_s(' 'E 4 7  r$)r=r9r& RowMaker[T]r)rerhs` r" class_rowrjs  r$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. cdfd }|S)Nc|Srr)rDfuncs r" args_row__z/args_row..args_row_..args_row__s = r$rgr)currorns r" args_row_zargs_row..args_row_s !r$)rpBaseCursor[Any, T]r&rir)rnrqs` r"args_rowrss 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. c>t|x dfd }|StS)Nc :ditt|SrdrA)rDrnrEs r" kwargs_row__z5kwargs_row..kwargs_row_..kwargs_row__s7d3uf#5677r$rgrG)r=rwrErns @r" kwargs_row_zkwargs_row..kwargs_row_s(' 'E 4 8  r$)r=rrr&rir)rnrxs` r" kwargs_rowrys  r$c|jx}r-t|x} |dkrtjddd}|StS)zR Generate a row factory returning the first column as a scalar value. r zat least one column expectedc |dS)NrrrDs r" scalar_row_zscalar_row..scalar_row_s !9 r$)rDr%r&r)rRrSeProgrammingErrorrI)r=rPrXr}s r" scalar_rowrsP  S0A%AW$N Q;$$%CD D 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)r~InterfaceErrorr|s r"rIrIs  = >>r$c|jx}rPt|x}C|j}t|Dcgc]"}|j |j |$c}Sycc}wr)rRrSrUrVrMr])r=rPrXr_rOs r"rHrHsh S0A%AW$N.3Gn .<CIIaL   $n    s'A c|j}|jtk(s;|jtk(s(|jtk(s|jt k(r|r|Sy)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)rXstatus TUPLES_OK SINGLE_TUPLE TUPLES_CHUNK COMMAND_OK)rPrXs r"rSrSsI kkG i :: % :: % JJ* $r$)r=r9r&zRowMaker[TupleRow])r=r9r&zRowMaker[DictRow])r=r9r&zRowMaker[NamedTuple])r_strrEbytesr&ztype[NamedTuple])reztype[T]r&BaseRowFactory[T])rnzCallable[..., T]r&r)r=r9r&z RowMaker[Any])rDr%r&r)r=r9r&zlist[str] | None)rPrr&z int | None)=r+ __future__r functoolstypingrrrrrr collectionsr collections.abcr r rr~r_compatr _encodingsrr=rpq.abcr _cursor_baser cursor_asyncr ExecStatusrrrrrrrr-r2r6r<r__annotations__rBrr:r>rJrZ lru_cacherTrjrsryrrIrHrSrr$r"rs\ #PP".- () ]] % % MM # # }})) }}))  C4 etZ8