test.py

import sys
import sqlite3
from PyQt6.QtWidgets import QApplication, QMainWindow, QLabel, QWidget, QVBoxLayout, QGridLayout, QLineEdit, QPushButton, QMessageBox, QListWidget
from PyQt6 import QtGui
from PyQt6.QtGui import QPalette, QColor
import popup
from calculation import calculate_costs
import database

# Maak verbinding met de database
conn = sqlite3.connect('calculatiedb.sqlite')
cursor = conn.cursor()

class general(QWidget):

    def __init__(self, color):
        super(general, self).__init__()
        self.setFixedSize(100,200)

    def create_general_info_section(self):
        widget = QWidget()
        layout = QVBoxLayout(widget)
        label_width = 100  # Vervang dit door de gewenste breedte van de labels
        field_width = 100  # Vervang dit door de gewenste breedte van de velden

        general_info_labels = ["Klant", "Artikelnummer", "Tekeningnummer", "Revisie", "Omschrijving", "Aantal stuks"]

        for label_text in general_info_labels:
            label = QLabel(label_text)
            label.setFixedWidth(100)
            layout.addWidget(label)
            label.setContentsMargins(0,0,0,0)

            entry = QLineEdit()
            layout.addWidget(entry)
            entry.setFixedWidth(100)
            entry.setContentsMargins(0,0,0,0)


            self.general_info_entries[label_text] = entry

        self.setCentralWidget(widget)
        

