File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff line change @@ -52,6 +52,7 @@ def __init__(self):
5252 self .databases = set ()
5353
5454 bd .projects .current_changed .connect (self .reset_metadata )
55+ bd .databases .metadata_changed .connect (self .check_databases )
5556
5657 def add_metadata (self , db_names_list : list ) -> None :
5758 """Include data from the brightway databases.
@@ -172,6 +173,12 @@ def reset_metadata(self) -> None:
172173 self .dataframe = pd .DataFrame ()
173174 self .databases = set ()
174175
176+ def check_databases (self ):
177+ removed_dbs = [db for db in self .databases if db not in bd .databases ]
178+ for db in removed_dbs :
179+ self .dataframe .drop (self .dataframe [self .dataframe .database == db ].index , inplace = True )
180+ self .databases .remove (db )
181+
175182 def get_existing_fields (self , field_list : list ) -> list :
176183 """Return a list of fieldnames that exist in the current dataframe."""
177184 return [fn for fn in field_list if fn in self .dataframe .columns ]
You can’t perform that action at this time.
0 commit comments