pymysql操作数据库

###
pymysql操作数据库
###
import pymysql

class General:
    def db_operation(query, db, host='localhost', user='***', password='***'):
        DB = pymysql.connect(db=db, host=host, user=user, password=password, port=3306)
        cur = DB.cursor(pymysql.cursors.DictCursor)

        cur.execute(query)
        data = cur.fetchall()  

        DB.commit()
        DB.close()

        return data