brightway-basic-explorer 0.9.dev3__py3-none-any.whl → 0.9.dev5__py3-none-any.whl

brightway_basic_explorer/__init__.py

@@ -1,8 +1,9 @@
 # coding=utf-8
 
 from IPython.external.qt_for_kernel import QtGui, QtCore
-from IPython.lib.guisupport import start_event_loop_qt4, get_app_qt4, is_event_loop_running_qt4
+from IPython.lib.guisupport import get_app_qt4, is_event_loop_running_qt4
 
+import re
 import os
 import sys
 import bw2data
@@ -26,7 +27,7 @@ def activity_to_json_with_params(act, params):
             _getAmountOrFormula,
         )
 
-        from sympy import Basic, Symbol
+        from sympy import Basic
     except:
         raise Exception("lca_algebraic not found, please install it before using show_activity_with_params")
 
@@ -39,11 +40,10 @@ def activity_to_json_with_params(act, params):
 
         # Params provided ? Evaluate formulas
         if isinstance(amount, Basic):
-            print(amount)
-            new_params = [(name, value) for name, value in _complete_and_expand_params(params, list(all_params().keys())).items()]
-            print(new_params)
-            amount = amount.subs(new_params).evalf()
-            print(amount)
+            new_params = list(_complete_and_expand_params(params, list(all_params().keys())).items())
+            amount = amount.subs(new_params)
+            if amount.is_number:
+                amount = float(amount.evalf())
             de["computed_amount"] = True
 
         de["amount"] = amount
@@ -52,6 +52,20 @@ def activity_to_json_with_params(act, params):
     a["exchanges"] = exs
     return a
 
+class MessageDialog(QtGui.QDialog):
+    def __init__(self, msg):
+        super().__init__()
+        self.msg = msg
+
+        layout = QtGui.QVBoxLayout()
+        self.setLayout(layout)
+        layout.addWidget(QtGui.QLabel(msg))
+
+        self.button = QtGui.QDialogButtonBox(QtGui.QDialogButtonBox.StandardButton.Ok)
+        layout.addWidget(self.button)
+        self.button.accepted.connect(self.close)
+
+
 class QStandardItemRO(QtGui.QStandardItem):
     def __init__(self, *args, data=None, **kwargs):
         super().__init__(*args, **kwargs)
@@ -59,8 +73,7 @@ class QStandardItemRO(QtGui.QStandardItem):
         self.setSelectable(True)
         self.setData(data)
 
-
-class TableModel(QtGui.QStandardItemModel):
+class ExchangeModel(QtGui.QStandardItemModel):
     def __init__(self, parent=None):
         super().__init__(parent)
         self.setColumnCount(6)
@@ -107,8 +120,13 @@ class ActionMenu(QtGui.QMenu):
         self.action_copy = self.addAction("Copy Full Reference")
         self.action_copy.triggered.connect(self.copy_full_reference_triggered)
 
-        self.action_explore = self.addAction("Explore")
-        self.action_explore.triggered.connect(self.explore_triggered)
+        if hasattr(self.parentWidget(), "explore"):
+            self.action_explore = self.addAction("Explore")
+            self.action_explore.triggered.connect(self.explore_triggered)
+
+        if hasattr(self.parentWidget(), "explore_in_new_window"):
+            self.action_explore = self.addAction("Explore in new window")
+            self.action_explore.triggered.connect(self.explore_in_new_window_triggered)
 
     def copy_triggered(self):
         QtGui.QGuiApplication.clipboard().setText(self.index.data())
@@ -122,22 +140,75 @@ class ActionMenu(QtGui.QMenu):
     def explore_triggered(self):
         self.parentWidget().explore(self.index)
 
+    def explore_in_new_window_triggered(self):
+        self.parentWidget().explore_in_new_window(self.index)
+
+class TabBar(QtGui.QTabBar):
+    def __init__(self):
+        super().__init__()
+
+    def tabSizeHint(self, index):
+        value = super().tabSizeHint(index)
+        return QtCore.QSize(200, value.height())
+
+    def minimumTabSizeHint(self, index):
+        value = super().minimumTabSizeHint(index)
+        return QtCore.QSize(100, value.height())
+
 
 class ActivityWindow(QtGui.QMainWindow):
-    keep = dict()
+    keep = list()
 
