Added PGconn.protocol_version

diff --git a/psycopg3/_pq_ctypes.py b/psycopg3/_pq_ctypes.py
index 1feb246ac33ef6b5241471e87dc9e5b9dd8daa92..da779cf40c5f03d9eb1472b5d4782d14137598df 100644 (file)
--- a/psycopg3/_pq_ctypes.py
+++ b/psycopg3/_pq_ctypes.py
@@ -141,6 +141,10 @@ PQparameterStatus = pq.PQparameterStatus
 PQparameterStatus.argtypes = [PGconn_ptr, c_char_p]
 PQparameterStatus.restype = c_char_p
 
+PQprotocolVersion = pq.PQprotocolVersion
+PQprotocolVersion.argtypes = [PGconn_ptr]
+PQprotocolVersion.restype = c_int
+
 PQerrorMessage = pq.PQerrorMessage
 PQerrorMessage.argtypes = [PGconn_ptr]
 PQerrorMessage.restype = c_char_p

diff --git a/psycopg3/pq_ctypes.py b/psycopg3/pq_ctypes.py
index 88c873f42a0d79390a3fb2878b8cbb1818cc4466..f7d3a42f39cf142770a56cbe98bff6fa926726c1 100644 (file)
--- a/psycopg3/pq_ctypes.py
+++ b/psycopg3/pq_ctypes.py
@@ -173,6 +173,10 @@ class PGconn:
         rv = impl.PQparameterStatus(self.pgconn_ptr, self._encode(name))
         return self._decode(rv)
 
+    @property
+    def protocol_version(self):
+        return impl.PQprotocolVersion(self.pgconn_ptr)
+
     @property
     def error_message(self):
         return self._decode(impl.PQerrorMessage(self.pgconn_ptr))

diff --git a/tests/test_pq.py b/tests/test_pq.py
index 5f1c22a8ee1f14fe21ab7c2474cd023821976904..5e02b54030fd6d3cb05120b74f93c97804295317 100644 (file)
--- a/tests/test_pq.py
+++ b/tests/test_pq.py
@@ -179,3 +179,7 @@ def test_parameter_status(pq, dsn, tempenv):
     pgconn = pq.PGconn.connect(dsn)
     assert pgconn.parameter_status('application_name') == "psycopg3 tests"
     assert pgconn.parameter_status('wat') is None
+
+
+def test_protocol_version(pgconn):
+    assert pgconn.protocol_version == 3