U ڲg@sUdZddlmZddlmZmZmZmZddlmZm Z m Z ddl m Z ddl mZddlmZmZe rdd l mZdd lmZmZdd lmZdd lmZmZdd lmZddlmZddlmZe dZ!de"d<dZ#de"d<dZ$de"d<dZ%de"d<edddZ&egdfZ'de"d<dZ(de"d<d Z)de"d!<d"Z*de"d#<ee+e)fZ,de"d$<ed%Z-ed&d'e-fZ.de"d(<ed)d'e-fZ/de"d*<Gd+d,d,eZ0eegd-fZ1de"d.<ee#gefZ2de"d/<Gd0d1d1eZ3Gd2d3d3eZ4Gd4d5d5eZ5Gd6d7d7eZ6dS)8zN Protocol objects representing different implementations of the same classes. ) annotations)AnyCallable GeneratorMapping)ProtocolSequence TYPE_CHECKING)pq)PyFormat) TypeAliasTypeVar)sql)RowRowMaker)PGresult)WaitReady) LiteralString) AdaptersMap)BaseConnectionNtypeNoneTypezbytes | bytearray | memoryviewr Bufferz.LiteralString | bytes | sql.SQL | sql.ComposedQueryz!Sequence[Any] | Mapping[str, Any]ParamsConnectionTypezBaseConnection[Any])boundPipelineCommandztype | tuple[DumperKey, ...] DumperKeyzstr | int | None ConnParamzdict[str, ConnParam]ConnDict ConnMappingRVztuple[int, Wait]z Ready | int PQGenConnrPQGenc@s&eZdZdZd ddddddd ZdS) WaitFunczU Wait on the connection which generated `PQgen` and return its final result. Nz PQGen[RV]intz float | Noner$)genfilenointervalreturncCsdSN)selfr)r*r+r.r.//tmp/pip-unpacked-wheel-b_ea6rx_/psycopg/abc.py__call__<szWaitFunc.__call__)N)__name__ __module__ __qualname____doc__r1r.r.r.r0r'7sr' Buffer | NoneDumpFuncLoadFuncc@s4eZdZdZeddddZeddddZd S) AdaptContextaO A context describing how types are adapted. Example of `~AdaptContext` are `~psycopg.Connection`, `~psycopg.Cursor`, `~psycopg.adapt.Transformer`, `~psycopg.adapt.AdaptersMap`. Note that this is a `~typing.Protocol`, so objects implementing `!AdaptContext` don't need to explicitly inherit from this class. rr,cCsdS)z1The adapters configuration that this object uses.Nr.r/r.r.r0adaptersSszAdaptContext.adaptersBaseConnection[Any] | NonecCsdS)zThe connection used by this object, if available. :rtype: `~psycopg.Connection` or `~psycopg.AsyncConnection` or `!None` Nr.r;r.r.r0 connectionXszAdaptContext.connectionN)r2r3r4r5propertyr<r>r.r.r.r0r9Gs  r9c@sxeZdZUdZded<ded<dddd d d Zd d dddZd ddddZd dddddZd dddddZ dS)DumperzM Convert Python objects of type `!cls` to PostgreSQL representation. pq.Formatformatr(oidNrAdaptContext | NoneclscontextcCsdSr-r.)r/rFrGr.r.r0__init__qzDumper.__init__rr6objr,cCsdS)zlConvert the object `!obj` to PostgreSQL representation. :param obj: the object to convert. Nr.r/rKr.r.r0dumpssz Dumper.dumprcCsdS)ziConvert the object `!obj` to escaped representation. :param obj: the object to convert. Nr.rLr.r.r0quotezsz Dumper.quoter r rKrBr,cCsdS)aReturn an alternative key to upgrade the dumper to represent `!obj`. :param obj: The object to convert :param format: The format to convert to Normally the type of the object is all it takes to define how to dump the object to the database. For instance, a Python `~datetime.date` can be simply converted into a PostgreSQL :sql:`date`. In a few cases, just the type is not enough. For example: - A Python `~datetime.datetime` could be represented as a :sql:`timestamptz` or a :sql:`timestamp`, according to whether it specifies a `!tzinfo` or not. - A Python int could be stored as several Postgres types: int2, int4, int8, numeric. If a type too small is used, it may result in an overflow. If a type too large is used, PostgreSQL may not want to cast it to a smaller type. - Python lists should be dumped according to the type they contain to convert them to e.g. array of strings, array of ints (and which size of int?...) In these cases, a dumper can implement `!get_key()` and return a new class, or sequence of classes, that can be used to identify the same dumper again. If the mechanism is not needed, the method should return the same `!cls` object passed in the constructor. If a dumper implements `get_key()` it should also implement `upgrade()`. Nr.r/rKrBr.r.r0get_keys"zDumper.get_keycCsdS)aReturn a new dumper to manage `!obj`. :param obj: The object to convert :param format: The format to convert to Once `Transformer.get_dumper()` has been notified by `get_key()` that this Dumper class cannot handle `!obj` itself, it will invoke `!upgrade()`, which should return a new `Dumper` instance, which will be reused for every objects for which `!get_key()` returns the same result. Nr.rPr.r.r0upgrades zDumper.upgrade)N) r2r3r4r5__annotations__rHrMrNrQrRr.r.r.r0r@as $r@c@s<eZdZUdZded<dddddd Zd d d d dZdS)LoaderzK Convert PostgreSQL values with type OID `!oid` to Python objects. rArBNr(rD)rCrGcCsdSr-r.)r/rCrGr.r.r0rHrIzLoader.__init__rr)datar,cCsdS)z| Convert the data returned by the database into a Python object. :param data: the data to convert. Nr.)r/rUr.r.r0loadsz Loader.load)N)r2r3r4r5rSrHrVr.r.r.r0rTs rTc@sFeZdZUded<ded<dKdddd Zeddd d d Zed dddZeddddZ eddddZ eddddZ ddddddddd d!Z d"d#dd$d%d&Z d"d#dd$d'd(Zd)d*d+d,d-d.Zd/d0d1d2d3Zd/d4d5d6d7d8Zd9d9d:d;d<d=d>Zd9d:d?d@dAdBZd+dCdDdEdFZd9d#dGdHdIdJZdS)L Transformerztuple[int, ...] | Nonetypeszlist[pq.Format] | NoneformatsNrD)rGcCsdSr-r.)r/rGr.r.r0rHrIzTransformer.__init__)rGr,cCsdSr-r.rEr.r.r0 from_contextszTransformer.from_contextr=r:cCsdSr-r.r;r.r.r0r>szTransformer.connectionstrcCsdSr-r.r;r.r.r0encodingszTransformer.encodingrcCsdSr-r.r;r.r.r0r<szTransformer.adapterszPGresult | NonecCsdSr-r.r;r.r.r0pgresultszTransformer.pgresultT) set_loadersrBboolzpq.Format | NoneNone)resultr^rBr,cCsdSr-r.)r/rar^rBr.r.r0 set_pgresultszTransformer.set_pgresultz Sequence[int]rA)rXrBr,cCsdSr-r.r/rXrBr.r.r0set_dumper_typesrIzTransformer.set_dumper_typescCsdSr-r.rcr.r.r0set_loader_typesrIzTransformer.set_loader_typesz Sequence[Any]zSequence[PyFormat]zSequence[Buffer | None])paramsrYr,cCsdSr-r.)r/rfrYr.r.r0 dump_sequenceszTransformer.dump_sequencerbytesrJcCsdSr-r.rLr.r.r0 as_literalrIzTransformer.as_literalr r@rOcCsdSr-r.rPr.r.r0 get_dumperrIzTransformer.get_dumperr(z RowMaker[Row]z list[Row])row0row1make_rowr,cCsdSr-r.)r/rkrlrmr.r.r0 load_rowsrIzTransformer.load_rowsz Row | None)rowrmr,cCsdSr-r.)r/rormr.r.r0load_rowrIzTransformer.load_rowztuple[Any, ...])recordr,cCsdSr-r.)r/rqr.r.r0 load_sequencerIzTransformer.load_sequencerT)rCrBr,cCsdSr-r.)r/rCrBr.r.r0 get_loaderrIzTransformer.get_loader)N)r2r3r4rSrH classmethodrZr?r>r\r<r]rbrdrergrirjrnrprrrsr.r.r.r0rWs2 rW)7r5 __future__rtypingrrrrrrr r Z_enumsr _compatr rrZrowsrrZpq.abcrZwaitingrrrZ _adapters_maprZ_connection_baserrrrSrrrrrr r!r"r[r#r$r%r&r'r7r8r9r@rTrWr.r.r.r0sD                S