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

Python3 MySQL 数据库连接 - PyMySQL 驱动

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

什么是 PyMySQL？

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

pip install pymysql

PyMysql的几个重要方法

connect函数：连接数据库，根据连接的数据库类型不同，该函数的参数也不相同。connect函数返回Connection对象。
cursor方法：获取操作数据库的Cursor对象。cursor方法属于Connection对象。
execute方法：用于执行SQL语句，该方法属于Cursor对象。
commit方法：在修改数据库后，需要调用该方法提交对数据库的修改，commit方法属于Cursor对象。
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"}]

查看数据库：