database.py

import sqlite3

class Database:
    def __init__(self, db_name="arztbriefe.db"):
        self.conn = sqlite3.connect(db_name)
        self.cursor = self.conn.cursor()
        self._create_table()

    def _create_table(self):
        self.cursor.execute('''
        CREATE TABLE IF NOT EXISTS arztbriefe (
            id INTEGER PRIMARY KEY AUTOINCREMENT,
            diagnose TEXT NOT NULL,
            text TEXT NOT NULL,
            therapie TEXT NOT NULL,
            massnahmen TEXT NOT NULL
        )
        ''')
        self.conn.commit()

    def insert_entry(self, diagnose, text, therapie, massnahmen):
        self.cursor.execute(
            "INSERT INTO arztbriefe (diagnose, text, therapie, massnahmen) VALUES (?, ?, ?)",
            (diagnose, text, therapie, massnahmen)
        )
        self.conn.commit()

    def search_by_diagnoses(self, diagnoses):
        # self.cursor.execute("SELECT therapie, massnahmen FROM arztbriefe WHERE diagnose LIKE ?", (f"%{diagnose}%",))
        #return self.cursor.fetchall()

        results = {}
        for diagnosis in diagnoses:
            query = "SELECT therapie, massnahmen FROM arztbriefe WHERE diagnose = ?"
            self.cursor.execute(query, (diagnosis,))
            entries = self.cursor.fetchall()
            if entries:
                results[diagnosis] = entries
        return results


    def get_all_diagnoses(self):
        """
        Fetch all unique diagnoses from the 'diagnose' column in the database.

        Returns:
            list: A list of diagnoses.
        """
        query = "SELECT DISTINCT diagnose FROM arztbriefe"
        self.cursor.execute(query)
        results = self.cursor.fetchall()
        return [row[0] for row in results]  # Extract diagnoses from query result
    
    def get_all_cases(self):
        query = "SELECT id, diagnose, text, therapie, massnahmen FROM arztbriefe"
        self.cursor.execute(query)
        return self.cursor.fetchall()    

    def close(self):
        self.conn.close()