first commit

modules/sale_rental/account.py

@@ -0,0 +1,88 @@
+# 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 Pool, PoolMeta
+from trytond.report import Report
+from trytond.tools import cached_property
+
+
+class Invoice(metaclass=PoolMeta):
+    __name__ = 'account.invoice'
+
+    sale_rentals = fields.Function(fields.Many2Many(
+            'sale.rental', None, None, "Rentals"),
+        'get_sale_rentals', searcher='search_sale_rentals')
+
+    def get_sale_rentals(self, name):
+        pool = Pool()
+        RentalLine = pool.get('sale.rental.line')
+        rentals = set()
+        for line in self.lines:
+            if isinstance(line.origin, RentalLine):
+                rentals.add(line.origin.rental.id)
+        return list(rentals)
+
+    @classmethod
+    def search_sale_rentals(cls, name, clause):
+        return [
+            ('lines.origin.rental' + clause[0][len(name):],
+                *clause[1:3], 'sale.rental.line', *clause[3:]),
+            ]
+
+
+class InvoiceLine(metaclass=PoolMeta):
+    __name__ = 'account.invoice.line'
+
+    @fields.depends('product', 'origin')
+    def on_change_product(self):
+        pool = Pool()
+        RentalLine = pool.get('sale.rental.line')
+        super().on_change_product()
+        if self.product and isinstance(self.origin, RentalLine):
+            # Prevent warning from Invoice._check_taxes
+            self.taxes = self.origin.taxes
+
+    @cached_property
+    def product_name(self):
+        pool = Pool()
+        RentalLine = pool.get('sale.rental.line')
+        Lang = pool.get('ir.lang')
+        lang = Lang.get()
+        name = super().product_name
+        if isinstance(self.origin, RentalLine):
+            converter = RentalLine.duration.converter
+            duration = Report.format_timedelta(
+                self.origin.duration_invoice, converter=converter, lang=lang)
+            start = lang.strftime(self.origin.start_invoice)
+            end = lang.strftime(self.origin.end_invoice)
+            name = f'[{start} -- {end}] {duration} × {name}'
+        return name
+
+    @property
+    def origin_name(self):
+        pool = Pool()
+        RentalLine = pool.get('sale.rental.line')
+        name = super().origin_name
+        if isinstance(self.origin, RentalLine):
+            name = self.origin.rental.rec_name
+        return name
+
+    @classmethod
+    def _get_origin(cls):
+        models = super()._get_origin()
+        models.append('sale.rental.line')
+        return models
+
+
+class InvoiceLine_Asset(metaclass=PoolMeta):
+    __name__ = 'account.invoice.line'
+
+    @fields.depends('origin')
+    def on_change_with_is_assets_depreciable(self, name=None):
+        pool = Pool()
+        RentalLine = pool.get('sale.rental.line')
+        depreciable = super().on_change_with_is_assets_depreciable(name=name)
+        if depreciable and isinstance(self.origin, RentalLine):
+            depreciable = False
+        return depreciable