From 812366f800ab2d6e29ff1796e614a68a20c2ed7a Mon Sep 17 00:00:00 2001 From: Sara Lago Date: Wed, 7 Jun 2023 11:19:42 +0200 Subject: [PATCH] [IMP] pms_api_rest: add filter in get_folios --- pms_api_rest/services/pms_folio_service.py | 9 +++------ 1 file changed, 3 insertions(+), 6 deletions(-) diff --git a/pms_api_rest/services/pms_folio_service.py b/pms_api_rest/services/pms_folio_service.py index 7cf8013fb..4ab2e00d9 100644 --- a/pms_api_rest/services/pms_folio_service.py +++ b/pms_api_rest/services/pms_folio_service.py @@ -51,9 +51,7 @@ class PmsFolioService(Component): partnerName=folio.partner_name if folio.partner_name else None, partnerPhone=folio.mobile if folio.mobile else None, partnerEmail=folio.email if folio.email else None, - state=dict(folio.fields_get(["state"])["state"]["selection"])[ - folio.state - ], + state=folio.state, amountTotal=round(folio.amount_total, 2), reservationType=folio.reservation_type, pendingAmount=folio.pending_amount, @@ -164,6 +162,7 @@ class PmsFolioService(Component): elif folio_search_param.filterByState == 'toAssign': subdomains = [ [("to_assign", "=", True)], + [("state", "in", ("draft", "confirm", "arrival_delayed"))], [("reservation_type", "!=", "out")], ] domain_filter.append(expression.AND(subdomains)) @@ -242,9 +241,7 @@ class PmsFolioService(Component): PmsFolioShortInfo( id=folio.id, name=folio.name, - state=dict(folio.fields_get(["state"])["state"]["selection"])[ - folio.state - ], + state=folio.state, partnerName=folio.partner_name if folio.partner_name else None, partnerPhone=folio.mobile if folio.mobile else None, partnerEmail=folio.email if folio.email else None,