from odoo import models, fields
from odoo.exceptions import UserError, ValidationError

class GroupPartner(models.Model):
    _name = 'group.partner'

    name = fields.Char(string='Name')

class ResPartner(models.Model):
    _inherit = 'res.partner'

    reference_number = fields.Char(string="Reference Number")
    company_type_id = fields.Many2one('res.partner.company_type', string='Company Type')
    custom_pricelist_id = fields.Many2one('product.pricelist', string='Price Matrix')
    group_partner_id = fields.Many2one('group.partner', string='Group Partner')
    customer_type = fields.Selection([
        ('pkp', 'PKP'),
        ('nonpkp', 'Non PKP')
    ])
    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')