Merge pull request #12 from AndreWohnsland/dev

Add proper plot alignement, migrate to qt6

Author: AndreWohnsland

pyproject.toml

@@ -12,20 +12,21 @@ python = ">=3.10,<3.12"
 GitPython = "^3.1.43"
 matplotlib = "^3.9.0"
 pandas = "^2.2.2"
-PyQt5 = "5.15.7"
 pyqt5-qt5 = "5.15.2"
 xlsxwriter = "^3.2.0"
 pyqtdarktheme = "^2.1.0"
 qtawesome = "^1.3.1"
 holidays = "^0.47"
 dataclasses-json = "^0.6.4"
+pyqt6 = "^6.4.2"
 
 [tool.poetry.group.dev.dependencies]
 pyinstaller = "^6.8.0"
 pip-chill = "^1.0.3"
 jupyterlab = "^4.2.2"
 mypy = "^1.10.0"
 ruff = "^0.4.8"
+pyqt6-tools = "^6.4.2.3.3"
 
 [build-system]
 requires = ["poetry-core"]

requirements.txt

@@ -1,7 +1,7 @@
 gitpython
 matplotlib
 pandas
-pyqt5
+pyqt6
 xlsxwriter
 pyqtdarktheme
 qtawesome

runme.py

@@ -6,7 +6,7 @@
 import logging
 import sys
 
-from PyQt5.QtWidgets import QApplication
+from PyQt6.QtWidgets import QApplication
 
 from src.ui_mainwindow import MainWindow
 from src.utils import get_additional_run_args, sync_theme
@@ -19,12 +19,12 @@
         app = QApplication(sys.argv + get_additional_run_args())
         w = MainWindow()
         # in case of active style change, also change theme, need to sync at start
-        app.paletteChanged.connect(sync_theme)
+        # app.paletteChanged.connect(sync_theme)
         # still keep the app running, even if the main window is closed (we use tray for the app)
         QApplication.setQuitOnLastWindowClosed(False)
         sync_theme()
         w.show()
-        sys.exit(app.exec_())
+        sys.exit(app.exec())
     except Exception as e:
         logger.exception(e)
         raise

scripts/compile_ui_to_python.ps1

@@ -6,7 +6,7 @@ $files = @(
 )
 
 foreach ($f in $files) {
-  pyuic5 -x .\$f.ui -o .\$f.py
+  pyuic6 -x .\$f.ui -o .\$f.py
 }
 
 cd ..

scripts/plotting.ipynb

@@ -17,8 +17,11 @@
     "country": "US",
     "subdiv": None,
     "workdays": [0, 1, 2, 3, 4],  # 0-6, 0=Monday, 6=Sunday
+    "plot_pause": True,
 }
-CONFIG_NAMES = Literal["name", "save_path", "country", "subdiv", "daily_hours", "weekly_hours", "workdays"]
+CONFIG_NAMES = Literal[
+    "name", "save_path", "country", "subdiv", "daily_hours", "weekly_hours", "workdays", "plot_pause"
+]
 
 
 @dataclass
@@ -31,6 +34,7 @@ class Config:
     country: str
     subdiv: str | None
     workdays: list[int]
+    plot_pause: bool
 
     @classmethod
     def from_kwargs(cls, **kwargs):

src/config_handler.py

