helpers.py

from compendium_v2 import db, survey_db
from compendium_v2.db import model


def init_db(config):
    dsn_prn = config['SQLALCHEMY_DATABASE_URI']
    db.init_db_model(dsn_prn)
    dsn_survey = config['SURVEY_DATABASE_URI']
    survey_db.init_db_model(dsn_survey)


def get_uppercase_nren_dict(session):
    """
    :param session: db session that is used to query the known NRENs
    :return: a dictionary of all known NRENs db entities keyed on the uppercased name
    """
    current_nrens = session.query(model.NREN).all()
    nren_dict = {nren.name.upper(): nren for nren in current_nrens}
    # add aliases that are used in the source data:
    nren_dict['ASNET'] = nren_dict['ASNET-AM']
    nren_dict['KIFU (NIIF)'] = nren_dict['KIFU']
    nren_dict['SURFNET'] = nren_dict['SURF']
    nren_dict['UOM'] = nren_dict['UOM/RICERKANET']
    return nren_dict