o 6 f@sdZddlZddlmZmZmZmZmZmZddl m Z ddl m Z ddlmZmZmZmZddlmZdd lmZdd lmZdd lmZdd lmZd dlmZerfddlm Z ddl!m"Z"ddl#m$Z$GdddZ%Gddde%eeZ&Gddde%eeZ'GdddeZ(dS)z# CockroachDB-specific connections. N)AnyOptionalTypeUnionoverload TYPE_CHECKING)errors) AdaptContext)Row RowFactoryAsyncRowFactoryTupleRow)Self) Connection) AdaptersMap)ConnectionInfo)AsyncConnection)adapters)PGconn)Cursor) AsyncCursorc@steZdZUeeed<ded<edeee e e dfde fddZ e defdd Ze dd d ZdddZd S)_CrdbConnectionMixin _adaptersrpgconnconnreturncCs"t|ttfr |j}t|dS)zS Return `!True` if the server connected to `!conn` is CockroachDB. s crdb_version) isinstancerrrboolparameter_status)clsrr"P/home/ertert/spirit/venv/lib/python3.10/site-packages/psycopg/crdb/connection.pyis_crdbsz_CrdbConnectionMixin.is_crdbcCs|jstt|_|jSN)rrrselfr"r"r#r*s z_CrdbConnectionMixin.adaptersCrdbConnectionInfocCs t|jSr%)r(rr&r"r"r#info2s z_CrdbConnectionMixin.infoNcCs||jr tddS)Nz/CockroachDB doesn't support prepared statements)r$reNotSupportedErrorr&r"r"r# _check_tpc6s  z_CrdbConnectionMixin._check_tpc)rr()rN)__name__ __module__ __qualname__rr__annotations__ classmethodrrrrrr$propertyrr)r,r"r"r"r#rs    rcseZdZdZdZee ddddddded ed ee d e e d d de e de de efddfddZee ddddddded ed e e d dde e de de efddfddZeddededeffdd ZZS)CrdbConnectionz= Wrapper for a connection to a CockroachDB database. psycopg.crdbFN autocommitprepare_thresholdcursor_factorycontextconninfor8 row_factoryr9r:zOptional[Type[Cursor[Row]]]r;kwargsrzCrdbConnection[Row]cKdSr%r")r!r<r8r=r9r:r;r>r"r"r#connectEs zCrdbConnection.connectzOptional[Type[Cursor[Any]]]zCrdbConnection[TupleRow]cKr?r%r"r!r<r8r9r:r;r>r"r"r#r@Ss c stj|fi|S)zZ Connect to a database server and return a new `CrdbConnection` instance. superr@r!r<r> __class__r"r#r@`sr5)r-r.r/__doc__rr1strrr r rintr rr@rr __classcell__r"r"rEr#r3;sj      &r3cseZdZdZdZee ddddddded ed ee d e e d d dee de de efddfddZee ddddddded ed ee d ddee de de efddfddZeddededeffdd ZZS)AsyncCrdbConnectionzD Wrapper for an async connection to a CockroachDB database. r4r5Fr6Nr7r<r8r9r=r:z Optional[Type[AsyncCursor[Row]]]r;r>rzAsyncCrdbConnection[Row]cdSr%r")r!r<r8r9r=r:r;r>r"r"r#r@rs zAsyncCrdbConnection.connectz Optional[Type[AsyncCursor[Any]]]zAsyncCrdbConnection[TupleRow]crMr%r"rAr"r"r#r@s cstj|fi|IdHSr%rBrDrEr"r#r@srG)r-r.r/rHrr1rIrrrJr r r rr@rrrKr"r"rEr#rLhsj      &rLc@sReZdZdZdZedefddZedefddZe dede efd d Z d S) r(zV `~psycopg.ConnectionInfo` subclass to get info about a CockroachDB database. r4rcCsdS)N CockroachDBr"r&r"r"r#vendorszCrdbConnectionInfo.vendorcCs>|d}|s td||}|durtd||S)z Return the CockroachDB server version connected. Return a number in the PostgreSQL format (e.g. 21.2.10 -> 210210). crdb_versionz''crdb_version' parameter status not setNz)couldn't parse CockroachDB version from: )r r* InternalErrorparse_crdb_versionInterfaceError)r'sververr"r"r#server_versions   z!CrdbConnectionInfo.server_versionrTcCsFtd|}|s dSt|ddt|ddt|dS)Nz\bv(\d+)\.(\d+)\.(\d+)ri'rd)researchrJgroup)r'rTmr"r"r#rRs 2z%CrdbConnectionInfo.parse_crdb_versionN) r-r.r/rHr2rIrOrJrVr1rrRr"r"r"r#r(sr())rHrYtypingrrrrrrr5r r*abcr rowsr r r r_compatr connectionr _adapters_mapr_connection_inforconnection_asyncr_typesrpq.abcrcursorr cursor_asyncrrr3rLr(r"r"r"r#s(            !-*