限制测量列数

quality_control/models/quality.py

@@ -206,24 +206,22 @@ class QualityCheck(models.Model):
     ], string='发布状态', default='draft')
 
     def add_measure_line(self):
-        self.column_nums = self.column_nums + 1
         """
         新增测量值,如果测量值有5列了，则提示“最多只能有5列测量值”
         """
-        pass
-        # self.ensure_one()
-        # self.env['quality.check.measure.line'].create({
-        #     'check_id': self.id,
-        #     'sequence': len(self.measure_line_ids) + 1,
-        # })
+        if self.column_nums >= 5:
+            raise UserError(_('最多只能有5列测量值'))
+        else:
+            self.column_nums = self.column_nums + 1
 
     def remove_measure_line(self):
-        self.column_nums = self.column_nums - 1
         """
         删除测量值
         """
-        pass
-        # self.ensure_one()
+        if self.column_nums <= 1:
+            raise UserError(_('最少要有1列测量值'))
+        else:
+            self.column_nums = self.column_nums - 1
 
     def do_preview(self):
         """