pymysql的使用-python连接MySQL数据库代码

一、pymysql使用

1. 安装：pip install pymysql
2. 导包：import pymysql
3. 调用connect()函数创建连接对象：conn = pymysql.connect(参数列表)，参数如下
   1. host：连接mysql的主机，本机为‘localhost’
   2. port：连接mysql主机的端口，默认3306
   3. user：连接的用户名
   4. password：连接的用户名密码
   5. database：连接的数据库名
   6. charset：通信采用的编码方式，推荐使用utf8
4. 获取游标对象：cursor = conn.crusor()，目标是要执行SQL语句，完成对数据库的增删改查操作
5. 执行SQL语句：count = cursor.execute(sql)，返回受影响的行数，主要用于执行insert、update、delete、select等SQL语句
   1. 获取查询结果集中的第一条数据：cursor.fetchone()，返回一个元组
   2. 获取查询结果集中的所有数据：cursor.fetchall()，返回一个元组
6. 获取查询结果集：result = cursor.fetchall()
7. 将修改操作提交到数据库：conn.commit()
8. 回滚/撤销数据：conn.rollback()
9. 关闭游标：cursor.close()
10. 关闭连接：conn.close()

 二、pymysql完成对数据库表中数据的查询操作

import pymysql
# 创建连接对象
conn = pymysql.connect(host="localhost", port=3306, user='root', password='******', database='python_sql', charset='utf8')
cur = conn.cursor()   # 获取游标对象
sql = 'select * from students;'  # 查询SQL语句
rowCount = cur.execute(sql)  # 执行SQL语句，返回执行过程中影响的行数
print('SQL语句执行过程中影响的行数为%d' % rowCount)
# print(cur.fetchone())  # 取出结果集中的一条数据，输出：(2, 'XY', 20, None, '男', b'\x00', 3)
for i in cur.fetchall():  # 取出所有数据
    print(i)
cur.close()  # 关闭游标
conn.close()  # 关闭连接
 
输出：
SQL语句执行过程中影响的行数为6
(2, 'XY', 20, None, '男', b'\x00', 3)
(3, 'YX', 18, None, '女', b'\x00', 2)
(4, 'XY', None, None, None, b'\x01', None)
(7, 'XYY', 18, None, '女', b'\x01', None)
(8, 'YXX', 18, None, '女', b'\x00', None)
(9, 'XX', 25, None, None, b'\x01', None)

三、 pymysql完成对数据库表中数据的增、删、改操作

import pymysql
# 创建连接对象
conn = pymysql.connect(host="localhost", port=3306, user='root', password='******', database='python_sql', charset='utf8')
cur = conn.cursor()   # 获取游标对象
# try:
# sql = "insert into students(id,name,age,gender,sid) values(10,'yy', 19, '男', 3);"   # 添加
# sql = "delete from students where id=7;"  # 删除
sql = "update students set age=21 where id=8;"  # 修改
rowCount = cur.execute(sql)  # 执行SQL语句，返回执行过程中影响的行数
print('SQL语句执行过程中影响的行数为%d' % rowCount)
conn.commit()
# except Exception as e:
    # conn.rollback()  # 撤销上面的SQL操作
cur.close()  # 关闭游标
conn.close()  # 关闭连接

四、防止SQL注入

SQL注入：用户提交带有恶意的数据与SQL语句进行字符串方式的拼接，从而影响了SQL语句的语义，最终产生数据泄露的现象

• 防止SQL注入：SQL语句参数化
  • SQL语句中的参数使用%s来占位，此处非字符串格式化操作
  • 将SQL语句%s占位所需要的参数存在一个列表中，把参数列表传递给execute方法中的第二个参数

import pymysql
def main():
    search_name = input('请输入姓名：')
    # 创建连接对象
    conn = pymysql.connect(host="localhost", port=3306, user='root', password='******', database='python_sql', charset='utf8')
    cur = conn.cursor()   # 获取游标对象
 
    # 非安全方式
    # sql = "select * from students where name='%s';" % search_name  # 查询SQL语句
    # print('sql语句为%s' % sql)
    # rowCount = cur.execute(sql)  # 执行SQL语句，返回执行过程中影响的行数
 
    # 安全方式
    params = [search_name]  # 构造参数列表
    rowCount = cur.execute("select * from students where name=%s", params)  # 若有多个参数，需参数化，params = [参数1, 参数2....]，此时sql语句中有多个%s，其中%s无需带引号
 
    print('SQL语句执行过程中影响的行数为%d' % rowCount)
    # result = cur.fetchone()
    result = cur.fetchall()
    print('查询结果为:\n', result)
    cur.close()  # 关闭游标
    conn.close()  # 关闭连接
 
if __name__ == '__main__':
    main()
 
输出：
------------------------------------------非安全方式--获取一条数据-------------------------
请输入姓名：XY
sql语句为select * from students where name='XY';
SQL语句执行过程中影响的行数为2
查询结果为:
 (2, 'XY', 20, None, '男', b'\x00', 3)
------------------------------------------安全方式--获取所有------------------------------
请输入姓名：XY
SQL语句执行过程中影响的行数为2
查询结果为:
 ((2, 'XY', 20, None, '男', b'\x00', 3), (4, 'XY', None, None, None, b'\x01', None))
-----------------------------------------------------------------------------------------

学习导航：http://xqnav.top/