def load_db()

in meter-data/generate_meter_data.py [0:0]

• 22 lines of code
• 8 McCabe index (conditional complexity)

def load_db(no_of_meters):
    # Open db file and load db in memory
    fh = open("db.json", 'r')
    db = json.load(fh)

    # If db file is empty, generate meterids and write it to db
    if len(db) == 0:
        logging.debug("Database file is empty!")

        # generate meterids and write it to the file
        meters = []
        meterids = getall_meterids_shuffled(1, no_of_meters)
        for mid in meterids:
            meter = {
                'meter_id': mid
            }
            for register_type in OBIS_CODES:
                for register in OBIS_CODES[register_type]:
                    meter[register] = 0
            meters.append(meter)

        fh = open("db.json", 'w')
        json.dump(meters, fh)
        db = meters

    logging.debug("####### Database entries ########")
    logging.debug(db)
    logging.debug('#################################')
    return db