def add_db_insert_and_cleanup(self, mongo_database, out_exts=["WFK"]):
spec = self.set_short_single_core_to_spec()
spec['mongo_database'] = mongo_database.as_dict()
insert_and_cleanup_fw = Firework([DatabaseInsertTask(),
FinalCleanUpTask(out_exts=out_exts)],
spec=spec,
name=(self.wf.name+"_insclnup")[:15])
append_fw_to_wf(insert_and_cleanup_fw, self.wf)
def add_final_cleanup(self, out_exts=["WFK"]):
cleanup_fw = Firework(FinalCleanUpTask(out_exts=out_exts), spec=self.set_short_single_core_to_spec(),
name=(self.wf.name+"_cleanup")[:15])
append_fw_to_wf(cleanup_fw, self.wf)