-    def __init__(self, activity_json):
+    def __init__(self, activity_json, params=None):
         super().__init__()
-        self.activity_key = (activity_json["database"], activity_json["code"])
+        self.tabs = dict()
+        self.xcount = 0
+        self.params = params
 
         self.setWindowTitle("Activity Viewer")
-        self.setGeometry(100, 100, 800, 600)
+        self.setMinimumSize(QtCore.QSize(800, 600))
+
+        self.tabs_widget = QtGui.QTabWidget()
+        self.tabs_widget.setTabBar(TabBar())
+        self.tabs_widget.setTabsClosable(True)
+        self.tabs_widget.setMovable(True)
+        self.tabs_widget.tabCloseRequested.connect(self.close_tab)
+        self.tabs_widget.setElideMode(QtCore.Qt.TextElideMode.ElideRight)
+        self.setCentralWidget(self.tabs_widget)
+        self.add_activity(activity_json)
+
+    def add_activity(self, activity_json):
+        key = (activity_json["database"], activity_json["code"])
+        if key in self.tabs:
+            activity_widget = self.tabs[key]
+            self.tabs_widget.setCurrentWidget(activity_widget)
+            return
+
+        activity_widget = ActivityTab(self, activity_json)
+        self.tabs[activity_widget.activity_key] = activity_widget
+        self.tabs_widget.addTab(activity_widget, f"#{self.xcount} "+activity_json["name"])
+        self.tabs_widget.setCurrentWidget(activity_widget)
+        activity_widget.tree_view.setColumnWidth(0, 400)
+        self.xcount += 1
+
+    def close_tab(self, index):
+        w = self.tabs_widget.widget(index)
+        del self.tabs[w.activity_key]
+        self.tabs_widget.removeTab(index)
+
+    def closeEvent(self, ev):
+        ActivityWindow.keep.remove(self)
+        super().closeEvent(ev)
+
+class ActivityTab(QtGui.QWidget):
+    def __init__(self, xparent, activity_json):
+        super().__init__()
+
+        self.xparent = xparent
+        self.activity_key = (activity_json["database"], activity_json["code"])
 
-        # Create central widget and layout
-        central_widget = QtGui.QWidget()
-        self.setCentralWidget(central_widget)
         layout = QtGui.QVBoxLayout()
-        central_widget.setLayout(layout)
+        self.setLayout(layout)
 
         grid = QtGui.QGridLayout()
         layout.addLayout(grid)
@@ -168,7 +239,7 @@ class ActivityWindow(QtGui.QMainWindow):
         self.tree_view.setSortingEnabled(True)
         self.tree_view.setAlternatingRowColors(True)
 
-        self.model = TableModel()
+        self.model = ExchangeModel()
         self.model.load(activity_json["exchanges"])
         self.tree_view.setModel(self.model)
         self.tree_view.doubleClicked.connect(self.doubleCliked)
@@ -183,11 +254,6 @@ class ActivityWindow(QtGui.QMainWindow):
         self.action_menu = ActionMenu(self, index)
         self.action_menu.exec(self.tree_view.viewport().mapToGlobal(point))
 
-    def closeEvent(self, ev):
-        if self.activity_key in ActivityWindow.keep:
-            del ActivityWindow.keep[self.activity_key]
-        super().closeEvent(ev)
-
     def update_filter(self, *args):
         text = self.filter_edit.text()
         if self.ignore_case.isChecked():
@@ -208,28 +274,169 @@ class ActivityWindow(QtGui.QMainWindow):
     def explore(self, index):
         r = index.row()
         v = self.model.root.child(r, 0).data()
-        show_activity(bw2data.get_activity((v["database"], v["code"])), self.params)
+        act = bw2data.get_activity((v["database"], v["code"]))
+        if self.xparent.params is None:
+            activity_json = activity_to_json(act)
+        else:
+            activity_json = activity_to_json_with_params(act, self.xparent.params)
+        self.xparent.add_activity(activity_json)
 
