def test(self): va_rows = sasi.util.va.read_va_from_csv(conf.conf['va_file']) va = VulnerabilityAssessment(rows = va_rows) f_by_h = va.get_features_by_habitats() features = va.get_features() substrates = va.get_substrates() g_by_h = va.get_gear_categories_by_habitats() gear_categories = va.get_gear_categories() h_by_g = va.get_habitats_by_gear_categories() habitats_for_gear = va.get_habitats_for_gear_category(gear_categories.keys().pop()) features_for_gear = va.get_features_for_gear_category(gear_categories.keys().pop()) f_by_g = va.get_features_by_gear_categories() habitats = va.get_habitats()
def main(): # Read features from vulernability assessment. va_rows = sasi.util.va.read_va_from_csv(conf.conf["va_file"]) va = VulnerabilityAssessment(rows=va_rows) features = va.get_features() # Get DB session. session = sa_session.get_session() # Clear features table. session.execute(sa_feature.table.delete()) session.commit() # Create Feature objects # note: might move this into the VA object itself later. feature_objs = [] for f in features.values(): f_obj = Feature(name=f["FEATURE"], id=f["FEATURE_CODE"], category=f["FEATURE_CLASS_CODE"]) feature_objs.append(f_obj) # Add feature objects to session and save to DB. session.add_all(feature_objs) session.commit()