]> git.ipfire.org Git - thirdparty/psycopg.git/commitdiff
Added format enum
authorDaniele Varrazzo <daniele.varrazzo@gmail.com>
Fri, 27 Mar 2020 08:13:46 +0000 (21:13 +1300)
committerDaniele Varrazzo <daniele.varrazzo@gmail.com>
Fri, 27 Mar 2020 08:13:46 +0000 (21:13 +1300)
psycopg3/pq/__init__.py
psycopg3/pq/enums.py
psycopg3/utils/queries.py

index b9387cca059547a8ae629ebb784db6106f3905af..98575ae6024534efb2307afd3ba49f1255391c48 100644 (file)
@@ -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",
index a8bbab7af18c0961acc3fa03f70e042093531541..e692fc14ed2b99c0a7824b4c13acd62b12feb9cd 100644 (file)
@@ -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
index 823d37c889e4f7db169e2cdea4dd3e9f369e9244..8cdd8cb3ee3952ae70dc02c10680c506497e2041 100644 (file)
@@ -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])