lims_industry: control duplication of Plants, Equipment and Components

lims_industry/industry.py

@@ -2,7 +2,7 @@
 # The COPYRIGHT file at the top level of this repository contains
 # the full copyright notices and license terms.
 
-from trytond.model import ModelSQL, ModelView, fields
+from trytond.model import ModelSQL, ModelView, fields, Unique
 from trytond.pool import Pool
 from trytond.pyson import Eval
 from trytond.transaction import Transaction
@@ -45,6 +45,11 @@ class Plant(ModelSQL, ModelView):
         super(Plant, cls).__setup__()
         cls._order.insert(0, ('party', 'ASC'))
         cls._order.insert(1, ('name', 'ASC'))
+        t = cls.__table__()
+        cls._sql_constraints = [
+            ('name_unique', Unique(t, t.party, t.name),
+                'lims_industry.msg_plant_name_unique'),
+            ]
 
     @staticmethod
     def default_country():
@@ -55,6 +60,19 @@ class Plant(ModelSQL, ModelView):
             if address and address.country:
                 return address.country.id
 
+    @classmethod
+    def copy(cls, plants, default=None):
+        if default is None:
+            default = {}
+        default = default.copy()
+
+        new_plants = []
+        for plant in plants:
+            default['name'] = plant.name + ' (copy)'
+            new_plant, = super(Plant, cls).copy([plant], default)
+            new_plants.append(new_plant)
+        return new_plants
+
 
 class EquipmentType(ModelSQL, ModelView):
     'Equipment Type'
@@ -156,6 +174,11 @@ class Equipment(ModelSQL, ModelView):
         super(Equipment, cls).__setup__()
         cls._order.insert(0, ('template', 'ASC'))
         cls._order.insert(1, ('name', 'ASC'))
+        t = cls.__table__()
+        cls._sql_constraints = [
+            ('name_unique', Unique(t, t.plant, t.name),
+                'lims_industry.msg_equipment_name_unique'),
+            ]
 
     def get_party(self, name):
         if self.plant:
@@ -193,6 +216,19 @@ class Equipment(ModelSQL, ModelView):
         self.power = power
         self.components = components
 
+    @classmethod
+    def copy(cls, equipments, default=None):
+        if default is None:
+            default = {}
+        default = default.copy()
+
+        new_equipments = []
+        for equipment in equipments:
+            default['name'] = equipment.name + ' (copy)'
+            new_equipment, = super(Equipment, cls).copy([equipment], default)
+            new_equipments.append(new_equipment)
+        return new_equipments
+
 
 class Component(ModelSQL, ModelView):
     'Component'
@@ -220,6 +256,11 @@ class Component(ModelSQL, ModelView):
         super(Component, cls).__setup__()
         cls._order.insert(0, ('equipment', 'ASC'))
         cls._order.insert(1, ('type', 'ASC'))
+        t = cls.__table__()
+        cls._sql_constraints = [
+            ('type_unique', Unique(t, t.equipment, t.type),
+                'lims_industry.msg_component_type_unique'),
+            ]
 
     def get_rec_name(self, name):
         res = self.type.rec_name

lims_industry/locale/es.po

@@ -466,6 +466,18 @@ msgctxt "model:ir.action,name:act_plant"
 msgid "Plants"
 msgstr "Plantas"
 
+msgctxt "model:ir.message,text:msg_component_type_unique"
+msgid "A component of the same type already exists for the equipment"
+msgstr "Ya existe un componente del mismo tipo para el equipo"
+
+msgctxt "model:ir.message,text:msg_equipment_name_unique"
+msgid "There is already an equipment with the same name for the plant"
+msgstr "Ya existe un equipo con el mismo nombre para la planta"
+
+msgctxt "model:ir.message,text:msg_plant_name_unique"
+msgid "There is already a plant with the same name for the party"
+msgstr "Ya existe una planta con el mismo nombre para el tercero"
+
 msgctxt "model:ir.ui.menu,name:menu_brand"
 msgid "Brands"
 msgstr "Marcas"

lims_industry/message.xml

@@ -0,0 +1,14 @@
+<?xml version="1.0"?>
+<tryton>
+    <data group="1">
+        <record model="ir.message" id="msg_plant_name_unique">
+            <field name="text">There is already a plant with the same name for the party</field>
+        </record>
+        <record model="ir.message" id="msg_equipment_name_unique">
+            <field name="text">There is already an equipment with the same name for the plant</field>
+        </record>
+        <record model="ir.message" id="msg_component_type_unique">
+            <field name="text">A component of the same type already exists for the equipment</field>
+        </record>
+    </data>
+</tryton>

lims_industry/tryton.cfg

@@ -5,3 +5,4 @@ depends:
 xml:
     industry.xml
     party.xml
+    message.xml