summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rwxr-xr-xassignment-list.py6
-rw-r--r--db_sqlite.py44
2 files changed, 50 insertions, 0 deletions
diff --git a/assignment-list.py b/assignment-list.py
index b336098..d5d436e 100755
--- a/assignment-list.py
+++ b/assignment-list.py
@@ -8,6 +8,7 @@ from add_group_form import addGroupForm
from edit_group_form import editGroupForm
from add_entry_form import addEntryForm
Globals = __import__("globals")
+DB = __import__("db_sqlite")
class AssignmentList(QMainWindow):
def __init__(self):
@@ -19,6 +20,7 @@ class AssignmentList(QMainWindow):
self.setWindowTitle("Assignment List")
self.createMenu()
self.createToolbar()
+ self.setupDB()
self.displayWidgets()
self.show()
@@ -47,6 +49,10 @@ class AssignmentList(QMainWindow):
tool_bar.addAction(self.add_group_act)
+ def setupDB(self):
+ # TODO should check for existing db instead of blindly creating it
+ DB.CreateTables("/tmp/test.db")
+
def displayWidgets(self):
main_widget = QWidget(self)
self.setCentralWidget(main_widget)
diff --git a/db_sqlite.py b/db_sqlite.py
new file mode 100644
index 0000000..9d47da8
--- /dev/null
+++ b/db_sqlite.py
@@ -0,0 +1,44 @@
+import sys
+from PyQt5.QtSql import QSqlDatabase, QSqlQuery
+Globals = __import__("globals")
+
+def CreateTables(db_path):
+ """
+ Create database at a specified directory
+ """
+ database = QSqlDatabase.addDatabase("QSQLITE") # SQlite version 3
+ database.setDatabaseName(db_path)
+
+ if not database.open():
+ print("Unable to open data source file.")
+ sys.exit(1) # Error out. TODO consider throwing a dialog instead
+
+ query = QSqlQuery()
+ # Erase database contents so that we don't have duplicates
+ query.exec_("DROP TABLE groups")
+ query.exec_("DROP TABLE entries")
+
+ query.exec_("""
+ CREATE TABLE groups (
+ id INTEGER PRIMARY KEY AUTOINCREMENT UNIQUE NOT NULL,
+ name VARCHAR(255) NOT NULL,
+ column TINYINT(1) DEFAULT FALSE,
+ link VARCHAR(255) NOT NULL,
+ hidden TINYINT(1) DEFAULT FALSE
+ )
+ """)
+
+ query.exec_("""
+ CREATE TABLE entries (
+ id INTEGER PRIMARY KEY AUTOINCREMENT UNIQUE NOT NULL,
+ parent_id REFERENCES groups (id),
+ description VARCHAR(255) NOT NULL,
+ due_date TEXT DEFAULT NULL,
+ alt_due_date VARCHAR(255) DEFAULT NULL,
+ link VARCHAR(255) DEFAULT NULL,
+ done TINYINT(1) DEFAULT FALSE,
+ hidden TINYINT(1) DEFAULT FALSE
+ )
+ """)
+
+ database.close()