first commit

modules/sale_gift_card/product.py

@@ -0,0 +1,74 @@
+# 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 trytond.i18n import gettext
+from trytond.model import fields
+from trytond.pool import Pool, PoolMeta
+from trytond.pyson import Eval, If
+from trytond.transaction import Transaction
+
+from .exceptions import GiftCardValidationError
+
+
+class Template(metaclass=PoolMeta):
+    __name__ = 'product.template'
+
+    gift_card = fields.Boolean(
+        "Gift Card",
+        domain=[
+            If(~Eval('type').in_(['service', 'goods']),
+                ('gift_card', '=', False),
+                ()),
+            ],
+        states={
+            'invisible': ~Eval('type').in_(['service', 'goods']),
+            })
+
+    @fields.depends('gift_card')
+    def on_change_gift_card(self):
+        pool = Pool()
+        ModelData = pool.get('ir.model.data')
+        if self.gift_card:
+            self.default_uom = ModelData.get_id('product', 'uom_unit')
+
+    @classmethod
+    def validate_fields(cls, templates, field_names):
+        pool = Pool()
+        ModelData = pool.get('ir.model.data')
+        UoM = pool.get('product.uom')
+        super().validate_fields(templates, field_names)
+        if field_names & {'gift_card', 'default_uom'}:
+            unit = UoM(ModelData.get_id('product', 'uom_unit'))
+            for template in templates:
+                if template.gift_card and template.default_uom != unit:
+                    raise GiftCardValidationError(
+                        gettext('sale_gift_card.msg_gift_card_invalid_uom',
+                            template=template.rec_name,
+                            unit=unit.rec_name))
+
+    @property
+    def account_expense_used(self):
+        pool = Pool()
+        Config = pool.get('account.configuration')
+        if self.gift_card:
+            transaction = Transaction()
+            with transaction.reset_context(), \
+                    transaction.set_context(self._context):
+                config = Config(1)
+            return config.gift_card_account_expense
+        return super().account_expense_used
+
+    @property
+    def account_revenue_used(self):
+        pool = Pool()
+        Config = pool.get('account.configuration')
+        if self.gift_card:
+            transaction = Transaction()
+            with transaction.reset_context(), \
+                    transaction.set_context(self._context):
+                config = Config(1)
+            return config.gift_card_account_revenue
+        return super().account_revenue_used
+
+
+class Product(metaclass=PoolMeta):
+    __name__ = 'product.product'