本文涵盖了四种方法,你可以用它们来连接你的Python应用程序和MySQL。这些例子将使用PlanetScale,一个用于MySQL的无服务器数据库平台。
本文将涵盖以下连接方法:
- mysqlclient
- mysql-connector-python
- PyMySQL
- aiomysql
设置PlanetScale
如果你需要一个免费的数据库来测试你的连接,PlanetScale是一个很好的选择。PlanetScale建立在Vitess之上,为你提供一个熟悉的工作流程分支来测试和部署不同模式。
Vitess建于2010年,用于解决YouTube的扩展问题。从那时起,这个开源项目持续增长,现在帮助一些公司,如Slack和Square处理他们的大规模数据扩展需求。
你可以从下面注册一个账户,获得一个免费的5GB数据库。
https://auth.planetscale.com/sign-up
创建一个数据库
当你进入PlanetScale的仪表板。
- 在你的组织概览仪表板上点击 "创建新数据库"。
- 给你的数据库起个名字。
- 从下拉菜单中选择一个区域。
- 点击创建数据库。
你的数据库将被创建为一个主要的开发分支。你可以在这里进行模式修改,一旦你准备好了,就可以把它推广到生产中。
获取连接凭证
要生成连接凭证,必须做到以下步骤:
- 在你的数据库概览页面上点击 "连接"。
- 从 "连接方式 "下拉菜单中选择 "Python"。
- 将凭证复制到你的Python应用程序文件夹中的.env文件。
为了确保连接的安全性,PlanetScale使用了来自证书颁发机构(CA)的证书,它是几乎所有平台上的系统根的一部分。当连接到数据库时,你必须指定你信任的CA证书的路径。注意,这个路径取决于你的平台。这里有一个跨越各种操作系统和发行版的常见路径列表。
https://planetscale.com/docs/concepts/secure-connections#ca-root-configuration
如果你从PlanetScale仪表板上复制了证书,你已经完成这一步,因为PlanetScale会自动检测你的操作系统。
如果你使用的是Windows系统,你需要下载根证书,然后用你所使用的驱动程序的正确配置选项指向它们的位置。
用mysqlclient连接到MySQL
mysqlclient包是用于MySQL的最流行的Python包之一。它包含MySQLdb模块,一个提供Python数据库API的MySQL接口。
- 首先安装mysqlclient。你可以通过两种方式来完成这件事。
- 使用来自MySQL官方网站的安装程序,它与你正在运行的操作系统和SQL版本相匹配。
使用 pip 来安装它。
pip install mysqlclient
在Linux上,你可能需要在安装mysqlclient之前安装Python3和MySQL开发头文件和库。
sudo apt-get install python3-dev default-libmysqlclient-dev build-essential
一旦你安装了mysqlclient,你可以使用下面的代码连接到数据库。
import os
import MySQLdb # import the MySQLdb module
from dotenv import load_dotenv
load_dotenv()
# Create the connection object
connection = MySQLdb.connect(
host=os.getenv("HOST"),
user=os.getenv("USERNAME"),
passwd=os.getenv("PASSWORD"),
db=os.getenv("DATABASE"),
ssl_mode="VERIFY_IDENTITY",
ssl={
'ca': os.getenv("SSL_CERT")
}
)
# Create cursor and use it to execute SQL command
cursor = connection.cursor()
cursor.execute("select @@version")
version = cursor.fetchone()
if version:
print('Running version: ', version)
else:
print('Not connected.')
注意,你必须安装 python-dotenv 并导入 os 模块以访问.env 文件中的数据库凭证。
Windows用户要注意,从wheel轮子文件中安装的mysqlclient不支持ssl模式。由于你想确保数据库连接是安全的,我们建议使用另一个驱动,比如mysql-connector-python。
这个例子使用连接来获取数据库的SQL版本,然后关闭连接。
其余的连接方法的语法与这个例子几乎相同。这种类似的语法源于 Python 数据库 API 规范 (PEP 249),它鼓励用于访问数据库的 Python 模块之间的一致性。
用MySQL connector连接到MySQL
MySQL connector/Python模块是Oracle支持的官方驱动,用于通过Python连接MySQL。该连接器完全是Python语言,而mysqlclient是用C语言编写的。它也是独立的,意味着它不需要MySQL客户端库或标准库以外的任何Python模块。
注意,MySQL Connector/Python不支持旧的MySQL服务器认证方法,这意味着4.1之前的MySQL版本不能工作。
首先安装mysql-connector-python模块。建议使用pip来安装它。
pip install mysql-connector-python
安装后,使用以下代码连接到MySQL:
import os
from dotenv import load_dotenv
from mysql.connector import Error
import mysql.connector
load_dotenv()
connection = mysql.connector.connect(
host=os.getenv("HOST"),
database=os.getenv("DATABASE"),
user=os.getenv("USERNAME"),
password=os.getenv("PASSWORD"),
ssl_ca=os.getenv("SSL_CERT")
)
try:
if connection.is_connected():
cursor = connection.cursor()
cursor.execute("select @@version ")
version = cursor.fetchone()
if version:
print('Running version: ', version)
else:
print('Not connected.')
except Error as e:
print("Error while connecting to MySQL", e)
finally:
connection.close()
用PyMySQL连接到MySQL
PyMySQL包是另一个连接器,你可以用它来连接Python和MySQL。如果你追求速度,这是一个很好的选择,因为它比mysql-connector-python快。
你可以使用 pip 来安装它。
pip install PyMySQL
然后,使用以下连接代码。
from dotenv import load_dotenv
import pymysql
import os
load_dotenv()
connection = pymysql.connect(
host=os.getenv("HOST"),
database=os.getenv("DATABASE"),
user=os.getenv("USERNAME"),
password=os.getenv("PASSWORD"),
ssl_ca=os.getenv("SSL_CERT")
)
cursor = connection.cursor()
cursor.execute("select @@version ")
version = cursor.fetchone()
if version:
print('Running version: ', version)
else:
print('Not connected.')
connection.close()
用aiomysql连接到MySQL
aiomysql库用于从asyncio框架访问MySQL数据库。除了是异步的特性之外,连接代码与PyMySQL相似。注意,使用aiomysql需要Python 3.7以上版本和PyMySQL。
要使用aiomysql,首先要用安装异步模块。
pip install asyncio
然后,用以下方法安装aiomysql。
pip install aiomysql
然后你可以使用以下代码连接到MySQL。
import os
import asyncio
import aiomysql
import ssl
from dotenv import load_dotenv
load_dotenv()
ctx = ssl.SSLContext(ssl.PROTOCOL_TLS_CLIENT)
ctx.load_verify_locations(cafile=os.getenv("SSL_CERT"))
loop = asyncio.get_event_loop()
async def connect_db():
connection = await aiomysql.connect(
host=os.getenv("HOST"),
port=3306,
user=os.getenv("USERNAME"),
password=os.getenv("PASSWORD"),
db=os.getenv("DATABASE"),
loop=loop,
ssl=ctx
)
cursor = await connection.cursor()
await cursor.execute("select @@version")
version = await cursor.fetchall()
print('Running version: ', version)
await cursor.close()
connection.close()
loop.run_until_complete(connect_db())
总结
在数据库中保存数据是应用程序的共同特征之一。像Python一样,大多数编程语言支持与不同数据库的连接和交互。本教程探讨了Python与MySQL的连接。
我们分享了可以使用的四种不同的常用连接方法,并在PlanetScale数据库上使用SSL对它们进行了测试,这是一个关键的步骤,因为它可以防止中间人攻击。我们回顾了每个连接方式的连接器是如何工作的,并展示了运行它所需的代码。现在你已经完成了这个实践概述,你已经准备好将MySQL数据库连接到Python应用程序中了。