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

Prettier indentation for CREATE TABLE

Browse source
This commit is contained in:
Simon Willison 2019-02-23 12:51:35 -08:00
commit ca225b02c0
2 changed files with 13 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

6
sqlite_utils/db.py
View file

@ -71,7 +71,7 @@ class Database:
None.__class__: "TEXT",
}
columns_sql = ",\n".join(
" [{col_name}] {col_type} {primary_key} {references}".format(
" [{col_name}] {col_type}{primary_key}{references}".format(
col_name=col_name,
col_type=col_type_mapping[col_type],
primary_key=" PRIMARY KEY" if (pk == col_name) else "",
@ -87,8 +87,8 @@ class Database:
for col_name, col_type in column_items
)
sql = """CREATE TABLE [{table}] (
{columns_sql}
){extra};
{columns_sql}
){extra};
""".format(
table=name, columns_sql=columns_sql, extra=extra
)

10
tests/test_create.py
View file

@ -28,6 +28,16 @@ def test_create_table(fresh_db):
{"name": "bytes_col", "type": "BLOB"},
{"name": "datetime_col", "type": "TEXT"},
] == [{"name": col.name, "type": col.type} for col in table.columns]
assert (
"CREATE TABLE [test_table] (\n"
" [text_col] TEXT,\n"
" [float_col] FLOAT,\n"
" [int_col] INTEGER,\n"
" [bool_col] INTEGER,\n"
" [bytes_col] BLOB,\n"
" [datetime_col] TEXT\n"
")"
) == table.schema
@pytest.mark.parametrize(