通过函数对pymysql进行极简单的封装

from pymysql.cursors import DictCursor
import pymysql


# 连接数据库
def get_conn():
    conn = pymysql.connect(
        user="root",
        password="1234",
        host="localhost",
        database="spider",  # 链接mysql里面的哪一个,要修改
        port=3306,
    )
    return conn


# 公共部分
def change(sql, isInsert=False):
    try:
        conn = get_conn()
        cursor = conn.cursor()
        count = cursor.execute(sql)
        conn.commit()
        if isInsert:
            # 新增的数据的id值拿出来
            new_id = cursor.lastrowid
            return new_id
        else:
            return count
    except Exception as e:
        print(e)
        conn.rollback()
    finally:
        if cursor:
            cursor.close()
        if conn:
            conn.close()


# 添加数据
def add(sql):
    return change(sql, isInsert=True)


# 更改数据
def upd(sql):
    return change(sql)


# 删除数据
def delete(sql):
    return change(sql)


# 查询一个
def get_one(sql):
    try:
        conn = get_conn()
        cursor = conn.cursor(cursor=DictCursor)
        cursor.execute(sql)
        return cursor.fetchone()
    except Exception as e:
        print(e)
    finally:
        if cursor:
            cursor.close()
        if conn:
            conn.close()


# 查全部
def get_all(sql):
    try:
        conn = get_conn()
        cursor = conn.cursor(cursor=DictCursor)
        cursor.execute(sql)
        return cursor.fetchall()
    except Exception as e:
        print(e)
    finally:
        if cursor:
            cursor.close()
        if conn:
            conn.close()


if __name__ == '__main__':
    # 一条sql语句，后面不加,
    # 添加数据
    # sql = 'insert into stu_new (sname, sgender, sage, score, class) values ("赵er山", 2, 65, 77, "一年二班")'
    # ret = add(sql)
    # print(ret)

    # 更改数据
    # sql = "update stu_new set sname='胡辣汤' where sno = 9"
    # ret = upd(sql)
    # print(ret)

    # 删除数据
    # sql = 'delete from stu_new where sno = 12'
    # ret = delete(sql)
    # print(ret)

    # 查询一个
    # sql = 'select * from stu_new where sno = 9'
    # ret = get_one(sql)
    # print(ret)

    # 查所有
    # sql = 'select * from stu_new'
    # ret = get_all(sql)
    # print(ret)