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

sqlite-utils add-foreign-key books.db books author_id authors id

Browse source 
Command for adding foreign keys to existing tables. Closes #2
This commit is contained in:
Simon Willison 2019-02-24 13:33:45 -08:00
commit f8d3b7cfe5
3 changed files with 76 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

26
sqlite_utils/cli.py
View file

@ -1,6 +1,7 @@
import click
from click_default_group import DefaultGroup
import sqlite_utils
from sqlite_utils.db import AlterError
import itertools
import json
import sys
@ -154,6 +155,31 @@ def add_column(path, table, col_name, col_type):
db[table].add_column(col_name, col_type)
@cli.command(name="add-foreign-key")
@click.argument(
"path",
type=click.Path(exists=True, file_okay=True, dir_okay=False, allow_dash=False),
required=True,
)
@click.argument("table")
@click.argument("column")
@click.argument("other_table")
@click.argument("other_column")
def add_foreign_key(path, table, column, other_table, other_column):
"""
Add a new foreign key constraint to an existing table. Example usage:
$ sqlite-utils add-foreign-key my.db books author_id authors id
WARNING: Could corrupt your database! Back up your database file first.
"""
db = sqlite_utils.Database(path)
try:
db[table].add_foreign_key(column, other_table, other_column)
except AlterError as e:
raise click.ClickException(e)
@cli.command(name="create-index")
@click.argument(
"path",