From 293c3ec4f63911cb0eab2d043109ae992dea32d2 Mon Sep 17 00:00:00 2001
From: Jose Luis <osotranquilo@gmail.com>
Date: Wed, 31 Jul 2019 20:00:31 +0200
Subject: [PATCH] [ADD] Segmentation and other fix

---
 .../models/inherited_hotel_checkin_partner.py  | 18 +++++++++++++++---
 hotel_roommatik/models/roommatik.py            | 17 +++++++++++++++++
 2 files changed, 32 insertions(+), 3 deletions(-)

diff --git a/hotel_roommatik/models/inherited_hotel_checkin_partner.py b/hotel_roommatik/models/inherited_hotel_checkin_partner.py
index 506d26be3..df12d86cb 100644
--- a/hotel_roommatik/models/inherited_hotel_checkin_partner.py
+++ b/hotel_roommatik/models/inherited_hotel_checkin_partner.py
@@ -25,7 +25,9 @@ class HotelFolio(models.Model):
                 'adults': stay['Adults'],
                 'arrival_hour': stay['Arrival_hour'],
                 'room_type_id': stay['RoomType']['Id'],
-                'partner_id': stay["Customers"][0]["Id"]
+                'partner_id': stay["Customers"][0]["Id"],
+                'segmentation_ids': stay['Segmentation'],
+                'channel_type': 'virtualdoor',
             }
             reservation_rm = reservation_obj.create(vals)
         else:
@@ -37,6 +39,13 @@ class HotelFolio(models.Model):
                          total_chekins,
                          reservation_rm.id)
             for room_partner in stay["Customers"]:
+                if room_partner['Address']['Nationality'] == 'ESP':
+                    code_ine = room_partner['Address']['Province']
+                else:
+                    code_ine = room_partner['Address']['Nationality']
+                province = self.env['code.ine'].search(
+                    [('name', '=', code_ine)], limit=1)
+                code_ine = province.id
 
                 checkin_partner_val = {
                     'folio_id': reservation_rm.folio_id.id,
@@ -44,6 +53,7 @@ class HotelFolio(models.Model):
                     'partner_id': room_partner["Id"],
                     'enter_date': stay["Arrival"],
                     'exit_date': stay["Departure"],
+                    'code_ine_id': code_ine,
                     }
                 try:
                     record = self.env['hotel.checkin.partner'].create(
@@ -57,8 +67,10 @@ class HotelFolio(models.Model):
                     stay['Id'] = record.id
                     stay['Room'] = reservation_rm.room_id.id
                     json_response = stay
-                except:
-                    json_response = {'Estate': 'Error not create Checkin'}
+                except Exception as e:
+                    error_name = 'Error not create Checkin '
+                    error_name += e.name
+                    json_response = {'Estate': error_name}
                     _logger.error('ROOMMATIK writing %s in reservation: %s).',
                                   checkin_partner_val['partner_id'],
                                   checkin_partner_val['reservation_id'])
diff --git a/hotel_roommatik/models/roommatik.py b/hotel_roommatik/models/roommatik.py
index bc4bf22f6..4bc380f7f 100755
--- a/hotel_roommatik/models/roommatik.py
+++ b/hotel_roommatik/models/roommatik.py
@@ -81,6 +81,23 @@ class RoomMatik(models.Model):
         apidata = self.env['hotel.room.type']
         return apidata.sudo().rm_get_prices(start_date, number_intervals, room_type, guest_number)
 
+    @api.model
+    def rm_get_segmentation(self):
+        # Gets segmentation list
+        # return ArrayOfSegmentation
+        segmentations = self.env['res.partner.category'].sudo().search([])
+        _logger.info('ROOMMATIK Get segmentation')
+        response = []
+        for segmentation in segmentations:
+            response.append({
+                "Segmentation": {
+                    "Id": segmentation.id,
+                    "Name": segmentation.display_name,
+                    },
+                })
+            json_response = json.dumps(response)
+        return json_response
+
     @api.model
     def _rm_add_payment(self, code, payment):
         apidata = self.env['account.payment']