o 6 f@s(UdZddlmZmZmZmZddlmZmZmZm Z m Z m Z ddlm Z ddl mZddlmZddlmZdd lmZmZe ridd lmZdd lmZmZdd lmZdd lmZmZddlm Z ddl!m"Z"e#dZ$e#e%d<e e&e'e(fZ)ee%d<e ee&ddfZ*ee%d<e eeee+effZ,ee%d<e dddZ-egdfZ.ee%d<e e#e dfZ/ee%d<e dZ0ee e1dfde0fZ2ee%d < edde0fZ3ee%d!< Gd"d#d#eZ4eege)fZ5ee%d$<ee)gefZ6ee%d%<Gd&d'd'eZ7Gd(d)d)eZ8Gd*d+d+eZ9Gd,d-d-eZ:dS).zN Protocol objects representing different implementations of the same classes. )AnyCallable GeneratorMapping)ListOptionalSequenceTupleTypeVarUnion) TYPE_CHECKING) TypeAlias)pq)PyFormat)Protocol LiteralString)sql)RowRowMaker)PGresult)WaitReady)BaseConnection) AdaptersMapNNoneTypeBufferzsql.SQLz sql.ComposedQueryParamsConnectionTypeBaseConnection[Any])boundPipelineCommand) DumperKey.r#RVrr PQGenConnPQGenc @s6eZdZdZ d deededeedefddZ dS) WaitFunczU Wait on the connection which generated `PQgen` and return its final result. NgenfilenotimeoutreturncCdSN)selfr(r)r*r.r.D/home/ertert/spirit/venv/lib/python3.10/site-packages/psycopg/abc.py__call__7zWaitFunc.__call__r-) __name__ __module__ __qualname____doc__r&r$intrfloatr1r.r.r.r0r'2sr'DumpFuncLoadFuncc@s4eZdZdZed ddZededfddZd 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. r+rcCdS)z1The adapters configuration that this object uses.Nr.r/r.r.r0adaptersNszAdaptContext.adaptersr cCr<)zThe connection used by this object, if available. :rtype: `~psycopg.Connection` or `~psycopg.AsyncConnection` or `!None` Nr.r=r.r.r0 connectionSzAdaptContext.connectionNr+r)r3r4r5r6propertyr>rr?r.r.r.r0r;Bs  r;c@seZdZUdZejed< eed< ddede e fddZ d e d e fd d Zd e d e fd dZd e ded efddZd e ded dfddZdS)DumperzM Convert Python objects of type `!cls` to PostgreSQL representation. formatoidNclscontextcCr,r-r.)r/rFrGr.r.r0__init__lzDumper.__init__objr+cCr<)zlConvert the object `!obj` to PostgreSQL representation. :param obj: the object to convert. Nr.r/rJr.r.r0dumpnz Dumper.dumpcCr<)ziConvert the object `!obj` to escaped representation. :param obj: the object to convert. Nr.rKr.r.r0quoteurMz Dumper.quotecCr<)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/rJrDr.r.r0get_key|s"zDumper.get_keycCr<)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.rOr.r.r0upgrades zDumper.upgrader-)r3r4r5r6rFormat__annotations__r7typerr;rHrrrLrNrr#rPrQr.r.r.r0rC\s  $rCc@sHeZdZUdZejed< d dedee fddZ de d e fd d Z dS) LoaderzK Convert PostgreSQL values with type OID `!oid` to Python objects. rDNrErGcCr,r-r.)r/rErGr.r.r0rHrIzLoader.__init__datar+cCr<)z| Convert the data returned by the database into a Python object. :param data: the data to convert. Nr.)r/rVr.r.r0loadr@z Loader.loadr-)r3r4r5r6rrRrSr7rr;rHrrrWr.r.r.r0rUs  rUc @seZdZUeeedfed<eeej ed<d9dee fddZ e dee ddfd d Z eded fd d ZedefddZed:ddZededfddZddddeddedeej ddfddZdeedej ddfddZdeedej ddfdd Zd!eedeedeeefd"d#Zd$edefd%d&Zd$ededefd'd(Z d)ed*ed+d,ded-fd.d/Z!d0ed+d,ded-fd1d2Z"d3eeedeedffd4d5Z#d6edej de$fd7d8Z%dS); Transformer.typesformatsNrGcCr,r-r.)r/rGr.r.r0rHrIzTransformer.__init__r+cCr,r-r.)rFrGr.r.r0 from_contextzTransformer.from_contextr cCr,r-r.r=r.r.r0r?r\zTransformer.connectioncCr,r-r.r=r.r.r0encodingr\zTransformer.encodingrcCr,r-r.r=r.r.r0r>r\zTransformer.adaptersrcCr,r-r.r=r.r.r0pgresultr\zTransformer.pgresultT) set_loadersrDresultr_rDcCr,r-r.)r/r`r_rDr.r.r0 set_pgresultr@zTransformer.set_pgresultcCr,r-r.r/rYrDr.r.r0set_dumper_typesrIzTransformer.set_dumper_typescCr,r-r.rbr.r.r0set_loader_typesrIzTransformer.set_loader_typesparamscCr,r-r.)r/rerZr.r.r0 dump_sequencer2zTransformer.dump_sequencerJcCr,r-r.rKr.r.r0 as_literalrIzTransformer.as_literalcCr,r-r.rOr.r.r0 get_dumperrIzTransformer.get_dumperrow0row1make_rowz RowMaker[Row]rcCr,r-r.)r/rirjrkr.r.r0 load_rowsr2zTransformer.load_rowsrowcCr,r-r.)r/rmrkr.r.r0load_rowrIzTransformer.load_rowrecordcCr,r-r.)r/ror.r.r0 load_sequencerIzTransformer.load_sequencerEcCr,r-r.)r/rErDr.r.r0 get_loaderrIzTransformer.get_loaderr-rA)&r3r4r5rr r7rSrrrRr;rH classmethodr[rBr?strr]r>r^boolrarrcrdrrrrfbytesrgrCrhrlrnrprUrqr.r.r.r0rXs`      "rX);r6typingrrrrrrrr r r r typing_extensionsr r_enumsr_compatrrrrowsrrpq.abcrwaitingrrr?r _adapters_maprrTrrSru bytearray memoryviewrrrsrrr"r#r$r7r%r&r'r9r:r;rCrUrXr.r.r.r0sD            S