psycopg/psycopg2
1
1
Fork 0
mirror of https://github.com/psycopg/psycopg2.git synced 2024-11-25 10:23:43 +03:00
Code Issues Packages Projects Releases Wiki Activity

Get CockroachDB version from the connection info

Browse Source
This commit is contained in:
Daniele Varrazzo 2020-07-21 21:48:11 +01:00
parent 7e1e801899
commit 9380f2a721
1 changed files with 10 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

23
tests/testutils.py
View File

@ -424,20 +424,17 @@ def crdb_version(conn, __crdb_version=[]):
if __crdb_version:
return __crdb_version[0]
with conn.cursor() as cur:
try:
cur.execute("show crdb_version")
except psycopg2.ProgrammingError:
__crdb_version.append(None)
else:
sver = cur.fetchone()[0]
m = re.search(r"\bv(\d+)\.(\d+)\.(\d+)", sver)
if not m:
raise ValueError(
"can't parse CockroachDB version from %s" % sver)
sver = conn.info.parameter_status("crdb_version")
if sver is None:
__crdb_version.append(None)
else:
m = re.search(r"\bv(\d+)\.(\d+)\.(\d+)", sver)
if not m:
raise ValueError(
"can't parse CockroachDB version from %s" % sver)
ver = int(m.group(1)) * 10000 + int(m.group(2)) * 100 + int(m.group(3))
__crdb_version.append(ver)
ver = int(m.group(1)) * 10000 + int(m.group(2)) * 100 + int(m.group(3))
__crdb_version.append(ver)
return __crdb_version[0]