From c7056a2e0d0a61b65199d8d0b5e7b79fb51c3de7 Mon Sep 17 00:00:00 2001
From: raylu <raylu@gridium.com>
Date: Tue, 22 Oct 2013 18:18:41 -0700
Subject: switch importer to requests, calculate kill costs

---
 db/__init__.py | 7 +++++--
 1 file changed, 5 insertions(+), 2 deletions(-)

(limited to 'db')

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
-- 
cgit v1.2.3