-    def show(self, *args, **kwargs):
-        super().show(*args, **kwargs)
+    def explore_in_new_window(self, index):
+        r = index.row()
+        v = self.model.root.child(r, 0).data()
+        act = bw2data.get_activity((v["database"], v["code"]))
+        show_activity(act, self.xparent.params)
+
+class SearchModel(QtGui.QStandardItemModel):
+    def __init__(self, parent=None):
+        super().__init__(parent)
+        self.setColumnCount(4)
+        self.setHeaderData(0, QtCore.Qt.Orientation.Horizontal, "Name")
+        self.setHeaderData(1, QtCore.Qt.Orientation.Horizontal, "Unit")
+        self.setHeaderData(2, QtCore.Qt.Orientation.Horizontal, "Category")
+        self.setHeaderData(3, QtCore.Qt.Orientation.Horizontal, "Location")
+
+    def load(self, data):
+        self.exchanges = [e for e in data]
+        self.root = self.invisibleRootItem()
+        for e in data:
+            row = [
+                QStandardItemRO(str(e.get(k, "-")), data=e)
+                for k in ["name", "unit", "categories", "location"]
+            ]
+
+            etype = e.get("type", "unknown")
+            if etype == "emission":
+                path = os.path.join(os.path.dirname(__file__), "icons", "emission.png")
+            elif etype in {"process", "processwithreferenceproduct"}:
+                path = os.path.join(os.path.dirname(__file__), "icons", "process.png")
+            elif etype == "natural resource":
+                path = os.path.join(os.path.dirname(__file__), "icons", "natural_resource.png")
+            else:
+                path = os.path.join(os.path.dirname(__file__), "icons", "unknown.png")
+
+            row[0].setIcon(QtGui.QIcon(path))
+            self.root.appendRow(row)
+
+class SearchWindow(QtGui.QMainWindow):
+    def __init__(self, db, keywords=""):
+        super().__init__()
+
+        self.db = db
+
+        self.setWindowTitle("Search Activity")
+        self.setMinimumSize(QtCore.QSize(800, 600))
+
+        central_widget = QtGui.QWidget()
+        self.setCentralWidget(central_widget)
+        layout = QtGui.QVBoxLayout()
+        central_widget.setLayout(layout)
+
+        grid = QtGui.QGridLayout()
+        layout.addLayout(grid)
+
+        next_row = grid.rowCount()
+        grid.addWidget(QtGui.QLabel("database:"), next_row, 0)
+        grid.addWidget(QtGui.QLabel(self.db.name), next_row, 1)
+
+        # Keyword query
+        next_row = grid.rowCount()
+        grid.addWidget(QtGui.QLabel("keywords:"), next_row, 0)
+        self.keywords = QtGui.QLineEdit("")
+        self.keywords.setText(keywords)
+        self.keywords.returnPressed.connect(self.update_search)
+        grid.addWidget(self.keywords, next_row, 1)
+        self.keywords_button = QtGui.QPushButton("Update")
+        self.keywords_button.clicked.connect(self.update_search)
+
+        grid.addWidget(self.keywords_button, next_row, 2)
+
+        next_row = grid.rowCount()
+        grid.addWidget(QtGui.QLabel("filter:"), next_row, 0)
+        self.filter_edit = QtGui.QLineEdit("")
+        self.filter_edit.textChanged.connect(self.update_filter)
+        grid.addWidget(self.filter_edit, next_row, 1)
+        grid.setColumnStretch(0, 0)
+        grid.setColumnStretch(1, 1)
+        self.ignore_case = QtGui.QCheckBox("Ignore Case")
+        self.ignore_case.setChecked(True)
+        if hasattr(self.ignore_case, "checkStateChanged"):
+            self.ignore_case.checkStateChanged.connect(self.update_filter)
+        else:
+            self.ignore_case.stateChanged.connect(self.update_filter)
+        next_row = grid.rowCount()
+        grid.addWidget(self.ignore_case, next_row, 1)
+
+        # Create tree view
+        self.tree_view = QtGui.QTreeView()
+        layout.addWidget(self.tree_view)
+        self.tree_view.setSortingEnabled(True)
+        self.tree_view.setAlternatingRowColors(True)
+        self.tree_view.doubleClicked.connect(self.doubleCliked)
+        self.tree_view.setExpandsOnDoubleClick(False)
+        self.tree_view.setSelectionBehavior(QtGui.QAbstractItemView.SelectionBehavior.SelectItems)
+        self.tree_view.setContextMenuPolicy(QtCore.Qt.ContextMenuPolicy.CustomContextMenu)
+        self.tree_view.customContextMenuRequested.connect(self.context_menu)
+
+        self.model = None
+
+        self.update_search()
+
+    def update_search(self):
+        text = self.keywords.text()
+        keywords = [re.sub("[^a-zA-Z0-9_-]", "", s) for s in text.split(" ") if len(s) > 0]
+        if all(len(x) < 3 for x in keywords):
+            w = MessageDialog("Too smalls keywords")
+            w.setWindowTitle("WARNING")
+            w.exec()
+            return
+
+        acts = self.db.search(" ".join(keywords), proxy=True, limit=200)
+        acts = [dict(a) for a in acts]
+
+        old_model = self.model
+        self.model = SearchModel()
+        self.model.load(acts)
+        self.tree_view.setModel(self.model)
+        if old_model is not None:
+            old_model.deleteLater()
         self.tree_view.setColumnWidth(0, 400)
 
