From 2af2c257ecac5f562d52edecb5ebd7812c1b43e3 Mon Sep 17 00:00:00 2001 From: miguelpadin Date: Tue, 15 Feb 2022 15:16:30 +0000 Subject: [PATCH] [IMP] pms: add user image base63 format as login response --- pms_api_rest/datamodels/pms_user.py | 7 +++++-- pms_api_rest/services/pms_login_service.py | 10 +++++++++- 2 files changed, 14 insertions(+), 3 deletions(-) diff --git a/pms_api_rest/datamodels/pms_user.py b/pms_api_rest/datamodels/pms_user.py index 83b03cef3f4..80683707ef5 100644 --- a/pms_api_rest/datamodels/pms_user.py +++ b/pms_api_rest/datamodels/pms_user.py @@ -11,6 +11,9 @@ class PmsApiRestUserInput(Datamodel): class PmsApiRestUserOutput(Datamodel): _name = "pms.api.rest.user.output" - # user = fields.String(required=False, allow_none=True) - # exp = fields.String(required=False, allow_none=True) token = fields.String(required=False, allow_none=True) + userId = fields.Integer(required=True, allow_none=False) + userName = fields.String(required=True, allow_none=False) + userImageBase64 = fields.String(required=False, allow_none=True) + defaultPropertyId = fields.Integer(required=True, allow_none=False) + defaultPropertyName = fields.String(required=True, allow_none=False) diff --git a/pms_api_rest/services/pms_login_service.py b/pms_api_rest/services/pms_login_service.py index a719cc18793..281c5ab53e9 100644 --- a/pms_api_rest/services/pms_login_service.py +++ b/pms_api_rest/services/pms_login_service.py @@ -52,4 +52,12 @@ def login(self, user): key="pms_secret_key_example", algorithm=jwt.ALGORITHMS.HS256, ) - return PmsApiRestUserOutput(token=token) + + return PmsApiRestUserOutput( + token=token, + userId=user_record.id, + userName=user_record.name, + defaultPropertyId=user_record.pms_property_id.id, + defaultPropertyName=user_record.pms_property_id.name, + userImageBase64=user_record.partner_id.image_1024, + )