@@ -104,7 +104,7 @@ def _generate_monthly_time(
         daily_minutes = CONFIG_HANDLER.config.daily_hours * 60
         for _day in full_month:
             days_data = df[df["date"] == _day.date()]
-            calculated_time = 1.0
+            calculated_time = 0.0
             # when we got a free day, we get the working time for this day
             if _day.date() in free_days:
                 calculated_time += daily_minutes

src/datastore.py

@@ -1,7 +1,7 @@
 from dataclasses import dataclass
 
 import qtawesome as qta
-from PyQt5.QtGui import QIcon
+from PyQt6.QtGui import QIcon
 
 from src.utils import get_background_color, get_font_color
 

src/icons.py

@@ -3,8 +3,8 @@
 from typing import TYPE_CHECKING
 
 import holidays
-from PyQt5.QtCore import Qt
-from PyQt5.QtWidgets import QWidget
+from PyQt6.QtCore import Qt
+from PyQt6.QtWidgets import QWidget
 
 from src.config_handler import CONFIG_HANDLER
 from src.icons import get_app_icon
@@ -21,8 +21,13 @@ def __init__(self, main_window: MainWindow):
         self.main_window = main_window
         self.setupUi(self)
         self.setWindowIcon(get_app_icon())
-        self.setWindowFlags(Qt.Window | Qt.CustomizeWindowHint | Qt.WindowTitleHint | Qt.WindowCloseButtonHint)  # type: ignore
-        self.setAttribute(Qt.WA_DeleteOnClose)  # type: ignore
+        self.setWindowFlags(
+            Qt.WindowType.Window
+            | Qt.WindowType.CustomizeWindowHint
+            | Qt.WindowType.WindowTitleHint
+            | Qt.WindowType.WindowCloseButtonHint
+        )
+        self.setAttribute(Qt.WidgetAttribute.WA_DeleteOnClose)
         self.country_list = holidays.list_supported_countries()
         self._update_country_list()
         self.set_config_values()
@@ -71,6 +76,7 @@ def set_config_values(self):
         self.input_name.setText(CONFIG_HANDLER.config.name)
         self.input_daily_hours.setValue(CONFIG_HANDLER.config.daily_hours)
         self.input_weekly_hours.setValue(CONFIG_HANDLER.config.weekly_hours)
+        self.input_plot_pause.setChecked(CONFIG_HANDLER.config.plot_pause)
         for day in CONFIG_HANDLER.config.workdays:
             getattr(self, f"radio_weekday_{day}").setChecked(True)
 
@@ -81,6 +87,7 @@ def apply_config(self):
         CONFIG_HANDLER.config.name = self.input_name.text()
         CONFIG_HANDLER.config.daily_hours = self.input_daily_hours.value()
         CONFIG_HANDLER.config.weekly_hours = self.input_daily_hours.value()
+        CONFIG_HANDLER.config.plot_pause = self.input_plot_pause.isChecked()
         selected_days: list[int] = []
         for day in range(7):
             if getattr(self, f"radio_weekday_{day}").isChecked():

src/ui_config_window.py

@@ -1,7 +1,7 @@
-from PyQt5.QtWidgets import QDialog, QFileDialog, QInputDialog, QLayout, QMessageBox, QTableWidget, QTableWidgetItem
+from PyQt6.QtWidgets import QDialog, QFileDialog, QInputDialog, QLayout, QMessageBox, QTableWidget, QTableWidgetItem
 
 from src import __version__
-from src.config_handler import CONFIG_HANDLER, CONFIG_NAMES
+from src.config_handler import CONFIG_HANDLER
 from src.filepath import HOME_PATH
 from src.icons import get_app_icon
 
@@ -13,12 +13,12 @@ def __init__(self):
     def show_message(self, message: str):
         """Prompt default messagebox, use a QMessageBox with OK-Button."""
         message_box = QMessageBox()
-        message_box.setStandardButtons(QMessageBox.Ok)  # type: ignore
+        message_box.setStandardButtons(QMessageBox.StandardButton.Ok)
         message_box.setText(str(message))
         message_box.setWindowIcon(get_app_icon())
         message_box.setWindowTitle("Information")
         message_box.show()
-        message_box.exec_()
+        message_box.exec()
 
     def report_choice(self):
         message_box = QMessageBox()
@@ -27,11 +27,11 @@ def report_choice(self):
         )
         message_box.setWindowTitle("Report Generation")
         message_box.setWindowIcon(get_app_icon())
-        overtime_button = message_box.addButton("Overtime", QMessageBox.YesRole)  # type: ignore
-        time_button = message_box.addButton("Time", QMessageBox.NoRole)  # type: ignore
-        message_box.addButton("Cancel", QMessageBox.RejectRole)  # type: ignore
+        overtime_button = message_box.addButton("Overtime", QMessageBox.ButtonRole.YesRole)
+        time_button = message_box.addButton("Time", QMessageBox.ButtonRole.NoRole)
+        message_box.addButton("Cancel", QMessageBox.ButtonRole.RejectRole)
 
-        message_box.exec_()
+        message_box.exec()
         if message_box.clickedButton() == overtime_button:
             return True
         if message_box.clickedButton() == time_button:
@@ -43,9 +43,9 @@ def user_okay(self, text: str):
         message_box.setText(text)
         message_box.setWindowTitle("Confirmation required")
         message_box.setWindowIcon(get_app_icon())
-        yes_button = message_box.addButton("Yes", QMessageBox.YesRole)  # type: ignore
-        message_box.addButton("No", QMessageBox.NoRole)  # type: ignore
-        message_box.exec_()
+        yes_button = message_box.addButton("Yes", QMessageBox.ButtonRole.YesRole)
+        message_box.addButton("No", QMessageBox.ButtonRole.NoRole)
+        message_box.exec()
         if message_box.clickedButton() == yes_button:
             return True
         return False
@@ -66,11 +66,11 @@ def get_folder(self, current_path: str, parent=None):
             current_path = str(HOME_PATH)
 
         dialog = QFileDialog(parent)
-        dialog.setFileMode(QFileDialog.DirectoryOnly)  # type: ignore
+        dialog.setFileMode(QFileDialog.FileMode.Directory)
         dialog.setDirectory(current_path)
 
-        if dialog.exec_() == QDialog.Accepted:  # type: ignore
-            return dialog.selectedFiles()[0]  # returns a list
+        if dialog.exec() == QDialog.DialogCode.Accepted:
+            return dialog.selectedFiles()[0]
         return ""
 
     def fill_table(self, table: QTableWidget, entry):
@@ -91,17 +91,6 @@ def set_header_names(self, table: QTableWidget, name1: str, name2: str):
         if header_2 is not None:
             header_2.setText(name2)
 
-    def get_user_data(self, parent):
-        needed_keys: list[CONFIG_NAMES] = ["name"]
-        # todo adjust to new class logic
-        for data in needed_keys:
-            text, ok = self.get_text(data, parent)
-            if not ok:
-                return
-            if text != "":
-                CONFIG_HANDLER.set_config_value(data, text, write=False)
-        CONFIG_HANDLER.write_config_file()
-
     def get_save_folder(self):
         user_path = CONFIG_HANDLER.config.save_path
         returned_path = self.get_folder(user_path)