-class ActivityWindowWithParams(ActivityWindow):
-    keep = dict()
+    def context_menu(self, point):
+        index = self.tree_view.indexAt(point)
+        self.action_menu = ActionMenu(self, index)
+        self.action_menu.exec(self.tree_view.viewport().mapToGlobal(point))
 
-    def __init__(self, activity_json, params):
-        super().__init__(activity_json)
-        self.params = params
+    def update_filter(self, *args):
+        if self.model is None:
+            return
 
-    def closeEvent(self, ev):
-        if self.activity_key in ActivityWindow.keep:
-            del ActivityWindowWithParams.keep[self.activity_key]
-        super().closeEvent(ev)
+        text = self.filter_edit.text()
+        if self.ignore_case.isChecked():
+            text = text.lower()
+            index = self.model.root.index()
+            for i in range(self.model.rowCount()):
+                v = self.model.root.child(i, 0).data()
+                self.tree_view.setRowHidden(i, index, text not in v["name"].lower())
+        else:
+            index = self.model.root.index()
+            for i in range(self.model.rowCount()):
+                v = self.model.root.child(i, 0).data()
+                self.tree_view.setRowHidden(i, index, text not in v["name"])
 
-    def explore(self, index):
+    def doubleCliked(self, index):
+        self.explore_in_new_window(index)
+
+    def explore_in_new_window(self, index):
         r = index.row()
         v = self.model.root.child(r, 0).data()
-        show_activity_with_params(bw2data.get_activity((v["database"], v["code"])), self.params)
+        act = bw2data.get_activity((v["database"], v["code"]))
+        show_activity(act)
+
+    def show(self, *args, **kwargs):
+        self.tree_view.setColumnWidth(0, 400)
+        super().show(*args, **kwargs)
+
 
 # Replace IPython version to one compatible with Qt6
 def start_event_loop_qt4(app=None):
@@ -246,7 +453,7 @@ def start_event_loop_qt4(app=None):
     else:
         app._in_event_loop = True
 
-def _show_activity(cls, activity_json, *args):
+def show_activity(act, params=None):
 
     if 'matplotlib' in sys.modules:
         import matplotlib
@@ -256,9 +463,13 @@ def _show_activity(cls, activity_json, *args):
 
     app = get_app_qt4()
 
-    if (window := cls.keep.get((activity_json["database"], activity_json["code"]), None)) is None:
-        window = cls(activity_json, *args)
-        cls.keep[(activity_json["database"], activity_json["code"])] = window
+    if params is None:
+        activity_json = activity_to_json(act)
+    else:
+        activity_json = activity_to_json_with_params(act, params)
+
+    window = ActivityWindow(activity_json, params)
+    ActivityWindow.keep.append(window)
     window.show()
 
     if not is_event_loop_running_qt4(app):
@@ -266,14 +477,33 @@ def _show_activity(cls, activity_json, *args):
 
     window.activateWindow()
 
-def show_activity(act):
-    activity_json = activity_to_json(act)
-    _show_activity(ActivityWindow, activity_json)
+def search(database, keywords=""):
+
+    if 'matplotlib' in sys.modules:
+        import matplotlib
+        if hasattr(matplotlib.backends, "backend"):
+            if 'qt' not in matplotlib.backends.backend:
+                print("WARNING: ActivityGUI will block, use `%matplotlib qt` to avoid blocking")
 