class MainWindow(QMainWindow):
    def __init__(self):
        super(MainWindow, self).__init__()
        self.setWindowTitle("Calculatieprogramma PSS")
        self.setGeometry(100, 100, 800, 600)
        self.setFixedSize(800,400)
        self.setWindowIcon(QtGui.QIcon("pss.ico"))
        
        self.general_info_entries = {}
        self.bewerkingen_entries = {}

        self.create_menu()
        
        layout = QGridLayout()
        layout.addWidget(general(), 0, 0)  # layout.addWidget(self.create_general_info_section(), 0, 0) # Corrected position (0, 0)
        layout.addWidget(self.create_bewerkingen_section(), 1, 0)  # Corrected position (1, 0)
        layout.addWidget(self.create_buttons(), 2, 0)  # Corrected position (2, 0)
        layout.addWidget(self.create_total_kosten_label(), 2, 1)

        # central_widget = QWidget()
        # central_widget.setLayout(layout)
        # self.setCentralWidget(central_widget)

    def create_menu(self):
        menubar = self.menuBar()

        # Maak het File-menu
        file_menu = menubar.addMenu("File")
        open_action = file_menu.addAction("Open")
        open_action.triggered.connect(self.do_something)
        save_action = file_menu.addAction("Save")
        save_action.triggered.connect(self.save)
        database_action = file_menu.addAction("Database")
        database_action.triggered.connect(self.open_database)
        exit_action = file_menu.addAction("Exit")
        exit_action.triggered.connect(QApplication.quit)

        # Maak het Edit-menu
        edit_menu = menubar.addMenu("Edit")
        edit_menu.addAction("Cut")
        edit_menu.addAction("Copy")
        edit_menu.addAction("Paste")

        # Maak het Help-menu
        help_menu = menubar.addMenu("Help")
        help_menu.addAction("About")

    

    def create_bewerkingen_section(self):
        widget = QWidget()
        layout = QVBoxLayout(widget)
        label_width = 100  # Vervang dit door de gewenste breedte van de labels
        field_width = 100  # Vervang dit door de gewenste breedte van de velden

        bewerkingen = [
            ("Zagen", self.do_something),
            ("Draaien", self.do_something),
            ("Frezen", popup.show_frezen_popup),
            ("Slijpen", self.do_something),
            ("Rondslijpen", self.do_something),
            ("Draadvonken", self.do_something),
            ("Gm/Eng", self.do_something)
        ]
        # for bewerking, entry in self.bewerkingen_entries.items():
        #     label = QPushButton(bewerking)
        #     layout.addWidget(label)

        for bewerking, command in bewerkingen:
            label = QPushButton(bewerking)
            label.clicked.connect(command)
            label.setFixedWidth(label_width)
            layout.addWidget(label)

            if bewerking == "Frezen":
                entry_frezen = QLineEdit()
                entry_frezen.setFixedWidth(field_width)
                layout.addWidget(entry_frezen)
                self.bewerkingen_entries[bewerking] = entry_frezen
            else:
                entry = QLineEdit()
                entry.setFixedWidth(field_width)
                layout.addWidget(entry)
                self.bewerkingen_entries[bewerking] = entry

        self.setCentralWidget(widget)

    def create_buttons(self):
        button_width = 100  # Vervang dit door de gewenste breedte van de labels
        calculate_button = QPushButton("Calculate")
        calculate_button.clicked.connect(self.calculate_hours)
        self.centralWidget().layout().addWidget(calculate_button)
        calculate_button.setFixedWidth(button_width)

        save_button = QPushButton("Save")
        save_button.clicked.connect(self.save)
        self.centralWidget().layout().addWidget(save_button)
        save_button.setFixedWidth(button_width)

    def create_total_kosten_label(self):
        self.totaal_kosten_label = QLabel("Totaal kosten:")
        self.centralWidget().layout().addWidget(self.totaal_kosten_label)

    def do_something(self):
        print("Do something")

    def save(self):
        # Roep de calculate_hours-functie aan om de totale kosten te berekenen
        self.calculate_hours()

        # Haal de totale kosten op
        totaal_kosten = float(self.totaal_kosten_label.text().replace("Totaal kosten: €", "").strip())

        # Controleer of er een regel is geselecteerd
        if database_listbox.selectedItems():
            # Haal de geselecteerde regel uit de lijst
            selected_item = database_listbox.selectedItems()[0]

            # Update de geselecteerde regel in de database
            query = "UPDATE calculatie SET klant=?, artikelnummer=?, tekeningnummer=?, revisie=?, omschrijving=?, stuks=?, prijs=? WHERE rowid=?"
            values = (
                self.general_info_entries["Klant"].text(),
                self.general_info_entries["Artikelnummer"].text(),
                self.general_info_entries["Tekeningnummer"].text(),
                self.general_info_entries["Revisie"].text(),
                self.general_info_entries["Omschrijving"].text(),
                self.general_info_entries["Aantal stuks"].text(),
                totaal_kosten,
                selected_item.rowid()
            )
            cursor.execute(query, values)
            conn.commit()

            QMessageBox.information(self, "Bijwerken", "De gegevens zijn succesvol bijgewerkt in de database!")
            print("Gegevens bijgewerkt in de database.")
        else:
            # Opslaan van de gegevens in de database
            query = "INSERT INTO calculatie (klant, artikelnummer, tekeningnummer, revisie, omschrijving, stuks, prijs) VALUES (?, ?, ?, ?, ?, ?, ?)"
            values = (
                self.general_info_entries["Klant"].text(),
                self.general_info_entries["Artikelnummer"].text(),
                self.general_info_entries["Tekeningnummer"].text(),
                self.general_info_entries["Revisie"].text(),
                self.general_info_entries["Omschrijving"].text(),
                self.general_info_entries["Aantal stuks"].text(),
                totaal_kosten
            )
            cursor.execute(query, values)
            conn.commit()

            QMessageBox.information(self, "Opslaan", "De gegevens zijn succesvol opgeslagen!")
            print("Gegevens opgeslagen in de database.")

    def calculate_hours(self):
        bewerkingsuren = {}
        for bewerking, entry in self.bewerkingen_entries.items():
            uren = entry.text()
            if uren:
                bewerkingsuren[bewerking] = float(uren)
        uurtarief = 97.5
        totaal_uren = sum(bewerkingsuren.values())
        totaal_kosten = totaal_uren * uurtarief

        # Toon de totale kosten in een label
        self.totaal_kosten_label.setText("Totaal kosten: € {:.2f}".format(totaal_kosten))

        calculate_costs(bewerkingsuren, uurtarief)

    def open_database(self):
        database.open_database_window(self, self.general_info_entries)


app = QApplication(sys.argv)
window = MainWindow()
window.show()
# Sluit de databaseverbinding
conn.close()
sys.exit(app.exec())