From 3e6130da150ed3a4aed243d9763e91fe51f4515f Mon Sep 17 00:00:00 2001
From: "qihao.gong@jikimo.com" <qihao.gong@jikimo.com>
Date: Wed, 5 Jul 2023 14:32:22 +0800
Subject: [PATCH] =?UTF-8?q?=E7=BB=B4=E4=BF=9D=E8=AE=A1=E5=88=92=E6=96=B0?=
 =?UTF-8?q?=E5=A2=9E=E7=A1=AE=E8=AE=A4=E7=BB=B4=E4=BF=9D=E8=AE=A1=E5=88=92?=
 =?UTF-8?q?=E5=92=8C=E6=A0=87=E8=AE=B0=E5=B7=B2=E5=AE=8C=E6=88=90=E6=8C=89?=
 =?UTF-8?q?=E9=92=AE=EF=BC=8C=E5=A4=84=E7=90=86=E8=87=AA=E5=8A=A8=E7=94=9F?=
 =?UTF-8?q?=E6=88=90=E7=BB=B4=E4=BF=9D=E8=AE=A1=E5=88=92=E7=9A=84=E6=96=B9?=
 =?UTF-8?q?=E6=B3=95=E6=8A=A5=E9=94=99=E7=9A=84bug?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 sf_maintenance/models/sf_maintenance.py            | 2 +-
 sf_maintenance/models/sf_maintenance_requests.py   | 6 ++++++
 sf_maintenance/views/maintenance_request_views.xml | 6 +++++-
 3 files changed, 12 insertions(+), 2 deletions(-)

diff --git a/sf_maintenance/models/sf_maintenance.py b/sf_maintenance/models/sf_maintenance.py
index 594c3bce..c5a31fb6 100644
--- a/sf_maintenance/models/sf_maintenance.py
+++ b/sf_maintenance/models/sf_maintenance.py
@@ -421,7 +421,7 @@ class SfMaintenanceEquipment(models.Model):
             'owner_user_id': self.owner_user_id.id,
             'user_id': self.technician_user_id.id,
             'maintenance_team_id': self.maintenance_team_id.id,
-            'duration': self.maintenance_duration,
+            'duration': self.overhaul_duration,
             'company_id': self.company_id.id or self.env.company.id,
             'equipment_maintenance_id': self.overhaul_id.id,
             'sf_maintenance_type': '检修'
diff --git a/sf_maintenance/models/sf_maintenance_requests.py b/sf_maintenance/models/sf_maintenance_requests.py
index 36cf2479..1de354fe 100644
--- a/sf_maintenance/models/sf_maintenance_requests.py
+++ b/sf_maintenance/models/sf_maintenance_requests.py
@@ -35,6 +35,12 @@ class SfMaintenanceEquipmentCategory(models.Model):
             if not record.equipment_maintenance_id:
                 raise UserError(_("设备维保标准不能为空，请选择后再保存"))
 
+    def confirm_maintenance(self):
+        self.write({'stage_id': 2})
+
+    def confirm_maintenance_done(self):
+        self.write({'stage_id': 3})
+
 
 
 
diff --git a/sf_maintenance/views/maintenance_request_views.xml b/sf_maintenance/views/maintenance_request_views.xml
index e1e934d9..cd5498c1 100644
--- a/sf_maintenance/views/maintenance_request_views.xml
+++ b/sf_maintenance/views/maintenance_request_views.xml
@@ -7,6 +7,10 @@
         <field name="model">maintenance.request</field>
         <field name="inherit_id" ref="maintenance.hr_equipment_request_view_form"/>
         <field name="arch" type="xml">
+            <xpath expr="//button[@name='archive_equipment_request']" position="before">
+                <button name="confirm_maintenance" string="确认维保计划" type="object" class="btn-primary" attrs="{'invisible': [('stage_id', '!=', 1)]}" />
+                <button name="confirm_maintenance_done" string="标记已完成" type="object" class="btn-primary" attrs="{'invisible': [('stage_id', '!=', 2)]}" />
+            </xpath>
             <xpath expr="//field[@name='maintenance_type']" position="replace">
                 <field name="sf_maintenance_type" widget="radio"/>
                 <field name="equipment_maintenance_id"/>
@@ -17,7 +21,7 @@
                 <notebook>
                     <page string="维保标准" attrs="{'invisible': [('equipment_maintenance_id', '=', False)]}">
                         <group>
-                             <field name="maintenance_standards" widget="ony2many">
+                            <field name="maintenance_standards" widget="ony2many">
                                 <tree create="False" string="维保项目">
                                     <field name="name"/>
                                     <field name="maintenance_standards"/>