main.py - fixed empty lease origin response

2023-01-17 11:18:07 +01:00 · 2023-01-17 11:18:07 +01:00 · a758d93970
parent a65687a082
commit a758d93970
1 changed files with 3 additions and 2 deletions
--- 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)