Python连接mysql数据库
pymsql介绍
PyMySQL 是在 Python3.x 版本中用于连接 MySQL 服务器的一个库，Python2中则使用mysqldb。
Django中也可以使用PyMySQL连接MySQL数据库。
pymsql安装
pip install pymysql
登陆验证中,之前学习的都是建立一个文件,通过把用户信息保存在这个文件中,然后一行一行的去读,做校验
# 获取用户输入 name=input(\'用户名:\') pwd=input(\'密码:\') # 校验用户输入的用户名和密码是否正确 with open("userinfo.txt") as f: for line in f: # 按照空格去分隔每一行,把值分别赋值给name_tmp和pwd_tmp name_tmp,pwd_tmp=line.strip().split(" ") # 如果能匹配上用户的输入,就表示登陆成功 if name_tmp ==name and pwd_tmp==pwd: print("登陆成功") break else: print(\'登录失败\')
上面的代码是通过文件操作来完成登陆验证的,但是有一个问题是每次都要一行一行的去读,而不能像使用数据库那样可以通过索引去找到要验证一条信息
下面引出通过pymsql让Python代码与mysql数据库建立连接
pymsql的使用
使用方法
# 导入pymysql模块 import pymysql # 连接database conn = pymysql.connect(host=“你的数据库地址”, user=“用户名”,password=“密码”,database=“数据库名”,charset=“utf8”) # 得到一个可以执行SQL语句的光标对象 cursor = conn.cursor() # 定义要执行的SQL语句 sql = """ CREATE TABLE USER1 ( id INT auto_increment PRIMARY KEY , name CHAR(10) NOT NULL UNIQUE, age TINYINT NOT NULL )ENGINE=innodb DEFAULT CHARSET=utf8; """ # 执行SQL语句 cursor.execute(sql) # 关闭光标对象 cursor.close() # 关闭数据库连接 conn.close()
import pymysql # 校验 用户输入的用户名和密码是否正确 # 去数据库里取数据做判断 # 1.连上数据库,获得的conn是连接 conn = pymysql.connect(host="localhost",database="s8",user="root",password="123456",charset="utf8") # 只有连接还不行,还需要获取光标,让我能够输入sql语句并执行 cursor=conn.cursor() #cursor是光标 # 2.执行sql语句 sql="select * from userinfo where id>3;" #要执行的sql语句写出来 ret = cursor.execute(sql) #execute是执行的意思,执行sql语句 # 3.关闭光标和连接 cursor.close() conn.close() print("%s row in set (0.00 sec)" % ret) \'\'\' 在这个例题中,mysql相当于是serversocket端,pycharm是客户端 客户端需要先于服务端建立连接,也就是本例题中的第1步 建立好连接后需要建立通信,也就是第2步,把我要执行的sql给执行了,并把 执行结果返回给变量ret,并保存在变量中 通信结束后需要断开连接,但在本例题中也获得了光标,需要把光标也关闭 \'\'\'
sql注入问题
import pymysql name=input(\'用户名:\') pwd=input(\'密码:\') # 校验用户输入的用户名和密码是否正确 # 去数据库里取数据做判断 # 1.连接上数据库 conn=pymysql.connect(host=\'localhost\',database=\'s8\',user=\'root\',password=\'123456\',charset=\'utf8\') # 2.获取光标 cursor=conn.cursor() # 3.写出要执行的sql语句 sql="select * from userinfo where name= \'%s\' and pwd=\'%s\';" % (name,pwd) print(sql) # 4.执行sql ret = cursor.execute(sql) #获取影响的行数 # 5.关闭光标和连接 cursor.close() conn.close() if ret: print(\'登陆成功\') else: print(\'登陆失败\') \'\'\' 这个例子会出现一个问题就是sql注入 就是当用户名上输入alex\' or 1=1 -- hehehe,密码不输入也会登陆成功 当我正常输入用户名和密码的时候是可以正常判断执行的 当我在用户名上输入alex\'or 1=1 --hehe时,实际在mysql数据库中的显示的sql 语句是select * from userinfo where name= \'alex\' or 1=1 -- hehe\' and pwd=\'\'; 1=1为真 --在mysql中是注释,所以这个sql语句会被当做select * from userinfo where name= \'alex\' or 1=1去执行,所以,在没有输入密码的情况下就可以显示登陆成功,这就是sql注入 \'\'\'
解决sql注入问题
# 原来是我们对sql进行字符串拼接 # sql="select * from userinfo where name=\'%s\' and password=\'%s\'" %(user,pwd) # print(sql) # res=cursor.execute(sql) #改写为（execute帮我们做字符串拼接，我们无需且一定不能再为%s加引号了） sql="select * from userinfo where name=%s and password=%s" #！！！注意%s需要去掉引号，因为pymysql会自动为我们加上 res=cursor.execute(sql,[user,pwd]) #pymysql模块自动帮我们解决sql注入的问题，只要我们按照pymysql的规矩来。
import pymysql # 获取用户输入 name= input(\'用户名:\') pwd =input(\'密码:\') # 校验用户输入的用户名和密码是否正确 # 去数据库里取数据做判断 # 1.连接上数据库 conn=pymysql.connect(host=\'localhost\',database=\'s8\',user=\'root\',password=\'123456\',charset=\'utf8\') # 2.获取光标 cursor=conn.cursor() # 3.要执行的sql语句 sql="select * from userinfo where name=\'%s\' and pwd=\'%s\';"%(name,pwd) print(sql) # 4.让pymysql拼接字符串 ret=cursor.execute(sql,[name,pwd]) #获取影响的行数 这里也可以写成ret=cursor.execute(sql,(name,pwd)) # 5.关闭光标和连接 cursor.close() conn.close() if ret: print(\'登陆成功\') else: print(\'登陆失败\') \'\'\' 第四步中execute中写上要执行的sql语句,要判断的字段,pymsql会自动的去判断输入的信息, 这样就避免了sql注入的问题 \'\'\'
通过pymsq增加数据库中的数据
import pymysql # 连接 conn=pymysql.connect(host="localhost",user=\'root\',password="123456",charset="utf8",database="s8") # 获取光标 cursor=conn.cursor() # 写sql语句 sql="insert into userinfo (name,pwd) VALUE (%s,%s);" username=\'egon2\' password=\'dashabi\' try: # 执行sql语句 cursor.execute(sql,(username,password)) conn.commit() #把修改的数据提交到数据库 except Exception as e: conn.rollback() #捕捉到错误就回滚 cursor.close() conn.close()
通过pymsql删除数据库中的数据
import pymysql # 连接 conn=pymysql.connect(host="localhost",database="s8",user="root",password="123456",charset="utf8") # 获取光标 cursor=conn.cursor() # 写sql语句 sql="delete from userinfo where name=%s;" username=\'egon2\' try: # 执行sql语句 cursor.execute(sql,(username,)) conn.commit() #把修改提交到数据库 except Exception as e: conn.rollback() cursor.close() conn.close()
通过pymsql修改数据库中的数据
import pymysql # 连接 username=input(\'用户名\') conn = pymysql.connect(host="localhost", user="root", password="root1234", database="s8", charset="utf8") # 没有- # 获取光标 cursor=conn.cursor() # 写sql语句 sql="update userinfo set pwd=%s where name=%s" print(sql) ret=cursor.execute(sql,(username,)) print(ret) # try: # # 执行sql语句 # cursor.execute(sql,(password,username)) # conn.commit() #把修改提交到数据库 # except Exception as e: # conn.rollback() # print(\'没有\') cursor.close() conn.close()
通过pymsql查询数据库中的数据
import pymysql # 连接 conn = pymysql.connect(host="localhost", user="root", password="123456", database="s8", charset="utf8") # 没有- # 获取光标 cursor = conn.cursor() # 写sql语句 sql = "select * from userinfo;" # 执行sql语句 ret=cursor.execute(sql) print(\'-->\',ret) # 一次取一条 print(cursor.fetchone()) print(cursor.fetchone()) print(cursor.fetchone()) # 一次取所有 # print(cursor.fetchall()) # 自定义取出多少条 # print(cursor.fetchmany(3)) # 进阶用法 print(cursor.fetchone()) print(cursor.fetchall()) print(cursor.fetchone()) #如果取不到会返回None # 移动取数据的光标 cursor.scroll(-2) #默认是相对移动 print(cursor.fetchone()) # 按照绝对位置去移动 cursor.scroll(4,mode=\'absolute\') print(cursor.fetchone()) cursor.close() conn.close()
通过pymsql批量操作数据库中的数据
import pymysql # 连接 conn=pymysql.connect(host=\'localhost\',database=\'s8\',user=\'root\',password=\'123456\',charset=\'utf8\') # 获取光标 cursor=conn.cursor() # 写sql语句 sql="insert into userinfo(name,pwd) values(%s,%s);" user1=\'dww1\' pwd1=\'123456\' user2=\'dww2\' pwd2=\'123456\' # 把所有要插入的信息保存在元祖或列表中 data=((user1,pwd1),(user2,pwd2)) try: # 执行sql语句 cursor.executemany(sql,data) #使用executemany做批量处理 conn.commit() #把修改提交到数据库 except Exception as e : conn.rollback() cursor.close() conn.close()
获取插入数据的ID(关联操作时会用到)
import pymysql # 连接 conn=pymysql.connect(host="localhost",user=\'root\',password=\'123456\',database=\'s8\',charset=\'utf8\') # 获取光标 cursor=conn.cursor() # 写sql语句 sql="insert into userinfo(name,pwd) values(%s,%s);" user1=\'dww\' pwd1=\'111111\' try: # 执行sql语句 cursor.execute(sql,(user1,pwd1)) conn.commit() #把修改提交到数据库 # 拿到我插入的这条数据的id last_id=cursor.lastrowid print(\'--->刚才插入的那条数据的id值是:\',last_id) except Exception as e: conn.rollback() cursor.close() conn.close()
插入数据失败回滚
# 导入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 = "Alex" age = 18 try: # 执行SQL语句 cursor.execute(sql, [username, age]) # 提交事务 conn.commit() except Exception as e: # 有异常，回滚事务 conn.rollback() cursor.close() conn.close()