02_pymysql增删改查

import pymysql


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


def change(sql, isInsert=False):
    try:
        conn = pymysql.connect(
            user="root",
            password="1234",
            host="localhost",
            database="spider",  # 链接mysql里面的哪一个,要修改
            port=3306,
        )
        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)


if __name__ == '__main__':

    sql = 'insert into stu_new (sname, sgender, sage, score, class) values ("赵本山", 1, 65, 90, "一年二班"),'
    ret = add(sql)
    print(ret)
    # ("范伟", 1, 60, 23, "一年二班"),
    # ("小沈阳", 1, 31, 9, "一年一班"),
    # ("沈春阳", 2, 28, 60, "一年一班"),
    # ("严启航", 1, 18, 99, "一年一班"),
    # ("赵四", 1, 38, -9, "一年三班"),
    # ("武则天", 2, 1300, 67, "一年三班")'