From e5d5b6bd7d83af6a78af59d662aa341924319cf3 Mon Sep 17 00:00:00 2001
From: John Ladan <jladan@uwaterloo.ca>
Date: Thu, 22 Mar 2012 16:26:57 -0400
Subject: [PATCH] more dbLayer category additions

---
 dbLayer.py | 26 ++++++++++++++++++++++++++
 1 file changed, 26 insertions(+)

diff --git a/dbLayer.py b/dbLayer.py
index a538ecf..0205247 100644
--- a/dbLayer.py
+++ b/dbLayer.py
@@ -4,6 +4,7 @@ dbFile = 'sqLibrary.db'
 bookTable = 'books'
 bookRemovedTable='books_deleted'
 bookCategoryTable='book_categories'
+categoryTable = 'categories'
 
 bookTableCreation = '''
 CREATE TABLE IF NOT EXISTS books
@@ -151,6 +152,7 @@ def getBookByID(bookid):
     c.close()
     return book
 
+
 # removes book from catalogue
 def removeBook(bookid):
     conn = sqlite3.connect(dbFile)
@@ -171,7 +173,30 @@ def deleteBook(bookid):
     conn.commit()
     c.close()
 
+#########################################
+# Category related functions
+########################################
+def getCategories():
+    conn = sqlite3.connect(dbFile)
+    c = conn.cursor()
+    query = "SELECT category FROM "+categoryTable+";"
+    c.execute(query)
+    cats = []
+    for category in c:
+        cats.append(category[0])
+    c.close()
+    return cats
 
+def addCategory(cat):
+    conn = sqlite3.connect(dbFile)
+    c = conn.cursor()
+    query = "INSERT OR IGNORE INTO "+categoryTable+" (category) VALUES ("+stringify(cat)");"
+    c.execte(query)
+    c.close()
+
+#########################################
+# Database initialization
+#########################################
 def createBooksTable():
     conn = sqlite3.connect(dbFile)
     c = conn.cursor()
@@ -186,5 +211,6 @@ def createTriggers():
     conn.commit()
     c.close()
 
+
 createBooksTable()
 createTriggers()