Py学习  »  DATABASE

如何使用PyMySQL操作mysql数据库?

CDA数据分析师 • 6 年前 • 344 次点击  

作者   Airy

本文转自公众号AiryData,转载需授权


在工作和生活中我们用Python处理数据的情况并不少见,而且很多情况是从数据库取数据,比如MySQL,这里我来分享下简单的Python操作MySQL的库pymysql。


安装与应用


适用环境
python版本 >=2.6或3.3
mysql版本>=4.1


安装
可以使用pip安装也可以手动下载安装。
使用pip安装,在命令行执行如下命令:
pip install PyMySQL(大写不行换小写)
手动安装,请先下载。下载地址:https://github.com/PyMySQL/PyMySQL/tarball/pymysql-X.X。
其中的X.X是版本(目前可以获取的最新版本是0.7.6)。
下载后解压压缩包。在命令行中进入解压后的目录,执行如下的指令:
python setup.py install
建议使用pip安装。


使用示例
连接数据库如下:

import pymysql.cursors
config = {
'host':'127.0.0.1',
'port':3306,
'user':'root',
'password':'zhyea.com',
'db':'employees',
'charset':'utf8mb4',
'cursorclass':pymysql.cursors.DictCursor,
}

# Connect to the database
connection = pymysql.connect(**config)


插入数据
执行sql语句前需要获取cursor,因为配置默认自动提交,故在执行sql语句后需要主动commit,最后不要忘记关闭连接:

from datetime import date, datetime, timedelta
import pymysql.cursors
#连接配置信息
config = {
'host':'127.0.0.1',
'port':3306,
'user':'root',
'password':'zhyea.com',
'db':'employees',
'charset':'utf8mb4',
'cursorclass':pymysql.cursors.DictCursor,
}
# 创建连接
connection = pymysql.connect(**config)

# 获取明天的时间
tomorrow = datetime.now().date() + timedelta(days=1)

# 执行sql语句
try:
    with connection.cursor() as cursor:
# 执行sql语句,插入记录
    sql = 'INSERT INTO employees (first_name, last_name, hire_date, gender, birth_date) VALUES (%s, %s, %s, %s, %s)'
    cursor.execute(sql, ('Robin', 'Zhyea', tomorrow, 'M', date(1989, 6, 14)));
# 没有设置默认自动提交,需要主动提交,以保存所执行的语句
    connection.commit()
finally:
    connection.close();


执行查询

import datetime
import pymysql.cursors
#连接配置信息
config = {
'host':'127.0.0.1',
'port':3306,
'user':'root',
'password':'zhyea.com',
'db':'employees',
'charset':'utf8mb4',
'cursorclass':pymysql.cursors.DictCursor,
}
# 创建连接
connection = pymysql.connect(**config)

# 获取雇佣日期
hire_start = datetime.date(1999, 1, 1)
hire_end = datetime.date(2016, 12, 31)
# 执行sql语句
try:
    with connection.cursor() as cursor:
    # 执行sql语句,进行查询
    sql = 'SELECT first_name, last_name, hire_date FROM employees WHERE hire_date BETWEEN %s AND %s'
    cursor.execute(sql, (hire_start, hire_end))
    # 获取查询结果
    result = cursor.fetchone()
    print(result)
    # 没有设置默认自动提交,需要主动提交,以保存所执行的语句
    connection.commit()
finally:
    connection.close();

这里的查询只取了一条查询结果,查询结果以字典的形式返回:

   result = cursor.fetchmany(2)


不过不建议这样使用,最好在sql语句中设置查询的记录总数。获取全部结果集可以使用fetchall方法:

result = cursor.fetchall()


因为只有两条记录,所以上面提到的这两种查询方式查到的结果是一样的:

   [{‘last_name’: ‘Vanderkelen’, ‘hire_date’: datetime.date(2015, 8, 12), ‘first_name’: ‘Geert’}, {‘last_name’: ‘Zhyea’, ‘hire_date’: datetime.date(2015, 8, 21), ‘first_name’: ‘Robin’}]


在django中使用


 在django中使用是我找这个的最初目的。目前同时支持python3.4、django1.8的数据库backend并不好找。这个是我目前找到的最好用的。


设置DATABASES和官方推荐使用的MySQLdb的设置没什么区别:

DATABASES = {
'default': {
'ENGINE': 'django.db.backends.mysql',
'NAME': 'mytest',
'USER': 'root',
'PASSWORD': 'zhyea.com',
'HOST': '127.0.0.1',
'PORT': '3306',
}
}

关键是这里:我们还需要在站点的__init__.py文件中添加如下的内容:
   import pymysql
   pymysql.install_as_MySQLdb()


以上就是Python中pymysql库的简单使用,谢谢。



今天看啥 - 高品质阅读平台
本文地址:http://www.jintiankansha.me/t/9vI5b05h6R
Python社区是高质量的Python/Django开发社区
本文地址:http://www.python88.com/topic/5028
 
344 次点击