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

add-column col_type now optional, defaults to str

Browse source
This commit is contained in:
Simon Willison 2019-02-24 14:24:00 -08:00
commit 3cab079d3e
6 changed files with 18 additions and 8 deletions
Download patch file Download diff file Expand all files Collapse all files

1
sqlite_utils/cli.py
View file

@ -148,6 +148,7 @@ def optimize(path, no_vacuum):
type=click.Choice(
["integer", "float", "blob", "text", "INTEGER", "FLOAT", "BLOB", "TEXT"]
),
required=False,
)
def add_column(path, table, col_name, col_type):
"Add a column to the specified table"

4
sqlite_utils/db.py
View file

@ -276,7 +276,9 @@ class Table:
self.db.conn.execute(sql)
return self
def add_column(self, col_name, col_type):
def add_column(self, col_name, col_type=None):
if col_type is None:
col_type = str
sql = "ALTER TABLE [{table}] ADD COLUMN [{col_name}] {col_type};".format(
table=self.name, col_name=col_name, col_type=COLUMN_TYPE_MAPPING[col_type]
)