first commit

modules/account_tax_rule_country/sale.py

@@ -0,0 +1,41 @@
+# 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.model import fields
+from trytond.pool import PoolMeta
+from trytond.pyson import Eval
+
+
+class Sale(metaclass=PoolMeta):
+    __name__ = 'sale.sale'
+
+    @classmethod
+    def __setup__(cls):
+        super().__setup__()
+        for field in (cls.shipment_party, cls.shipment_address, cls.warehouse):
+            field.states['readonly'] |= (
+                Eval('lines', [0]) & Eval('shipment_address'))
+
+
+class Line(metaclass=PoolMeta):
+    __name__ = 'sale.line'
+
+    @fields.depends('sale', 'warehouse', '_parent_sale.shipment_address')
+    def _get_tax_rule_pattern(self):
+        pattern = super()._get_tax_rule_pattern()
+
+        from_country = from_subdivision = to_country = to_subdivision = None
+        if self.warehouse and self.warehouse.address:
+            from_country = self.warehouse.address.country
+            from_subdivision = self.warehouse.address.subdivision
+        if self.sale and self.sale.shipment_address:
+            to_country = self.sale.shipment_address.country
+            to_subdivision = self.sale.shipment_address.subdivision
+
+        pattern['from_country'] = from_country.id if from_country else None
+        pattern['from_subdivision'] = (
+            from_subdivision.id if from_subdivision else None)
+        pattern['to_country'] = to_country.id if to_country else None
+        pattern['to_subdivision'] = (
+            to_subdivision.id if to_subdivision else None)
+        return pattern