*Jf UdZddlmZmZmZmZddlmZmZmZm Z m Z m Z ddlm Z ddl mZddlmZddlmZdd lmZmZe r(dd 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 ddZ-egdfZ.ee%d<e e#e dfZ/ee%d<e dZ0ee e1dfde0fZ2ee%d < edde0fZ3ee%d!< Gd"d#eZ4eege)fZ5ee%d$<ee)gefZ6ee%d%<Gd&d'eZ7Gd(d)eZ8Gd*d+eZ9Gd,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 FeZdZdZ ddeededeedefdZ dS) WaitFunczU Wait on the connection which generated `PQgen` and return its final result. NgenfilenotimeoutreturncdSN)selfr*r+r,s N/home/alex/cs2snipeproduction/venv/lib/python3.11/site-packages/psycopg/abc.py__call__zWaitFunc.__call__7s Sr/) __name__ __module__ __qualname____doc__r'r%intrfloatr3r0r4r2r)r)2sf GK9&)4r>Bsk     X  H%:;   X   r4r>ceZdZUdZejed< eed< ddede e fdZ de d e fd Zde d e fd Zde ded efd Zde ded dfd ZdS)DumperzM Convert Python objects of type `!cls` to PostgreSQL representation. formatoidNclscontextcdSr/r0)r1rKrLs r2__init__zDumper.__init__lr4objr-cdS)zlConvert the object `!obj` to PostgreSQL representation. :param obj: the object to convert. Nr0r1rPs r2dumpz Dumper.dumpn r4cdS)ziConvert the object `!obj` to escaped representation. :param obj: the object to convert. Nr0rRs r2quotez Dumper.quoteurTr4cdS)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()`. Nr0r1rPrIs r2get_keyzDumper.get_key|s D r4cdS)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. Nr0rXs r2upgradezDumper.upgrades  r4r/)r5r6r7r8rFormat__annotations__r9typerr>rNrrrSrVrr$rYr[r0r4r2rHrH\s I HHHQNNDN8L+ANNNN            " 3" " Y" " " " H  3    X       r4rHcXeZdZUdZejed< d dedee fdZ de de fd Z dS) LoaderzK Convert PostgreSQL values with type OID `!oid` to Python objects. rINrJrLcdSr/r0)r1rJrLs r2rNzLoader.__init__rOr4datar-cdS)z| Convert the data returned by the database into a Python object. :param data: the data to convert. Nr0)r1rbs r2loadz Loader.loadrDr4r/)r5r6r7r8rr\r]r9rr>rNrrrdr0r4r2r`r`sx INMCM(<*@MMMM  C      r4r`c eZdZUeeedfed<eeej ed<d)dee fdZ e dee ddfdZ eded fd Zedefd Zed*d ZededfdZddddeddedeej ddfdZdeedej ddfdZdeedej ddfdZdeedeedeeefdZdedefdZdededefdZ dededd ded!fd"Z!d#edd ded!fd$Z"d%eeedeedffd&Z#d'edej de$fd(Z%dS)+ Transformer.typesformatsNrLcdSr/r0)r1rLs r2rNzTransformer.__init__rOr4r-cdSr/r0)rKrLs r2 from_contextzTransformer.from_contextsMPSr4r!cdSr/r0r@s r2rCzTransformer.connections=@Sr4cdSr/r0r@s r2encodingzTransformer.encodings"sr4rcdSr/r0r@s r2rAzTransformer.adapterss),r4rcdSr/r0r@s r2pgresultzTransformer.pgresults03r4T) set_loadersrIresultrrrIcdSr/r0)r1rsrrrIs r2 set_pgresultzTransformer.set_pgresults sr4cdSr/r0r1rgrIs r2set_dumper_typeszTransformer.set_dumper_typesrOr4cdSr/r0rws r2set_loader_typeszTransformer.set_loader_typesrOr4paramscdSr/r0)r1r{rhs r2 dump_sequencezTransformer.dump_sequences%(Sr4rPcdSr/r0rRs r2 as_literalzTransformer.as_literalrOr4cdSr/r0rXs r2 get_dumperzTransformer.get_dumperrOr4row0row1make_rowz RowMaker[Row]rcdSr/r0)r1rrrs r2 load_rowszTransformer.load_rowsscr4rowcdSr/r0)r1rrs r2load_rowzTransformer.load_rowrOr4recordcdSr/r0)r1rs r2 load_sequencezTransformer.load_sequencerOr4rJcdSr/r0)r1rJrIs r2 get_loaderzTransformer.get_loaderrOr4r/rE)&r5r6r7rr r9r]rrr\r>rN classmethodrkrFrCstrrnrArqboolrur rxrzrrrr}bytesrrHrrrrr`rr0r4r2rfrfs E#s(O $$$$ d29o &&&&CC 6CCCCP8L#9PmPPP[P @H%:;@@@X@ "#"""X" ,,,X, 3(:.333X3 !&* $  #   UhsmTRYT4TTTTThsmTRYT4TTTT)sm).6x.@) (6" #))))1c0e0000CcC8CCCCC".= eTCS?SxSSSSWHXf-=$>W5c?WWWWDcD29DDDDDDDr4rf);r8typingrrrrrrr r r r r typing_extensionsrr_enumsr_compatrrrrowsrrpq.abcrwaitingrrrCr _adapters_maprr^rr]r bytearray memoryviewrrrrr r#r$r%r9r&r'r)r;r<r>rHr`rfr0r4r2rs 544444444444BBBBBBBBBBBBBBBB ''''''((((((,,,,,,,,+########      $$$$$$$$************d$%J67 777y.HIyIII(3-c)::; ;;;)1FGGG%b$h////T5)9#::; 9;;; WT]] sF{!3Wb!@A 9AAA VWb01y111xuf}-)---x}-)---     8   4P P P P P XP P P f     X   01E1E1E1E1E(1E1E1E1E1Er4