Allow cleanup (2)

CederGroupHub · Mar 1, 2024 · 49043e0 · 49043e0
1 parent 7188618
commit 49043e0
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/alab_management/scripts/cleanup_lab.py b/alab_management/scripts/cleanup_lab.py
@@ -42,6 +42,9 @@ def cleanup_lab(
                 f"If you want to remove the simulation database then type in "
                 f"{config['general']['name']}_sim:  "
             )
+            print(f"Removing database {database_name}")
+            _GetMongoCollection.init()
+            _GetMongoCollection.client.drop_database(database_name)  # type: ignore 
 
         # if sim_mode != AlabOSConfig().is_sim_mode() or database_name == "Alab":
         #     print("Wrong name of database. Hence, not removed.")