在Python3中操作MySQL資料庫

在Python3中使用mysql資料庫需要安裝pymysql庫

pip install pymysql

操作MySQL

導包

import pymysql

第一步：打開資料庫連接

db = pymysql.connect(host="資料庫地址", 
                     user="用戶名", 
                     password="密碼", 
                     port="埠", 
                     database="資料庫名", 
                     charset='utf8')

第二步：創建游標

cursor = db.cursor()

第三步：操作資料庫

1、創建表

# 如果數據表已經存在使用execute()方法刪除表。
cursor.execute("DROP TABLE IF EXISTS EMPLOYEE")

# 創建數據表SQL語句
sql = """CREATE TABLE EMPLOYEE (
         FIRST_NAME  CHAR(20) NOT NULL,
         LAST_NAME  CHAR(20),
         AGE INT,  
         SEX CHAR(1),
         INCOME FLOAT )"""

cursor.execute(sql)

2、查詢數據

1. Python查詢Mysql使用 fetchone() 方法獲取單條數據, 使用fetchall() 方法獲取多條數據。
2. fetchone(): 該收全部的返回結果行.
3. rowcount: 這是方法獲取下一個查詢結果集。結果集是一個對象
4. fetchall():接一個只讀屬性，並返回執行execute()方法後影響的行數。

# SQL 查詢語句
sql = "SELECT * FROM EMPLOYEE WHERE INCOME > {}".format(1000)
try:
    # 執行SQL語句
    cursor.execute(sql)
    # 獲取所有記錄列表
    results = cursor.fetchall()
    for row in results:
        fname = row[0]
        lname = row[1]
        age = row[2]
        sex = row[3]
        income = row[4]
        # 列印結果
        print("fname={},lname={},age={},sex={},income={}".format(fname, lname, age, sex, income))

except:
    print("Error: unable to fecth data")

3、添加數據

# SQL 插入語句
sql = """INSERT INTO EMPLOYEE(FIRST_NAME,
         LAST_NAME, AGE, SEX, INCOME)
         VALUES ('Mac', 'Mohan', 20, 'M', 2000)"""
try:
    cursor.execute(sql)
    # 提交到資料庫執行
    db.commit()
except:
    # 發生錯誤時回滾
    db.rollback()

4、修改數據

# SQL 更新語句
sql = "UPDATE EMPLOYEE SET AGE = AGE + 1 WHERE SEX = '{}'".format('M')
try:
    cursor.execute(sql)
    # 提交到資料庫執行
    db.commit()
except:
    # 發生錯誤時回滾
    db.rollback()

5、刪除數據

# SQL 刪除語句
sql = "DELETE FROM EMPLOYEE WHERE AGE > {}".format(20)
try:
    cursor.execute(sql)
    # 提交到資料庫執行
    db.commit()
except:
    # 發生錯誤時回滾
    db.rollback()

第四步：關閉游標，資料庫連接

cursor.close()
db.close()