[FIX] controller function get_partner_child_ids

1 files changed, 4 insertions, 4 deletions

diff --git a/indoteknik_api/controllers/controller.py b/indoteknik_api/controllers/controller.py
index 0b08a58f..c90d3ff1 100644
--- a/indoteknik_api/controllers/controller.py
+++ b/indoteknik_api/controllers/controller.py
@@ -135,10 +135,10 @@ class Controller(http.Controller):
             return False
     
     def get_partner_child_ids(self, partner_id):
-        parent_partner_id = request.env['res.partner'].search([('id', '=', partner_id)], limit=1)
-        parent_partner_id = parent_partner_id.parent_id.id or parent_partner_id.id 
-        partner_childs = request.env['res.partner'].search([('parent_id', '=', int(parent_partner_id))])
-        partner_child_ids = [v['id'] for v in partner_childs] + [partner_id]
+        partner = request.env['res.partner'].search([('id', '=', partner_id)], limit=1)
+        partner_child_ids = [x['id'] for x in partner.child_ids] + [partner.id]
+        if partner.parent_id:
+            partner_child_ids += [x['id'] for x in partner.parent_id.child_ids]
         return partner_child_ids
 
     @http.route('/api/token', auth='public', methods=['GET', 'OPTIONS'])