pymysql 库的正确打开姿势 ——Python

 

PyMySQL 是在 Python3.x 版本中用于连接 MySQL 服务器的一个库，Python2中则使用mysqldb

使用示例

连接数据库如下：

 

importpymysql.cursors
 
# Connect to the database
connection =pymysql.connect(host='127.0.0.1',
                            port=3306,
                             user='root',
                            password='zhyea.com',
                             db='employees',
                            charset='utf8mb4',
                             cursorclass=pymysql.cursors.DictCursor)

#!/usr/bin/env python
#   --coding = utf-8
#   Author Allen Lee
import pymysql
#创建链接对象
conn = pymysql.connect(host='127.0.0.1', port=3306, user='root', passwd='123', db='Allen')
#创建游标
cursor = conn.cursor()
#执行sql，更新单条数据，并返回受影响行数
effect_row = cursor.execute("update hosts set host = '1.1.1.2'")
#插入多条，并返回受影响的函数
effect_row = cursor.executemany("insert into hosts(host,color_id)values(%s,%s)",[("1.0.0.1",1,),("10.0.0.3",2)])
#获取最新自增ID
new_id = cursor.lastrowid
#查询数据
cursor.execute("select * from hosts")
#获取一行
row_1 = cursor.fetchone()
#获取多(3)行
row_2 = cursor.fetchmany(3)
#获取所有
row_3 = cursor.fetchall()
#重设游标类型为字典类型
cursor = conn.cursor(cursor=pymysql.cursors.DictCursor)
#提交，保存新建或修改的数据
conn.commit()
#关闭游标
cursor.close()
#关闭连接
conn.close()

 

也可以使用字典进行连接参数的管理，我觉得这样子更优雅一些：

 

 

importpymysql.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，最后不要忘记关闭连接：

 

 

fromdatetime importdate, 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();

 

执行查询：

 

importdatetime
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:
    withconnection.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();

 

 

这里的查询支取了一条查询结果，查询结果以字典的形式返回：

从结果集中获取指定数目的记录，可以使用fetchmany方法：

 

 

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'}]

importpymysql
pymysql.install_as_MySQLdb()

 

在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文件中添加如下的内容：

 

importpymysql
pymysql.install_as_MySQLdb()