diff options
Diffstat (limited to 'python/common.py')
| -rw-r--r-- | python/common.py | 19 |
1 files changed, 10 insertions, 9 deletions
diff --git a/python/common.py b/python/common.py index 4124b669..07d44a29 100644 --- a/python/common.py +++ b/python/common.py @@ -16,15 +16,16 @@ def query_db(conn, sqlstr, **kwargs): params = kwargs.get('params', None) one = kwargs.get('one', True) try: - with conn.cursor() as c: - if params: - c.execute(sqlstr, params) - else: - c.execute(sqlstr) - if one: - return c.fetchone() - else: - return c.fetchall() + c = conn.cursor() + if params: + c.execute(sqlstr, params) + else: + c.execute(sqlstr) + if one: + return c.fetchone() + else: + return c.fetchall() + c.close() except psycopg2.Error: c.close() conn.rollback() |
