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


class EvaluationPlanWizard(models.Model):
    _name = 'evaluation.plan.wizard'
    _description = '评估方案弹窗'

    property_ids = fields.Many2many('yuthon.property', string='物业')

    def confirm(self):
        evaluation_area = 0
        name_list = []
        first_id = self.property_ids[0]
        for i in self.property_ids:
            name = i.name
            name_list.append(name)
            evaluation_area += i.evaluation_area
        name = ','.join(name_list)
        self.env['property.evaluation.plan'].create({
            'name': name,
            'assessment_category': 'rent_money',
            'evaluation_area': evaluation_area,
            'owner_id': first_id.title_information_ids[0].owner_id.id if first_id.title_information_ids else None,
            'registered_id': first_id.title_information_ids[0].registered_id.id if first_id.title_information_ids else None,
            'rent_id': first_id.reference_ids[0].rent_id.id if first_id.reference_ids else None,
            'lease_period': first_id.reference_ids[0].lease_period if first_id.reference_ids else None,
            'annual_increase_percentage': first_id.reference_ids[0].annual_increase_percentage if first_id.reference_ids else None,
            'remake': first_id.remark,
            'property_ids': self.property_ids.ids
        })