pymysql 增

1、重点：数据库插入数据需要进行---------事物提交

其它：事物回滚

import pymysql
# 注册用户
print("=========注册用户===========")
username = input("请输入用户名：")
pwd = input("请输入密码:")
# 连接数据库
conn = pymysql.connect(
    host='localhost',
    port=3306,
    user='root',
    password='@WSX3edc',
    database='userinfo',
    charset='utf8'
)
# 获取光标
cursor = conn.cursor()
sql = "insert into info(name, password) value (%s, %s);"
try:
    cursor.execute(sql, [username, pwd])
    conn.commit()   # 添加数据，需要进行事物提交
except Exception as e:
    print(e)
    conn.rollback()     # 事物回滚

cursor.close()
conn.close()

2、注意：lastrowid是 光标的一个属性

import pymysql
# 连接数据库
conn = pymysql.connect(
    host='localhost',
    port=3306,
    user='root',
    password='@WSX3edc',
    database='userinfo',
    charset='utf8'
)
cursor = conn.cursor()
sql1 = "insert into class(title) value (%s);"
sql2 = "insert into student(name, cid) value (%s, %s);"
try:
    cursor.execute(sql1, "E班")
    curr_cid = cursor.lastrowid   # 获取光表最后的row id
    cursor.execute(sql2, ['wt', curr_cid])
    conn.commit()       # 事物提交
except Exception as e:
    print(e)
    conn.rollback()     # 事物回滚

cursor.close()
conn.close()

3、批量添加数据

注意：批量添加的数据类型

import pymysql
# 连接数据库
conn = pymysql.connect(
    host='localhost',
    port=3306,
    user='root',
    password='@WSX3edc',
    database='userinfo',
    charset='utf8'
)
cursor = conn.cursor()
sql = "insert into info(name, password) value (%s, %s);"
data = [('a', 1), ('b', 2), ('c', 3)]
try:
    cursor.executemany(sql, data)   # 批量添加
    conn.commit()
except Exception as e:
    print(e)
    conn.rollback()

cursor.close()
conn.close()