diff options
Diffstat (limited to 'python/db.py')
| -rw-r--r-- | python/db.py | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/python/db.py b/python/db.py index cc73b5ac..f04cfdc4 100644 --- a/python/db.py +++ b/python/db.py @@ -45,7 +45,7 @@ def dbconn(dbname, cursor_factory=DictCursor): cursor_factory=cursor_factory, options="-c extra_float_digits=3") -def dbengine(dbname): +def dbengine(dbname, cursor_factory=DictCursor): if dbname in ['rmbs_model', 'corelogic']: uri = URL(drivername="mysql+mysqlconnector", host="debian", database=dbname, @@ -61,7 +61,8 @@ def dbengine(dbname): username=user_name, database=dbname, query={"options": "-c extra_float_digits=3"}) - return create_engine(uri, paramstyle="format") + return create_engine(uri, paramstyle="format", + connect_args={'cursor_factory': cursor_factory}) def with_connection(dbname): def decorator(f): |