Ommodulering

assignment.py

@@ -62,7 +62,8 @@ class AssignmentDlg(QDialog):
         deliveredLabel = QLabel(self.trUtf8("De&livered"))
         
         dateEdit = QLineEdit()
-        courseEdit = QLineEdit()
+        courseEdit = QComboBox()
+	courseEdit.addItems(MainWindow.getCoursesString())
         descriptionEdit = QLineEdit()
         availableEdit = QCheckBox()
         begunEdit = QCheckBox()

course.py

@@ -53,3 +53,46 @@ class AddCourseDlg(CourseDlg):
         
         self.setWindowTitle(self.trUtf8("Add new course"))
 
+    def accept(self):
+		courseCode = unicode(self.codeEdit.text())
+		courseTitle = unicode(self.titleEdit.text())
+		courseShort = unicode(self.shortEdit.text())
+		course = CourseModel(courseCode, courseTitle, courseShort)
+		
+
+class CourseModel():
+	
+	code = ""
+	title = ""
+	short = ""
+	full = ""
+
+	def __init__(self, code, title, short):
+		self.code = code
+		self.title = title
+		self.short = short
+		setFull(code, title)
+
+	def setCode(self, code):
+		self.code = code
+
+	def getCode(self):
+		return self.code
+
+	def setTitle(self, title):
+		self.title = title
+	
+	def getTitle(self):
+		return self.title
+
+	def setShort(self, short):
+		self.short = short
+	
+	def getShort(self):
+		return self.short
+
+	def setFull(self, code, title):
+		self.full = code + ' ' + title
+
+	def getFull(self):
+		return self.full

db.py

@@ -16,6 +16,8 @@ from calendar import *
 from book import *
 from course import *
 
+class DB:
+
     def initDB():
         conn = sqlite.connect('egon.db')
         curs = conn.cursor()
@@ -145,6 +147,20 @@ def addNewCourse(code, title, short):
     
         exitDB(conn)
    
+    def getCourses():
+        cursor, conn = initDB()
+
+        courseQuery = '''
+            SELECT *
+            FROM Course
+        '''
+
+        cursor.execute(courseQuery)
+
+        courses = []
+        for row in cursor.fetchall():
+	        courses.append(CourseModel(row[0], row[1], row[2]))
+
     def exitDB(conn):
         conn.commit()
         conn.close()

main.py

@@ -15,17 +15,26 @@ from schedule import *
 from calendar import *
 from book import *
 from course import *
+from db import *
 
 __version__ = "0.0.1"
 
 class MainWindow(QMainWindow):
 
+    courses = []
+    coursesString = []
+    books = []
+    booksString = []
+
     def __init__(self, parent=None):
         super(MainWindow, self).__init__(parent)
         
         # The program name
         self.title = "Egon"
        
+        # The database
+        self.db = DB()
+
         # The tabs
         assignment = AssignmentTab()
         reading = ReadingTab()
@@ -106,6 +115,9 @@ class MainWindow(QMainWindow):
         editToolbar.setObjectName("EditToolBar")
         self.addActions(editToolbar, (editAddCourse, editAddBook, None, editShowCalendar))
         
+        self.courses = self.db.getCourses()
+        makeCoursesString()
+
     def addCourse(self):##, code, title, short):
         self.acdlg = AddCourseDlg()
         self.acdlg.show()
@@ -206,8 +218,25 @@ class MainWindow(QMainWindow):
     def refreshTable(self):
         pass
 
+    def addNewCourse(self, course):
+        self.courses.append(course)
+        db.addNewCourse(course.getCode(), course.getTitle(), course.getShort())
+        self.addNewCourseString(course)
 
-class Model():
+    def getCourses(self):
+        return self.courses
+
+    def addNewCourseString(self, course):
+        self.coursesString.append(course.getFull())
+
+    def makeCoursesString(self, courses):
+        for c in courses:
+            addNewCourseString(course)
+
+    def getCoursesString(self):
+        return self.coursesString
+
+class Model:
     
     def __init__(self, parent=None):
         self.assignmentModel = AssignmentModel()