python使用pymysql实现操作mysql

pymsql是Python中操作MySQL的模块，其使用方法和MySQLdb几乎相同。但目前pymysql支持python3.x而后者不支持3.x版本。

适用环境

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.6.6）。

下载后解压压缩包。在命令行中进入解压后的目录，执行如下的指令：

python setup.py install

建议使用pip安装。

使用示例

连接数据库如下：

    import pymysql.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)

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

    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();

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

从结果集中获取指定数目的记录，可以使用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'}]

在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()

最后给大家附上pymysql实现增删改查的代码，希望大家能够喜欢

    #!/usr/bin/python
    #coding:gbk
    import pymysql
    from builtins import int

    #将MysqlHelper的几个函数写出来

    def connDB():               #连接数据库
      conn=pymysql.connect(host="localhost",user="root",passwd="zx69728537",db="student");
      cur=conn.cursor();
      return (conn,cur);

    def exeUpdate(conn,cur,sql):        #更新或插入操作
      sta=cur.execute(sql);
      conn.commit();
      return (sta);

    def exeDelete(conn,cur,IDs):        #删除操作
      sta=0;
      for eachID in IDs.split(' '):
        sta+=cur.execute("delete from students where Id=%d"%(int(eachID)));
      conn.commit();
      return (sta);

    def exeQuery(cur,sql):           #查找操作
      cur.execute(sql);
      return (cur);

    def connClose(conn,cur):          #关闭连接，释放资源
      cur.close();
      conn.close();

    result=True;
    print("请选择以上四个操作：1、修改记录，2、增加记录，3、查询记录，4、删除记录.(按q为退出)");
    conn,cur=connDB();
    number=input();
    while(result):
      if(number=='q'):
        print("结束操作");
        break;
      elif(int(number)==1):
        sql=input("请输入更新语句：");
        try:
          exeUpdate(conn, cur, sql);
          print("更新成功");
        except Exception:
          print("更新失败");
          raise;
      elif(int(number)==2):
          sql=input("请输入新增语句:");
          try:
            exeUpdate(conn, cur, sql);
            print("新增成功");
          except Exception:
            print("新增失败");
            raise;
      elif(int(number)==3):
        sql=input("请输入查询语句：");
        try:
          cur=exeQuery(cur, sql);
          for item in cur:
            print("Id="+str(item[0])+" name="+item[1]);
        except Exception:
          print("查询出错");
          raise;
      elif(int(number)==4):
        Ids=input("请输入Id，并用空格隔开");
        try:
          exeDelete(conn, cur, Ids);
          print("删除成功");
        except Exception:
          print("删除失败");
          raise;
      else:
        print("非法输入，将结束操作!");
        result=False;
        break;
      print("请选择以上四个操作：1、修改记录，2、增加记录，3、查询记录，4、删除记录.(按q为退出)");
      number=input("请选择操作");