From db126e496e719d6cc608b6f9fd6e647cc695458f Mon Sep 17 00:00:00 2001 From: sinergia Date: Thu, 7 Jul 2022 11:04:56 -0500 Subject: [PATCH] charge company for default in equipment --- equipment.py | 14 +++++++++----- 1 file changed, 9 insertions(+), 5 deletions(-) diff --git a/equipment.py b/equipment.py index 2f81ff0..75e49e8 100644 --- a/equipment.py +++ b/equipment.py @@ -3,6 +3,7 @@ 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): @@ -20,10 +21,11 @@ class OpticalEquipment(Workflow, ModelSQL, ModelView): ], "State", required=True, readonly=True, sort=False) - company = fields.Many2One('company.company', "Company") + 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) + propietary_address = fields.Many2One('party.address', + "Propietary Address", required=True) product = fields.Many2One('product.product', "Product") refurbish = fields.Boolean("Refurbish", readonly=True) equipment_type = fields.Char('type', readonly=True) @@ -89,19 +91,21 @@ class OpticalEquipment(Workflow, ModelSQL, ModelView): @classmethod def default_state(cls): return 'draft' + + @staticmethod + def default_company(): + return Transaction().context.get('company') @classmethod @ModelView.button @Workflow.transition('draft') def draft(cls, equipments): pass - #raise userError(str('draft')) @classmethod @ModelView.button @Workflow.transition('registred') def registred(cls, equipments): - #raise UserError(str(equipments)) cls.set_code(equipments) - #state = 'registred' +