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

查看數據庫：