关于pymysql
PyMySQL 是在 Python3.x 版本中用于连接 MySQL 服务器的一个库，Python2中则使用mysqldb。
Django中也可以使用PyMySQL连接MySQL数据库。
安装：
pip install pymysql
注意：
在进行本文以下内容之前需要注意：
• 你有一个MySQL数据库，并且已经启动。
• 你有可以连接该数据库的用户名和密码
• 你有一个有权限操作的database
连接数据库：
方式一：（有bug）
import pymysql inputName = input('name:') inputPwd = input('pwd:') # 连接database conn = pymysql.connect(host="localhost", database="db1", user="root", password="123456", charset="utf8") cursor = conn.cursor() # 得到一个可以执行SQL语句的光标对象 sql = "select * from userinfo where name='%s' and pwd = '%s';" % (inputName, inputPwd) # '%s'这个引号要带上 print(sql) ret = cursor.execute(sql) # 执行SQL语句 cursor.close() # 关闭光标 conn.close() # 关闭数据库连接 if ret: print('登录成功') else: print('登录失败')
运行：
第一种情况: 输入: name: zhou' -- kitty12322343 # 相当于把后面的都给注释了 pwd: abcd 输出: select * from userinfo where name='zhou' -- kitty12322343' and pwd = 'abcd'; 登录成功 第二种情况: 输入: name:kitty' or 1=1 -- haha # 1=1就是True pwd:abcd 输出: select * from userinfo where name='kitty' or 1=1 -- haha' and pwd = 'abcd'; 登录成功
正确的链接数据库姿势:
import pymysql inputName = input('name:') inputPwd = input('pwd:') # 连接database conn = pymysql.connect(host="localhost", database="db1", user="root", password="123456", charset="utf8") cursor = conn.cursor() # 得到一个可以执行SQL语句的光标对象 sql = "select * from userinfo where name=%s and pwd=%s;" # %s也不用加引号了 print(sql) ret = cursor.execute(sql, [inputName, inputPwd]) # 执行SQL语句 # 通过这种方式传参数,就没有bug了 ============================ 字典方式: cursor.execute("select * from userinfo where name=%(user)s and pwd=(pwd)%s;" , {'user'='zhou', 'pwd'='123'} ) ============================ cursor.close() # 关闭光标 conn.close() # 关闭数据库连接 if ret: print('登录成功') else: print('登录失败')
增删改查操作
1、增
# 导入pymysql模块 import pymysql # 连接database conn = pymysql.connect(host=“你的数据库地址”, user=“用户名”,password=“密码”,database=“数据库名”,charset=“utf8”) # 得到一个可以执行SQL语句的光标对象 cursor = conn.cursor() sql = "INSERT INTO USER1(name, age) VALUES (%s, %s);" username = "zhou" age = 18 # 执行SQL语句 cursor.execute(sql, [username, age]) # 提交事务 conn.commit() cursor.close() conn.close()
插入数据失败回滚:
-------------- conn.rollback()
# 导入pymysql模块 import pymysql # 连接database conn = pymysql.connect(host=“你的数据库地址”, user=“用户名”,password=“密码”,database=“数据库名”,charset=“utf8”) # 得到一个可以执行SQL语句的光标对象 cursor = conn.cursor() sql = "INSERT INTO USER1(name, age) VALUES (%s, %s);" username = "zhou" age = 18 try: # 执行SQL语句 cursor.execute(sql, [username, age]) # 提交事务 conn.commit() except Exception as e: # 有异常，回滚事务 conn.rollback() cursor.close() conn.close()
批量执行操作:
-------------executemany()
# 导入pymysql模块 import pymysql # 连接database conn = pymysql.connect(host=“你的数据库地址”, user=“用户名”,password=“密码”,database=“数据库名”,charset=“utf8”) # 得到一个可以执行SQL语句的光标对象 cursor = conn.cursor() sql = "INSERT INTO USER1(name, age) VALUES (%s, %s);" data = [("zhou", 18), ("zhi", 20), ("long", 21)] try: # 批量执行多条插入SQL语句 cursor.executemany(sql, data) # 提交事务 conn.commit() except Exception as e: # 有异常，回滚事务 conn.rollback() cursor.close() conn.close()
获取插入数据的ID(关联表格操作时会用到)
------------- cursor.lastrowid
# 导入pymysql模块 import pymysql # 连接database conn = pymysql.connect(host=“你的数据库地址”, user=“用户名”,password=“密码”,database=“数据库名”,charset=“utf8”) # 得到一个可以执行SQL语句的光标对象 cursor = conn.cursor() sql = "INSERT INTO USER1(name, age) VALUES (%s, %s);" username = "zhou" age = 18 try: # 执行SQL语句 cursor.execute(sql, [username, age]) # 提交事务 conn.commit() # 提交之后，获取刚插入的数据的ID last_id = cursor.lastrowid except Exception as e: # 有异常，回滚事务 conn.rollback() cursor.close() conn.close()
2、删
# 导入pymysql模块 import pymysql # 连接database conn = pymysql.connect(host=“你的数据库地址”, user=“用户名”,password=“密码”,database=“数据库名”,charset=“utf8”) # 得到一个可以执行SQL语句的光标对象 cursor = conn.cursor() sql = "DELETE FROM USER1 WHERE id=%s;" try: cursor.execute(sql, [4]) # 提交事务 conn.commit() except Exception as e: # 有异常，回滚事务 conn.rollback() cursor.close() conn.close()
3、改
# 导入pymysql模块 import pymysql # 连接database conn = pymysql.connect(host=“你的数据库地址”, user=“用户名”,password=“密码”,database=“数据库名”,charset=“utf8”) # 得到一个可以执行SQL语句的光标对象 cursor = conn.cursor() # 修改数据的SQL语句 sql = "UPDATE USER1 SET age=%s WHERE name=%s;" username = "kitty" age = 88 try: # 执行SQL语句 cursor.execute(sql, [age, username]) # 提交事务 conn.commit() except Exception as e: # 有异常，回滚事务 conn.rollback() cursor.close() conn.close()
4、查
查询单条数据：
# 导入pymysql模块 import pymysql # 连接database conn = pymysql.connect(host=“你的数据库地址”, user=“用户名”,password=“密码”,database=“数据库名”,charset=“utf8”) # 得到一个可以执行SQL语句的光标对象 cursor = conn.cursor() # 查询数据的SQL语句 sql = "SELECT id,name,age from USER1 WHERE id=1;" # 执行SQL语句 cursor.execute(sql) # 获取单条查询数据 ret = cursor.fetchone() cursor.close() conn.close() # 打印下查询结果 print(ret) # (1, 'zhou', '123456') 从光标开始处,拿到一条数据,拿到数据后,光标向下移动一行
查询多条数据：
# 导入pymysql模块 import pymysql # 连接database conn = pymysql.connect(host=“你的数据库地址”, user=“用户名”,password=“密码”,database=“数据库名”,charset=“utf8”) # 得到一个可以执行SQL语句的光标对象 cursor = conn.cursor() # 查询数据的SQL语句 sql = "SELECT id,name,age from USER1;" # 执行SQL语句 cursor.execute(sql) # 这个得到一个数据总数 # 获取多条查询数据 ret = cursor.fetchall() # 拿到所有的数据 cursor.close() conn.close() # 打印下查询结果 print(ret)
进阶用法：
# 可以获取指定数量的数据 cursor.fetctmany(3) # 从光标开始处,拿到3条数据, 拿到数据后,光标向下移动3行 # 光标按绝对位置移动1 绝对定位:从0开始移动光标 cursor.scroll(1, mode="absolute") 只能是: 正数 # 光标按照相对位置(当前位置)移动1 相对定位:相对于当前光标位置 (默认) cursor.scroll(1, mode="relative") 正数:向下 负数:向上
示例：
import pymysql inputName = input('name:') inputPwd = input('pwd:') conn = pymysql.connect(host="localhost", database="db1", user="root", password="123456", charset="utf8") cursor = conn.cursor() # 得到一个可以执行SQL语句的光标对象 sql = "select * from userinfo;" cursor.execute(sql) ret1 = cursor.fetchmany(3) # 取2个,光标位置准备取第3条数据 cursor.scroll(1,mode='absolute') ret2 = cursor.fetchone() # 光标绝对移动,从0开始移动,所以取第2条数据 cursor.close() # 关闭光标 conn.close() # 关闭数据库连接 print(ret1) # ((1, 'zhou', '123456'), (2, 'long', '123456')), (3, 'zhi', '123456')) print(ret2) # (2, 'long', '123456')