PYTHON——数据存储：MySQL数据库

Python3 MySQL 数据库连接 - PyMySQL 驱动

本文我们为大家介绍 Python3 使用 PyMySQL 连接数据库，并实现简单的增删改查。

什么是 PyMySQL？

PyMySQL 模块中提供的API与SQLite3模块中提供的API类似，因为他们都是遵循Python DB API 2.0标准。在标准Windows环境下，安装PyMySQL的方法是：

pip install pymysql

PyMysql的几个重要方法

1. connect函数：连接数据库，根据连接的数据库类型不同，该函数的参数也不相同。connect函数返回Connection对象。
2. cursor方法：获取操作数据库的Cursor对象。cursor方法属于Connection对象。
3. execute方法：用于执行SQL语句，该方法属于Cursor对象。
4. commit方法：在修改数据库后，需要调用该方法提交对数据库的修改，commit方法属于Cursor对象。
5. rollback方法：如果修改数据库失败，一般需要调用该方法进行数据库回滚操作，也就是将数据库恢复成修改之前的样子。

实例：

from pymysql import *
import json

def connectDB():
    db = connect('127.0.0.1','root','password','databasename')
    return db
db = connectDB()

def creatTable(db):
    cursor = db.cursor()
    sql = '''
    CREATE TABLE Persons
        (
            id INT PRIMARY KEY NOT NULL,
            name TEXT NOT NULL,
            age INT NOT NULL,
            address CHAR(50),
            salary REAL
        );    
    '''
    try:
        cursor.execute(sql)
        db.commit()
        return True
    except:
        db.rollback()
    return False

def insertRecords(db):
    cursor = db.cursor()
    try:
        cursor.execute('DELETE FROM persons')
        cursor.execute("INSERT INTO persons(id,name,age,address,salary)
                       VALUES(1,'Paul',32,'California',2000.00)");
        cursor.execute("INSERT INTO persons(id,name,age,address,salary)
                           VALUES(2,'Allen',25,'Texas',3000.00)");
        cursor.execute("INSERT INTO persons(id,name,age,address,salary)
                           VALUES(3,'Teddy',23,'Norway',2500.00)");
        cursor.execute("INSERT INTO persons(id,name,age,address,salary)
                           VALUES(4,'Mark',19,'Rich',5000.00)");
        db.commit()
        return True
    except Exception as e:
        print(e)
        db.rollback()
    return False

def selectRecords(db):
    cursor = db.cursor()
    sql = 'SELECT name,age,address,salary FROM Persons ORDER BY age DESC'
    cursor.execute(sql)
    results = cursor.fetchall()
    print(results)
    fields = ['name','age','address','salary']
    records = []
    for row in results:
        records.append(dict(zip(fields,row)))
    return json.dumps(records)

if creatTable(db):
    print('成功创建Persons表')
else:
    print('persons表已经存在')

if insertRecords(db):
    print('成功插入数据')
else:
    print('插入记录失败')

print(selectRecords(db))
db.close()

运行返回结果：

persons表已经存在
成功插入数据
(('Paul', 32, 'California', 2000.0), ('Allen', 25, 'Texas', 3000.0), ('Teddy', 23, 'Norway', 2500.0), ('Mark', 19, 'Rich', 5000.0))
[{"age": 32, "name": "Paul", "salary": 2000.0, "address": "California"}, {"age": 25, "name": "Allen", "salary": 3000.0, "address": "Texas"}, {"age": 23, "name": "Teddy", "salary": 2500.0, "address": "Norway"}, {"age": 19, "name": "Mark", "salary": 5000.0, "address": "Rich"}]

查看数据库：