python链接数据库工具包

简单介绍

使用pymysql导包，编写一个工具类，减少代码开发，增强代码的通用性

操作步骤

1.新建一个py文件，命名con_mysql
2.使用pymysql链接数据库，建立游标对象，使其去调用sql
3.使用捕获异常来判断提交事务，回滚事务，finally最后关闭链接
4.判断是查询还是写入数据
5.整合代码

代码如下

con_mysql.py

import pymysql


def connect(sql):
    conn = ""
    cur = ""
    try:
        conn = pymysql.Connect(
            host='localhost',
            user='root',
            password='root',
            database='test',
            port=3306,
            charset='utf8'
        )
        # 3.创建游标对象
        cur = conn.cursor()
        # 4.编写sql
        # 5.使用游标对象去调用sql
        cur.execute(sql)

        # 6.获取查询的结果
        if sql.find('select') == 0:
            result = cur.fetchall()
            return result
        data_list = ['insert', 'update', 'delete']
        for i in data_list:
            if sql.find(i) == 0:
                print("写入数据要提交")
                # 提交操作
                conn.commit()

    except Exception as e:
        print(e)
        conn.rollback()
    finally:
        # 7.关闭游标对象
        cur.close()

测试.py

import con_mysql
# sql = "insert into stu values (8,'测试',32)"
sql = "select * from stu"
result = con_mysql.connect(sql)
if result:
    print(result)

结果如图

最后

上面的数据库链接信息替换成自己的，sql根据自己数据库里的表来定义即可