pymysql模块

Posted 2021-01-25 %华&仔%

pymysql模块

一、模块方法

1.安装pymysql模块

pip install pymysql

2.卸载pymysql模块

pip uninstall pymysql

 

3.连接

db = pymsql.connect()    创建一个连接

参数：
host - 数据库服务器所在的主机
user - 以登录身份登录的用户名
password - 要使用的密码。
database - 要使用的数据库，None不使用特定的数据库。
port - 要使用的MySQL端口，默认通常都可以。（默认值：3306）
bind_address - 当客户端具有多个网络接口时，请指定从中连接到主机的接口。参数可以是主机名或IP地址。
unix_socket - 您可以选择使用unix套接字而不是TCP / IP。
read_timeout - 以秒为单位读取连接的超时（默认值：无 - 无超时）
write_timeout - 以秒为单位写入连接的超时（默认值：无 - 无超时）
charset - 你要使用的Charset。
sql_mode - 要使用的默认SQL_MODE。
read_default_file - 指定my.cnf文件以从[client]部分下读取这些参数。
conv - 使用转换字典而不是默认字典。这用于提供类型的自定义编组和解组。见转换器。
use_unicode - 是否默认为unicode字符串。对于Py3k，此选项默认为true。
client_flag - 要发送给MySQL的自定义标志。在constants.CLIENT中查找潜在值。
cursorclass - 要使用的自定义游标类。
init_command - 建立连接时要运行的初始SQL语句。
connect_timeout - 连接时抛出异常之前的超时。（默认值：10，最小值：1，最大值：31536000）
ssl - 类似于mysql_ssl_set（）参数的参数的dict。
read_default_group - 要在配置文件中读取的组。
compress - 不支持
named_pipe - 不支持
autocommit - 自动提交模式。无表示使用服务器默认值。（默认值：False）
local_infile - 允许使用LOAD DATA LOCAL命令的布尔值。（默认值：False）
max_allowed_pa​​cket - 发送到服务器的最大数据包大小（以字节为单位）。（默认值：16MB）仅用于限制小于默认值（16KB）的“LOAD LOCAL INFILE”数据包的大小。
defer_connect - 不要明确连接contruction - 等待连接调用。（默认值：False）
auth_plugin_map - 插件名称的一个字典，用于处理该插件的类。该类将Connection对象作为构造函数的参数。该类需要一个认证方法，将认证包作为参数。对于对话框插件，可以使用提示（echo，prompt）方法（如果没有authenticate方法）从用户返回字符串。（实验）
server_public_key - SHA256 authenticnticaiton插件公钥值。（默认：无）
db - 数据库的别名。（与MySQLdb兼容）
passwd - 密码的别名。（与MySQLdb兼容）
binary_prefix - 在字节和bytearray上添加_binary前缀。（默认值：False）

常用连接参数：
    host:数据库服务器的主机名或者IP地址（localhost,IP,10.127.1.1）
    user:数据库服务器的登录用户名
    password:密码
    database:数据库名称
    port:   数据库端口，默认3306
    charset:设置字符集名称

　　

