]> git.ipfire.org Git - thirdparty/psycopg.git/commitdiff
fix: rename QueuedLibpqWriter object
authorDaniele Varrazzo <daniele.varrazzo@gmail.com>
Thu, 10 Aug 2023 00:37:22 +0000 (01:37 +0100)
committerDaniele Varrazzo <daniele.varrazzo@gmail.com>
Thu, 10 Aug 2023 00:42:39 +0000 (01:42 +0100)
It was mistakenly named Driver, not Writer.

psycopg/psycopg/copy.py
tests/test_copy.py

index df2975f6e1498b3163c2701fefb572626e335868..10701e1480ce41df50c0d70103453e655e1c35ff 100644 (file)
@@ -389,7 +389,7 @@ class LibpqWriter(Writer):
             self.cursor._results = [res]
 
 
-class QueuedLibpqDriver(LibpqWriter):
+class QueuedLibpqWriter(LibpqWriter):
     """
     A writer using a buffer to queue data to write to a Postgres database.
 
index 29fad4581b882b4c951fb9fc923382ab92e7bae1..2c21368ae5c7d9915ddcbff0a3149cbb0edf3fbe 100644 (file)
@@ -12,7 +12,7 @@ from psycopg import pq
 from psycopg import sql
 from psycopg import errors as e
 from psycopg.pq import Format
-from psycopg.copy import Copy, LibpqWriter, QueuedLibpqDriver, FileWriter
+from psycopg.copy import Copy, LibpqWriter, QueuedLibpqWriter, FileWriter
 from psycopg.adapt import PyFormat
 from psycopg.types import TypeInfo
 from psycopg.types.hstore import register_hstore
@@ -643,7 +643,7 @@ def test_worker_life(conn, format, buffer):
     cur = conn.cursor()
     ensure_table(cur, sample_tabledef)
     with cur.copy(
-        f"copy copy_in from stdin (format {format.name})", writer=QueuedLibpqDriver(cur)
+        f"copy copy_in from stdin (format {format.name})", writer=QueuedLibpqWriter(cur)
     ) as copy:
         assert not copy.writer._worker
         copy.write(globals()[buffer])
@@ -663,7 +663,7 @@ def test_worker_error_propagated(conn, monkeypatch):
     cur = conn.cursor()
     cur.execute("create temp table wat (a text, b text)")
     with pytest.raises(ZeroDivisionError):
-        with cur.copy("copy wat from stdin", writer=QueuedLibpqDriver(cur)) as copy:
+        with cur.copy("copy wat from stdin", writer=QueuedLibpqWriter(cur)) as copy:
             copy.write("a,b")