pymysql 增

一、重點：數據庫插入數據須要進行---------事物提交

其它：事物回滾

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()

二、注意：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()

三、批量添加數據

注意：批量添加的數據類型

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()