jUdZddlmZddlZddlmZmZmZmZm Z m Z ddl m Z ddl mZmZddlmZdd lmZdd lmZdd lmZerdd lmZdd lmZddlmZddlmZej j!Z!ej j"Z"ej j#Z#ej j$Z$eddZ%edddZ&Gdde e&Z'Gdde e&Z(Gdde e&Z)Gdde e&Z*e+edfZ,de-d< e.e/efZ0de-d < dEd%Z1dFd'Z2dGd)Z3ej4d*dHd0Z5dId4Z6dJd7Z7dJd8Z8dKd:Z9dLd>Z:dMd@Z;dNdDZs r$ tuple_rowrBds  Lr'RowMaker[DictRow]cDt|xdfd }|StS)zRow factory to represent rows as dictionaries. The dictionary keys are taken from the column names of the returned columns. Nvaluesrrdict[str, Any]c>tt|Sr!dictzip)rEnamess r$ dict_row_zdict_row..dict_row_xsE6**++ +r')rErrrF _get_names no_result)r>rLrKs @r$dict_rowrPqsF F###0 , , , , , ,r'RowMaker[NamedTuple]c|jxrCtx}2t|jgfdt |DR}|jSt 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. Nc3BK|]}|VdSr!)fname).0iress r$ z!namedtuple_row..s-)O)O1#))A,,)O)O)O)O)O)Or')pgresult _get_nfields_make_nt _encodingrange_makerO)r>nfieldsntrWs @r$namedtuple_rowrasf  S0A0A%AW$N f& P)O)O)O)Og)O)O)O P P Pxr'iencstrrKbytestype[NamedTuple]cZtfd|D}td|S)Nc3\K|]&}t|V'dSr!)rdecode)rUnrbs r$rXz_make_nt..s6GGA(#77GGGGGGr'r)rAr )rbrKsnamess` r$r[r[s5 GGGGGGG G GF eV $ $$r'clstype[T]BaseRowFactory[T]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]] r>r;r RowMaker[T]cHt|x dfd }|StS)NrErrrc Jditt|SNr"rH)rErkrKs r$ class_row__z2class_row..class_row_..class_row__s+s66T#eV"4"455666r'rErrrrM)r>rsrKrks @r$ class_row_zclass_row..class_row_sJ'' 'E 4 7 7 7 7 7 7 7  r')r>r;rror")rkrus` r$ class_rowrvs)       r'funcCallable[..., T]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. curBaseCursor[Any, T]rrocdfd }|S)NrErrrc|Sr!r")rErws r$ args_row__z/args_row..args_row_..args_row__s4= r'rtr")rzr~rws r$ args_row_zargs_row..args_row_s) ! ! ! ! ! !r')rzr{rror")rwrs` r$args_rowrs) 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>r{rrocHt|x dfd }|StS)NrErrrc Jditt|SrrrH)rErwrKs r$ kwargs_row__z5kwargs_row..kwargs_row_..kwargs_row__s+t77d3uf#5#566777r'rtrM)r>rrKrws @r$ kwargs_row_zkwargs_row..kwargs_row_sJ'' 'E 4 8 8 8 8 8 8 8  r')r>r{rror")rwrs` r$ kwargs_rowrs)       r' RowMaker[Any]c|jx}r1t|x} |dkrtjdd d}|StS) zR Generate a row factory returning the first column as a scalar value. Nr zat least one column expectedrErrrc|dS)Nrr"rEs r$ scalar_row_zscalar_row..scalar_row_s !9 r')rErrr)rYrZeProgrammingErrorrO)r>rWr_rs r$ scalar_rowrsd   S0A0A%AW$N Q;;$%CDD D    r'rErrc*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)rInterfaceErrorrs r$rOrOs = > >>r'list[str] | Nonec|jxr4tx}#|jfdt|DSdS)Nc`g|]*}|+Sr")rTrh)rUrVrbrWs r$ z_get_names..s>   )*CIIaLL   $ $   r')rYrZr\r])r>r_rbrWs @@r$rNrNsj S0A0A%AW$N     .3Gnn    tr'rWr int | Nonec|j}|jtks2|jtks"|jtks|jt kr|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 TUPLES_CHUNK COMMAND_OK)rWr_s r$rZrZsP kG i : % % : % % J* $ $ $tr')r>r;rr?)r>r;rrC)r>r;rrQ)rbrcrKrdrre)rkrlrrm)rwrxrrm)r>r;rr)rErrr)r>r;rr)rWrrr)=r, __future__r functoolstypingrrrrrr collectionsr collections.abcr r rrr_compatr _encodingsrr>rpq.abcr _cursor_baser cursor_asyncr ExecStatusrrrrrrrr.r4r9rAr__annotations__rIrcr=rBrPra lru_cacher[rvrrrrOrNrZr"r'r$rs #"""""PPPPPPPPPPPPPPPP""""""........------*      (((((()))))) ] % M # }) })  GC4   getZ888 < < < <