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