三种驱动mysql.connector、PyMySQL、mysqldb使用和比较
mysql-connector-python
此为廖雪峰老师教程中所用驱动，为python官方驱动。安装命令：
pip install mysql-connector-python --allow-external mysql-connector-python
或者
pip install mysql-connector
使用mysql.connector访问数据库：
import mysql.connector
# password设为mysql的root口令
conn = mysql.connector.connect(user='root', password='password', database='test')
cursor = conn.cursor()
# 创建user表
cursor.execute('要执行的sql语句')
# 提交事务
conn.commit()
# 关闭Cursor和Connection:
cursor.close()
conn.close()
PyMySQL 驱动
PyMySQL仅限于 Python3.x 版本中连接 MySQL 。
- 安装命令：
pip3 install PyMySQL
若pip安装失败，可使用git命令或者手动下载源安装包后，进入工程目录使用python命令手动安装：
cd PyMySQL/
python3 setup.py install
- 使用PyMySQL访问数据库-建表：
#!/usr/bin/python3
import pymysql
# 打开数据库连接
db = pymysql.connect("localhost","testuser","test123","TESTDB" )
# 使用 cursor() 方法创建一个游标对象 cursor
cursor = db.cursor()
# 使用 execute() 方法执行 SQL，如果表存在则删除
cursor.execute("DROP TABLE IF EXISTS EMPLOYEE")
# 使用预处理语句创建表
sql = """CREATE TABLE EMPLOYEE (
FIRST_NAME CHAR(20) NOT NULL,
LAST_NAME CHAR(20),
AGE INT,
SEX CHAR(1),
INCOME FLOAT )"""
cursor.execute(sql)
# 关闭数据库连接
db.close()
- 使用PyMySQL访问数据库-插入、更新、删除数据：
注意事务提交以及错误回滚处理
#!/usr/bin/python3
import pymysql
# 打开数据库连接
db = pymysql.connect("localhost","testuser","test123","TESTDB" )
# 使用cursor()方法获取操作游标
cursor = db.cursor()
# SQL 插入语句
# 此处可将sql换为update、delete等语句
sql = "INSERT INTO EMPLOYEE(FIRST_NAME, \
LAST_NAME, AGE, SEX, INCOME) \
VALUES ('%s', '%s', %s, '%s', %s)" % \
('Mac', 'Mohan', 20, 'M', 2000)
try:
# 执行sql语句
cursor.execute(sql)
# 执行sql语句
db.commit()
except:
# 发生错误时回滚
db.rollback()
# 关闭数据库连接
db.close()
- 使用PyMySQL访问数据库-查询：
可使用fetchall()等方法对查询结果进行获取和进一步处理，见代码中注释
#!/usr/bin/python3
import pymysql
# 打开数据库连接
db = pymysql.connect("localhost","testuser","test123","TESTDB" )
# 使用cursor()方法获取操作游标
cursor = db.cursor()
# SQL 查询语句
sql_query = "SELECT * FROM EMPLOYEE \
WHERE INCOME > %s" % (1000)
try:
# 执行SQL语句
cursor.execute(sql_query)
# fetchone(): 该方法获取下一个查询结果集。结果集是一个对象
# fetchall(): 接收全部的返回结果行.
# rowcount: 这是一个只读属性，并非方法，直接返回执行execute()方法后影响的行数。
results = cursor.fetchall()
for row in results:
pass
except:
print ("Error: unable to fetch data")
# 关闭数据库连接
db.close()
MySQLdb驱动
仅支持linux的python2.x版本。用法同上类似，安装：
pip install MySQLdb
比较和总结
- mysql.connector作为官方版本，在Python中重新实现MySQL协议，速度相对最慢，但因为不依赖其它库，因此最便携；
- PyMySQL 限制于Python3.x 版本；
- MySQLdb限制于Python2.x 版本，且不支持windows。
因此推荐使用PyMySQL。
SSH连接Mysql
基于pymysql与sshtunnel，实现mysql跳转连接，用于服务器无法直接本地连接的场景。即：在本地（记为地址A），通过SSH中间服务器B，连接远程MYSQL服务器C。
下述代码中地址及端口，根据实际网络策略进行配置：
from pymysql import connect
from sshtunnel import SSHTunnelForwarder
# 指定SSH远程跳转
server = SSHTunnelForwarder(ssh_address_or_host=('IP地址B', B端口号), # 指定SSH中间登录地址和端口号
ssh_username='user_name_b', # 指定地址B的SSH登录用户名
ssh_password='pwd_b', # 指定地址B的SSH登录密码
local_bind_address=('127.0.0.1', A端口号), # 绑定本地地址A（默认127.0.0.1）及与B相通的端口（根据网络策略配置，若端口全放，则此行无需配置，使用默认即可）
remote_bind_address=('IP地址C', 3306) # 指定最终目标C地址，端口号为mysql默认端口号3306
)
server.start()
# 打印本地端口，以检查是否配置正确
print(server.local_bind_port)
# 设置mysql连接参数，地址与端口均必须设置为本地地址与端口
# 用户名和密码以及数据库名根据自己的数据库进行配置
db = connect(host="127.0.0.1", port=server.local_bind_port, user="root", passwd="root_pwd", db="test_db")
cursor = db.cursor()
sql = "select COUNT(1) from table1"
try:
# 执行SQL语句检查是否连接成功
cursor.execute("SELECT VERSION()")
result = cursor.fetchone()
print("Database version : %s " % result)
#执行查询语句
cursor.execute(sql)
result = cursor.fetchone()
print("sql result : %s" % result)
except:
print("Error: unable to fetch data")
# 关闭连接
db.close()
server.close()
