From a758d9397008f05a0d35fb23734e77471d4ee796 Mon Sep 17 00:00:00 2001 From: Oscar Krause Date: Tue, 17 Jan 2023 11:18:07 +0100 Subject: [PATCH] main.py - fixed empty lease origin response --- app/main.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/app/main.py b/app/main.py index 573cb42..46d8a65 100644 --- a/app/main.py +++ b/app/main.py @@ -170,8 +170,9 @@ async def _leases(request: Request, origin: bool = False): for lease in session.query(Lease).all(): x = lease.serialize() if origin: - # assume that each lease has a valid origin record - x['origin'] = session.query(Origin).filter(Origin.origin_ref == lease.origin_ref).first().serialize() + lease_origin = session.query(Origin).filter(Origin.origin_ref == lease.origin_ref).first() + if lease_origin is not None: + x['origin'] = lease_origin.serialize() response.append(x) session.close() return JSONr(response)