深入解析SQLAlchemy中的原生SQL查询用法，兼具优势与技巧

后端

2024-01-08 20:02:46

原生SQL查询：SQLAlchemy中的高级工具

何时使用原生SQL查询？

SQLAlchemy的ORM通常可以轻松处理数据库操作。然而，在某些情况下，使用原生SQL查询会带来优势：

优化性能： 原生SQL查询通常比ORM查询更快，因为它们绕过了ORM的开销。对于高性能应用程序，这一点至关重要。
访问复杂数据结构： ORM有时无法处理复杂数据结构，例如存储在JSON列中的数据。原生SQL查询可轻松访问此类数据。
使用存储过程或函数： 某些数据库支持存储过程或函数。这些对象只能通过原生SQL查询调用。

如何使用原生SQL查询？

SQLAlchemy提供了两种执行原生SQL查询的方法：

execute()方法： 此方法可执行任何SQL查询，并返回一个ResultProxy对象，可迭代以获取查询结果。

from sqlalchemy import create_engine

engine = create_engine('postgresql://user:password@host:port/database')

with engine.connect() as connection:
    result = connection.execute('SELECT * FROM users')

    for row in result:
        print(row)

scalar()方法： 此方法可执行返回单个值的SQL查询，并返回该值。

from sqlalchemy import create_engine

engine = create_engine('postgresql://user:password@host:port/database')

with engine.connect() as connection:
    user_count = connection.execute('SELECT COUNT(*) FROM users').scalar()

    print(user_count)

如何使用参数？

在原生SQL查询中使用参数可防止SQL注入并提高性能。SQLAlchemy提供了两种方法：

问号占位符（？）： 将问号作为参数占位符，然后使用execute()方法的params参数指定参数值。

from sqlalchemy import create_engine

engine = create_engine('postgresql://user:password@host:port/database')

with engine.connect() as connection:
    result = connection.execute('SELECT * FROM users WHERE id = ?', (1,))

    for row in result:
        print(row)

命名参数： 将冒号（：）作为参数占位符，然后使用execute()方法的params参数指定参数值。

from sqlalchemy import create_engine

engine = create_engine('postgresql://user:password@host:port/database')

with engine.connect() as connection:
    result = connection.execute('SELECT * FROM users WHERE id = :user_id', {'user_id': 1})

    for row in result:
        print(row)