Python資料庫編程之conn.execute

一、概述

在Python中，為了操作資料庫，我們需要先連接資料庫。連接成功後可以利用execute()方法執行SQL語句。conn.execute()是一個用於執行對資料庫的增、刪、改、查等操作的方法。

二、執行SQL語句

conn.execute()方法用於執行SQL語句。我們可以使用該方法執行增、刪、改、查表數據等操作。例如，如果我們需要查詢一個表中的所有數據，並列印出每一行的所有欄位，則可以使用如下代碼:

import sqlite3

conn = sqlite3.connect('example.db')
cursor = conn.execute("SELECT * from COMPANY")
for row in cursor:
    print(row)
conn.close()

上述代碼中，我們連接了一個SQLite資料庫，並執行了一個SELECT語句查詢了一個名為COMPANY的表中的所有數據。並通過cursor變數迭代每一行數據進行輸出。

三、插入數據

使用Python編程語言，我們可以在資料庫中執行插入語句。例如，如果我們需要將一些數據插入到一個名為COMPANY的表中，則可以使用如下代碼:

import sqlite3

conn = sqlite3.connect('example.db')
conn.execute("INSERT INTO COMPANY (ID,NAME,AGE,ADDRESS,SALARY) \
      VALUES (1, 'Paul', 32, 'California', 20000.00 )")

conn.execute("INSERT INTO COMPANY (ID,NAME,AGE,ADDRESS,SALARY) \
      VALUES (2, 'Allen', 25, 'Texas', 15000.00 )")

conn.execute("INSERT INTO COMPANY (ID,NAME,AGE,ADDRESS,SALARY) \
      VALUES (3, 'Teddy', 23, 'Norway', 20000.00 )")

conn.execute("INSERT INTO COMPANY (ID,NAME,AGE,ADDRESS,SALARY) \
      VALUES (4, 'Mark', 25, 'Rich-Mond', 65000.00 )")

conn.commit()
print("Records created successfully")
conn.close()

上述代碼中，我們定義了4個要插入的數據行，然後通過conn.execute()方法使用INSERT INTO語句進行插入操作，最後還需要通過commit()方法進行提交操作，才能真正將數據插入到表中。

四、更新數據

我們可以使用Python的conn.execute()方法更新一條或多條數據。例如，如果需要更新名為COMPANY表中ID為1的記錄的地址和薪水數據，則可以使用如下代碼：

import sqlite3

conn = sqlite3.connect('example.db')
conn.execute("UPDATE COMPANY set ADDRESS = 'Texas' where ID = 1")
conn.execute("UPDATE COMPANY set SALARY = 25000.00 where ID = 1")
conn.commit()
print("Total number of rows updated :", conn.total_changes)

cursor = conn.execute("SELECT id, name, address, salary from COMPANY")
for row in cursor:
    print(row)
conn.close()

上述代碼中，我們使用UPDATE語句將ID為1的地址更新為Texas，將薪水更新為25000.00，並通過cursor變數迭代每一行數據進行輸出。

五、刪除數據

我們可以使用Python的conn.execute()方法刪除一行或多行數據。例如，如果我們要刪除名為COMPANY中ID為2的記錄，則可以使用如下代碼：

import sqlite3

conn = sqlite3.connect('example.db')
conn.execute("DELETE from COMPANY where ID = 2;")
conn.commit()
print("Total number of rows deleted :", conn.total_changes)

cursor = conn.execute("SELECT id, name, address, salary from COMPANY")
for row in cursor:
    print(row)
conn.close()

上述代碼中，我們使用DELETE語句刪除ID為2的記錄，並通過cursor變數迭代每一行數據進行輸出。

六、事務

如果有一個以上的操作需要在一個事務中完成，那麼我們可以使用Python的事務支持，將多個操作組合在一起。例如，如果我們需要同時更新一個名為COMPANY的表和一個名為DEPARTMENT的表，則可以使用如下代碼：

import sqlite3

conn = sqlite3.connect('example.db')
print("Opened database successfully")

conn.execute("UPDATE COMPANY set SALARY = 25000.00 where ID = 1")
conn.execute("DELETE from COMPANY where ID = 2;")

cursor = conn.execute("SELECT id, name, address, salary from COMPANY")
for row in cursor:
    print(row)

conn.execute("INSERT INTO DEPARTMENT (ID,DEPT,EMP_ID) \
      VALUES (1, 'IT', 1 )")
conn.execute("INSERT INTO DEPARTMENT (ID,DEPT,EMP_ID) \
      VALUES (2, 'HR', 2 )")

cursor = conn.execute("SELECT id, dept, emp_id from DEPARTMENT")
for row in cursor:
    print(row)

conn.commit()
print("Transaction completed successfully")
conn.close()

上述代碼中，我們在一個事務中進行了多個操作，即通過UPDATE、DELETE語句更新和刪除COMPANY表的數據，再通過INSERT INTO語句向DEPARTMENT表中插入數據和查詢DEPARTMENT表中的數據。

七、錯誤處理

在編寫代碼的過程中，我們需要注意異常處理，以確保代碼在運行時不會出現不必要的錯誤。如果接收到問題，Python將引發異常，我們可以使用try … except語句處理它。

import sqlite3

try:
    conn = sqlite3.connect('example.db')

    conn.execute("UPDATE COMPANY set SALARY = 35000.00 where ID = 1")
    conn.execute("DELETE from COMPANY where ID = 2;")

    cursor = conn.execute("SELECT id, name, address, salary from COMPANY")
    for row in cursor:
        print(row)

    conn.execute("INSERT INTO DEPARTMENT (ID,DEPT,EMP_ID) \
          VALUES (1, 'IT', 1 )")
    conn.execute("INSERT INTO DEPARTMENT (ID,DEPT,EMP_ID) \
          VALUES (2, 'HR', 2 )")

    conn.commit()
    print("Transaction completed successfully")
except Exception as e:
    print("Transaction failed", e)
finally:
    conn.close()

在上述代碼中，我們在try塊中執行多個操作，如果其中任何一個操作出現問題，整個事務都會下跌到except塊中，最後使用finally塊關閉連接。

八、總結

conn.execute()方法是一個用於執行對資料庫的增、刪、改、查等操作的SQL語句的方法。我們可以通過該方法執行各種SQL語句，如SELECT、INSERT、UPDATE、DELETE等操作，也可以利用Python的事務支持將多個操作組合在一起。在編寫代碼時，我們需要注意異常處理，以確保代碼在運行時不會出現不必要的錯誤。