sqlite中的变量表名

Question

问题：是否可以使用变量作为表名而不必使用字符串构造函数？

---

信息：

我现在正在开展一个项目，该项目对来自我的恒星模拟的数据进行分类。为此，我将所有数据加载到 sqlite 数据库中。它工作得很好，但我决定为我的数据库增加更多的灵活性、效率和可用性。我计划稍后将小行星添加到模拟中，并希望为每颗恒星提供一个表格。这样我就不必为每个太阳系中的 1-4k 查询 20m 一些小行星的表。

有人告诉我，使用字符串构造函数很糟糕，因为它让我容易受到 SQL 注入攻击。虽然这在这里没什么大不了的，因为我是唯一可以访问这些数据库的人，但我想遵循最佳实践。而且这样，如果我做一个类似情况的向公众开放的项目，我知道该怎么做。

目前我正在这样做：

cursor.execute("CREATE TABLE StarFrame"+self.name+" (etc etc)")

这可行，但我想做更多类似的事情：

cursor.execute("CREATE TABLE StarFrame(?) (etc etc)",self.name)

虽然我知道这可能是不可能的。虽然我会满足于类似的东西

cursor.execute("CREATE TABLE (?) (etc etc)",self.name)

如果这根本不可能，我会接受这个答案，但如果有人知道这样做的方法，请告诉。 :)

我正在用 python 编码。

Answer 1

很遗憾，表格不能作为参数替换的目标（我没有找到任何确定的来源，但我在一些网络论坛上看到过）。

如果您担心注入（您可能应该担心），您可以编写一个函数，在传递字符串之前对其进行清理。由于您只是在寻找一个表名，因此您应该安全地接受字母数字，去掉所有标点符号，例如 )(][;, 和空格。基本上，只需保留A-Z a-z 0-9。

def scrub(table_name):
    return ''.join( chr for chr in table_name if chr.isalnum() )

scrub('); drop tables --')  # returns 'droptables'

Answer 2

对于寻找将表格作为变量的方法的人，我从另一个回复相同的问题here 中得到这个：

它说了以下内容并且有效。全部引自mhawke：

您不能对表名使用参数替换。您需要自己将表名添加到查询字符串中。像这样的：

query = 'SELECT * FROM {}'.format(table)
c.execute(query)

需要注意的一点是表名的值的来源。如果那来自不受信任的来源，例如一个用户，那么您需要验证表名以避免潜在的 SQL 注入攻击。一种方法可能是构造一个参数化查询，从 DB 目录中查找表名：

import sqlite3

def exists_table(db, name):
    query = "SELECT 1 FROM sqlite_master WHERE type='table' and name = ?"
    return db.execute(query, (name,)).fetchone() is not None

Answer 3

我不会将数据分成多个表。如果您在星列上创建索引，则高效访问数据不会有任何问题。

Answer 4

尝试字符串格式：

sql_cmd = '''CREATE TABLE {}(id, column1, column2, column2)'''.format(
            'table_name')
db.execute(sql_cmd)

用你的愿望替换'table_name'。

Answer 5

为了避免硬编码表名，我使用了：

table = "sometable"
c = conn.cursor()

c.execute('''CREATE TABLE IF NOT EXISTS {} (
                importantdate DATE,
                somename VARCHAR,
                )'''.format(table))

c.execute('''INSERT INTO {} VALUES (?, ?)'''.format(table),
                  (datetime.strftime(datetime.today(), "%Y-%m-%d"),
                   myname))

Answer 6

正如在其他答案中所说，“表不能成为参数替换的目标”但是如果您发现自己处于没有选择的绑定中，这是一种测试提供的表名是否为的方法有效。
注意：我已将表名设为真正的猪，以尝试涵盖所有基础。

import sys
import sqlite3
def delim(s):
  delims="\"'`"
  use_delim = []
  for d in delims:
   if d not in s:
    use_delim.append(d)
  return use_delim

db_name = "some.db"
db = sqlite3.connect(db_name)
mycursor = db.cursor()
table = 'so""m ][ `etable'
delimiters = delim(table)
if len(delimiters) < 1:
    print "The name of the database will not allow this!"
    sys.exit()
use_delimiter = delimiters[0]
print "Using delimiter ", use_delimiter
mycursor.execute('SELECT name FROM sqlite_master where (name = ?)', [table])
row = mycursor.fetchall()
valid_table = False
if row:
    print (table,"table name verified")
    valid_table = True
else:
    print (table,"Table name not in database", db_name)

if valid_table:
    try:
        mycursor.execute('insert into ' +use_delimiter+ table +use_delimiter+ ' (my_data,my_column_name) values (?,?) ',(1,"Name"));
        db.commit()
    except Exception as e:
        print "Error:", str(e)
    try:
        mycursor.execute('UPDATE ' +use_delimiter+ table +use_delimiter+ ' set my_column_name = ? where my_data = ?', ["ReNamed",1])
        db.commit()
    except Exception as e:
        print "Error:", str(e)
db.close()

Answer 7

你可以使用这样的东西 conn = sqlite3.connect() createTable = '''CREATE TABLE %s (# );''' %dateNow) conn.execute(createTable)

基本上，如果我们现在想根据日期将数据分成几个表，例如，您想根据日期监控系统。

createTable = '''CREATE TABLE %s (# );''' %dateNow) 表示您创建一个带有变量 dateNow 的表，根据您的编码语言，您可以将 dateNow 定义为一个变量来检索当前日期来自您的编码语言。

Answer 8

您可以将查询保存在 .sql 或 txt 文件中，并使用 open().replace() 方法在查询的任何部分使用变量。长期阅读，但第一次发帖，如果这里有任何问题，我深表歉意。

```SQL in yoursql.sql```
Sel *
From yourdbschema.tablenm


```SQL to run```
tablenm = 'yourtablename'

cur = connect.cursor() 

query = cur.execute(open(file = yoursql.sql).read().replace('tablenm',tablenm))

Answer 9

您可以将字符串作为 SQL 命令传递：

import sqlite3
conn = sqlite3.connect('db.db')
c = conn.cursor()
tablename, field_data = 'some_table','some_data'
query = 'SELECT * FROM '+tablename+' WHERE column1=\"'+field_data+"\""
c.execute(query)