Question

我想在我的网页上创建一个搜索功能，该功能链接到5个表的数据库。

这个想法是，用户可以选择要搜索的表并输入搜索词，它将返回结果列表。

很明显，我可以使用5个if语句来做到这一点，但似乎应该有一种更简单的方法。

这段代码是我尝试过的，但是由于某种原因，当我将item和table变量放入查询中而不是实际的列名时，它不起作用。仅当我在table和table.item中键入特定术语时，它才有效。

   if 'form.submitted' in request.params:
        #search term
        search = request.params['body']
        #table to be searched
        table = request.params['table']
        list = eval(table).__table__.columns._data.keys()

        results = {}

        for item in list:
          try:

            searchdb = request.dbsession.query(table).filter(table.item==search).all()                                     

            results[item]=searchdb

          except:

              continue

        return (results)

我想知道这是否可行，或者我应该放弃并写下5条if语句

Answer 1

请不要使用eval()，it is mostly evil。您无法安全地在用户输入上使用它，如"Eval really is dangerous"所示。想象一下一个对手正在调用您的API，而不是使用您的表单或类似的东西，只是向您发送了一些内容，

body=byebye&table=__import__("os").system("rm -rf /")

根据您的设置，您可能会丢失整个系统，容器或所拥有的东西。这不是他们唯一能做的。它们具有完整的Python表达式。

在您的情况下，处理用户选择的表（模型）的正确方法是进行查找：

the_5_tables = {
    "table_1": Table1,
    "table_2": Table2,
    # etc.
}

然后您要做的就是

#table to be searched
model = the_5_tables[request.params['table']]

这具有将用户可以使用的表列入白名单的附加好处，即使当前作用域可以访问其他表。

使用实际的Column对象而不是其键，可以更轻松地制作过滤器：

results = {}

for col in model.__table__.columns:
    try:
        searchdb = request.dbsession.query(model).filter(col == search).all()                                     
        results[col.key] = searchdb

    except Exception as e:
        print(f"Unhandled error: {e}")
        continue

return results

不幸的是，这将使您与表中的列一样多的数据库往返次数。现在，我猜测裸except:的存在是为了掩盖类型不匹配引起的错误，例如，当您尝试搜索数字列时。您可以进一步检查这些列，以避免出现这种情况：

from sqlalchemy.types import String

# ...

results = {}

for col in model.__table__.columns:
    if isinstance(col.type, String):
        searchdb = request.dbsession.query(model).filter(col == search).all()                                     
        results[col.key] = searchdb

return results

更进一步，如果您实际上对匹配哪些列不感兴趣，则可以构成一个查询，例如：

from sqlalchemy import literal
from sqlalchemy.types import String

# ...

str_cols = [c for c in model.__table__.c if isinstance(c.type, String)]
results = request.dbsession.query(model).filter(literal(search).in_(str_cols)).all()                                     
return results

尽管有点hacky，但仍然可以在单个查询中获取匹配的列（这没什么用；在查询后在Python中执行相同的操作很简单）：

from sqlalchemy import func

# ...

results = request.dbsession.query(
        model,
        func.concat_ws(",", *[
            func.if_(c == search, c.key, None)
            for c in str_cols
        ]).label("in_columns")).\
    filter(literal(search).in_(str_cols)).all()                                     
return results

in_columns将是用逗号分隔的匹配名称的字符串。

Answer 2

请不要使用\b\p{N}+\s+(?:an|year)s?（内置类型）来命名变量/标识符。

list

使用SQL Alchemy查询，有一种方法可以遍历表中的所有列

2 个答案: