如果仅用于原始SQL查询,SQLAlchemy仍然是推荐的吗?

时间:2012-07-18 14:05:52

标签: python orm sqlalchemy flask

使用Flask,我很想知道SQLAlchemy是否仍然是使用原始SQL(直接SELECT x FROM table WHERE ...)而不是使用ORM查询数据库的最佳方式,或者是否有更简单但更强大的替代方案?

感谢您的回复。

2 个答案:

答案 0 :(得分:17)

我一直使用SQLAlchemy进行直接查询。

主要优势:它为您提供针对SQL注入攻击的最佳保护。 SQLAlchemy做了正确的事,无论你抛出什么参数。

我发现它也可以根据条件调整生成的SQL。显示上面有多个过滤器控件的结果集?只需在一组if / elif / else结构中构建查询,您就知道您的SQL仍然是金色的。

以下是一些实时代码的摘录(较旧的SA版本,因此语法可能略有不同):

# Pull start and end dates from form
# ...
# Build a constraint if `start` and / or `end` have been set.
created = None
if start and end:
    created = sa.sql.between(msg.c.create_time_stamp, 
        start.replace(hour=0, minute=0, second=0),
        end.replace(hour=23, minute=59, second=59))
elif start:
    created = (msg.c.create_time_stamp >= 
               start.replace(hour=0, minute=0, second=0))
elif end:
    created = (msg.c.create_time_stamp <= 
               end.replace(hour=23, minute=59, second=59))

# More complex `from_` object built here, elided for example
# [...]
# Final query build
query = sa.select([unit.c.eli_uid], from_obj=[from_])
query = query.column(count(msg.c.id).label('sent'))
query = query.where(current_store)
if created:
    query = query.where(created)

来自此代码的代码要复杂得多,但我想在此处突出显示日期范围代码。如果我必须使用字符串格式化构建SQL,我可能已经在某处引入了SQL注入漏洞,因为它更容易忘记引用值。

答案 1 :(得分:2)

在我参与了一个小项目后,我决定尝试使用MySQLDB,而不使用SQL Alchemy。

它工作正常并且使用起来很方便,这是一个例子(我创建了一个处理数据库所有工作的小类)

import MySQLdb
from MySQLdb.cursors import DictCursor

class DatabaseBridge():
    def __init__(self, *args, **kwargs):
        kwargs['cursorclass'] = DictCursor
        self.cnx = MySQLdb.connect (**kwargs)
        self.cnx.autocommit(True)
        self.cursor = self.cnx.cursor()

    def query_all(self, query, *args):
        self.cursor.execute(query, *args)
        return self.cursor.fetchall()

    def find_unique(self, query, *args):
        rows = self.query_all(query, *args);
        if len(rows) == 1:
            return rows[0]

        return None

    def execute(self, query, params):
        self.cursor.execute(query, params)
        return self.cursor.rowcount

    def get_last_id(self):
        return self.cnx.insert_id()

    def close(self):
        self.cursor.close()
        self.cnx.close()

database = DatabaseBridge(**{
        'user': 'user',
        'passwd': 'password',
        'db': 'my_db'
    })

rows = database.query_all("SELECT id, name, email FROM users WHERE is_active = %s AND project = %s", (1, "My First Project"))

(这是一个愚蠢的例子)。

它的作用就像魅力你必须考虑到这些:

  • 不支持多线程!如果你不使用Python的multiprocessing,那就没关系。
  • 您将无法拥有SQLAlchemy(数据库到类(模型)包装器,查询生成(select,where,order_by等))的所有优点。 这是您希望如何使用数据库的关键点。

但另一方面,和SQLAlchemy一样,还有保护SQL注入攻击:

基本查询将是这样的:

cursor.execute("SELECT * FROM users WHERE data = %s" % "Some value") # THIS IS DANGEROUS

但你应该这样做:

cursor.execute("SELECT * FROM users WHERE data = %s", ("Some value",)) # This is secure!

看到了区别?再读一遍;)

不同之处在于我将%替换为,:我们将参数作为...参数传递给执行,并将这些参数转义。使用%时,参数不会被转义,从而启用SQL注入攻击!

这里的最后一句话是,它取决于您的使用情况以及您计划如何处理项目。对我来说,SQLAlchemy过度使用(这是一个基本的shell脚本!),所以MysqlDB很完美。