X-Git-Url: https://harishankar.org/repos/?p=biaweb_qt.git;a=blobdiff_plain;f=biaweb_db.py;fp=biaweb_db.py;h=5496453c19904f8793046a377fff069f98925626;hp=9be184dd2d900dc103255e3633e0ec43a2450efe;hb=35c21d64e5d7554d2c610599a87c7f236b83d992;hpb=51a6ece9a63764e7a57230081d299f36d3186918

diff --git a/biaweb_db.py b/biaweb_db.py
index 9be184d..5496453 100644
--- a/biaweb_db.py
+++ b/biaweb_db.py
@@ -2,9 +2,32 @@
 # Database handling functions
 
 import sqlite3
+import os
+import os.path
 
+# function to create a category
+def create_category (dbname, category_name, category_desc, category_stub):
+	try:
+		conn = sqlite3.connect (dbname)
+		c = conn.cursor ()
+		c.execute ("INSERT INTO categories (name, desc, stub) VALUES (?, ?, ?);",
+							(category_name, category_desc, category_stub))
+		conn.commit ()
+		conn.close ()
+		return True
+	except sqlite3.Error:
+		return False
+
+
+# function to create a new site database
 def create_db (dbname, site_title, site_url, keywords, description, copyright,
 					num_rss, dest_path):
+	try:
+		if os.path.exists (dbname):
+			os.remove (dbname)
+	except OSError:
+		return False
+
 	try:
 		conn = sqlite3.connect (dbname)
 		c = conn.cursor ()