[TOC] ## 一 介紹 在python程序中操作數據庫要用到了pymysql模塊，該模塊本質就是一個套接字客戶端軟件，使用前需要先安裝 ~~~ pip3 install pymysql ~~~ ## 二 **鏈接、執行sql、關閉** ``` # 1.導入模塊 import pymysql user=input('用戶名: ').strip() pwd=input('密碼: ').strip() # 2.創建鏈接 conn=pymysql.connect(host='localhost',user='root',password='123',database='testdb',charset='utf8') # 3.設置游標 cursor=conn.cursor() #結果集默認以元組顯示 #cursor=conn.cursor(cursor=pymysql.cursors.DictCursor) #字典結果集 # 4.執行sql語句[查詢] sql='select * from userinfo where name=%s and password=%s' res=cursor.execute(sql,[user,pwd]) #此寫法可解決sql注入問題。 print(res) # 5.關閉游標和連接 cursor.close() conn.close() # 6.其他 if res: print('登錄成功') else: print('登錄失敗') ``` ## 三 **數據查詢：** 執行查詢后,默認返回的是查詢出的行數,如果要查看查詢出的數據,需要用到下列4個函數來實現 1. fetchone(n) 顯示游標指向的那條數據,每查詢一次游標下移一次 2. fetchmany(n) 顯示到第n條數據 3. fetchall() 顯示所有查詢出的數據 4. scroll(n) 移動游標位置 ~~~ import pymysql conn=pymysql.connect(host='localhost',user='root',password='123',database='testdb',charset='utf8') cursor=conn.cursor() sql='select * from userinfo;' #1. 獲取所有查詢出的數據 res=cursor.execute(sql) res1=cursor.fetchall() print('所有:',res1) #2. 獲取單行數據 res=cursor.execute(sql) #查看了數據后,結果集游標就變化了,所以每次都查詢執行sql res2=cursor.fetchone() res3=cursor.fetchone() print('單行:',res2,res3) #3. 獲取查詢出的前三行數據 res=cursor.execute(sql) res4=cursor.fetchmany(3) print('前三行:',res4) #4.修改游標后獲取單行數據 res=cursor.execute(sql) cursor.scroll(3,mode='absolute') # 相對絕對位置移動 # cursor.scroll(3,mode='relative') # 相對當前位置移動 res=cursor.execute(sql) res5=cursor.fetchone() print('游標:',res2,res3) conn.commit() cursor.close() conn.close() #執行結果 所有: ((1, 'luogang', 22), (2, 'luo', 23), (3, 'gang1', 23), (4, 'gang2', 23), (5, 'gang3', 23)) 單行: (1, 'luogang', 22) (2, 'luo', 23) 前三行: ((1, 'luogang', 22), (2, 'luo', 23), (3, 'gang1', 23)) 游標: (1, 'luogang', 22) (2, 'luo', 23) ~~~ ## 四 **數據修改** 數據修改和數據查詢方法是一樣的,寫好sql然后用`cursor.execute()`執行sql,只返回影響的行數. 如果是插入語句,且有自增ID,可以通過`lastrowid`方法顯示最后 ~~~ import pymysql conn=pymysql.connect(host='localhost',user='root',password='123',database='testdb',charset='utf8') cursor=conn.cursor() #1.插入語句 sql='insert into userinfo(name,age) values(%s,%s);' res=cursor.execute(sql,("root",36)) print('插入:',res) print('最后ID:',cursor.lastrowid) #2. 更新語句:略 conn.commit() #提交后才發現表中插入記錄成功 cursor.close() conn.close() #執行結果: 插入: 1 最后ID: 7 ~~~