Python 数据库代码封装

#coding:utf-8
import pymysql


class MysqlHelp(object):

    config = {"host": "localhost", "user": "root", "password": "CENTos_", "db": "test", "charset": "utf8"}

    def __init__(self):

        self.connection = None
        self.cursor = None

#从数据库中查询一行数据

    def getOne(self, sql, *args):

        try:
            self.connection = pymysql.connect(**MysqlHelp.config)
            self.cursor = self.connection.cursor()
            self.cursor.execute(sql, args)
            return self.cursor.fetchone()
        except Exception as ex:
            print(ex, ex)
        finally:
            self.close()

#从数据库中查询多行数据

    def getList(self, sql, *args):

        try:
            self.connection = pymysql.connect(**MysqlHelp.config)
            self.cursor = self.connection.cursor()
            self.cursor.execute(sql, args)
            return self.cursor.fetchall()
        except Exception as ex:
            print(ex, ex)
        finally:
            self.close()

#从数据库中获取多行数据

    def executeDML(self, sql, *args):

        try:
            self.connection = pymysql.connect(**MysqlHelp.config)
            self.cursor = self.connection.cursor()
            num = self.cursor.execute(sql, args) #返回执行语句后影响的行数
            self.connection.commit()
            return num
        except Exception as ex:
            self.connection.rollback()
            print(ex, ex)
        finally:
            self.close()

    def close(self):

        if self.cursor:
            self.cursor.close()
        if self.connection:
            self.connection.close()


if __name__ == "__main__":

    helper = MysqlHelp()
    print(helper.executeDML('delete from t_user where id = %s', 8))