From 08cec77887546737a17233bb5ad42a088b0f7609 Mon Sep 17 00:00:00 2001 From: Dichi13 <diciganteng01@icloud.com> Date: Sun, 28 Apr 2019 14:41:16 +0700 Subject: [PATCH] fixed subdata admin --- modules/page.py | 27 ++++++++++++++++----------- 1 file changed, 16 insertions(+), 11 deletions(-) diff --git a/modules/page.py b/modules/page.py index dbafce7..54895d8 100644 --- a/modules/page.py +++ b/modules/page.py @@ -40,24 +40,29 @@ def get_page(): } if return_admin: - admins = db.User.objects(page_list__in = [vdata.id]) - if len(admins) > 0: - admin = admins[0] - del admin.password - del admin.page_list - vdata["admin"] = admin - else: - vdata["admin"] = None - + resp["admin_subdata"] = [] + i = 0 for x in subdata: admins = db.User.objects(page_list__in = [x.id]) if len(admins) > 0: admin = admins[0] del admin.password del admin.page_list - x["admin"] = admin + print(admin) + resp["admin_subdata"].append(admin) else: - x["admin"] = None + resp["admin_subdata"].append(None) + i += 1 + + if return_admin: + admins = db.User.objects(page_list__in = [vdata.id]) + if len(admins) > 0: + admin = admins[0] + del admin.password + del admin.page_list + resp["admin"] = admin + else: + resp["admin"] = None return jsonify(resp) else: -- GitLab