person + database improved

Implementace/controllers/base_controller.py

-
-
-class BaseController:
-    def __init__(self, db):
-        self.db = db

Implementace/controllers/mainwindow_controller.py

-from controllers.base_controller import BaseController
 from windows import mainwindow
 from PyQt5 import QtWidgets as Qt
-class Mainwindow(Qt.QMainWindow, mainwindow.Ui_MainWindow, BaseController):
+class Mainwindow(Qt.QMainWindow, mainwindow.Ui_MainWindow):
     def __init__(self, db, parent=None):
         super(Mainwindow, self).__init__(parent)
-        BaseController.__init__(self, db)
+        self.db = db
         self.setupUi(self)
         print(self.db.get_groups())
         for group in self.db.get_groups():

Implementace/database.py

 import sqlite3 as sqlite
 from model.group import Group
+from model.person import Person
 class Database:
@@ -9,10 +10,21 @@ class Database:
     def get_groups(self, person_id = None):
         if person_id is not None:
-            curs = self.connection.execute("SELECT * FROM GROUPS WHERE person_id=?", person_id)
+            curs = self.connection.execute("SELECT * FROM GROUPS WHERE person_id = ?", person_id)
         else:
             curs = self.connection.execute("SELECT * FROM GROUPS")
         res = []
         for row in curs:
             res.append(Group(row))
-        return res
+        return res
\ No newline at end of file
+
+    def get_group_members(self, group_id):
+        curs = self.connection.execute('''SELECT *
+                                        FROM person
+                                            JOIN person_to_group
+                                            JOIN groups
+                                        WHERE groups.group_id = ?''', group_id)
+        res = []
+        for row in curs:
+            print(row)
+            #res.append(Person())
\ No newline at end of file

Implementace/model/person.py

+
+
+class Person:
+    def __init__(self, id, name, surname):
+        self.id = id
+        self.name = name
+        self.surname = surname
\ No newline at end of file