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():
pp = pprint.PrettyPrinter(indent=4)
# Read features from vulernability assessment.
va_rows = sasi.util.va.read_va_from_csv(conf.conf['va_file'])
va = VulnerabilityAssessment(rows = va_rows)
# Get features by habitat.
f_by_h = va.get_features_by_habitats()
# Combine geo and bio features into one list for each habitat.
combined_f_by_h = {}
for h, features in f_by_h.items():
combined_features = list(features['1'].union(features['2']))
combined_f_by_h[h] = combined_features
print "assignments = %s" % pp.pformat(combined_f_by_h)
def main():
# Read habitat types and features from vulnerability assessment.
va_rows = sasi.util.va.read_va_from_csv(conf.conf['va_file'])
va = VulnerabilityAssessment(rows = va_rows)
valid_habitat_types = va.get_habitats()
features_by_habs = va.get_features_by_habitats()
# Get DB session.
session = sa_session.get_session()
# Clear habitat_types table.
session.execute(sa_habitat_type.habitat_type_feature_table.delete())
session.execute(sa_habitat_type.habitat_type_table.delete())
session.commit()
# For each valid habitat_type...
for h in valid_habitat_types:
(substrate_id,energy) = h.split(',')
# Get substrate object.
substrate = session.query(Substrate).filter(Substrate.id == substrate_id).one()
# Get features.
features_for_hab = features_by_habs[h]
features = []
for category_features in features_for_hab.values():
features.extend([session.query(Feature).filter(Feature.id == f).one() for f in category_features])
# Create habitat_type object.
habitat_type_obj = Habitat_Type(substrate=substrate, energy=energy, features=features)
# Add to session.
session.add(habitat_type_obj)
# Save to db.
session.commit()
def main(): # Read substrates from vulernability assessment. 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() substrates = va.get_substrates() # Get DB session. session = sa_session.get_session() # Clear substrate table. session.execute(sa_substrate.table.delete()) session.commit() # Create Substrate objects # note: might move this into the VA object itself later. substrate_objs = [] for s in substrates.values(): s_obj = Substrate(name=s['SUBSTRATE'], id=s['SUBSTRATE_CODE']) substrate_objs.append(s_obj) # Add substrate objects to session and save to DB. session.add_all(substrate_objs) session.commit()
