1010
1111from activity_browser import app
1212from activity_browser .ui import icons , widgets
13+ from activity_browser .ui import core
1314
1415
1516
@@ -264,8 +265,8 @@ def __init__(self, index: int, parent=None):
264265 self .load_btn .setToolTip ("Load (new) data for this scenario table" )
265266 self .remove_btn = QtWidgets .QPushButton (icons .qicons .delete , "Delete" )
266267 self .remove_btn .setToolTip ("Remove this scenario table" )
267- self .view = widgets . ABTreeView (self )
268- self .model = widgets . ABItemModel ( self )
268+ self .view = ScenarioImportView (self )
269+ self .model = ScenarioImportModel ( parent = self )
269270 self .view .setModel (self .model )
270271 self .scenario_df = pd .DataFrame (columns = ss .SUPERSTRUCTURE )
271272
@@ -373,7 +374,7 @@ def sync_superstructure(self, df: pd.DataFrame) -> None:
373374 return
374375 self .scenario_df = df
375376 cols = ss .scenario_names_from_df (self .scenario_df )
376- self .model .setDataFrame (pd .DataFrame (cols , columns = ["Scenarios" ]))
377+ self .model .set_dataframe (pd .DataFrame (cols , columns = ["Scenarios" ]))
377378 self ._parent .combined_dataframe ()
378379
379380 def scenario_db_check (self , df : pd .DataFrame ) -> pd .DataFrame :
@@ -399,6 +400,14 @@ def dataframe(self) -> pd.DataFrame:
399400 return self .scenario_df
400401
401402
403+ class ScenarioImportView (widgets .ABTreeView ):
404+ """Tree view for scenario imports."""
405+
406+
407+ class ScenarioImportModel (core .ABTreeModel ):
408+ """Model for displaying imported scenario names."""
409+
410+
402411class ExcelReadDialog (QtWidgets .QDialog ):
403412 SUFFIXES = {
404413 ".xls" ,
0 commit comments