Refactor due extension

1 files changed, 8 insertions, 0 deletions

diff --git a/indoteknik_custom/models/res_partner.py b/indoteknik_custom/models/res_partner.py
index 5dec8da8..1695639d 100644
--- a/indoteknik_custom/models/res_partner.py
+++ b/indoteknik_custom/models/res_partner.py
@@ -20,6 +20,14 @@ class ResPartner(models.Model):
     sppkp = fields.Char(string="SPPKP")
     counter = fields.Integer(string="Counter", default=0)
 
+    def get_child_ids(self):
+        partner = self.env['res.partner'].search([('id', '=', self.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]
+            partner_child_ids += [partner.parent_id.id]
+        return partner_child_ids
+
     def unlink(self):
         if self._name == 'res.partner':
             raise UserError('Maaf anda tidak bisa delete contact')