kontor/gui/kontor.py

"""
PyQT6 GUI for Kontor
"""
import sys
from pathlib import Path

import yaml
from PySide6.QtGui import QAction, QIcon
from PySide6.QtWidgets import QWidget, QVBoxLayout, QMenu, QMessageBox, QTabWidget, QTableView, QHBoxLayout, QCheckBox
from PySide6.QtWidgets import QApplication, QLabel, QMainWindow
from platformdirs import PlatformDirs

from comic_model import ComicTableModel
from dialogs import ExportKontorDialog, ImportKontorDialog
from data import KontorDB
from model_config import KontorModelConfig
from table_model import KontorTableModel
from media_file_model import MediaFileTableModel


class MainWindow(QMainWindow):

    def __init__(self, config):
        super().__init__()

        self.tick = QIcon('res/tick.png')
        self.cross = QIcon('res/cross.png')
        self.import_icon = QIcon("res/application-import.png")
        self.export_icon = QIcon("res/application-export.png")
        self.circle_icon = QIcon("res/arrow-circle-double.png")

        self.setWindowTitle("Kontor")
        self.setMinimumSize(800, 500)
        self._create_actions()
        self._create_menubar()
        self._create_toolbars()
        self._create_statusbar()

        self.data = []
        self.filter = {}
        self.kontor_db = KontorDB(config)
        self.central_widget = QWidget()
        parent_layout = QVBoxLayout()
        self.central_widget.setLayout(parent_layout)
        self.tabs = QTabWidget()
        self.tabs.addTab(self.generate_data_tab("comic"), "Comics")
        self.tabs.addTab(self.generate_data_tab("media_file"), "MediaFile")
        self.tabs.currentChanged.connect(self._tab_changed)
        #label.setAlignment(Qt.AlignmentFlag.AlignCenter)
        parent_layout.addWidget(self.tabs)

        self.setCentralWidget(self.central_widget)

    def _create_actions(self):
        self.newAction = QAction("&New", self)
        self.aboutAction = QAction("&Über...", self)
        self.aboutAction.triggered.connect(self.about)
        self.importAction = QAction(self.import_icon, "&Import", self)
        self.importAction.triggered.connect(self.import_from_file)
        self.exportAction = QAction(self.export_icon, "&Export", self)
        self.exportAction.triggered.connect(self.export_to_file)
        self.refreshAction = QAction(self.circle_icon, "&Refresh", self)
        self.refreshAction.triggered.connect(self.refresh)
        self.updateTitleAction = QAction("&Update Titles", self)
        self.downloadAction = QAction("&Download Videos", self)
        self.exitAction = QAction("&Beenden", self)
        self.exitAction.setShortcut("Alt+F4")
        self.exitAction.triggered.connect(self.close)

    def _create_menubar(self):
        menu_bar = self.menuBar()
        # File menu
        file_menu = QMenu("&Datei")
        menu_bar.addMenu(file_menu)
        file_menu.addAction(self.exitAction)
        # Kontor menu
        kontor_menu = QMenu("&Kontor")
        menu_bar.addMenu(kontor_menu)
        kontor_menu.addAction(self.importAction)
        kontor_menu.addAction(self.exportAction)
        media_file_menu = QMenu("&MediaFile")
        media_file_menu.addAction(self.updateTitleAction)
        media_file_menu.addAction(self.downloadAction)
        kontor_menu.addMenu(media_file_menu)
        # Help menu
        help_menu = QMenu("&Hilfe")
        menu_bar.addMenu(help_menu)
        help_menu.addAction(self.aboutAction)

    def _create_toolbars(self):
        # Kontor toolbar
        kontor_tool_bar = self.addToolBar("Kontor")
        kontor_tool_bar.addAction(self.importAction)
        kontor_tool_bar.addAction(self.exportAction)
        kontor_tool_bar.addAction(self.refreshAction)

    def _create_statusbar(self):
        self.statusBar = self.statusBar()
        self.statusBar.showMessage("Kontor ready", 6000)
        self.status_label = QLabel("")
        self.statusBar.addPermanentWidget(self.status_label)

    def about(self):
        QMessageBox.about(self.central_widget, "Über Kontor", f"Python: 3.11\nKontor: 0.1.0")

    def import_from_file(self):
        import_dlg = ImportKontorDialog(self)
        if import_dlg.exec():
            print(f"import DB from file {import_dlg.file_name}")
        else:
            print("no nothing for import")
        pass

    def export_to_file(self):
        export_dlg = ExportKontorDialog(self, self.kontor_db)
        if export_dlg.exec():
            print(export_dlg.get_tables_to_export())
            print(f"export DB to {export_dlg.file_name}")
            self.statusBar.showMessage(f"export DB to {export_dlg.file_name}", 3000)
        else:
            self.statusBar.showMessage("Export cancelled", 3000)

    def refresh(self):
        self.data[self.tabs.currentIndex()].refresh()

    def _tab_changed(self, tab_index):
        self.data[tab_index].refresh()

    def generate_data_tab(self, table_name):
        data_tab = QWidget()
        table_config = KontorModelConfig(self.kontor_db, self, table_name)
        model = KontorTableModel(table_config)
        layout = QVBoxLayout()
        self.data.append(model)
        data_tab.setLayout(layout)
        table_view = QTableView()
        table_view.setModel(model)
        layout.addLayout(table_config.get_filter_layout())
        layout.addWidget(table_view)
        model.refresh()
        return data_tab


if __name__ == '__main__':
    app = QApplication(sys.argv)
    dirs = PlatformDirs("kontor")
    database_config = Path(dirs.user_config_dir, 'database-config.yaml')
    with open(database_config, 'rt') as f:
        db_config = yaml.safe_load(f.read())
    window = MainWindow(db_config)
    window.show()
    app.exec()