database.py

import sqlite3 as sqlite
from model.group import Group
from model.person import Person


class Database:
    def __init__(self):
        self.connection = sqlite.connect('database.db')
        self.connection.isolation_level = None  # so we don't have to use self.connection.commit()

    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)
        else:
            curs = self.connection.execute("SELECT * FROM GROUPS")
        res = []
        for row in curs:
            res.append(Group(row))
        return res

    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())