UI functionality added to main view

[biaweb_qt.git] / biaweb_db.py

diff --git a/biaweb_db.py b/biaweb_db.py
index 5496453..576f284 100644 (file)
--- a/biaweb_db.py
+++ b/biaweb_db.py
@@ -4,6 +4,7 @@
 import sqlite3
 import os
 import os.path
 import sqlite3
 import os
 import os.path

+import pprint

 
 # function to create a category
 def create_category (dbname, category_name, category_desc, category_stub):
 
 # function to create a category
 def create_category (dbname, category_name, category_desc, category_stub):


@@ -18,6 +19,33 @@ def create_category (dbname, category_name, category_desc, category_stub):
        except sqlite3.Error:
                return False
 
        except sqlite3.Error:
                return False
 

+# Function to get list of articles (either full list or just in a category
+def get_articles (dbname, category_id=None):
+       try:
+               conn = sqlite3.connect (dbname)
+               c = conn.cursor ()
+               if category_id == None:
+                       c.execute ("SELECT * FROM articles;")
+               else:
+                       c.execute ("SELECT * FROM articles WHERE cid=?", (category_id,))
+               conn.commit ()
+               rows = c.fetchall ()
+               return rows
+       except sqlite3.Error:
+               return False
+
+# Function to get list of categories and return a (category_id, category_name) array
+def get_categories (dbname):
+       try:
+               conn = sqlite3.connect (dbname)
+               c = conn.cursor ()
+               c.execute ("SELECT * FROM categories;")
+               conn.commit ()
+               recs = c.fetchall ()
+               conn.close ()
+               return recs
+       except sqlite3.Error:
+               return False

 
 # function to create a new site database
 def create_db (dbname, site_title, site_url, keywords, description, copyright,
 
 # function to create a new site database
 def create_db (dbname, site_title, site_url, keywords, description, copyright,