Category creation completed

[biaweb_qt.git] / biaweb_db.py

diff --git a/biaweb_db.py b/biaweb_db.py
index 9be184d..5496453 100644 (file)
--- 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 ()