python操作MySQL数据库简单示例

通过python的pymysql模块，实现数据库表的创建、插入以及查询的简单示例。

1. 数据库表创建。

# -*- coding:utf-8 -*-
"""
使用python操作MySQL数据库示例
执行sql_create语句建立数据库表books
"""

import pymysql


def main():
    # 首先建立数据库连接对象db,参数分别为地址用户名密码及数据库，端口默认为3306。
    db = pymysql.connect(host='localhost', user='xxxx', password='xxxx', database='ditto2008')
    # 建立游标对象
    cursor = db.cursor()
    # 如果存在该表则先删除，防止多次运行报错。
    cursor.execute('DROP TABLE IF EXISTS books')
    # 创建数据库表books的sql语句，包含3列，id为非空且自动排序，并且为主键
    sql_create = '''
    CREATE TABLE books (
    id int(8) NOT NULL AUTO_INCREMENT,
    name varchar(50) NOT NULL,
    price decimal(10,2) DEFAULT NULL,
    PRIMARY KEY (id)
    )
    '''
    try:
        cursor.execute(sql_create)
        print('创建成功')
    except Exception as e:
        print('创建表失败', e)
    finally:
        cursor.close()


if __name__ == '__main__':
    main()

控制台打印：创建成功

通过Navicat查看如下所示：

2. 数据库表中插入信息。

# -*- coding:utf-8 -*-
"""
使用python操作mysql数据库示例。
向数据库表books中插入信息。
"""
import pymysql


def main():
    # 首先建立数据库连接对象db
    db = pymysql.connect(host='localhost', user='xxxx', password='xxxx', database='ditto2008')
    # 建立游标对象
    cursor = db.cursor()
    # 要插入的书籍信息，为一个list，每一项分别包含name和price
    book_data = [("国富论", '89.27'),
                 ("明朝那些事儿", '99.88'),
                 ("传习录", '66.77')
                 ]
    try:
        # 执行批量插入的sql语句，使用%s作为占位符防止sql注入。
        cursor.executemany('insert into books(name, price) values (%s, %s)', book_data)
        db.commit()
        print('数据插入成功。')
    except Exception as e:
        print('数据插入失败', e)
        # 如果出现异常则回滚
        db.rollback()
    finally:
        # 任何情况下都关闭连接
        db.close()


if __name__ == '__main__':
    main()

控制台打印：数据插入成功。

通过Navicat查看插入结果如下所示：

3. 数据库查询

# -*- coding:utf-8 -*-
"""
使用python操作mysql数据库示例。
数据库查询示例，通过书名查询价格。
"""

import pymysql


def main():
    # 首先建立数据库连接对象db
    db = pymysql.connect(host='localhost', user='xxxx', password='xxxx', database='ditto2008')
    # 建立游标对象
    cursor = db.cursor()
    # 查询语句，name使用占位符
    query = 'select price from books where name = %s'
    # 要查询的book_name，也可以直接通过方法传参获取。
    book_name = '传习录'
    try:
        # 执行sql语句，传参给占位符。
        cursor.execute(query, (book_name,))
        # 获取查询结果，通过fetchone获取price
        result = cursor.fetchone()
        # 查询结果为tuple，获取第1项。
        print(result[0])
    except Exception as e:
        print('查询失败', e)
    finally:
        # 关闭数据库连接。
        db.close()


if __name__ == '__main__':
    main()

控制台打印结果：66.77