mysql之pymsql的使用

# -*- coding:utf-8 -*-

import pymysql

user = input('請輸入用戶名：')
pwd = input('請輸入密碼：')

# 1.連接
conn = pymysql.connect(host='127.0.0.1', port=3306, user='root', password='123', db='t1', charset='utf8')
print(conn)

# 2.創建游標
cursor = conn.cursor()

#注意%s需要加引號
sql = "select * from t1.userinfo where username='%s' and pwd='%s'" %(user, pwd)
print(sql)

# 3.執行sql語句
cursor.execute(sql)

result=cursor.execute(sql) #執行sql語句，返回sql查詢成功的記錄數目
print(result)

# 關閉連接，游標和連接都要關閉
cursor.close()
conn.close()

if result:
    print('登陸成功')
else:
    print('登錄失敗')

下面是執行過程

請輸入用戶名：lisi
請輸入密碼：123
<pymysql.connections.Connection object at 0x000001BEFABFE240>
select * from t1.userinfo where username='lisi' and pwd='123'
1
登陸成功

二、execute()之sql注入

方式

#1、sql注入之：用戶存在，繞過密碼
lisi' -- 任意字符

#2、sql注入之：用戶不存在，繞過用戶與密碼
xxx' or 1=1 -- 任意字符

請輸入用戶名：sdj;fja;' or 1=1 -- ;j;j;jj;jjkdsjfjsd
請輸入密碼：123
<pymysql.connections.Connection object at 0x000001EF2BE3E240>
select * from t1.userinfo where username='sdj;fja;' or 1=1 -- ;j;j;jj;jjkdsjfjsd' and pwd='123'
1
登陸成功

解決：

1采用列表的方式

# 原來是我們對sql進行字符串拼接
# sql="select * from userinfo where name='%s' and password='%s'" %(username,pwd)
# print(sql)
# result=cursor.execute(sql)

#改寫為（execute幫我們做字符串拼接，我們無需且一定不能再為%s加引號了）
sql="select * from userinfo where name=%s and pwd=%s" #！！！注意%s需要去掉引號，因為pymysql會自動為我們加上
result=cursor.execute(sql,[user,pwd]) #pymysql模塊自動幫我們解決sql注入的問題，只要我們按照pymysql的規矩來。

 2采用字典的方法

# -*- coding:utf-8 -*-

import pymysql
user = input('請輸入用戶名').strip()
pwd = input('請輸入密碼').strip()

# 連接服務端
conn = pymysql.connect(
    host='127.0.0.1',
    user='root',
    password="123",
    database='t1',
    port=3306,
    charset='utf8'

)
# -- ddadad

# 創建游標對象
cur = conn.cursor()

sql = "select * from userinfo where username = %(name)s and pwd = %(password)s"
print(sql)
# resultNum = cur.execute(sql,[user,pwd])
resultNum = cur.execute(sql,{"name":user,"password":pwd})
print(resultNum)

cur.close()
conn.close()

if resultNum:
    print('登陸成功')
else:
    print('登陸失敗')

 三、增、刪、改：conn.commit()

commit()方法：在數據庫里增、刪、改的時候，必須要進行提交，否則插入的數據不生效。

基本框架

import pymysql
username = input('請輸入用戶名：')
pwd = input('請輸入密碼：')

# 1.連接
conn = pymysql.connect(host='localhost', port=3306, user='root', password='123', db='db8', charset='utf8')

# 2.創建游標
cursor = conn.cursor()

--------增刪改操作----------------

#一定記得commit
conn.commit()

# 4.關閉游標
cursor.close()

# 5.關閉連接
conn.close()

增--》一組數據

sql='insert into userinfo(username,pwd) values(%s,%s)'

effect_row=cursor.execute(sql,(username,pwd))   # effect_row=1

增---》多組數據

sql='insert into userinfo(username,pwd) values(%s,%s)'
effect_row=cursor.executemany(sql,[('趙八','112'),('劉九','114'),('封十','911')])  #effect_row=3

 刪

sql='delete from userinfo  where pwd like "%2"'
effect_row=cursor.execute(sql)

改

sql='update userinfo set username = %s  where pwd="114"'
effect_row=cursor.execute(sql,'niu')   #effect_row=2

 **上面的變量 effect_row=cursor.execute(...) ,返回的是成功改變的條目數字

四、查：fetchone、fetchmany、fetchall

表的內容

mysql> select * from userinfo;
+----+----------+-----+
| id | username | pwd |
+----+----------+-----+
|  1 | mjj      | 123 |
|  3 | 張三     | 110 |
|  4 | 李四     | 119 |
+----+----------+-----+
3 rows in set (0.00 sec)

固定格式

import pymysql

# 1.連接
conn = pymysql.connect(host='localhost', port=3306, user='root', password='', db='db8', charset='utf8')


# 2.創建游標
cursor = conn.cursor()

sql = 'select * from userinfo'
cursor.execute(sql)

------------執行的查詢--------------


# 4.關閉游標
cursor.close()

# 5.關閉連接
conn.close()

fetchone查看一條符合條件的數據，可以連續使用，查詢的是上一個fetchone的后面一條

# 查詢第一行的數據
row = cursor.fetchone()
print(row) # (1, 'mjj', '123')

# 查詢第二行數據
row = cursor.fetchone()
print(row) # (3, '張三', '110')

fetchall():查詢所有符合條件的數據

# 獲取所有的數據
rows = cursor.fetchall()
print(rows)

#運行結果
((1, 'mjj', '123'), (3, '張三', '110'), (4, '李四', '119'))  取到的返回值是元組

fetchmany:獲取指定的條數數據

row=cursor.fetchmany(3)
print(row)

 cursor.scroll(num,mode='relative|absolute')  當mode=absolute時，num不能小於0

cursor.scroll(1,mode='relative')  # 相對當前位置移動
cursor.scroll(2,mode='absolute') # 相對絕對位置移動

# 查詢第一行的數據
row = cursor.fetchone()
print(row) # (1, 'mjj', '123')

# 查詢第二行數據
row = cursor.fetchone() # (3, '張三', '110')
print(row)

cursor.scroll(-1,mode='relative') #設置之后，光標相對於當前位置往前移動了一行，所以打印的結果為第二行的數據
row = cursor.fetchone() 
print(row)

cursor.scroll(0,mode='absolute') #設置之后，光標相對於首行沒有任何變化，所以打印的結果為第一行數據
row = cursor.fetchone() 
print(row)