# This file is part of Tryton. The COPYRIGHT file at the top level of # this repository contains the full copyright notices and license terms. from secrets import token_hex from sql.conditionals import NullIf from sql.operators import Equal from trytond.i18n import gettext from trytond.model import Exclude, fields from trytond.modules.sale.exceptions import SaleValidationError from trytond.pool import Pool, PoolMeta from trytond.pyson import Bool, Eval class Sale(metaclass=PoolMeta): __name__ = 'sale.sale' web_shop = fields.Many2One('web.shop', "Web Shop", ondelete='RESTRICT') web_id = fields.Char( "Web ID", strip=False, states={ 'required': Bool(Eval('web_shop')), 'readonly': ~Eval('web_id'), }) @classmethod def __setup__(cls): super().__setup__() t = cls.__table__() cls._sql_constraints += [ ('web_id', Exclude(t, (NullIf(t.web_id, ''), Equal)), 'web_shop.msg_sale_web_id_unique') ] @fields.depends('web_shop', 'web_id') def on_change_web_shop(self, nbytes=None): if self.web_shop and not self.web_id: self.web_id = token_hex(nbytes) @classmethod def copy(cls, sales, default=None): if default is None: default = {} else: default = default.copy() default.setdefault('web_shop', None) default.setdefault('web_id', None) return super().copy(sales, default=default) @classmethod def validate_fields(cls, sales, field_names): pool = Pool() WebShop = pool.get('web.shop') super().validate_fields(sales, field_names) if field_names & {'state', 'party'}: web_shops = WebShop.search([]) guests = {s.guest_party for s in web_shops} for sale in sales: if (sale.state not in {'draft', 'cancelled'} and sale.party in guests): raise SaleValidationError( gettext('web_shop.msg_sale_invalid_party', sale=sale.rec_name, party=sale.party.rec_name))