Update promotion program scheme

1 files changed, 29 insertions, 0 deletions

diff --git a/indoteknik_custom/models/promotion/promotion_product.py b/indoteknik_custom/models/promotion/promotion_product.py
new file mode 100644
index 00000000..2fad0f0d
--- /dev/null
+++ b/indoteknik_custom/models/promotion/promotion_product.py
@@ -0,0 +1,29 @@
+from odoo import fields, models
+
+
+class PromotionProduct(models.Model):
+    _name = "promotion.product"
+    _rec_name = "product_id"
+
+    product_id = fields.Many2one(comodel_name="product.product", string="Product variant")
+    qty = fields.Integer(string="Quantity")
+    program_line_id = fields.Many2one(comodel_name="promotion.program.line", string="Program line")
+
+    def formats(self, purchase_qty=1):
+        ir_attachment = self.env['ir.attachment']
+        
+        result = []
+        for rec in self:
+            weight = rec.product_id.weight or 0
+            result.append({
+                'id': rec.product_id.id,
+                'image': ir_attachment.api_image('product.template', 'image_256', rec.product_id.product_tmpl_id.id),
+                'display_name': rec.product_id.display_name,
+                'name': rec.product_id.name,
+                'code': rec.product_id.code,
+                'price': rec.product_id.calculate_website_price(),
+                'qty': rec.qty * purchase_qty,
+                'weight': weight,
+                'package_weight': weight * rec.qty
+            })
+        return result
\ No newline at end of file