Python使用笔记27--mysql操作封装类

1、面向过程

import pymysql

#面向过程
mysql_info = {'host':'127.0.0.1',
              'port':3306,
              'user':'root',
              'password':'123456',
              'autocommit':True,
              'db':'db001',
              'charset':'utf8'}


def connect():#连接数据库
    connect = pymysql.connect(**mysql_info)
    cursor = connect.cursor()
    return connect,cursor


def execute(cursor,sql):#执行sql
    result = cursor.execute(sql)
    return result


def close(connect,cursor):#关闭连接
    cursor.close()
    connect.close()


connect,cursor = connect()
sql = 'select * from user_info'
result = execute(cursor,sql)
close(connect,cursor)

2、面向对象

class MysqlDB:
    def __init__(self,mysql_info):
        self.mysql_info = mysql_info
        self.__connect()

    def __del__(self):#析构函数，实例被销毁的时候自动执行的函数
        self.__close()

    def __connect(self):#私有方法
        self.conn = pymysql.connect(**self.mysql_info)
        self.cur = self.conn.cursor(pymysql.cursors.DictCursor)

    def execute(self,sql):
        self.execute(sql)

    def execute_one(self):#select * from user_info where name = 'xxx';
        return self.cur.fetchone()

    def execute_all(self):#select * from user_info;
        return self.cur.fetchall()

    def execute_many(self,limit):
        return self.cur.fetchmany(limit)

    def __close(self):
        self.cur.close()
        self.conn.close()


m = MysqlDB(mysql_info)
m.execute('select * from user_info;')
result = m.execute_all()
print(result)