#coding=utf-8import pymysqlclass MysqlUntil:def __init__(self, host, user, passwd, db, port, charset):self.host = hostself.user = userself.passwd = passwdself.db = dbself.port = int(port)self.charset = charsetdef conn_mysql(self):# 连接数据库try:conn = pymysql.connect(host=self.host, user=self.user, passwd=self.passwd, db=self.db, port=self.port,charset=self.charset)except Exception:print("连接MYSQL失败")else:return conndef mysql_select(self, select_sql):# 执行查询语句，解析返回结果conn = self.conn_mysql()try:cur = conn.cursor() # 获取一个游标cur.execute(select_sql)data = cur.fetchall()cols = cur.descriptioncur.close() # 关闭游标conn.close() # 释放数据库资源except Exception as e:conn.rollback()print(e)else:col = []for i in cols:col.append(i[0])datas = list(map(list, data))data_list = []for data_one in datas:data_dict = {}for i in range(len(data_one)):data_dict[col[i]] = data_one[i]data_list.append(data_dict)return data_listdef mysql_insert(self, insert_sql):# 执行插入语句conn = self.conn_mysql()try:cur = conn.cursor() # 获取一个游标sta = cur.execute(insert_sql)if sta == 1:print('DONE:' + insert_sql)else:print('FAIL')mit()cur.close() # 关闭游标conn.close() # 释放数据库资源except Exception as e:conn.rollback()print(e)
