from odoo import models, fields, api

class ResPartnerSite(models.Model):
    _name = 'res.partner.site'
    
    partner_ids = fields.One2many('res.partner', 'site_id', string='Partners')
    partner_id = fields.Many2one('res.partner', string='Partner')
    name = fields.Char(string='Name')
    street = fields.Char(string='Street')
    street2 = fields.Char(string='Street 2')
    city = fields.Char(string='City')
    state_id = fields.Many2one('res.country.state', string='State')
    country_id = fields.Many2one('res.country', string='Country')
    zip = fields.Char(string='Zip')
    
    @api.model
    def create(self, vals):
        rec = super(ResPartnerSite, self).create(vals)
        rec.partner_id = rec.partner_id.get_main_parent()
        return rec