From fa454c234ae04126b903b442cb7a2ec560c3ba8f Mon Sep 17 00:00:00 2001 From: miguelpadin Date: Tue, 6 Feb 2024 19:31:18 +0000 Subject: [PATCH] [IMP] pms-api-rest: add field 'createdBy' to folio & reservation service --- pms_api_rest/datamodels/pms_folio.py | 2 ++ pms_api_rest/datamodels/pms_reservation.py | 1 + pms_api_rest/services/pms_folio_service.py | 1 + pms_api_rest/services/pms_reservation_service.py | 2 ++ 4 files changed, 6 insertions(+) diff --git a/pms_api_rest/datamodels/pms_folio.py b/pms_api_rest/datamodels/pms_folio.py index 91c0943e6..271489d8d 100644 --- a/pms_api_rest/datamodels/pms_folio.py +++ b/pms_api_rest/datamodels/pms_folio.py @@ -29,6 +29,8 @@ class PmsFolioInfo(Datamodel): firstCheckin = fields.String(required=False, allow_none=True) lastCheckout = fields.String(required=False, allow_none=True) createDate = fields.String(required=False, allow_none=True) + createdBy = fields.String(required=False, allow_none=True) + pmsPropertyId = fields.Integer(required=False, allow_none=True) partnerId = fields.Integer(required=False, allow_none=True) reservations = fields.List( diff --git a/pms_api_rest/datamodels/pms_reservation.py b/pms_api_rest/datamodels/pms_reservation.py index e396172e6..9bf4da99b 100644 --- a/pms_api_rest/datamodels/pms_reservation.py +++ b/pms_api_rest/datamodels/pms_reservation.py @@ -91,6 +91,7 @@ class PmsReservationInfo(Datamodel): partnerRequests = fields.String(required=False, allow_none=True) nights = fields.Integer(required=False, allow_none=True) isReselling = fields.Boolean(required=False, allow_none=True) + createdBy = fields.String(required=False, allow_none=True) # TODO: Refact # messages = fields.List(fields.Dict(required=False, allow_none=True)) diff --git a/pms_api_rest/services/pms_folio_service.py b/pms_api_rest/services/pms_folio_service.py index 6fbdf4ed1..3ce6a51cc 100644 --- a/pms_api_rest/services/pms_folio_service.py +++ b/pms_api_rest/services/pms_folio_service.py @@ -62,6 +62,7 @@ class PmsFolioService(Component): firstCheckin=str(folio.first_checkin), lastCheckout=str(folio.last_checkout), createDate=folio.create_date.isoformat(), + createdBy=folio.create_uid.name, internalComment=folio.internal_comment if folio.internal_comment else None, diff --git a/pms_api_rest/services/pms_reservation_service.py b/pms_api_rest/services/pms_reservation_service.py index f6b3ec72f..6ee3af037 100644 --- a/pms_api_rest/services/pms_reservation_service.py +++ b/pms_api_rest/services/pms_reservation_service.py @@ -103,6 +103,7 @@ class PmsReservationService(Component): isSplitted=reservation.splitted, pendingCheckinData=reservation.pending_checkin_data, createDate=reservation.create_date.isoformat(), + createdBy=reservation.create_uid.name, segmentationId=reservation.segmentation_ids[0].id if reservation.segmentation_ids else None, @@ -130,6 +131,7 @@ class PmsReservationService(Component): isReselling=any( line.is_reselling for line in reservation.reservation_line_ids ), + ) return res