From dcaa315884537b3ff2ac3cff096b8e33873dfc44 Mon Sep 17 00:00:00 2001
From: Julien <dev@ingeoveritas.com>
Date: Tue, 11 Jan 2022 18:09:24 +0100
Subject: [PATCH] Warn user about dependencies and disable action

---
 roof_classify/roof_classify.py | 34 +++++++++++++++++++++++++++++++---
 1 file changed, 31 insertions(+), 3 deletions(-)

diff --git a/roof_classify/roof_classify.py b/roof_classify/roof_classify.py
index ad72ab5..cd85e4c 100644
--- a/roof_classify/roof_classify.py
+++ b/roof_classify/roof_classify.py
@@ -22,19 +22,25 @@
 
 # standard
 import os.path
+from functools import partial
 from pathlib import Path
 
 # Initialize Qt resources from file resources.py
 from qgis.core import QgsSettings
+from qgis.PyQt.Qt import QUrl
 from qgis.PyQt.QtCore import QCoreApplication, QLocale, QTranslator
-from qgis.PyQt.QtGui import QIcon
+from qgis.PyQt.QtGui import QDesktopServices, QIcon
 from qgis.PyQt.QtWidgets import QAction, QFileDialog
 
 # Import the code for the dialog
-from roof_classify.__about__ import __icon_path__, __title__
+from roof_classify.__about__ import __icon_path__, __title__, __uri_homepage__
 from roof_classify.gui.dlg_settings import PlgOptionsFactory
 from roof_classify.gui.roof_classify_dialog import RoofClassifyDialog
-from roof_classify.logic import DataClassifier
+
+try:
+    from roof_classify.logic import DataClassifier
+except ImportError:
+    DataClassifier = None
 from roof_classify.toolbelt import PlgLogger
 
 
@@ -84,6 +90,25 @@ def __init__(self, iface):
         self.dlg.lineEdit_4.clear()
         self.dlg.pushButton_4.clicked.connect(self.select_output_folder)
 
+    def check_dependencies(self) -> None:
+        """Check if all dependencies are satisfied. If not, warn the user and disable plugin."""
+        # if import failed
+        if DataClassifier is None:
+            self.log(
+                message=self.tr("Error importing Scikit Learn. Plugin disabled."),
+                log_level=2,
+                push=True,
+                button=True,
+                button_connect=partial(
+                    QDesktopServices.openUrl,
+                    QUrl(f"{__uri_homepage__}/usage/installation"),
+                ),
+            )
+            for action in self.actions:
+                action.setEnabled(False)
+        else:
+            self.log(message=self.tr("Dependencies satisfied"), log_level=4)
+
     def tr(self, message):
         """Get the translation for a string using Qt translation API.
 
@@ -183,6 +208,9 @@ def initGui(self):
             parent=self.iface.mainWindow(),
         )
 
+        # -- Post UI initialization
+        self.iface.initializationCompleted.connect(self.check_dependencies)
+
     def unload(self):
         """Removes the plugin menu item and icon from QGIS GUI."""
         for action in self.actions: