Converting between sqlite and functional_sqlite backends

Author: mrvisscher

activity_browser/actions/database/database_duplicate.py

@@ -1,11 +1,17 @@
+import copy
+
 from qtpy import QtWidgets
 
+import bw2data as bd
+import bw_functional as bf
+
 from activity_browser import application
 from activity_browser.actions.base import ABAction, exception_dialogs
-from activity_browser.mod import bw2data as bd
 from activity_browser.ui.icons import qicons
 from activity_browser.ui.threading import ABThread
 
+from .database_new import NewDatabaseDialog
+
 
 class DatabaseDuplicate(ABAction):
     """
@@ -23,28 +29,34 @@ class DatabaseDuplicate(ABAction):
     @exception_dialogs
     def run(db_name: str):
         assert db_name in bd.databases
+        backend = bd.databases[db_name].get("backend", "undefined")
 
-        new_name, ok = QtWidgets.QInputDialog.getText(
-            application.main_window,
-            f"Copy {db_name}",
-            "Name of new database:" + " " * 25,
-        )
-        if not new_name or not ok:
+        if backend not in ["sqlite", "functional_sqlite"]:
+            QtWidgets.QMessageBox.information(
+                application.main_window,
+                "Not possible",
+                f"Unsupported database backend {backend}",
+            )
             return
 
-        if new_name in bd.databases:
+        name, backend, ok = NewDatabaseDialog.get_new_database_data(window_title="Duplicate database", backend=backend)
+
+        if not name or not ok:
+            return
+
+        if name in bd.databases:
             QtWidgets.QMessageBox.information(
                 application.main_window,
                 "Not possible",
                 "A database with this name already exists.",
             )
             return
 
-        DuplicateDatabaseDialog(db_name, new_name, application.main_window)
+        DuplicateDatabaseDialog(db_name, name, backend, application.main_window)
 
 
 class DuplicateDatabaseDialog(QtWidgets.QProgressDialog):
-    def __init__(self, from_db: str, to_db: str, parent=None):
+    def __init__(self, from_db: str, to_db: str, backend: str, parent=None):
         super().__init__(parent=parent)
         self.setWindowTitle("Duplicating database")
         self.setLabelText(
@@ -53,12 +65,12 @@ def __init__(self, from_db: str, to_db: str, parent=None):
         self.setModal(True)
         self.setRange(0, 0)
 
-        self.dup_thread = DuplicateDatabaseThread(from_db, to_db, self)
+        self.dup_thread = DuplicateDatabaseThread(application)
         self.dup_thread.finished.connect(self.thread_finished)
 
         self.show()
 
-        self.dup_thread.start()
+        self.dup_thread.start(from_db, to_db, backend)
 
     def thread_finished(self) -> None:
         self.dup_thread.exit(0)
@@ -67,11 +79,24 @@ def thread_finished(self) -> None:
 
 
 class DuplicateDatabaseThread(ABThread):
-    def __init__(self, from_db, to_db, parent=None):
-        super().__init__(parent=parent)
-        self.copy_from = from_db
-        self.copy_to = to_db
 
-    def run_safely(self):
-        database = bd.Database(self.copy_from)
-        database.copy(self.copy_to)
+    def run_safely(self, copy_from, copy_to, backend):
+        database = bd.Database(copy_from)
+
+        data = database.load()
+        data = database.relabel_data(data, copy_from, copy_to)
+
+        new_database = bd.Database(copy_to, backend=backend)
+
+        metadata = copy.copy(database.metadata)
+        metadata["format"] = f"Copied from '{copy_from}'"
+        metadata["backend"] = backend
+        new_database.register(**metadata)
+
+        if database.backend == "sqlite" and backend == "functional_sqlite":
+            data = bf.convert_sqlite_to_functional_sqlite(data)
+        elif database.backend == "functional_sqlite" and backend == "sqlite":
+            data = bf.convert_functional_sqlite_to_sqlite(data)
+
+        new_database.write(data, searchable=metadata.get("searchable"))
+        return new_database

activity_browser/actions/database/database_new.py

@@ -58,9 +58,9 @@ class NewDatabaseDialog(QtWidgets.QDialog):
     A dialog for creating a new database.
     """
 
-    def __init__(self, parent=None):
+    def __init__(self, window_title="New Database", backend="functional_sqlite", parent=None):
         super().__init__(parent)
-        self.setWindowTitle("New Database")
+        self.setWindowTitle(window_title)
         self.setModal(True)
 
         self.name_input = QtWidgets.QLineEdit(self)
@@ -69,6 +69,7 @@ def __init__(self, parent=None):
 
         self.backend_dropdown = QtWidgets.QComboBox(self)
         self.backend_dropdown.addItems(["functional_sqlite", "sqlite"])
+        self.backend_dropdown.setCurrentText(backend)
 
         self.create_button = QtWidgets.QPushButton("Create", self)
         self.create_button.setDisabled(True)
@@ -77,7 +78,7 @@ def __init__(self, parent=None):
         self.build_layout()
 
     @classmethod
-    def get_new_database_data(cls) -> tuple[str, str, bool]:
+    def get_new_database_data(cls, window_title="New Database", backend="functional_sqlite") -> tuple[str, str, bool]:
         """
         Opens a dialog to collect data for creating a new database.
 
@@ -87,7 +88,7 @@ def get_new_database_data(cls) -> tuple[str, str, bool]:
                 - The selected backend type (str).
                 - A boolean indicating whether the dialog was accepted (True) or canceled (False).
         """
-        dialog = cls(application.main_window)
+        dialog = cls(window_title, backend, application.main_window)
         result = dialog.exec_()
 
         return dialog.name_input.text(), dialog.backend_dropdown.currentText(), result == QtWidgets.QDialog.Accepted

pyproject.toml

@@ -42,7 +42,7 @@ dependencies = [
     "bw_simapro_csv >=0.2.6",
     "ecoinvent_interface",
     "matrix_utils>=0.5",
-    "bw-functional==0b82",
+    "bw-functional==0b84",
     "networkx",
     "numpy>=1.23.5,<2",
     "pandas>=2.2.1",

recipe/meta.yaml

@@ -27,7 +27,7 @@ requirements:
     - bw_graph_tools >=0.5
     - bw_processing >=1.0
     - bw_simapro_csv >=0.2.6
-    - bw_functional=0.b.82
+    - bw_functional=0.b.84
     - ecoinvent_interface
     - matrix_utils >=0.5
     - numpy >=1.23.5, <2