diff options
Diffstat (limited to 'db')
| -rw-r--r-- | db/__init__.py | 9 |
1 files changed, 9 insertions, 0 deletions
diff --git a/db/__init__.py b/db/__init__.py index 92837a2..66566d5 100644 --- a/db/__init__.py +++ b/db/__init__.py @@ -13,3 +13,12 @@ def execute(cursor, sql, *values): def query(cursor, sql, *values): execute(cursor, sql, *values) return cursor.fetchall() + +def get(cursor, sql, *values): + execute(cursor, sql, *values) + result = cursor.fetchone() + if result is None: + raise Exception('no rows returned for query: {} with values {}'.format(sql, values)) + if cursor.fetchone() is not None: + raise Exception('multiple results returned for query'.format(sql, values)) + return result |