Update promotion program scheme

1 files changed, 114 insertions, 0 deletions

diff --git a/indoteknik_custom/models/promotion/promotion_program_line.py b/indoteknik_custom/models/promotion/promotion_program_line.py
new file mode 100644
index 00000000..a16f426d
--- /dev/null
+++ b/indoteknik_custom/models/promotion/promotion_program_line.py
@@ -0,0 +1,114 @@
+from odoo import models, fields
+from datetime import datetime
+
+
+class PromotionProgramLine(models.Model):
+    _name = 'promotion.program.line'
+
+    program_id = fields.Many2one('promotion.program', 'Program')
+    name = fields.Char('Name')
+    type = fields.Selection([
+        ("special_price", "Special Price"),
+        ("bundling", "Bundling"),
+        ("discount_loading", "Discount Loading"),
+        ("merchandise", "Merchandise")
+    ], 'Type')
+    image = fields.Binary(string="Image")
+
+    package_limit = fields.Integer('Package limit')
+    package_limit_user = fields.Integer('Package limit / user')
+    package_limit_trx = fields.Integer('Package limit / transaction')
+
+    product_ids = fields.One2many('promotion.product', 'program_line_id', 'Product')
+    free_product_ids = fields.One2many('promotion.free_product', 'program_line_id', 'Free Product')
+
+    price = fields.Float('Price')
+
+    order_promotion_ids = fields.One2many('sale.order.promotion', 'program_line_id', 'Promotions')
+
+    def get_active_promotions(self, product_id):
+        current_time = datetime.now().strftime('%Y-%m-%d %H:%M:%S')
+        return self.search([
+            ('program_id.start_time', '<=', current_time),
+            ('program_id.end_time', '>=', current_time),
+            ('product_ids.product_id', '=', product_id)
+        ])
+    
+    def _res_limit_qty(self):
+        return {
+            'all': self.package_limit,
+            'user': self.package_limit_user,
+            'transaction': self.package_limit_trx,
+        }
+    
+    def _get_remaining_qty(self, user):
+        remaining_qty = {
+            'all': self.package_limit,
+            'user': self.package_limit_user,
+            'transaction': self.package_limit_trx
+        }
+
+        for promotion in self.order_promotion_ids:
+            order = promotion.order_id
+            if order.state != 'cancel':
+                remaining_qty['all'] -= promotion.quantity
+            if user and order.partner_id.id == user['partner_id']:
+                remaining_qty['user'] -= promotion.quantity
+        
+        if remaining_qty['all'] < remaining_qty['user']:
+            remaining_qty['user'] = remaining_qty['all']
+        if remaining_qty['user'] < remaining_qty['transaction']:
+            remaining_qty['transaction'] = remaining_qty['user']
+        
+        return remaining_qty
+    
+    def _res_type(self):
+        return {
+            'value': self.type,
+            'label': dict(self._fields['type'].selection).get(self.type)
+        }
+    
+    def _get_remaining_time(self):
+        calculate_time = self.program_id.end_time - datetime.now()
+        return round(calculate_time.total_seconds())
+    
+    def formats(self, user):
+        return [x.format(user) for x in self]
+    
+    def format(self, user = None, qty = 1):
+        ir_attachment = self.env['ir.attachment']
+        limit_qty = self._res_limit_qty()
+        remaining_qty = self._get_remaining_qty(user)
+        
+        percent_remaining = 0
+        if limit_qty['all'] > 0:
+            percent_remaining = (limit_qty['all'] - remaining_qty['all']) / limit_qty['all'] * 100
+
+        products = self.product_ids.formats(purchase_qty=qty)
+        free_products = self.free_product_ids.formats(purchase_qty=qty)
+
+        merged_products = products + free_products
+        weight = 0
+        if not any(x['package_weight'] == 0 for x in merged_products):
+            weight = sum(x['package_weight'] for x in merged_products)
+
+        response = {
+            'id': self.id,
+            'name': self.name,
+            'image': ir_attachment.api_image(self._name, 'image', self.id),
+            'remaining_time': self._get_remaining_time(),
+            'type': self._res_type(),
+            'limit_qty': limit_qty,
+            'remaining_qty': remaining_qty,
+            'used_percentage': percent_remaining,
+            'price': {
+                'price': self.price,
+                'price_discount': self.price,
+                'discount_percentage': 0
+            },
+            'products': products,
+            'free_products': free_products,
+            'weight': weight
+        }
+
+        return response
\ No newline at end of file