def store_fragility_model(self):
"""Load fragility model and write it to database."""
new_models = []
fmis = models.inputs4job(self.job_ctxt.job_id, "fragility")
for fmi in fmis:
if fmi.fragilitymodel_set.all().count() > 0:
continue
path = os.path.join(self.job_ctxt.base_path, fmi.path)
parser = fragility.FragilityModelParser(path)
writer = FragilityDBWriter(fmi, parser)
writer.serialize()
new_models.append(writer.model)
return new_models if new_models else None
def setUp(self):
[self.input] = models.inputs4job(self.job.id, input_type="fragility")
self.parser = FragilityModelParser(self.path)
self.writer = FragilityDBWriter(self.input, self.parser)
