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

table.m2m(..., m2m_table=x) argument

Browse source
This commit is contained in:
Simon Willison 2019-08-03 20:51:22 +03:00
commit b6b92980c0
2 changed files with 13 additions and 7 deletions
Download patch file Download diff file Expand all files Collapse all files

9
sqlite_utils/db.py
View file

@ -1048,7 +1048,9 @@ class Table:
self.create_index(column_values.keys(), unique=True)
return pk
def m2m(self, other_table, record_or_list=None, pk=None, lookup=None):
def m2m(
self, other_table, record_or_list=None, pk=None, lookup=None, m2m_table=None
):
our_id = self.last_pk
if lookup is not None:
assert record_or_list is None, "Provide lookup= or record, not both"
@ -1056,9 +1058,8 @@ class Table:
assert record_or_list is not None, "Provide lookup= or record, not both"
tables = list(sorted([self.name, other_table]))
columns = ["{}_id".format(t) for t in tables]
m2m_table = self.db.table(
"{}_{}".format(*tables), pk=columns, foreign_keys=columns
)
m2m_table_name = m2m_table or "{}_{}".format(*tables)
m2m_table = self.db.table(m2m_table_name, pk=columns, foreign_keys=columns)
if lookup is None:
records = (
[record_or_list]

11
tests/test_m2m.py
View file

@ -74,9 +74,14 @@ def test_m2m_requires_either_records_or_lookup(fresh_db):
people.m2m("tags", {"tag": "hello"}, lookup={"foo": "bar"})
def test_m2m_explicit_argument(fresh_db):
# .m2m("humans", ..., m2m_table="relationships")
assert False
def test_m2m_explicit_table_name_argument(fresh_db):
people = fresh_db.table("people", pk="id")
people.insert({"name": "Wahyu"}).m2m(
"tags", lookup={"tag": "Coworker"}, m2m_table="tagged"
)
assert fresh_db["tags"].exists
assert fresh_db["tagged"].exists
assert not fresh_db["people_tags"].exists
def test_uses_existing_m2m_table_if_exists(fresh_db):