Ytj dZddlmZddlZddlmZmZddlmZ ddl m Z dd l m Z dd lmZdd lmZdd lmZdd lmZerddlmZGddZGddeee ZGddeee ZGddeZdS)z# CockroachDB-specific connections. ) annotationsN) TYPE_CHECKINGAny)errors)Row)adapters) Connection) AdaptersMap)ConnectionInfo)AsyncConnection)PGconncteZdZUded<ded<edd Zedd Zedd ZddZ dS)_CrdbConnectionMixinzAdaptersMap | None _adaptersrpgconnconn/Connection[Any] | AsyncConnection[Any] | PGconnreturnboolct|ttfr|j}t |dS)zS Return `!True` if the server connected to `!conn` is CockroachDB. s crdb_version) isinstancer rrrparameter_status)clsrs U/var/www/html/CA-Chatbot/venv/lib/python3.11/site-packages/psycopg/crdb/connection.pyis_crdbz_CrdbConnectionMixin.is_crdbs> dZ9 : : ;DD))/::;;;r cP|jstt|_|jSN)rr r selfs rr z_CrdbConnectionMixin.adapters&s#~ 3(22DN~rCrdbConnectionInfoc*t|jSr )r#rr!s rinfoz_CrdbConnectionMixin.info.s!$+...rNonecb||jrtjddS)Nz/CockroachDB doesn't support prepared statements)rreNotSupportedErrorr!s r _check_tpcz_CrdbConnectionMixin._check_tpc2s7 << $ $ Y%&WXX X Y YrN)rrrr)rr )rr#)rr&) __name__ __module__ __qualname____annotations__ classmethodrpropertyr r%r*rrrrs!!!!NNN<<<[<X///X/YYYYYYrrceZdZdZdZdS)CrdbConnectionz= Wrapper for a connection to a CockroachDB database. psycopg.crdbNr+r,r-__doc__r1rrr3r37 JJJrr3ceZdZdZdZdS)AsyncCrdbConnectionzD Wrapper for an async connection to a CockroachDB database. r4Nr5r1rrr9r9?r7rr9c^eZdZdZdZed dZed dZedd Zd S)r#zV `~psycopg.ConnectionInfo` subclass to get info about a CockroachDB database. r4rstrcdS)N CockroachDBr1r!s rvendorzCrdbConnectionInfo.vendorNs}rintc|dx}stjd||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 setNz)couldn't parse CockroachDB version from: )rr( InternalErrorparse_crdb_versionInterfaceError)r"sververs rserver_versionz!CrdbConnectionInfo.server_versionRsi--n=== M/"KLL L**400 0C 9"#Wt#W#WXX X rrE int | Nonectjd|x}sdSt|ddzt|ddzzt|dzS)Nz\bv(\d+)\.(\d+)\.(\d+)r i'rd)researchr?group)r"rEms rrCz%CrdbConnectionInfo.parse_crdb_versionaskY8$??? 41771::&QWWQZZ3)>>QWWQZZPPrN)rr;)rr?)rEr;rrH) r+r,r-r6r0r>rGr/rCr1rrr#r#Gs J X   X QQQ[QQQrr#)r6 __future__rrLtypingrrrr(rowsr_typesr connectionr _adapters_mapr _connection_infor connection_asyncrpq.abcrrr3r9r#r1rrrZs #""""" %%%%%%%%######''''''------...... YYYYYYYY>     ):c?        .0D   QQQQQQQQQQr