qj dZddlmZddlZddlmZmZddlmZ ddl m Z ddl m Z dd lmZdd lmZdd lmZd d lmZerddlmZGddZGddee e ZGddeee ZGddeZy)z# CockroachDB-specific connections. ) annotationsN)Any TYPE_CHECKING)errors)Row) Connection) AdaptersMap)ConnectionInfo)AsyncConnection)adapters)PGconnc`eZdZUded<ded<ed dZed dZed dZd dZ y )_CrdbConnectionMixinzAdaptersMap | None _adaptersrpgconnczt|ttfr |j}t |j dS)zS Return `!True` if the server connected to `!conn` is CockroachDB. s crdb_version) isinstancer r rboolparameter_status)clsconns X/var/www/html/kelly-backend/venv/lib/python3.12/site-packages/psycopg/crdb/connection.pyis_crdbz_CrdbConnectionMixin.is_crdbs2 dZ9 :;;DD))/:;;cZ|jstt|_|jSN)rr rselfs rrz_CrdbConnectionMixin.adapters&s~~(2DN~~rc,t|jSr)CrdbConnectionInforrs rinfoz_CrdbConnectionMixin.info.s!$++..rcd|j|jrtjdy)Nz/CockroachDB doesn't support prepared statements)rreNotSupportedErrorrs r _check_tpcz_CrdbConnectionMixin._check_tpc2s) << $%%&WX X %rN)rz/Connection[Any] | AsyncConnection[Any] | PGconnreturnr)r(r )r(r")r(None) __name__ __module__ __qualname____annotations__ classmethodrpropertyrr#r'rrrrsM!! N<<//YrrceZdZdZdZy)CrdbConnectionz= Wrapper for a connection to a CockroachDB database. psycopg.crdbNr*r+r,__doc__r0rrr2r27 Jrr2ceZdZdZdZy)AsyncCrdbConnectionzD Wrapper for an async connection to a CockroachDB database. r3Nr4r0rrr8r8?r6rr8cJeZdZdZdZeddZeddZed dZy) r"zV `~psycopg.ConnectionInfo` subclass to get info about a CockroachDB database. r3cy)N CockroachDBr0rs rvendorzCrdbConnectionInfo.vendorNsrc|jd}|stjd|j|}|tjd||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 setz)couldn't parse CockroachDB version from: )rr% InternalErrorparse_crdb_versionInterfaceError)r sververs rserver_versionz!CrdbConnectionInfo.server_versionRsZ$$^4//"KL L%%d+ ;""%Nth#WX X rctjd|}|syt|jddzt|jddzzt|jdzS)Nz\bv(\d+)\.(\d+)\.(\d+)r i'rd)researchintgroup)r rBms rr@z%CrdbConnectionInfo.parse_crdb_versioncsW II/ 61771:&QWWQZ3)>>QWWQZPPrN)r(str)r(rJ)rBrMr(z int | None) r*r+r,r5r/r<rDr.r@r0rrr"r"GsJ J  QQrr")r5 __future__rrHtypingrrrr%rowsr connectionr _adapters_mapr _connection_infor connection_asyncr _typesrpq.abcrrr2r8r"r0rrrXsq # %#'-.YY> ):c?  .0D "Q"Qr