Python 連線 MySQL 的幾種姿勢

儘管很多 NoSQL 資料庫近幾年大放異彩，但是像 MySQL 這樣的關係型資料庫依然是網際網路的主流資料庫之一，每個學 Python 的都有必要學好一門資料庫，不管你是做資料分析，還是網路爬蟲，Web 開發、亦或是機器學習，你都離不開要和資料庫打交道，而 MySQL 又是最流行的一種資料庫，這篇文章介紹 Python 操作 MySQL 的幾種方式，你可以在實際開發過程中根據實際情況合理選擇。

1、MySQL-python

MySQL-python 又叫 MySQLdb，是 Python 連線 MySQL 最流行的一個驅動，很多框架都也是基於此庫進行開發，遺憾的是它只支援 Python2.x，而且安裝的時候有很多前置條件，因為它是基於C開發的庫，在 Windows 平臺安裝非常不友好，經常出現失敗的情況，現在基本不推薦使用，取代的是它的衍生版本。

# 前置條件
sudo apt-get install python-dev libmysqlclient-dev # Ubuntu
sudo yum install python-devel mysql-devel # Red Hat / CentOS

# 安裝
pip install MySQL-python

Windows 直接通過下載 exe 檔案安裝

#!/usr/bin/python
import MySQLdb

db = MySQLdb.connect(
     host="localhost",    # 主機名
     user="john",         # 使用者名稱
     passwd="megajonhy",  # 密碼
     db="jonhydb")        # 資料庫名稱

# 查詢前，必須先獲取遊標
cur = db.cursor()

# 執行的都是原生SQL語句
cur.execute("SELECT * FROM YOUR_TABLE_NAME")

for row in cur.fetchall():
    print(row[0])

db.close()

2、mysqlclient

由於 MySQL-python 年久失修，後來出現了它的 Fork 版本 mysqlclient，完全相容 MySQLdb，同時支援 Python3.x，是 Django ORM的依賴工具，如果你想使用原生 SQL 來運算元據庫，那麼推薦此驅動。安裝方式和 MySQLdb 是一樣的，Windows 可以在 https://www.lfd.uci.edu/~gohlke/pythonlibs/#mysqlclient 網站找到 對應版本的 whl 包下載安裝。

image.png

# Windows安裝
pip install some-package.whl

# linux 前置條件
sudo apt-get install python3-dev # debian / Ubuntu
sudo yum install python3-devel # Red Hat / CentOS
brew install mysql-connector-c # macOS (Homebrew)

pip install mysqlclient

3、PyMySQL

PyMySQL 是純 Python 實現的驅動，速度上比不上 MySQLdb，最大的特點可能就是它的安裝方式沒那麼繁瑣，同時也相容 MySQL-python

pip install PyMySQL
# 為了相容mysqldb，只需要加入
pymysql.install_as_MySQLdb()

一個例子

import pymysql
conn = pymysql.connect(host='127.0.0.1', user='root', passwd="xxx", db='mysql')
cur = conn.cursor()
cur.execute("SELECT Host,User FROM user")
for r in cur:
    print(r)
cur.close()
conn.close()

4、peewee

寫原生 SQL 的過程非常繁瑣，程式碼重複，沒有物件導向思維，繼而誕生了很多封裝 wrapper 包和 ORM 框架，ORM 是 Python 物件與資料庫關係表的一種對映關係，有了 ORM 你不再需要寫 SQL 語句。提高了寫程式碼的速度，同時相容多種資料庫系統，如sqlite, mysql、postgresql，付出的代價可能就是效能上的一些損失。如果你對 Django 自帶的 ORM 熟悉的話，那麼 peewee的學習成本幾乎為零。它是 Python 中是最流行的 ORM 框架。

pip install peewee

一個例子

import peewee
from peewee import *

db = MySQLDatabase('jonhydb', user='john', passwd='megajonhy')

class Book(peewee.Model):
    author = peewee.CharField()
    title = peewee.TextField()

    class Meta:
        database = db

Book.create_table()
book = Book(author="me", title='Peewee is cool')
book.save()
for book in Book.filter(author="me"):
    print(book.title)

官方文件：http://docs.peewee-orm.com/en/latest/peewee/installation.html

5、SQLAlchemy

如果想找一種既支援原生 SQL，又支援 ORM 的工具，那麼 SQLAlchemy 是最好的選擇，它非常接近 Java 中的 Hibernate 框架。

from sqlalchemy import create_engine
from sqlalchemy.orm import sessionmaker
 
from sqlalchemy_declarative import Address, Base, Person

class Address(Base):
    __tablename__ = 'address'
    id = Column(Integer, primary_key=True)
    street_name = Column(String(250))
 
engine = create_engine('sqlite:///sqlalchemy_example.db')
Base.metadata.bind = engine
 
DBSession = sessionmaker(bind=engine)
session = DBSession()
 
# Insert a Person in the person table
new_person = Person(name='new person')
session.add(new_person)
session.commit()

現在差不多搞明白了這幾種資料庫驅動的優劣，接下來你就可以選擇其中的一個進行系統的學習再把它應用到專案中去了。