SQLAlchemy执行SQL语句|极客教程

SQLAlchemy执行SQL语句

SQLAlchemy是一个Python的ORM（对象关系映射）工具，它可以让开发者通过Python代码来操作数据库，而不用直接编写SQL语句。但是有时候我们仍然需要执行原生的SQL语句来完成某些特定的操作，例如一些复杂的查询、索引创建等。

在本文中，我们将介绍如何使用SQLAlchemy执行原生的SQL语句，包括执行查询、更新和删除操作等。

连接数据库

首先，我们需要创建一个与数据库的连接。在SQLAlchemy中，可以使用create_engine函数来创建一个引擎对象，然后使用engine.connect()方法来获取一个连接对象。

from sqlalchemy import create_engine

# 创建一个MySQL数据库连接
engine = create_engine('mysql+pymysql://username:password@localhost/mydatabase')
conn = engine.connect()

在上面的代码中，我们使用create_engine函数创建了一个连接到名为mydatabase的MySQL数据库的引擎对象，并使用engine.connect()方法获取了一个连接对象conn。

执行查询操作

接下来，我们将介绍如何使用SQLAlchemy执行查询操作。SQLAlchemy提供了execute方法来执行原生的SQL查询语句，并可以使用fetchall方法获取查询结果。

result = conn.execute("SELECT * FROM users")
rows = result.fetchall()

for row in rows:
    print(row)

在上面的代码中，我们使用conn.execute方法执行了一个简单的SELECT查询语句，并使用fetchall方法获取了查询结果，然后遍历输出了每一行的结果。你可以根据实际需求编写更复杂的查询语句。

执行更新和删除操作

除了查询操作，我们还可以使用SQLAlchemy执行更新和删除操作。下面分别介绍如何执行更新和删除操作。

更新操作

conn.execute("UPDATE users SET name = 'Alice' WHERE id = 1")

在上面的代码中，我们使用conn.execute方法执行了一个更新操作，将users表中id为1的记录的name字段更新为Alice。你可以根据实际需求编写更新操作的SQL语句。

删除操作

conn.execute("DELETE FROM users WHERE id = 1")

在上面的代码中，我们使用conn.execute方法执行了一个删除操作，删除了users表中id为1的记录。同样，你可以根据实际需求编写删除操作的SQL语句。

使用事务

在执行多个SQL语句时，我们通常会使用事务来确保操作的原子性，即要么所有操作成功提交，要么所有操作失败回滚。在SQLAlchemy中，可以使用begin方法来开始一个事务，commit方法提交事务，rollback方法回滚事务。

trans = conn.begin()

try:
    conn.execute("UPDATE users SET name = 'Bob' WHERE id = 2")
    conn.execute("DELETE FROM users WHERE id = 3")
    trans.commit()
except:
    trans.rollback()

在上面的代码中，我们使用begin方法开始一个事务，然后执行了一个更新和一个删除操作，最后根据操作是否成功提交或回滚事务。

执行存储过程和函数

除了执行原生的SQL语句，我们还可以使用SQLAlchemy执行数据库中的存储过程和函数。

result = conn.execute("CALL sp_get_users()")
rows = result.fetchall()

for row in rows:
    print(row)

在上面的代码中，我们使用conn.execute方法执行了一个名为sp_get_users的存储过程，并输出了查询结果。同样，你可以根据实际需求执行其他存储过程或函数。

结束连接

最后，我们需要在程序结束时关闭连接。

conn.close()

在这篇文章中，我们介绍了如何使用SQLAlchemy执行原生的SQL语句，包括执行查询、更新、删除操作以及使用事务和执行存储过程和函数。

SQLAlchemy执行SQL语句