Python MySQL – mysql-connector 驅動
Python MySQL - mysql-connector 驅動
MySQL 是最流行的關係型資料庫管理系統,如果你不熟悉 MySQL,可以閱讀我們的 MySQL 教程。
本章節我們為大家介紹使用 mysql-connector 來連線使用 MySQL, mysql-connector 是 MySQL 官方提供的驅動器。
我們可以使用 pip 命令來安裝 mysql-connector:
python -m pip install mysql-connector
使用以下程式碼測試 mysql-connector 是否安裝成功:
執行以上程式碼,如果沒有產生錯誤,表明安裝成功。
注意:如果你的 MySQL 是 8.0 版本,密碼外掛驗證方式發生了變化,早期版本為 mysql_native_password,8.0 版本為 caching_sha2_password,所以需要做些改變:
先修改 my.ini 配置:
[mysqld]
default_authentication_plugin=mysql_native_password
然後在 mysql 下執行以下命令來修改密碼:
ALTER USER 'root'@'localhost' IDENTIFIED WITH mysql_native_password BY '新密碼';
更多內容可以參考:Python MySQL8.0 連結問題。
建立資料庫連線
可以使用以下程式碼來連線資料庫:
demo_mysql_test.py:
import mysql.connector
mydb = mysql.connector.connect(
host="localhost",
user="yourusername",
passwd="yourpassword"
)
print(mydb)
建立資料庫
建立資料庫使用 "CREATE DATABASE" 語句,以下建立一個名為 itread01_db 的資料庫:
demo_mysql_test.py:
import mysql.connector
mydb = mysql.connector.connect(
host="localhost",
user="root",
passwd="123456"
)
mycursor = mydb.cursor()
mycursor.execute("CREATE DATABASE itread01_db")
建立資料庫前我們也可以使用 "SHOW DATABASES" 語句來檢視資料庫是否存在:
demo_mysql_test.py:
輸出所有資料庫列表:
import mysql.connector
mydb = mysql.connector.connect(
host="localhost",
user="root",
passwd="123456"
)
mycursor = mydb.cursor()
mycursor.execute("SHOW DATABASES")
for x in mycursor:
print(x)
或者我們可以直接連線資料庫,如果資料庫不存在,會輸出錯誤資訊:
demo_mysql_test.py:
import mysql.connector
mydb = mysql.connector.connect(
host="localhost",
user="root",
passwd="123456",
database="itread01_db"
)
建立資料表
建立資料表使用 "CREATE TABLE" 語句,建立資料表前,需要確保資料庫已存在,以下建立一個名為 sites 的資料表:
demo_mysql_test.py:
import mysql.connector
mydb = mysql.connector.connect(
host="localhost",
user="root",
passwd="123456",
database="itread01_db"
)
mycursor = mydb.cursor()
mycursor.execute("CREATE TABLE sites (name VARCHAR(255), url VARCHAR(255))")
執行成功後,我們可以看到資料庫建立的資料表 sites,欄位為 name 和 url。
我們也可以使用 "SHOW TABLES" 語句來檢視資料表是否已存在:
demo_mysql_test.py:
import mysql.connector
mydb = mysql.connector.connect(
host="localhost",
user="root",
passwd="123456",
database="itread01_db"
)
mycursor = mydb.cursor()
mycursor.execute("SHOW TABLES")
for x in mycursor:
print(x)
主鍵設定
建立表的時候我們一般都會設定一個主鍵(PRIMARY KEY),我們可以使用 "INT AUTO_INCREMENT PRIMARY KEY" 語句來建立一個主鍵,主鍵起始值為 1,逐步遞增。
如果我們的表已經建立,我們需要使用 ALTER TABLE 來給表新增主鍵:
demo_mysql_test.py:
給 sites 表新增主鍵。
import mysql.connector
mydb = mysql.connector.connect(
host="localhost",
user="root",
passwd="123456",
database="itread01_db"
)
mycursor = mydb.cursor()
mycursor.execute("ALTER TABLE sites ADD COLUMN id INT AUTO_INCREMENT PRIMARY KEY")
如果你還未建立 sites 表,可以直接使用以下程式碼建立。
demo_mysql_test.py:
給表建立主鍵。
import mysql.connector
mydb = mysql.connector.connect(
host="localhost",
user="root",
passwd="123456",
database="itread01_db"
)
mycursor = mydb.cursor()
mycursor.execute("CREATE TABLE sites (id INT AUTO_INCREMENT PRIMARY KEY, name VARCHAR(255), url VARCHAR(255))")
插入資料
插入資料使用 "INSERT INTO" 語句:
demo_mysql_test.py:
向 sites 表插入一條記錄。
import mysql.connector
mydb = mysql.connector.connect(
host="localhost",
user="root",
passwd="123456",
database="itread01_db"
)
mycursor = mydb.cursor()
sql = "INSERT INTO sites (name, url) VALUES (%s, %s)"
val = ("itread01", "https://www.itread01.com")
mycursor.execute(sql, val)
mydb.commit()
print(mycursor.rowcount, "記錄插入成功。")
執行程式碼,輸出結果為:
批量插入
批量插入使用 executemany() 方法,該方法的第二個引數是一個元組列表,包含了我們要插入的資料:
demo_mysql_test.py:
向 sites 表插入多條記錄。
import mysql.connector
mydb = mysql.connector.connect(
host="localhost",
user="root",
passwd="123456",
database="itread01_db"
)
mycursor = mydb.cursor()
sql = "INSERT INTO sites (name, url) VALUES (%s, %s)"
val = [
('Google', 'https://www.google.com'),
('Github', 'https://www.github.com'),
('Taobao', 'https://www.taobao.com'),
('stackoverflow', 'https://www.stackoverflow.com/')
]
mycursor.executemany(sql, val)
mydb.commit()
print(mycursor.rowcount, "記錄插入成功。")
執行程式碼,輸出結果為:
執行以上程式碼後,我們可以看看資料表的記錄:
如果我們想在資料記錄插入後,獲取該記錄的 ID ,可以使用以下程式碼:
demo_mysql_test.py:
import mysql.connector
mydb = mysql.connector.connect(
host="localhost",
user="root",
passwd="123456",
database="itread01_db"
)
mycursor = mydb.cursor()
sql = "INSERT INTO sites (name, url) VALUES (%s, %s)"
val = ("Zhihu", "https://www.zhihu.com")
mycursor.execute(sql, val)
mydb.commit()
print("1 條記錄已插入, ID:", mycursor.lastrowid)
執行程式碼,輸出結果為:
查詢資料
查詢資料使用 SELECT 語句:
demo_mysql_test.py:
import mysql.connector
mydb = mysql.connector.connect(
host="localhost",
user="root",
passwd="123456",
database="itread01_db"
)
mycursor = mydb.cursor()
mycursor.execute("SELECT * FROM sites")
myresult = mycursor.fetchall()
for x in myresult:
print(x)
執行程式碼,輸出結果為:
(1, 'itread01', 'https://www.itread01.com')
(2, 'Google', 'https://www.google.com')
(3, 'Github', 'https://www.github.com')
(4, 'Taobao', 'https://www.taobao.com')
(5, 'stackoverflow', 'https://www.stackoverflow.com/')
(6, 'Zhihu', 'https://www.zhihu.com')
也可以讀取指定的欄位資料:
demo_mysql_test.py:
import mysql.connector
mydb = mysql.connector.connect(
host="localhost",
user="root",
passwd="123456",
database="itread01_db"
)
mycursor = mydb.cursor()
mycursor.execute("SELECT name, url FROM sites")
myresult = mycursor.fetchall()
for x in myresult:
print(x)
執行程式碼,輸出結果為:
('itread01', 'https://www.itread01.com')
('Google', 'https://www.google.com')
('Github', 'https://www.github.com')
('Taobao', 'https://www.taobao.com')
('stackoverflow', 'https://www.stackoverflow.com/')
('Zhihu', 'https://www.zhihu.com')
如果我們只想讀取一條資料,可以使用 fetchone() 方法:
demo_mysql_test.py:
import mysql.connector
mydb = mysql.connector.connect(
host="localhost",
user="root",
passwd="123456",
database="itread01_db"
)
mycursor = mydb.cursor()
mycursor.execute("SELECT * FROM sites")
myresult = mycursor.fetchone()
print(myresult)
執行程式碼,輸出結果為:
(1, 'itread01', 'https://www.itread01.com')
where 條件語句
如果我們要讀取指定條件的資料,可以使用 where 語句:
demo_mysql_test.py
讀取 name 欄位為 itread01 的記錄:
import mysql.connector
mydb = mysql.connector.connect(
host="localhost",
user="root",
passwd="123456",
database="itread01_db"
)
mycursor = mydb.cursor()
sql = "SELECT * FROM sites WHERE name ='itread01'"
mycursor.execute(sql)
myresult = mycursor.fetchall()
for x in myresult:
print(x)
執行程式碼,輸出結果為:
(1, 'itread01', 'https://www.itread01.com')
也可以使用萬用字元 %:
demo_mysql_test.py
import mysql.connector
mydb = mysql.connector.connect(
host="localhost",
user="root",
passwd="123456",
database="itread01_db"
)
mycursor = mydb.cursor()
sql = "SELECT * FROM sites WHERE url LIKE '%oo%'"
mycursor.execute(sql)
myresult = mycursor.fetchall()
for x in myresult:
print(x)
執行程式碼,輸出結果為:
(1, 'itread01', 'https://www.itread01.com')
(2, 'Google', 'https://www.google.com')
為了防止資料庫查詢發生 SQL 注入的攻擊,我們可以使用 %s 佔位符來轉義查詢的條件:
demo_mysql_test.py
import mysql.connector
mydb = mysql.connector.connect(
host="localhost",
user="root",
passwd="123456",
database="itread01_db"
)
mycursor = mydb.cursor()
sql = "SELECT * FROM sites WHERE name = %s"
na = ("itread01", )
mycursor.execute(sql, na)
myresult = mycursor.fetchall()
for x in myresult:
print(x)
排序
查詢結果排序可以使用 ORDER BY 語句,預設的排序方式為升序,關鍵字為 ASC,如果要設定降序排序,可以設定關鍵字 DESC。
demo_mysql_test.py
按 name 欄位字母的升序排序:
import mysql.connector
mydb = mysql.connector.connect(
host="localhost",
user="root",
passwd="123456",
database="itread01_db"
)
mycursor = mydb.cursor()
sql = "SELECT * FROM sites ORDER BY name"
mycursor.execute(sql)
myresult = mycursor.fetchall()
for x in myresult:
print(x)
執行程式碼,輸出結果為:
(3, 'Github', 'https://www.github.com')
(2, 'Google', 'https://www.google.com')
(1, 'itread01', 'https://www.itread01.com')
(5, 'stackoverflow', 'https://www.stackoverflow.com/')
(4, 'Taobao', 'https://www.taobao.com')
(6, 'Zhihu', 'https://www.zhihu.com')
降序排序例項:
demo_mysql_test.py
按 name 欄位字母的降序排序:
import mysql.connector
mydb = mysql.connector.connect(
host="localhost",
user="root",
passwd="123456",
database="itread01_db"
)
mycursor = mydb.cursor()
sql = "SELECT * FROM sites ORDER BY name DESC"
mycursor.execute(sql)
myresult = mycursor.fetchall()
for x in myresult:
print(x)
執行程式碼,輸出結果為:
(6, 'Zhihu', 'https://www.zhihu.com')
(4, 'Taobao', 'https://www.taobao.com')
(5, 'stackoverflow', 'https://www.stackoverflow.com/')
(1, 'itread01', 'https://www.itread01.com')
(2, 'Google', 'https://www.google.com')
(3, 'Github', 'https://www.github.com')
Limit
如果我們要設定查詢的資料量,可以通過 "LIMIT" 語句來指定
demo_mysql_test.py
讀取前 3 條記錄:
import mysql.connector
mydb = mysql.connector.connect(
host="localhost",
user="root",
passwd="123456",
database="itread01_db"
)
mycursor = mydb.cursor()
mycursor.execute("SELECT * FROM sites LIMIT 3")
myresult = mycursor.fetchall()
for x in myresult:
print(x)
執行程式碼,輸出結果為:
(1, 'itread01', 'https://www.itread01.com')
(2, 'Google', 'https://www.google.com')
(3, 'Github', 'https://www.github.com')
也可以指定起始位置,使用的關鍵字是 OFFSET:
demo_mysql_test.py
從第二條開始讀取前 3 條記錄:
import mysql.connector
mydb = mysql.connector.connect(
host="localhost",
user="root",
passwd="123456",
database="itread01_db"
)
mycursor = mydb.cursor()
mycursor.execute("SELECT * FROM sites LIMIT 3 OFFSET 1")
myresult = mycursor.fetchall()
for x in myresult:
print(x)
執行程式碼,輸出結果為:
(2, 'Google', 'https://www.google.com')
(3, 'Github', 'https://www.github.com')
(4, 'Taobao', 'https://www.taobao.com')
刪除記錄
刪除記錄使用 "DELETE FROM" 語句:
demo_mysql_test.py
刪除 name 為 stackoverflow 的記錄:
import mysql.connector
mydb = mysql.connector.connect(
host="localhost",
user="root",
passwd="123456",
database="itread01_db"
)
mycursor = mydb.cursor()
sql = "DELETE FROM sites WHERE name = 'stackoverflow'"
mycursor.execute(sql)
mydb.commit()
print(mycursor.rowcount, " 條記錄刪除")
執行程式碼,輸出結果為:
注意:要慎重使用刪除語句,刪除語句要確保指定了 WHERE 條件語句,否則會導致整表資料被刪除。
為了防止資料庫查詢發生 SQL 注入的攻擊,我們可以使用 %s 佔位符來轉義刪除語句的條件:
demo_mysql_test.py
import mysql.connector
mydb = mysql.connector.connect(
host="localhost",
user="root",
passwd="123456",
database="itread01_db"
)
mycursor = mydb.cursor()
sql = "DELETE FROM sites WHERE name = %s"
na = ("stackoverflow", )
mycursor.execute(sql, na)
mydb.commit()
print(mycursor.rowcount, " 條記錄刪除")
執行程式碼,輸出結果為:
更新表資料
資料表更新使用 "UPDATE" 語句:
demo_mysql_test.py
將 name 為 Zhihu 的欄位資料改為 ZH:
import mysql.connector
mydb = mysql.connector.connect(
host="localhost",
user="root",
passwd="123456",
database="itread01_db"
)
mycursor = mydb.cursor()
sql = "UPDATE sites SET name = 'ZH' WHERE name = 'Zhihu'"
mycursor.execute(sql)
mydb.commit()
print(mycursor.rowcount, " 條記錄被修改")
執行程式碼,輸出結果為:
注意:UPDATE 語句要確保指定了 WHERE 條件語句,否則會導致整表資料被更新。
為了防止資料庫查詢發生 SQL 注入的攻擊,我們可以使用 %s 佔位符來轉義更新語句的條件:
demo_mysql_test.py
import mysql.connector
mydb = mysql.connector.connect(
host="localhost",
user="root",
passwd="123456",
database="itread01_db"
)
mycursor = mydb.cursor()
sql = "UPDATE sites SET name = %s WHERE name = %s"
val = ("Zhihu", "ZH")
mycursor.execute(sql, val)
mydb.commit()
print(mycursor.rowcount, " 條記錄被修改")
執行程式碼,輸出結果為:
刪除表
刪除表使用 "DROP TABLE" 語句, IF EXISTS 關鍵字是用於判斷表是否存在,只有在存在的情況才刪除:
demo_mysql_test.py
import mysql.connector
mydb = mysql.connector.connect(
host="localhost",
user="root",
passwd="123456",
database="itread01_db"
)
mycursor = mydb.cursor()
sql = "DROP TABLE IF EXISTS sites"
mycursor.execute(sql)