ems/Connector/modules/database.py

'''
Created on 26.11.2013

@author: Philipp Rauch
'''
from profile.database import Database

### Dict of Tables to sort after ###
_sort_lookup = { 'EA_Last' : 'DateTime',
                'GRIDManager_I_IST' : 'DateTime',
                'GRIDManager_I_PH' : 'DateTime',
                'GRIDManager_Status' : 'DateTime',
                'GRIDManager_Temperatur': 'DateTime',
                'GRIDManager_U_IST': 'DateTime',
                'PAC' : 'date_/_time',
                'pCharger': 'DateTime'}

def setup(conf):
    '''
    establishes a connection to the database and returns a Database object

    @return: Database object
    '''

    db = Database()
    db.loadDatabase(strHost = conf['mySQL_server'],
                     intPort = int(conf['mySQL_port']),
                     strUser = conf['mySQL_user'],
                     strPasswd = conf['mySQL_pass'],
                     strDatabase = conf['mySQL_database'],
                     strTable = None)
    return db

def loop(db, item):
    '''
    get the last line of the table 'item' in database
    and return it
    '''
    path = item.split('/')
    if path[0] == '':
        path.remove('')

    table = path[len(path) -1]
    if table == 'device':
        tmp = { table : {} }
        for i in db.getInformation()['Table'].keys():
            tmp[table][i] = {}
        return tmp

    try:
        sort = db.getInformation()['Table_Prime'][table][0]
    except:
        try:
            sort = _sort_lookup[table]
        except:
            return { 'error/db_error' : { table : 'no Prime-Key / Sort defined' } }

    res = db.readDatabase(intRowOffset=0, intRowNumber=1, strTable=table, strSort=sort, bSortInvert=True)
    result = res.to_dict()

    for k in result:
        result[k] = result[k][0]

    return {item : result}

def close(cursor):
    pass