from odoo import fields, models


class MaintenanceRequestWizard(models.TransientModel):
    _name = 'maintenance.request.wizard'
    _description = '维保二次确认弹窗'

    name = fields.Char('')

    def submit(self):
        context = self.env.context
        equipment_id = self.env['maintenance.equipment'].sudo().search([('id', '=', context['equipment_id'])])
        request_ids = self.env['maintenance.request'].search([('stage_id.done', '=', False),
                                                              ('equipment_id', '=', equipment_id.id),
                                                              ('maintenance_type', '=', 'preventive'),
                                                              ('sf_maintenance_type', '=', context['type'])])
        request_ids.write({'active': False})
        return equipment_id.create_maintenance_request(context['type'])

    def cancel(self):
        context = self.env.context
        equipment_id = self.env['maintenance.equipment'].sudo().search([('id', '=', context['equipment_id'])])
        if context['type'] == '保养':
            equipment_id.initial_action_date = equipment_id.initial_action_date_old
        elif context['type'] == '检修':
            equipment_id.initial_overhaul_date = equipment_id.initial_overhaul_date_old