from collections import defaultdict from trytond.pool import Pool from trytond.model import ( Workflow, ModelSQL, ModelView, Unique, fields) from trytond.pyson import Eval, If from trytond.transaction import Transaction from trytond.exceptions import UserError class OpticalEquipment(Workflow, ModelSQL, ModelView): 'Optical Equipment' __name__ = 'optical_equipment.equipment' code = fields.Char( "Code", select=True,states={'readonly': True }) state = fields.Selection([ ('draft', "Draft"), ('registred', "Registred"), ('contrated', "Contrated") ], "State", required=True, readonly=True, sort=False) company = fields.Many2One('company.company', "Company", readonly=True) location = fields.Many2One('stock.location', "Location") propietary = fields.Many2One('party.party', "Propietary") propietary_address = fields.Many2One('party.address', "Propietary Address", required=True, domain=[('party', '=', Eval('propietary'))] ) product = fields.Many2One('product.product', "Product", domain=[('equipment', '=', True)], depends=['equipment'] ) refurbish = fields.Boolean("Refurbish", readonly=True) equipment_type = fields.Char('type', readonly=True) risk = fields.Char('Type risk') use = fields.Char('Use') biomedical_class = fields.Char('Biomedical Class') main_tecnology = fields.Char('Main tecnology') calibration = fields.Boolean("Apply calibration", readonly=True) mark_category = fields.Many2One('product.category', 'Mark', required=True, domain=[('parent', '=', None), ('accounting', '=', False)], ) model_category = fields.Many2One('product.category', "Model", required=True, domain=[('parent', '=', Eval('mark_category')), ('accounting', '=', False)],) reference = fields.Char("Reference", size=None) origin_country = fields.Many2One('country.country',"Origin Country") software_version = fields.Char("Software version", size=None) useful_life = fields.Integer("Useful life") warranty = fields.Integer("Warranty") serial = fields.Char("Serial", size=None) health_register = fields.Char("Health Register", size=None) subscription_history = fields.Many2Many('sale.subscription-optical_equipment.equipment', 'equipment','subscription', "Subscriptions", states={'readonly': True}) current_subscription = fields.Many2One('sale.subscription') @classmethod def __setup__(cls): super(OpticalEquipment, cls).__setup__() t = cls.__table__() cls._sql_constraints = [ ('serial_unique', Unique(t, t.serial), 'optical_equipment.msg_serial_unique') ] cls._transitions = ({ ('draft', 'registred'), ('registred', 'contrated'), }) cls._buttons.update({ 'draft': { 'invisible': Eval('state') == 'draft'}, 'registred': { 'invisible': Eval('state').in_(['registred', 'contrated'])}} ) @classmethod def set_code(cls, equipments): pool = Pool() Config = pool.get('optical_equipment.configuration') config = Config(1) for equipment in equipments: if not equipment.code: try: equipment.code = config.equipment_sequence.get() equipment.state = 'registred' cls.save(equipments) #Revisar except UserError: raise UserError(str('Validation Error')) @classmethod def default_state(cls): return 'draft' @staticmethod def default_company(): return Transaction().context.get('company') @fields.depends('product', 'equipment_type','use', 'biomedical_class', 'calibration', 'mark_category', 'model_category') def on_change_product(self): if self.product: self.equipment_type = self.product.equipment_type self.use = self.product.use self.biomedical_class = self.product.biomedical_class self.calibration = self.product.calibration self.mark_category = self.product.mark_category self.model_category = self.product.model_category else: self.equipment_type = None self.use = None self.biomedical_class = None self.calibration = None self.mark_category = None self.model_category = None @classmethod @ModelView.button @Workflow.transition('draft') def draft(cls, equipments): pass @classmethod @ModelView.button @Workflow.transition('registred') def registred(cls, equipments): cls.set_code(equipments)