make arraysize config option

segasai · Sep 15, 2020 · ced7bca · ced7bca
1 parent d7bcc93
commit ced7bca
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/py/sqlutilpy/sqlutil.py b/py/sqlutilpy/sqlutil.py
@@ -27,6 +27,9 @@
 from select import select
 from psycopg2.extensions import POLL_OK, POLL_READ, POLL_WRITE
 
+class config:
+    arraysize = 100000
+
 
 class SqlUtilException(Exception):
     pass
@@ -120,7 +123,7 @@ def getCursor(conn, driver=None, preamb=None, notNamed=False):
         if notNamed:
             return cur
         cur = conn.cursor(name='sqlutilcursor')
-        cur.arraysize = 100000
+        cur.arraysize = config.arraysize
     elif driver == 'sqlite3':
         cur = conn.cursor()
     return cur