-def show_activity_with_params(act, params):
-    activity_json = activity_to_json_with_params(act, params)
-    _show_activity(ActivityWindowWithParams, activity_json, params)
+    app = get_app_qt4()
+
+    if isinstance(database, str):
+        if database not in bw2data.databases:
+            w = MessageDialog(f"Database {database} not found !")
+            w.setWindowTitle("WARNING")
+            w.exec()
+            return
+        database = bw2data.Database(database)
+
+    window = SearchWindow(database, keywords)
+    window.show()
+
+    if not is_event_loop_running_qt4(app):
+        start_event_loop_qt4(app)
+
+    window.activateWindow()
+    return window
 
 def close_all():
-    for w in list(ActivityWindow.keep.values()):
+    for w in list(ActivityWindow.keep):
         w.close()

{brightway_basic_explorer-0.9.dev3.dist-info → brightway_basic_explorer-0.9.dev5.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: brightway-basic-explorer
-Version: 0.9.dev3
+Version: 0.9.dev5
 Summary: Implement basic GUI to explorer brightway ativities
 Author-email: Benoît GSCHWIND <benoit.gschwind@minesparis.psl.eu>, OIE - Mines Paris PSL <benoit.gschwind@minesparis.psl.eu>
 Maintainer-email: Benoît GSCHWIND <benoit.gschwind@minesparis.psl.eu>

{brightway_basic_explorer-0.9.dev3.dist-info → brightway_basic_explorer-0.9.dev5.dist-info}/RECORD

@@ -1,10 +1,10 @@
-brightway_basic_explorer/__init__.py,sha256=N-B9Yqpz3GRtC1WBAeMS9yHJlRqCcr57zPb_BmpnE1o,10404
+brightway_basic_explorer/__init__.py,sha256=3ll2rWXQZhiBORKKPpLcSaJU3EzrpTb8WEnbncSgpFM,18908
 brightway_basic_explorer/icons/emission.png,sha256=js9yLt3jihZ1x7pXfuL-cXQ0HtC57Nrfxg58YfbWpfU,154
 brightway_basic_explorer/icons/natural_resource.png,sha256=vbXPQful4vuibRNUoM4qze4hV2-eW3OkmkePUOnWsRI,162
 brightway_basic_explorer/icons/process.png,sha256=zlJJ3J-2geqybqiL6YwGHFka0kqEIwKZqTeUzijPcE0,240
 brightway_basic_explorer/icons/unknown.png,sha256=92LyvebtGvewcaLjEojQOWXJzeIicIVgWqvov9XhOrc,165
-brightway_basic_explorer-0.9.dev3.dist-info/licenses/COPYING,sha256=b8nnCcy_4Nd_v_okJ6mDKCvi64jkexzbSfIag7TR5mU,13827
-brightway_basic_explorer-0.9.dev3.dist-info/METADATA,sha256=AEvmc5NG8d0QtKuDBLtDAu9kD7ujrKg8xRNlyfNho78,1601
-brightway_basic_explorer-0.9.dev3.dist-info/WHEEL,sha256=YVMoNqKzERt-wjUZwJ33xBGAwnFl-4cqbYkTtWa4itE,91
-brightway_basic_explorer-0.9.dev3.dist-info/top_level.txt,sha256=YrW7Qnd1cgs1TDGBcL5wNmDg7XSiiwYsFymP7cfL0hU,25
-brightway_basic_explorer-0.9.dev3.dist-info/RECORD,,
+brightway_basic_explorer-0.9.dev5.dist-info/licenses/COPYING,sha256=b8nnCcy_4Nd_v_okJ6mDKCvi64jkexzbSfIag7TR5mU,13827
+brightway_basic_explorer-0.9.dev5.dist-info/METADATA,sha256=j_mDzpAwZKnvX6dapQifPagoErK2ZnOJQvVbhPYuFvI,1601
+brightway_basic_explorer-0.9.dev5.dist-info/WHEEL,sha256=YVMoNqKzERt-wjUZwJ33xBGAwnFl-4cqbYkTtWa4itE,91
+brightway_basic_explorer-0.9.dev5.dist-info/top_level.txt,sha256=YrW7Qnd1cgs1TDGBcL5wNmDg7XSiiwYsFymP7cfL0hU,25
+brightway_basic_explorer-0.9.dev5.dist-info/RECORD,,