把数据库的操作封装成类，以及一些简便方法

为了提高数据库的操作速度，避免重复的进入和退出，我们可以把一些常用的数据库的操作封装成一个类。

 

把数据库操作写成函数：每次都得连接数据库一次连接一次提交，重复操作浪费时间

把数据库操作写成类：一次连接可以多次提交，最佳重复操作时不需要关闭数据库

 

import pymysql

class SQLManager(object):

    def __init__(self): #实例化后自动执行此函数
        self.connect()

    def connect(self): #此时进入数据库，游标也已经就绪
        self.conn = pymysql.connect(host="127.0.0.1", port=3306, user="root", passwd="123", db="day66", charset="utf8")
        self.cursor = self.conn.cursor(cursor=pymysql.cursors.DictCursor)

    def get_list(self, sql, args=None):
        self.cursor.execute(sql, args)
        result = self.cursor.fetchall()
        return result

    def get_one(self, sql, args=None):
        self.cursor.execute(sql, args)
        result = self.cursor.fetchone()
        return result


    def run(self, sql, args=None):
        self.cursor.execute(sql, args)
        self.conn.commit()

    def create(self, sql, args=None):
        self.cursor.execute(sql, args)
        self.conn.commit()
        ret = self.cursor.lastrowid  # 取到刚才创建的记录的ID
        return ret

    def m_create(self, sql, args=None):
        # self.cursor.executemany(sql, [(1, 36), (1, 38), (1, 36)])
        self.cursor.executemany(sql, args)  # 批量执行
        self.conn.commit()
        ret = self.cursor.lastrowid  # 取到刚才创建的记录的ID
        return ret

    def close(self):
        self.cursor.close()
        self.conn.close()

 

 有时候可能会忘记关闭数据库，

With 定义函数的时候有两个方法，一个叫__enter__

另一个叫__exit__

后面的参数不用管

 

 

 

import pymysql
class SQLManager(object):

    def __init__(self):
        self.connect()

    def connect(self):
        self.conn = pymysql.connect(host="127.0.0.1", port=3306, user="root", passwd="", db="mysite", charset="utf8")
        self.cursor = self.conn.cursor(cursor=pymysql.cursors.DictCursor)

    def get_list(self, sql, args=None):
        self.cursor.execute(sql, args)
        result = self.cursor.fetchall()
        return result

    def get_one(self, sql, args=None):
        self.cursor.execute(sql, args)
        result = self.cursor.fetchone()
        return result


    def run(self, sql, args=None):
        self.cursor.execute(sql, args)
        self.conn.commit()

    def create(self, sql, args=None):
        self.cursor.execute(sql, args)
        self.conn.commit()
        ret = self.cursor.lastrowid  # 取到刚才创建的记录的ID
        return ret

    def m_create(self, sql, args=None):
        # self.cursor.executemany(sql, [(1, 36), (1, 38), (1, 36)])
        self.cursor.executemany(sql, args)  # 批量执行
        self.conn.commit()
        ret = self.cursor.lastrowid  # 取到刚才创建的记录的ID
        return ret

    def close(self):
        self.cursor.close()
        self.conn.close()

    def __enter__(self):
        print("startttttttttttttttt........")
        return self

    def __exit__(self, exc_type, exc_val, exc_tb):
        self.close()
        print("closeeeeeeeeeeeeeeeeeeee.....")

执行的例子：

def modal_teacher_list(request):
    teacher_id = request.GET.get("teacher_id")
    with SQLManager() as db:
        class_ids_o = db.get_list("select class_id from teacher2class WHERE teacher_id=%s", [teacher_id])
    print(class_ids_o)
    class_ids = [i["class_id"] for i in class_ids_o]
    return HttpResponse(json.dumps(class_ids))