api: don't iterate over all formdefs when getting user forms (#56518)
This commit is contained in:
parent
80660e37d9
commit
522d605701
22
wcs/api.py
22
wcs/api.py
|
@ -53,7 +53,7 @@ from .qommon.errors import (
|
|||
UnknownNameIdAccessForbiddenError,
|
||||
)
|
||||
from .qommon.form import ComputedExpressionWidget
|
||||
from .qommon.storage import Contains, Equal, NotEqual
|
||||
from .qommon.storage import Contains, Equal, Intersects, NotEqual, Or
|
||||
from .qommon.template import Template, TemplateError
|
||||
|
||||
|
||||
|
@ -888,11 +888,9 @@ class ApiUserDirectory(Directory):
|
|||
if category_slugs:
|
||||
categories = Category.select([Contains('url_name', category_slugs)])
|
||||
|
||||
if get_publisher().is_using_postgresql() and not get_request().form.get('full') == 'on':
|
||||
if get_publisher().is_using_postgresql():
|
||||
from wcs import sql
|
||||
|
||||
from .qommon.storage import Equal
|
||||
|
||||
order_by = 'receipt_time'
|
||||
if get_request().form.get('sort') == 'desc':
|
||||
order_by = '-receipt_time'
|
||||
|
@ -905,6 +903,22 @@ class ApiUserDirectory(Directory):
|
|||
offset=misc.get_int_or_400(get_request().form.get('offset')),
|
||||
order_by=order_by,
|
||||
)
|
||||
if get_request().form.get('full') == 'on':
|
||||
# load full objects
|
||||
formdefs = {x.formdef_id: x.formdef for x in user_forms}
|
||||
formdef_user_forms = {}
|
||||
for formdef_id, formdef in formdefs.items():
|
||||
formdef_user_forms.update(
|
||||
{
|
||||
(formdef_id, x.id): x
|
||||
for x in formdef.data_class().select(
|
||||
[Contains('id', [x.id for x in user_forms if x.formdef_id == formdef_id])]
|
||||
)
|
||||
}
|
||||
)
|
||||
# and put them back in order
|
||||
sorted_user_forms_tuples = [(x.formdef_id, x.id) for x in user_forms]
|
||||
user_forms = [formdef_user_forms.get(x) for x in sorted_user_forms_tuples]
|
||||
else:
|
||||
formdefs = FormDef.select()
|
||||
user_forms = []
|
||||
|
|
Loading…
Reference in New Issue