Pymysql的简便使用

平常方式，就在代码字段里直接导入Pymysql，然后进行连接数据库创建游标等操作。

真正写SQL语句的时间可能还没写数据库连接的时间多

如果有写多个.py文件都需要重复的连接数据库导出数据，这样写代码重复率过高，非常影响效率

 附上一张平常我写连接 Pymysql 的代码图

改进

思路 : 创建一个类，通过导入类调用里面的方法省去写连接代码的时间

创建一个 公用模板的 package ，里面写sql模板语句

这里我创建的是 utils 公用模板包 里面存放我需要用到的一些模板代码文件

在我的 utils 下我创建了一个 sql.py 文件 里面写了一个 SQLHelper 类

在以后我需要用到连接 MySQL的时候就可以调用这个类了

import pymysql

class SQlHelper(object):

    @staticmethod
    def open():
        conn = pymysql.connect(user="root",host="localhost",port=3306,charset="utf8",
                               password='your_password',database="your_database")
        cursor = conn.cursor(cursor=pymysql.cursors.DictCursor)
        return conn,cursor

    @staticmethod
    def close(conn,cursor):
        conn.commit()
        cursor.close()
        conn.close()

    @staticmethod
    def fetch_one(sql,args):
        conn,cursor = SQlHelper.open()
        cursor.execute(sql,args)
        obj = cursor.fetchone()
        SQlHelper.close(conn,cursor)
        return obj

    @staticmethod
    def fetch_many(sql,args):
        conn,cursor = SQlHelper.open()
        cursor.execute(sql,args)
        obj = cursor.fetchmany()
        SQlHelper.close(conn,cursor)
        return obj

    @staticmethod
    def fetch_all(sql,args):
        conn,cursor = SQlHelper.open()
        cursor.execute(sql,args)
        obj = cursor.fetchall()
        SQlHelper.close(conn,cursor)
        return obj

里面写了 fetch_one , fetch_many , fetch_all方法

注意: def open 中 returen 返回的是个元组 conn,cursor，在下面引入的时候顺序位置不能反

sql语句查询到的最终结果返回到控制台上是元组的形式，如果要想看的方便，可以在创建游标的时候这么写 cursor = conn.cursor(cursor=pymysql.cursors.DictCursor) ，这样返回的就是一个字典 带有键值对的类型了，方便后续的操作。

SQLHelper 类还可以这么写

import pymysql

class SQlHelper(object):

    @staticmethod
    def open():
        conn = pymysql.connect(host='localhost', user="root", charset="utf8",
                               password="your_password", database="your_database", port=3306)
        cursor = conn.cursor(cursor=pymysql.cursors.DictCursor)
        return conn,cursor

    @staticmethod
    def close(conn, cursor):
        conn.commit()
        cursor.close()
        conn.close()

    @classmethod
    def fetch_many(cls,sql,args):
        conn,cursor = cls.open()
        cursor.execute(sql,args)
        obj = cursor.fetchmany(2)
        cls.close(conn,cursor)
        return obj

    @classmethod
    def fetch_one(cls,sql,args):
        conn,cursor = cls.open()
        cursor.execute(sql,args)
        obj = cursor.fetchone()
        cls.close(conn,cursor)
        return obj

    @classmethod
    def fetch_all(cls,sql,args):
        conn,cursor = cls.open()
        cursor.execute(sql,args)
        obj = cursor.fetchall()
        cls.close(conn,cursor)
        return obj

代码部分 :

from utils.sql import SQlHelper


def login():
    num = 20
    obj = SQlHelper.fetch_all("select * from emp where deptno = %s",[num])
    print(obj)

if __name__ == '__main__':
    login()

返回 dict 类型结果，查询到了我的emp表下deptno（部门编号)为20的员工信息