diff options
Diffstat (limited to 'python')
| -rw-r--r-- | python/db.py | 9 | ||||
| -rw-r--r-- | python/intex_scenarios.py | 2 |
2 files changed, 7 insertions, 4 deletions
diff --git a/python/db.py b/python/db.py index 1fe42b85..a47c44c6 100644 --- a/python/db.py +++ b/python/db.py @@ -28,7 +28,10 @@ def query_db(conn, sqlstr, params=None, one=True): else: c.execute(sqlstr) if one: - return c.fetchone() + r = c.fetchone() + c.close() + return r else: - return c.fetchall() - c.close() + r = c.fetchall() + c.close() + return r diff --git a/python/intex_scenarios.py b/python/intex_scenarios.py index 58516799..1b80efff 100644 --- a/python/intex_scenarios.py +++ b/python/intex_scenarios.py @@ -7,7 +7,7 @@ import json import re
import psycopg2
from common import root
-from db import conn, query_db, with_connection
+from db import query_db, with_connection
import sys
import yaml
|