首页
学习
活动
专区
圈层
工具
发布
首页
学习
活动
专区
圈层
工具
社区首页 >问答首页 >使用表名作为给定参数构建SQL查询字符串

使用表名作为给定参数构建SQL查询字符串
EN

Stack Overflow用户
提问于 2022-01-10 19:39:02
回答 1查看 637关注 0票数 1

我试图在SQL中使用字符串格式。但是在传入变量时,这些变量是用引号插入的,并破坏了语法。

示例

这里,我试图将表名传递给函数。

代码语言:javascript
代码运行次数:0
运行
复制
def see_results(cur, table):
    print("complete")
    cur.execute(''' SELECT * from %s ''', (table,))
    results = cur.fetchall()
    print(results)

问题

如果我将"temp_yellow_pages"作为参数传递,则结果查询是:''' SELECT * from "temp_yellow_pages" '''。这个破了。

我想不出一种将任何东西分配给变量table而不使用"的方法,因为query = temp_yellow_pages也会中断。

EN

回答 1

Stack Overflow用户

回答已采纳

发布于 2022-01-10 20:29:33

字符串构建(易于SQL注入)

什么赫尔伍德

代码语言:javascript
代码运行次数:0
运行
复制
def selectFrom(table):
    return 'SELECT * FROM ' + table


def see_results(cur, table):
    print("complete")
    cur.execute(selectFrom(table))
    results = cur.fetchall()
    print(results)

甚至直接使用f字符串cur.execute(f"SELECT * FROM {table}"

但是,如果传递的参数table中有恶意输入,比如附加的DROPTRUNCATE语句( statement ),怎么办?

查询构建(更安全)

使用https://pypi.org/project/psycopg/之类的具有SQL功能的库( safe或https://pypi.org/project/psycopg/),您可以使用应用输入验证的安全方法来构建SQL。

请参阅模块psycopg2.sql中的示例,以便为给定的表参数组合SQL-语句。

代码语言:javascript
代码运行次数:0
运行
复制
from psycopg2 import sql

cur.execute(
    sql.SQL("SELECT * FROM {} WHERE values IN (%s, %s)")
        .format(sql.Identifier('my_table')),
    [10, 20])
票数 2
EN
页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接:

https://stackoverflow.com/questions/70658045

复制
相关文章

相似问题

领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档