mirror of
https://github.com/simonw/sqlite-utils.git
synced 2026-07-23 01:14:31 +02:00
Create table if_not_exists=True argument, closes #397
This commit is contained in:
parent
79b5b58354
commit
aa24903113
3 changed files with 33 additions and 3 deletions
|
|
@ -522,7 +522,9 @@ This will create a table with the following schema:
|
|||
Explicitly creating a table
|
||||
---------------------------
|
||||
|
||||
You can directly create a new table without inserting any data into it using the ``.create()`` method::
|
||||
You can directly create a new table without inserting any data into it using the ``.create()`` method:
|
||||
|
||||
.. code-block:: python
|
||||
|
||||
db["cats"].create({
|
||||
"id": int,
|
||||
|
|
@ -534,6 +536,17 @@ The first argument here is a dictionary specifying the columns you would like to
|
|||
|
||||
This method takes optional arguments ``pk=``, ``column_order=``, ``foreign_keys=``, ``not_null=set()`` and ``defaults=dict()`` - explained below.
|
||||
|
||||
A ``sqlite_utils.utils.sqlite3.OperationalError`` will be raised if a table of that name already exists.
|
||||
|
||||
To do nothing if the table already exists, add ``if_not_exists=True``:
|
||||
|
||||
.. code-block:: python
|
||||
|
||||
db["cats"].create({
|
||||
"id": int,
|
||||
"name": str,
|
||||
}, pk="id", if_not_exists=True)
|
||||
|
||||
.. _python_api_compound_primary_keys:
|
||||
|
||||
Compound primary keys
|
||||
|
|
|
|||
|
|
@ -658,6 +658,7 @@ class Database:
|
|||
defaults: Optional[Dict[str, Any]] = None,
|
||||
hash_id: Optional[str] = None,
|
||||
extracts: Optional[Union[Dict[str, str], List[str]]] = None,
|
||||
if_not_exists: bool = False,
|
||||
) -> str:
|
||||
"Returns the SQL ``CREATE TABLE`` statement for creating the specified table."
|
||||
foreign_keys = self.resolve_foreign_keys(name, foreign_keys or [])
|
||||
|
|
@ -748,11 +749,14 @@ class Database:
|
|||
pks=", ".join(["[{}]".format(p) for p in pk])
|
||||
)
|
||||
columns_sql = ",\n".join(column_defs)
|
||||
sql = """CREATE TABLE [{table}] (
|
||||
sql = """CREATE TABLE {if_not_exists}[{table}] (
|
||||
{columns_sql}{extra_pk}
|
||||
);
|
||||
""".format(
|
||||
table=name, columns_sql=columns_sql, extra_pk=extra_pk
|
||||
if_not_exists="IF NOT EXISTS " if if_not_exists else "",
|
||||
table=name,
|
||||
columns_sql=columns_sql,
|
||||
extra_pk=extra_pk,
|
||||
)
|
||||
return sql
|
||||
|
||||
|
|
@ -767,6 +771,7 @@ class Database:
|
|||
defaults: Optional[Dict[str, Any]] = None,
|
||||
hash_id: Optional[str] = None,
|
||||
extracts: Optional[Union[Dict[str, str], List[str]]] = None,
|
||||
if_not_exists: bool = False,
|
||||
) -> "Table":
|
||||
"""
|
||||
Create a table with the specified name and the specified ``{column_name: type}`` columns.
|
||||
|
|
@ -783,6 +788,7 @@ class Database:
|
|||
defaults=defaults,
|
||||
hash_id=hash_id,
|
||||
extracts=extracts,
|
||||
if_not_exists=if_not_exists,
|
||||
)
|
||||
self.execute(sql)
|
||||
table = self.table(
|
||||
|
|
@ -1300,6 +1306,7 @@ class Table(Queryable):
|
|||
defaults: Optional[Dict[str, Any]] = None,
|
||||
hash_id: Optional[str] = None,
|
||||
extracts: Optional[Union[Dict[str, str], List[str]]] = None,
|
||||
if_not_exists: bool = False,
|
||||
) -> "Table":
|
||||
"""
|
||||
Create a table with the specified columns.
|
||||
|
|
@ -1318,6 +1325,7 @@ class Table(Queryable):
|
|||
defaults=defaults,
|
||||
hash_id=hash_id,
|
||||
extracts=extracts,
|
||||
if_not_exists=if_not_exists,
|
||||
)
|
||||
return self
|
||||
|
||||
|
|
|
|||
|
|
@ -1114,3 +1114,12 @@ def test_create(fresh_db):
|
|||
" [bytes] BLOB\n"
|
||||
")"
|
||||
)
|
||||
|
||||
|
||||
def test_create_if_not_exists(fresh_db):
|
||||
fresh_db["t"].create({"id": int})
|
||||
# This should error
|
||||
with pytest.raises(sqlite3.OperationalError):
|
||||
fresh_db["t"].create({"id": int})
|
||||
# This should not
|
||||
fresh_db["t"].create({"id": int}, if_not_exists=True)
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue