change request api list site

1 files changed, 4 insertions, 11 deletions

diff --git a/indoteknik_api/controllers/api_v1/partner.py b/indoteknik_api/controllers/api_v1/partner.py
index 1e7220cc..a6e14a19 100644
--- a/indoteknik_api/controllers/api_v1/partner.py
+++ b/indoteknik_api/controllers/api_v1/partner.py
@@ -8,19 +8,12 @@ class Partner(controller.Controller):
     prefix = '/api/v1/'
 
     def get_partner_child_ids(self, partner_id):
-        def _get_all_child_ids(partner):
-            child_ids = [partner.id]
-            for child in partner.child_ids:
-                child_ids += _get_all_child_ids(child)
-            return child_ids
-        
         partner = request.env[self._name].search([('id', '=', partner_id)], limit=1)
-        partner_child_ids = _get_all_child_ids(partner)
-        
+        if not partner.parent_id:
+            partner_child_ids = [x['id'] for x in partner.child_ids] + [partner.id]
         if partner.parent_id:
-            partner_child_ids += _get_all_child_ids(partner.parent_id)
-            
-        return list(set(partner_child_ids))
+            partner_child_ids = [x['id'] for x in partner.parent_id.child_ids]
+        return partner_child_ids
 
     @http.route(prefix + 'partner/<id>/list/site', auth='public', methods=['GET', 'OPTIONS'])
     @controller.Controller.must_authorized()