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

Added create_index(..., if_not_exists=True), refs #14

Browse source
This commit is contained in:
Simon Willison 2019-02-24 10:46:44 -08:00
commit 7784b924a1
3 changed files with 18 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

5
sqlite_utils/db.py
View file

@ -236,19 +236,20 @@ class Table:
self.exists = True
return self
def create_index(self, columns, index_name=None, unique=False):
def create_index(self, columns, index_name=None, unique=False, if_not_exists=False):
if index_name is None:
index_name = "idx_{}_{}".format(
self.name.replace(" ", "_"), "_".join(columns)
)
sql = """
CREATE {unique}INDEX {index_name}
CREATE {unique}INDEX {if_not_exists}{index_name}
ON {table_name} ({columns});
""".format(
index_name=index_name,
table_name=self.name,
columns=", ".join(columns),
unique="UNIQUE " if unique else "",
if_not_exists="IF NOT EXISTS " if if_not_exists else "",
)
self.db.conn.execute(sql)
return self