data_dump.py

import ConfigParser
from datetime import datetime
import MySQLdb
from ports_info import portInfo

CONFIG_FILE = "connection.conf"
config = ConfigParser.RawConfigParser()


def ConfigSectionMap(section):
    config.read(CONFIG_FILE)
    dict1 = {}
    try:
        options = config.options(section)
        for option in options:
            try:
                dict1[option] = config.get(section, option)
            except:
                print "exception on {}".format(option)
                dict1[option] = None
    except:
        print "exception on {}".format(section)
    return dict1

def main():
    db = MySQLdb.connect(host=ConfigSectionMap("Controller")['address'],
                         user=ConfigSectionMap("Controller")['username'],
                         passwd=ConfigSectionMap("Controller")['password'],
                         port=3306)

    cur = db.cursor()

    cur.execute("create database if not exists bandwidth")
    cur.execute("create database if not exists tenant_mapper");

    portInfoObj = portInfo()
    portInfoObj.sshConnect()
    tenantPorts = portInfoObj.instanceToPortNumber()
    bandwidthDict = portInfoObj.instanceToBandwidth()

    tenant_db = MySQLdb.connect(host=ConfigSectionMap("Controller")['address'],
                         user=ConfigSectionMap("Controller")['username'],
                         passwd=ConfigSectionMap("Controller")['password'],
                         db="tenant_mapper",
                         port=3306)

    ten_cur = tenant_db.cursor()
    ten_cur.execute("SET sql_notes=0; ")
    ten_cur.execute("create table if not exists mapper \
                     (tenantid VARCHAR(50), tenantname VARCHAR(50), PRIMARY KEY (tenantid));")
    i = 0
    ten_cur.execute("SET sql_notes=1; ")
    for key in tenantPorts:
        try:
            ten_cur.execute("INSERT INTO mapper (tenantid,tenantname) values ('{}','{}');"
                            .format(key, "tenant" + str(i)))
        except MySQLdb.Error:
            # do nothing
            pass

        tenant_db.commit()
        i = i + 1

    tenant_db.close()

    db = MySQLdb.connect(host=ConfigSectionMap("Controller")['address'],
                         user=ConfigSectionMap("Controller")['username'],
                         passwd=ConfigSectionMap("Controller")['password'],
                         db="bandwidth",
                         port=3306)

    cur = db.cursor()

    i = 0
    for instance, bandwidthList in bandwidthDict.iteritems():
        cur.execute("SET sql_notes=0; ")
        cur.execute("create table if not exists {} \
    				(currentdatetime VARCHAR(50),bandwidthusage INTEGER);"
                    .format("tenant" + str(i)))
        cur.execute("SET sql_notes=1; ")
        cur.execute("INSERT INTO {} (currentdatetime,bandwidthusage) values ('{}',{});"
                    .format("tenant" + str(i), datetime.now().strftime('%Y/%m/%d %H:%M:%S'),
                            sum(bandwidthList)))
        db.commit()
        i = i + 1

    db.close()

if __name__ == "__main__":
    main()