py-PDBC

import pymysql
class DB:
    #连接数据库，并返回连接到的数据库对象
    def con_db(self,hostname='qzx',username='root',password='123456',database='ATM',port=3306):#默认参数
        db = pymysql.connect(hostname,username,password,database,port)
        return db

    #查询数据库
    def search_db(self,sql):
        db = self.con_db()#连接数据库
        cu = db.cursor()#得到一个游标
        cu.execute(sql)#通过游标执行sql语句
        result = cu.fetchall()#获取sql执行结果
        db.commit()#提交数据库
        db.close()#关闭数据库连接对象
        return result

    #增删改数据库
    def fix_db(self,sql):
        db = self.con_db()
        cu = db.cursor()  # 得到一个游标
        cu.execute(sql)  # 通过游标执行sql语句
        db.commit()  # 提交数据库
        db.close()  # 关闭数据库连接对象

if __name__ == '__main__':
    d =DB()
    d.fix_db("delete from userinfo where username='q2'")
    result = d.search_db("select * from userinfo where username='q1'")
    print(result)