init

ui/startup/databaseChecker.py

@@ -0,0 +1,83 @@
+import traceback
+
+from arcaea_offline.database import Database
+from PySide6.QtCore import QDir, QFile, Qt, QTimer, Slot
+from PySide6.QtWidgets import QDialog, QMessageBox
+
+from ui.extends.settings import Settings
+
+from .databaseChecker_ui import Ui_DatabaseChecker
+
+
+class DatabaseChecker(Ui_DatabaseChecker, QDialog):
+    def __init__(self, parent=None):
+        super().__init__(parent)
+        self.setupUi(self)
+        self.setWindowFlag(Qt.WindowType.WindowMinimizeButtonHint, False)
+        self.setWindowFlag(Qt.WindowType.WindowMaximizeButtonHint, False)
+        self.setWindowFlag(Qt.WindowType.WindowCloseButtonHint, True)
+        self.dbFileSelector.setMode(self.dbFileSelector.getExistingDirectory)
+        self.dbFileSelector.filesSelected.connect(self.fileSelected)
+
+        self.settings = Settings(self)
+        dbDir = self.settings.value("Default/DbDir", None, str)
+        if dbDir and QFile(QDir(dbDir).filePath(Database.dbFilename)).exists():
+            self.dbFileSelector.selectFile(dbDir)
+            result = self.checkDbVersion()
+            if result:
+                QTimer.singleShot(50, self.accept)
+        else:
+            self.dbFileSelector.selectFile(QDir.currentPath())
+
+    def fileSelected(self):
+        self.checkDbVersion()
+
+    def checkDbVersion(self) -> str | None:
+        dbQDir = QDir(self.dbFileSelector.selectedFiles()[0])
+        dbDir = dbQDir.absolutePath()
+
+        dbDir = self.dbFileSelector.selectedFiles()[0]
+        dbQFile = QFile(QDir(dbDir).filePath(Database.dbFilename))
+        if not dbQFile.exists():
+            result = QMessageBox.question(self, "Database", "Create database file now?")
+            if result != QMessageBox.StandardButton.Yes:
+                return
+            dbQFile.open(QFile.OpenModeFlag.WriteOnly)
+            dbQFile.close()
+
+        Database.dbDir = dbDir
+
+        try:
+            with Database().conn as conn:
+                version = conn.execute(
+                    "SELECT value FROM properties WHERE key = 'db_version'"
+                ).fetchone()[0]
+                self.dbVersionLabel.setText(version)
+                self.continueButton.setEnabled(True)
+
+                self.dbCheckConnLabel.setText('<font color="green">OK</font>')
+                self.settings.setValue("Default/DbDir", dbDir)
+            return version
+        except Exception as e:
+            QMessageBox.critical(
+                self, "Database Error", "\n".join(traceback.format_exception(e))
+            )
+            self.dbInitButton.setEnabled(True)
+            self.continueButton.setEnabled(False)
+            self.dbCheckConnLabel.setText('<font color="red">Error</font>')
+            return False
+
+    @Slot()
+    def on_dbInitButton_clicked(self):
+        try:
+            Database().init()
+        except Exception as e:
+            QMessageBox.critical(
+                self, "Database Error", "\n".join(traceback.format_exception(e))
+            )
+        finally:
+            self.checkDbVersion()
+
+    @Slot()
+    def on_continueButton_clicked(self):
+        self.accept()