2iUdZddlmZddlZddlmZmZmZmZm 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,Z.dict_row_xsE6*+ +r%)rEr&r'zdict[str, Any] _get_names no_result)r>rGrFs @r"dict_rowrKqs* 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,)Os)pgresult _get_nfields_make_nt _encodingrange_makerJ)r>nfieldsntrQs @r"namedtuple_rowr[sQ  S0A%AW$N f&& P)Og)O Pxxr%icDtfd|D}td|S)Nc3RK|]}t|j ywr)rdecode)rOnencs r"rRz_make_nt..sGA(#7Gs$'r)r=r )r`rFsnamess` r"rUrUs 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|SNrrB)rEclsrFs r" class_row__z2class_row..class_row_..class_row__s6T#eV"4566r%rEr&r'rrH)r>rgrFrfs @r" class_row_zclass_row..class_row_s(' 'E 4 7  r%)r>r:r' RowMaker[T]r)rfris` r" class_rowrks  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)rEfuncs r" args_row__z/args_row..args_row_..args_row__s = r%rhr)currpros r" args_row_zargs_row..args_row_s !r%)rqBaseCursor[Any, T]r'rjr)rorrs` r"args_rowrts 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|SrerB)rErorFs r" kwargs_row__z5kwargs_row..kwargs_row_..kwargs_row__s7d3uf#5677r%rhrH)r>rxrFros @r" kwargs_row_zkwargs_row..kwargs_row_s(' 'E 4 8  r%)r>rsr'rjr)rorys` r" kwargs_rowrzs  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)NrrrEs r" scalar_row_zscalar_row..scalar_row_s !9 r%)rEr&r'r)rSrTeProgrammingErrorrJ)r>rQrYr~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)rInterfaceErrorr}s r"rJrJs  = >>r%c|jx}rPt|x}C|j}t|Dcgc]"}|j |j |$c}Sycc}wr)rSrTrVrWrNr^)r>rQrYr`rPs r"rIrIsd S0A%AW$N.3Gn )*CIIaL   $    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)rYstatus TUPLES_OK SINGLE_TUPLE TUPLES_CHUNK COMMAND_OK)rQrYs r"rTrTsI kkG i :: % :: % JJ* $r%)r>r:r'zRowMaker[TupleRow])r>r:r'zRowMaker[DictRow])r>r:r'zRowMaker[NamedTuple])r`strrFbytesr'ztype[NamedTuple])rfztype[T]r'BaseRowFactory[T])rozCallable[..., T]r'r)r>r:r'z RowMaker[Any])rEr&r'r)r>r:r'zlist[str] | None)rQrr'z int | None)=r, __future__r functoolstypingrrrrrr collectionsr collections.abcr r rrr_compatr _encodingsrr>rpq.abcr _cursor_baser cursor_asyncr ExecStatusrrrrrrrr.r3r7r=r__annotations__rCrr;r?rKr[ lru_cacherUrkrtrzrrJrIrTrr%r"rs\ #PP".- () ]] % % MM # # }})) }}))  C4 etZ8