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

--ignore for sqlite-utils drop-table and drop-view, closes #237

Browse source
This commit is contained in:
Simon Willison 2021-02-25 09:11:37 -08:00
commit bba6e241be
3 changed files with 36 additions and 9 deletions
Download patch file Download diff file Expand all files Collapse all files

18
sqlite_utils/cli.py
View file

@ -946,14 +946,15 @@ def create_table(
required=True,
)
@click.argument("table")
@click.option("--ignore", is_flag=True)
@load_extension_option
def drop_table(path, table, load_extension):
def drop_table(path, table, ignore, load_extension):
"Drop the specified table"
db = sqlite_utils.Database(path)
_load_extensions(db, load_extension)
if table in db.table_names():
db[table].drop()
else:
try:
db[table].drop(ignore=ignore)
except sqlite3.OperationalError:
raise click.ClickException('Table "{}" does not exist'.format(table))
@ -1002,14 +1003,15 @@ def create_view(path, view, select, ignore, replace, load_extension):
required=True,
)
@click.argument("view")
@click.option("--ignore", is_flag=True)
@load_extension_option
def drop_view(path, view, load_extension):
def drop_view(path, view, ignore, load_extension):
"Drop the specified view"
db = sqlite_utils.Database(path)
_load_extensions(db, load_extension)
if view in db.view_names():
db[view].drop()
else:
try:
db[view].drop(ignore=ignore)
except sqlite3.OperationalError:
raise click.ClickException('View "{}" does not exist'.format(view))