写法1：
conn = pymysql.connect(host=\'localhost\',user=\'root\',password=\'\',database=\'userinfo\',charset=\'utf8\')

写法2：
db_config = {
    \'host\':\'localhost\',
    \'user\':\'root\',
    \'password\':\'\',
    \'database\':\'userinfo\',
    \'charset\':\'utf8\'
}
conn = pymysql.connect(**db_config)

　　

4.conn(连接)对象支持的方法

conn = pymysql.connect(host=\'localhost\',user=\'root\',password=\'\',database=\'test\',charset=\'utf8\')

conn.cursor(cursor=None)   #创建一个游标
    cursor默认为None,设置为pymysql.cursors.DictCursor，返回值为字典形式
conn.commit()   #提交当前事务 在增、删、改数据时，才会保存数据。
conn.rollback() #回滚当前事务
conn.close()    #关闭连接

　　

5.cursor(游标)对象支持的方法

cursor.execute(query,args=None)    #执行一个SQL语句
    参数：
        query 要执行查询的SQL语句
        args  传递的参数可为 元组、列表或字典
    返回值：
        返回值为受影响的行数，类型为int;
        受影响的行数也可以用cursor.rowcount获得

cursor.executemany()    #执行多条语句

cursor.fetchone()   #取得结果的一条数据
cursor.fetchmany(size=None)  #取得结果的多条数据，指定size的值可以设置取多少数据
cursor.fetchall()   #取得结果的所有数据

cursor.lastrowid    #获取上次插入数据的自增ID值

cursor.close()  #关闭游标对象

　　

二、pymysql操作mysql数据库

创建数据库 userinfo
create database userinfo default charset utf8;

创建数据表 user
Create Table: CREATE TABLE `user` (
  `id` int(11) NOT NULL AUTO_INCREMENT,
  `username` char(32) NOT NULL,
  `password` char(64) NOT NULL,
  PRIMARY KEY (`id`)
) ENGINE=MyISAM AUTO_INCREMENT=4 DEFAULT CHARSET=utf8

插入数据
insert into user(username,password) values (\'felix\',\'123\');

　

 

1.增

 (1)增加1条记录

import pymysql
conn = pymysql.connect(host=\'localhost\',user=\'root\',password=\'\',database=\'userinfo\',charset=\'utf8\') #连接数据库获得一个连接
cursor = conn.cursor()  #创建一个游标
# sql = "insert into user(username,password) values(\'test\',\'123\')"  #SQL语句
sql = "insert into user(username,password) values(%s,%s)"
try:
    # res = cursor.execute(sql)
    res = cursor.execute(sql, (\'test\', 123))    #执行一个SQL语句
    print(\'受影响的行数：\',res)
    conn.commit()   #提交数据，此操作使用与增、删、改操作
except BaseException as err:
    print(\'错误：\',err)
    conn.rollback() #事务回滚
finally:
    cursor.close()  #关闭游标
    conn.close()    #关闭连接

　　 

执行结果：

 

(2)增加多条记录

import pymysql
conn = pymysql.connect(host=\'localhost\',user=\'root\',password=\'\',database=\'userinfo\',charset=\'utf8\')
cursor = conn.cursor()
# sql = "insert into user(username,password) values(\'test\',\'123\')"
sql = "insert into user(username,password) values(%s,%s)"
try:
    # res = cursor.execute(sql)
    res = cursor.executemany(sql, [(\'test1\', \'123\'), (\'test2\', \'123\'), (\'test3\', \'123\')])
    print(\'受影响的行数：\',res)
    conn.commit()   #提交数据
except BaseException as err:
    print(\'错误：\',err)
    conn.rollback()
finally:
    cursor.close()
    conn.close()

　

 

(3)获取自增ID值

　　cursor.lastrowid　获取自增ID值

import pymysql
conn = pymysql.connect(host=\'localhost\',user=\'root\',password=\'\',database=\'userinfo\',charset=\'utf8\') #连接数据库获得一个连接
cursor = conn.cursor()  #创建一个游标
sql = "insert into user(username,password) values(%s,%s)"
try:
    res = cursor.execute(sql, (\'test\', 123))    #执行一个SQL语句
    print(\'受影响的行数：\',res)
    conn.commit()
    last_insert_id = cursor.lastrowid   #使用cursor.lastrowid 获取刚刚插入的自增ID
    print(\'刚刚插入的自增ID为：{}\'.format(last_insert_id))
except BaseException as err:
    print(\'错误：\',err)
    conn.rollback() #事务回滚
finally:
    cursor.close()  #关闭游标
    conn.close()    #关闭连接

执行结果：
受影响的行数： 1
刚刚插入的自增ID为：11

　　　　　  

2.删

 (1)删除一个记录

import pymysql
conn = pymysql.connect(host=\'localhost\',user=\'root\',password=\'\',database=\'userinfo\',charset=\'utf8\') #连接数据库获得一个连接
cursor = conn.cursor()  #创建一个游标
# sql = "delete from"  #SQL语句
# sql = "delete from user where =3"  #SQL语句
sql = "delete from user where username = %s"  #SQL语句

try:
    # res = cursor.execute(sql)
    res = cursor.execute(sql, \'test2\')    #执行一个SQL语句
    print(\'受影响的行数：\',res)
    conn.commit()   #提交数据，此操作使用与增、删、改操作
except BaseException as err:
    print(\'错误：\',err)
    conn.rollback() #事务回滚
finally:
    cursor.close()  #关闭游标
    conn.close()    #关闭连接

 

(2)删除多个记录

import pymysql
conn = pymysql.connect(host=\'localhost\',user=\'root\',password=\'\',database=\'userinfo\',charset=\'utf8\') #连接数据库获得一个连接
cursor = conn.cursor()  #创建一个游标
sql = "delete from user where username = %s"  #SQL语句
try:
    res = cursor.executemany(sql, (\'alina\',\'felix\'))    #执行多个SQL语句
    print(\'受影响的行数：\',res)
    conn.commit()   #提交数据，此操作使用与增、删、改操作
except BaseException as err:
    print(\'错误：\',err)
    conn.rollback() #事务回滚
finally:
    cursor.close()  #关闭游标
    conn.close()    #关闭连接

 

3.改

 (1)更改1条记录

import pymysql
conn = pymysql.connect(host=\'localhost\',user=\'root\',password=\'\',database=\'userinfo\',charset=\'utf8\')
cursor = conn.cursor()
# sql = "update user set username=\'alina\',password=\'t123\' where id = 3"
sql = "update user set username= %s,password = %s where id = %s "
try:
    #res = cursor.execute(sql)
    res = cursor.execute(sql, (\'alina\',\'t123\',3))
    print(\'受影响的行数：\',res)
    conn.commit()   #提交数据
except BaseException as err:
    print(\'错误：\',err)
    conn.rollback()
finally:
    cursor.close()
    conn.close()

 

(2)更改多条记录

import pymysql
conn = pymysql.connect(host=\'localhost\',user=\'root\',password=\'\',database=\'userinfo\',charset=\'utf8\')
cursor = conn.cursor()
sql = "update user set password = %s where id = %s "
try:
    res = cursor.executemany(sql, [(\'123456\',1),(\'123456\',2),(\'123456\',3)])
    print(\'受影响的行数：\',res)
    conn.commit()   #提交数据
except BaseException as err:
    print(\'错误：\',err)
    conn.rollback()
finally:
    cursor.close()
    conn.close()

　　

4.查

 (1)获取1条数据　　

cursor.fetchone()

import pymysql
conn = pymysql.connect(host=\'localhost\',user=\'root\',password=\'\',database=\'userinfo\',charset=\'utf8\')
cursor = conn.cursor()
sql = "select * from user where username=%s and password=%s"
try:
    user = input(\'>>>\')
    passwd = input(\'>>>\')
    res = cursor.execute(sql,(user,passwd))
    print(\'受影响的行数：\',res)
    ret = cursor.fetchone()
    print(ret)
    if ret:
        print(\'登录成功\')
        print(\'用户名：{} 密码：{}\'.format(ret[1],ret[2]))
    else:
        print(\'用户名密码错误！登录失败！\')
except BaseException as err:
    print(\'错误：\',err)
finally:
    cursor.close()
    conn.close()

　　

注意：
不要使用字符串拼接方式使用SQL语句，防止SQL注入问题，如下实例：uuuu\' or 1=1 --

SQL注入问题：

import pymysql
conn = pymysql.connect(host=\'localhost\',user=\'root\',password=\'\',database=\'userinfo\',charset=\'utf8\')
cursor = conn.cursor()

try:
    user = input(\'>>>\')
    passwd = input(\'>>>\')
    sql = "select * from user where username=\'%s\' and password=\'%s\'" % (user, passwd)
    res = cursor.execute(sql)
    print(\'受影响的行数：\',res)
    ret = cursor.fetchone()
    print(ret)
    if ret:
        print(\'登录成功\')
        print(\'用户名：{} 密码：{}\'.format(ret[1],ret[2]))
    else:
        print(\'用户名密码错误！登录失败！\')
except BaseException as err:
    print(\'错误：\',err)
finally:
    cursor.close()
    conn.close()

执行结果：
>>>uuuuu\' or 1=1 -- 
>>>
受影响的行数： 3
(1, \'felix\', \'123456\')
登录成功
用户名：felix 密码：123456

 

(2)获取多条数据　　

　　cursor.fetchmany()　　获取多条数据

　　cursor.fetchall()　　获取所有数据

 

import pymysql
conn = pymysql.connect(host=\'localhost\',user=\'root\',password=\'\',database=\'userinfo\',charset=\'utf8\')
cursor = conn.cursor()

try:
    user = input(\'>>>\')
    passwd = input(\'>>>\')
    sql = "select * from user where username=\'%s\' and password=\'%s\'" % (user, passwd)
    res = cursor.execute(sql)
    print(\'受影响的行数：\',res)
    # ret = cursor.fetchone()   #获取1条数据
    # ret = cursor.fetchmany(3) #获取指定条数据
    # ret = cursor.fetchmany(size=4)    #获取指定条数据
    ret = cursor.fetchall()     #获取所有数据
    print(ret)
    if ret:
        print(\'登录成功\')
        print(\'用户名：{} 密码：{}\'.format(ret[0][1],ret[0][2]))
    else:
        print(\'用户名密码错误！登录失败！\')
except BaseException as err:
    print(\'错误：\',err)
finally:
    cursor.close()
    conn.close()
执行结果：
>>>test
>>>123456
受影响的行数： 4
((3, \'test\', \'123456\'), (4, \'test\', \'123456\'), (5, \'test\', \'123456\'), (6, \'test\', \'123456\'))
登录成功
用户名：test 密码：123456

 

(3)fetch获取数据类型

　　fetch默认获取的数据是元祖类型，如果想要获取字典类型的数据，则设置游标为字典类型

　　cursor = conn.cursor(cursor=pymysql.cursors.DictCursor) 　　　参数cursor=pymysql.cursors.DictCursor 结果以字典形式返回

import pymysql
conn = pymysql.connect(host=\'localhost\',user=\'root\',password=\'\',database=\'userinfo\',charset=\'utf8\')
cursor = conn.cursor(cursor=pymysql.cursors.DictCursor) #cursor=pymysql.cursors.DictCursor 结果以字典形式返回
try:
    sql = "select * from user"
    res = cursor.execute(sql)
    print(\'受影响的行数：\',res)
    ret = cursor.fetchall()
    print(ret)
except BaseException as err:
    print(\'错误：\',err)
finally:
    cursor.close()
    conn.close()
执行结果：
受影响的行数： 3
[{\'id\': 1, \'username\': \'felix\', \'password\': \'123456\'}, {\'id\': 2, \'username\': \'alina\', \'password\': \'123456\'}, {\'id\': 3, \'username\': \'test\', \'password\': \'123456\'}]

　　

(4)指定游标位置

　　在fetch数据时按照顺序进行，可以使用cursor.scroll(num,mode)来移动游标位置，如：

cursor.scroll(3,mode=\'absolute\')　　 #相对绝对位置移动 即都是相对首行位置移动，不受当前位置影响
cursor.scroll(3, mode=\'relative\')　　 #相对当前位置移动，受当前位置影响

　　

import pymysql
db_config = {
    \'host\':\'localhost\',
    \'user\':\'root\',
    \'password\':\'\',
    \'database\':\'userinfo\',
    \'charset\':\'utf8\'
}
conn = pymysql.connect(**db_config)
cursor = conn.cursor(cursor=pymysql.cursors.DictCursor)

sql = "select * from user"
res = cursor.execute(sql)
print(\'受影响的行数：\',res)
ret = cursor.fetchone()
print(\'未指定游标位置获取第1条数据\',ret)
ret = cursor.fetchone()
print(\'未指定游标位置获取第2条数据\',ret)

cursor.scroll(3,mode=\'absolute\')  #相对绝对位置移动 即相对首行位置移动
ret = cursor.fetchone()
print(\'mode=absolute 相对首行位置移动后\',ret)

cursor.scroll(-2, mode=\'relative\')   #相对当前位置移动 当前位置为第4行
ret = cursor.fetchone()
print(\'mode=relative 相对当前位置移动后\',ret)

cursor.scroll(3, mode=\'relative\')   #相对当前位置移动 当前位置为第4行
ret = cursor.fetchone()
print(\'mode=relative 相对当前位置移动后\',ret)

cursor.scroll(3,mode=\'absolute\')  #相对绝对位置移动 即相对首行位置移动
ret = cursor.fetchone()
print(\'mode=absolute 相对首行位置移动后\',ret)

cursor.close()
conn.close()
执行结果：
受影响的行数： 9
未指定光标位置获取第1条数据 {\'id\': 1, \'username\': \'felix\', \'password\': \'123456\'}
未指定光标位置获取第2条数据 {\'id\': 2, \'username\': \'alina\', \'password\': \'123456\'}
mode=absolute 相对首行位置移动后 {\'id\': 4, \'username\': \'test\', \'password\': \'123\'}
mode=relative 相对当前位置移动后 {\'id\': 3, \'username\': \'test\', \'password\': \'123456\'}
mode=relative 相对当前位置移动后 {\'id\': 7, \'username\': \'test\', \'password\': \'123\'}
mode=absolute 相对首行位置移动后 {\'id\': 4, \'username\': \'test\', \'password\': \'123\'}