From: Daniele Varrazzo <daniele.varrazzo@gmail.com>
Date: Fri, 27 Mar 2020 08:13:46 +0000 (+1300)
Subject: Added format enum
X-Git-Tag: 3.0.dev0~667
X-Git-Url: http://git.ipfire.org/gitweb.cgi?a=commitdiff_plain;h=4c647b8fed7fc5c8a42a68c6150a91592dcf29a6;p=thirdparty%2Fpsycopg.git

Added format enum
---

diff --git a/psycopg3/pq/__init__.py b/psycopg3/pq/__init__.py
index b9387cca0..98575ae60 100644
--- a/psycopg3/pq/__init__.py
+++ b/psycopg3/pq/__init__.py
@@ -16,6 +16,7 @@ from .enums import (
     TransactionStatus,
     Ping,
     DiagnosticField,
+    Format,
 )
 from .encodings import py_codecs
 from .misc import error_message
@@ -35,6 +36,7 @@ __all__ = (
     "ExecStatus",
     "Ping",
     "DiagnosticField",
+    "Format",
     "PGconn",
     "Conninfo",
     "PQerror",
diff --git a/psycopg3/pq/enums.py b/psycopg3/pq/enums.py
index a8bbab7af..e692fc14e 100644
--- a/psycopg3/pq/enums.py
+++ b/psycopg3/pq/enums.py
@@ -79,3 +79,8 @@ class DiagnosticField(IntEnum):
     SOURCE_FILE = ord("F")
     SOURCE_LINE = ord("L")
     SOURCE_FUNCTION = ord("R")
+
+
+class Format(IntEnum):
+    TEXT = 0
+    BINARY = 1
diff --git a/psycopg3/utils/queries.py b/psycopg3/utils/queries.py
index 823d37c88..8cdd8cb3e 100644
--- a/psycopg3/utils/queries.py
+++ b/psycopg3/utils/queries.py
@@ -8,6 +8,7 @@ import re
 from collections.abc import Sequence, Mapping
 
 from .. import exceptions as exc
+from ..pq import Format
 
 
 def query2pg(query, vars, codec):
@@ -149,7 +150,7 @@ def split_query(query, encoding="ascii"):
             part[1] = m.group(1)
 
         # Binary format
-        part[2] = ph[-1:] == b"b"
+        part[2] = Format(ph[-1:] == b"b")
 
         if phtype is None:
             phtype = type(part[1])