mirror of
https://github.com/fastapi-admin/fastapi-admin.git
synced 2025-08-15 03:04:37 +08:00
32 lines
865 B
Python
32 lines
865 B
Python
from copy import deepcopy
|
|
|
|
|
|
async def handle_m2m_fields_create_or_update(body, m2m_fields, model, create=True, pk=None):
|
|
"""
|
|
handle m2m update or create
|
|
:param body:
|
|
:param m2m_fields:
|
|
:param model:
|
|
:param create:
|
|
:param pk:
|
|
:return:
|
|
"""
|
|
copy_body = deepcopy(body)
|
|
m2m_body = {}
|
|
for k, v in body.items():
|
|
if k in m2m_fields:
|
|
m2m_body[k] = copy_body.pop(k)
|
|
if create:
|
|
obj = await model.create(**copy_body)
|
|
else:
|
|
await model.filter(pk=pk).update(**copy_body)
|
|
obj = await model.get(pk=pk)
|
|
for k, v in m2m_body.items():
|
|
m2m_related = getattr(obj, k)
|
|
if not create:
|
|
await m2m_related.clear()
|
|
m2m_model = m2m_related.model
|
|
m2m_objs = await m2m_model.filter(pk__in=v)
|
|
await m2m_related.add(*m2m_objs)
|
|
return obj
|