remote/sqlite-utils
1
0
Fork 0
mirror of https://github.com/simonw/sqlite-utils.git synced 2026-09-19 06:54:25 +02:00
Code Issues Projects Releases Packages Wiki Activity

Add tests for escaping table names.

Browse source
This commit is contained in:
Amjith Ramanujam 2019-09-02 10:21:26 -07:00
commit e01943271b
3 changed files with 55 additions and 11 deletions
Download patch file Download diff file Expand all files Collapse all files

15
sqlite_utils/db.py
View file

@ -646,7 +646,7 @@ class Table(Queryable):
return self
def drop(self):
self.db.conn.execute("DROP TABLE {}".format(self.name))
self.db.conn.execute("DROP TABLE [{}]".format(self.name))
def guess_foreign_table(self, column):
column = column.lower()
@ -760,21 +760,20 @@ class Table(Queryable):
def detect_fts(self):
"Detect if table has a corresponding FTS virtual table and return it"
rows = self.db.conn.execute(
"""
sql = """
SELECT name FROM sqlite_master
WHERE rootpage = 0
AND (
sql LIKE '%VIRTUAL TABLE%USING FTS%content="{table}"%'
sql LIKE '%VIRTUAL TABLE%USING FTS%content=%{table}%'
OR (
tbl_name = "{table}"
AND sql LIKE '%VIRTUAL TABLE%USING FTS%'
)
)
""".format(
table=self.name
)
).fetchall()
table=self.name
)
rows = self.db.conn.execute(sql).fetchall()
if len(rows) == 0:
return None
else:
@ -1166,7 +1165,7 @@ class View(Queryable):
)
def drop(self):
self.db.conn.execute("DROP VIEW {}".format(self.name))
self.db.conn.execute("DROP VIEW [{}]".format(self.name))
def chunks(sequence, size):