Python数据库连接

# -- coding: utf-8 --

"""
Created on Mon Aug  27 11:14:15 2018
@author: Nemo
"""
#%%
class GetDatumFromDatabase:
    """ database information """
    def __init__(self, host,user,pwd,dbName,sqlString):
        self.host = host
        self.user = user
        self.pwd = pwd
        self.dbName = dbName
        self.sqlString = sqlString

    """ get data from Mysql database """
    def getDataFromMysql(self,charset='utf8'):
        import pandas as pd
        import pymysql as mysql
        db = mysql.connect(self.host,
                           self.user,
                           self.pwd,
                           self.dbName,
                           charset=charset)
        #cursor = db.cursor(cursor=mysql.cursors.DictCursor)
        cursor = db.cursor()
        cursor.execute(self.sqlString)
        colDescription=cursor.description
        InformalParam=cursor.fetchall()
        db.close()

        sourceData=pd.DataFrame(list(InformalParam))
        colName=[]
        for i in range(len(colDescription)):
             colName.append(colDescription[i][0])
        sourceData.columns = colName
        return sourceData
#%%
if __name__ == '__main__':
    sqlString="SELECT * from t_result where checksystem ='' and predicttime > '2018-08-31 00:00:00'"
    dbInfo=GetDatumFromDatabase(host="192.0.0.1*",
                                user="root",
                                pwd="root",
                                dbName="db_ai_data",
                                sqlString=sqlString)
    sourceData=dbInfo.getDataFromMysql()
    #修改sql
    dbInfo.sqlString="select distinct paramcode,paramname from t_information"
    paramBaseData=dbInfo.getDataFromMysql()