i dZddlmZddlZddlmZmZddlmZ ddl m Z dd l m Z dd lmZdd lmZdd lmZdd lmZerddlmZGddZGddeee ZGddeee ZGddeZy)z# CockroachDB-specific connections. ) annotationsN) TYPE_CHECKINGAny)errors)Row)adapters) Connection) AdaptersMap)ConnectionInfo)AsyncConnection)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 rrboolparameter_status)clsconns w/var/www/html/hwPaymentPortal-be-dev/hw-payment-portal-api/venv/lib/python3.12/site-packages/psycopg/crdb/connection.pyis_crdbz_CrdbConnectionMixin.is_crdbs2 dZ9 :;;DD))/:;;cZ|jstt|_|jSN)rr r selfs rr z_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__ classmethodrpropertyr r#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|jdx}stjd|j|x}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_versionRs^--n===//"KL L**40 0C 9""%Nth#WX X rctjd|x}syt|jddzt|jddzzt|jdzS)Nz\bv(\d+)\.(\d+)\.(\d+)r i'rd)researchintgroup)r rBms rr@z%CrdbConnectionInfo.parse_crdb_versionasZYY8$???1771:&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_typesr connectionr _adapters_mapr _connection_infor connection_asyncrpq.abcrrr2r8r"r0rrrXsq # %#'-.YY> ):c?  .0D QQr