diff options
Diffstat (limited to 'python/db.py')
| -rw-r--r-- | python/db.py | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/python/db.py b/python/db.py index 81f1cc4e..8ede3756 100644 --- a/python/db.py +++ b/python/db.py @@ -1,7 +1,7 @@ import datetime import os import psycopg2 -from psycopg2.extras import DictCursor +from psycopg2.extras import DictCursor, NamedTupleCursor from psycopg2 import IntegrityError, DataError from psycopg2.extensions import DateFromPy, register_adapter, AsIs from psycopg2.pool import ThreadedConnectionPool @@ -33,7 +33,7 @@ register_adapter(datetime.date, InfDateAdapter) register_adapter(np.int64, lambda x: AsIs(x)) register_adapter(np.float, nan_to_null) -def dbconn(dbname, cursor_factory=DictCursor): +def dbconn(dbname, cursor_factory=NamedTupleCursor): if dbname == 'etdb': dbname = 'ET' user_name = 'et_user' @@ -45,7 +45,7 @@ def dbconn(dbname, cursor_factory=DictCursor): cursor_factory=cursor_factory, options="-c extra_float_digits=3") -def dbengine(dbname, cursor_factory=DictCursor): +def dbengine(dbname, cursor_factory=NamedTupleCursor): if dbname in ['rmbs_model', 'corelogic']: uri = URL(drivername="mysql+mysqlconnector", host="debian", database=dbname, |