diff options
Diffstat (limited to 'db/__init__.py')
| -rw-r--r-- | db/__init__.py | 7 |
1 files changed, 5 insertions, 2 deletions
diff --git a/db/__init__.py b/db/__init__.py index 66566d5..3173d58 100644 --- a/db/__init__.py +++ b/db/__init__.py @@ -14,11 +14,14 @@ def query(cursor, sql, *values): execute(cursor, sql, *values) return cursor.fetchall() +class NoRowsException(Exception): pass +class MultipleRowsException(Exception): pass + 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)) + raise NoRowsException('no rows returned', sql, values) if cursor.fetchone() is not None: - raise Exception('multiple results returned for query'.format(sql, values)) + raise MultipleRowsException('multiple results returned', sql, values) return result |