Migrations fix (#1360)

* Fixed migrations install during project setup

* Added default migration install action

* Added test for migration install action

Author: mrvisscher

activity_browser/actions/__init__.py

@@ -58,3 +58,4 @@
 from .biosphere_update import BiosphereUpdate
 from .plugin_wizard_open import PluginWizardOpen
 from .settings_wizard_open import SettingsWizardOpen
+from .migrations_install import MigrationsInstall

activity_browser/actions/migrations_install.py

@@ -0,0 +1,40 @@
+from PySide2 import QtWidgets
+
+from activity_browser import application
+from activity_browser.actions.base import ABAction, exception_dialogs
+from activity_browser.ui import icons, threading
+from activity_browser.mod.bw2io.migrations import ab_create_core_migrations
+
+
+class MigrationsInstall(ABAction):
+    """
+    ABAction to install the default migrations from bw2io
+    """
+
+    icon = icons.qicons.import_db
+    text = "Install default migrations"
+
+    @staticmethod
+    @exception_dialogs
+    def run():
+        def update_dialog_slot(progress: int, label: str):
+            dialog.setValue(progress)
+            dialog.setLabelText(label)
+
+
+        dialog = QtWidgets.QProgressDialog(application.main_window)
+        dialog.setWindowTitle("Installing migrations")
+        dialog.setMaximum(100)
+        dialog.setCancelButton(None)
+
+        thread = MigrationsInstallThread(application)
+
+        thread.status.connect(update_dialog_slot)
+
+        thread.start()
+        dialog.exec_()
+
+
+class MigrationsInstallThread(threading.ABThread):
+    def run_safely(self):
+        ab_create_core_migrations()

activity_browser/mod/bw2io/ecoinvent.py

@@ -15,8 +15,8 @@ def ab_import_ecoinvent_release(version, system_model):
     from bw2io import migrations
     from .migrations import ab_create_core_migrations
 
-    # if not len(migrations):
-    #     ab_create_core_migrations()
+    if not len(migrations):
+        ab_create_core_migrations()
 
     # downloading a release through a AB version of ecoinvent_interface that implements a progress_slot
     release = ABEcoinventRelease(ei.Settings())

activity_browser/ui/menu_bar.py

@@ -57,6 +57,8 @@ def setup_file_menu(self) -> None:
         self.file_menu.addAction(self.export_db_action)
         self.file_menu.addAction(self.update_biosphere_action)
         self.file_menu.addSeparator()
+        self.file_menu.addMenu(MigrationsMenu(self))
+        self.file_menu.addSeparator()
         self.file_menu.addAction(self.manage_settings_action)
 
     def setup_view_menu(self) -> None:
@@ -121,3 +123,16 @@ def biosphere_exists(self) -> None:
         exists = True if bd.config.biosphere in bd.databases else False
         self.update_biosphere_action.setEnabled(exists)
         self.import_db_action.setEnabled(exists)
+
+
+class MigrationsMenu(QtWidgets.QMenu):
+    def __init__(self, parent=None) -> None:
+        super().__init__(parent)
+
+        self.setTitle("Migrations")
+        self.install_migrations_action = actions.MigrationsInstall.get_QAction()
+
+        self.addAction(self.install_migrations_action)
+
+
+

tests/actions/test_various_actions.py

@@ -1,6 +1,7 @@
 import os
 
 import bw2data as bd
+import bw2io as bi
 import pytest
 from PySide2 import QtWidgets
 
@@ -66,3 +67,17 @@ def test_settings_wizard_open(ab_app):
     assert application.main_window.findChild(SettingsWizard).isVisible()
 
     application.main_window.findChild(SettingsWizard).destroy()
+
+
+def test_migrations_install(ab_app, qtbot):
+    from activity_browser.actions.migrations_install import MigrationsInstallThread
+
+    project_name = "migrations_project"
+    bd.projects.set_current(project_name)
+
+    assert not len(bi.migrations)
+
+    actions.MigrationsInstall.run()
+
+    assert len(bi.migrations)
+