博客

如何在Python中执行数据库事务？

  • Post category:Python

以下是如何在Python中执行数据库事务的完整使用攻略。

使用数据库事务的前提条件

在使用Python执行数据库事务之前，需要确保已经安装并启动了支持事务的数据库，例如MySQL或PostgreSQL，并且需要安装Python的相应数据库驱动程序，例如mysql-connector-python或psycopg2。

步骤1：导入模块

在Python中使用应的数据库驱动程序连接数据库。以下是导入mysql-connector-python模块的基本语法：

import mysql.connector

以下是导入psycopg2模块的基本语法：

import psycopg2

步骤2：连接数据库

在Python中，可以使用相应的数据库驱动程序连接数据库。以下是连接MySQL数据库的基本语法：

mydb = mysql.connector.connect(
  host="localhost",
  user="yourusername",
  password="yourpassword",
  database="mydatabase"
)

以下是连接PostgreSQL数据库的基本语法：

mydb = psycopg2.connect(
  host="localhost",
  user="yourusername",
  password="yourpassword",
  database="mydatabase"
)

在上面的语法中，localhost是数据库服务器的主机名，yourusername和yourpassword是数据库的用户名和密码，mydatabase是要使用的数据库的名称。

步骤3：执行事务

在Python中，可以使用commit()和rollback()方法执行事务。以下是执行事务的基本语法：

mycursor = mydb.cursor()

mycursor.execute("START TRANSACTION")

try:
  # 执行事务操作
  mycursor.execute("INSERT INTO customers (name, address) VALUES (%s, %s)", ("John", "Highway 21"))
  mycursor.execute("UPDATE customers SET address = 'Park Lane 38' WHERE address = 'Highway 21'")
  mydb.commit()
except:
  mydb.rollback()

在上面的语法中，START TRANSACTION语句开始一个新的事务。在try块中，执行事务操作，如果所有操作都成功，则使用commit()方法提交事务。如果任何操作失败，则使用rollback()方法回滚事务。

示例1

在这个示例中，我们使用Python连接到MySQL数据库，并执行一个简单的事务，将两个表中的数据插入到第三个表中。

以下是Python代码：

import mysql.connector

mydb = mysql.connector.connect(
  host="localhost",
  user="yourusername",
  password="yourpassword",
  database="mydatabase"
)

mycursor = mydb.cursor()

mycursor.execute("START TRANSACTION")

try:
  mycursor.execute("INSERT INTO customers (name, address) VALUES (%s, %s)", ("John", "Highway 21"))
  mycursor.execute("INSERT INTO orders (customer_id, product, price) VALUES (%s, %s, %s)", (mycursor.lastrowid, "Apple", 1.99))
  mydb.commit()
except:
  mydb.rollback()

mycursor.execute("SELECT * FROM orders")

for x in mycursor:
  print(x)

在上面的代码中，我们首先使用mysql-connector-python模块连接到MySQL数据库。然后，使用START TRANSACTION语句开始一个新的事务。在try块中，我们将数据插入到customers表和orders表中。如果所有操作都成功，则使用commit()方法提交事务。如果任何操作失败，则使用rollback()方法回滚事务。最后，使用SELECT语句从orders表中检索数据，并使用print()函数打印检索到的数据。

示例2

在这个示例中，我们使用Python连接到PostgreSQL数据库，并执行一个简单的事务，将两个表中的数据插入到第三个表中。

以下是Python代码：

import psycopg2

mydb = psycopg2.connect(
  host="localhost",
  user="yourusername",
  password="yourpassword",
  database="mydatabase"
)

mycursor = mydb.cursor()

mycursor.execute("START TRANSACTION")

try:
  mycursor.execute("INSERT INTO customers (name, address) VALUES (%s, %s)", ("John", "Highway 21"))
  mycursor.execute("INSERT INTO orders (customer_id, product, price) VALUES (%s, %s, %s)", (mycursor.lastrowid, "Apple", 1.99))
  mydb.commit()
except:
  mydb.rollback()

mycursor.execute("SELECT * FROM orders")

for x in mycursor:
  print(x)

在上面的代码中，我们首先使用psycopg2模块连接到PostgreSQL数据库。然后，使用START TRANSACTION语句开始一个新的事务。在try块中，我们将数据插入到customers表和orders表中。如果所有操作都成功，则使用commit()方法提交事务。如果任何操作失败，则使用rollback()方法回滚事务。最后，使用SELECT语句从orders表中检索数据，并使用print()函数打印检索到的数据。

以上是如何在Python中执行数据库事务的完整使用攻略，包括导入模块、连接数据库、执行事务等步骤。提供了两个示例以便更好地理解如何在Python中执行数据库事务。

Tags: ,