o "4g @sdZddlmZddlZddlmZmZddlmZ ddl m Z ddl m Z dd lmZdd lmZdd lmZd d lmZerFddlmZGdddZGdddee e ZGdddeee ZGdddeZdS)z# CockroachDB-specific connections. ) annotationsN)Any TYPE_CHECKING)errors)Row) Connection) AdaptersMap)ConnectionInfo)AsyncConnection)adapters)PGconnc@sReZdZUded<ded<edd d Zedd d ZedddZdddZ dS)_CrdbConnectionMixinzAdaptersMap | None _adaptersrpgconnconn/Connection[Any] | AsyncConnection[Any] | PGconnreturnboolcCs"t|ttfr |j}t|dS)zS Return `!True` if the server connected to `!conn` is CockroachDB. s crdb_version) isinstancerr rrparameter_status)clsrr]/var/www/html/notificationserver/venv/lib/python3.10/site-packages/psycopg/crdb/connection.pyis_crdbsz_CrdbConnectionMixin.is_crdbr cCs|jstt|_|jSN)rr r selfrrrr &s z_CrdbConnectionMixin.adaptersCrdbConnectionInfocCs t|jSr)rrrrrrinfo.s z_CrdbConnectionMixin.infoNonecCs||jr tddS)Nz/CockroachDB doesn't support prepared statements)rreNotSupportedErrorrrrr _check_tpc2s  z_CrdbConnectionMixin._check_tpcN)rrrr)rr )rr)rr!) __name__ __module__ __qualname____annotations__ classmethodrpropertyr r r$rrrrrs    rc@eZdZdZdZdS)CrdbConnectionz= Wrapper for a connection to a CockroachDB database. psycopg.crdbNr%r&r'__doc__rrrrr,7r,c@r+)AsyncCrdbConnectionzD Wrapper for an async connection to a CockroachDB database. r-Nr.rrrrr1?r0r1c@s>eZdZdZdZedddZeddd Zedd d ZdS)rzV `~psycopg.ConnectionInfo` subclass to get info about a CockroachDB database. r-rstrcCsdS)N CockroachDBrrrrrvendorNszCrdbConnectionInfo.vendorintcCs>|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: )rr" InternalErrorparse_crdb_versionInterfaceError)rsververrrrserver_versionRs   z!CrdbConnectionInfo.server_versionr: int | NonecCsFtd|}|s dSt|ddt|ddt|dS)Nz\bv(\d+)\.(\d+)\.(\d+)r i'rd)researchr5group)rr:mrrrr8cs 2z%CrdbConnectionInfo.parse_crdb_versionN)rr2)rr5)r:r2rr=) r%r&r'r/r*r4r<r)r8rrrrrGs  r)r/ __future__rr@typingrrrr"rowsr connectionr _adapters_mapr _connection_infor connection_asyncr _typesr pq.abcrrr,r1rrrrrs"