75 lines
2.3 KiB
Python
75 lines
2.3 KiB
Python
# This file is part of Tryton. The COPYRIGHT file at the top level of
|
|
# this repository contains the full copyright notices and license terms.
|
|
import binascii
|
|
import os
|
|
try:
|
|
from secrets import token_hex
|
|
except ImportError:
|
|
def token_hex(nbytes=None):
|
|
if nbytes is None:
|
|
nbytes = 32
|
|
return binascii.hexlify(os.urandom(nbytes)).decode('ascii')
|
|
|
|
from sql.conditionals import NullIf
|
|
from sql.operators import Equal
|
|
|
|
from trytond.i18n import gettext
|
|
from trytond.model import Exclude, fields
|
|
from trytond.pool import PoolMeta, Pool
|
|
from trytond.pyson import Bool, Eval
|
|
|
|
from trytond.modules.sale.exceptions import SaleValidationError
|
|
|
|
|
|
class Sale(metaclass=PoolMeta):
|
|
__name__ = 'sale.sale'
|
|
|
|
web_shop = fields.Many2One('web.shop', "Web Shop", ondelete='RESTRICT')
|
|
web_id = fields.Char(
|
|
"Web ID",
|
|
states={
|
|
'required': Bool(Eval('web_shop')),
|
|
'readonly': ~Eval('web_id'),
|
|
},
|
|
depends=['web_shop'])
|
|
|
|
@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(cls, sales):
|
|
pool = Pool()
|
|
WebShop = pool.get('web.shop')
|
|
super().validate(sales)
|
|
|
|
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))
|