From 71bb9694bc021ea0d35dd766f3737bd257ff45eb Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Madis=20M=C3=A4gi?= Date: Wed, 2 Aug 2023 00:17:59 +0300 Subject: [PATCH] Update red highlight check --- inventory-app/main.py | 14 ++++++-------- 1 file changed, 6 insertions(+), 8 deletions(-) diff --git a/inventory-app/main.py b/inventory-app/main.py index e1a413d..f4cd3f5 100755 --- a/inventory-app/main.py +++ b/inventory-app/main.py @@ -51,14 +51,12 @@ from oidc import page_oidc, login_required, read_user from api import page_api def check_foreign_key_format(item): - try: - if type(item["inventory"]["owner"]["foreign_id"]) == ObjectId: - return True - if type(item["inventory"]["user"]["foreign_id"]) == ObjectId: - return True - except: - pass - return False + owner = item.get("inventory", {}).get("owner", {}) + user = item.get("inventory", {}).get("user", {}) + if owner.get("foreign_id", False): + return not bool(owner.get("username")) + if user.get("foreign_id", False): + return not bool(user.get("username")) def render_markdown(text): if not text: