Add apply voucher on checkout api

3 files changed, 112 insertions, 46 deletions

diff --git a/indoteknik_api/controllers/api_v1/sale_order.py b/indoteknik_api/controllers/api_v1/sale_order.py
index aa211b75..e036751e 100644
--- a/indoteknik_api/controllers/api_v1/sale_order.py
+++ b/indoteknik_api/controllers/api_v1/sale_order.py
@@ -58,25 +58,28 @@ class SaleOrder(controller.Controller):
         limit = params['value']['limit']
         offset = params['value']['offset']
         if not params['valid']:
-            return self.response(code=400, description=params) 
+            return self.response(code=400, description=params)
 
-        partner_child_ids = self.get_partner_child_ids(params['value']['partner_id'])
+        partner_child_ids = self.get_partner_child_ids(
+            params['value']['partner_id'])
         domain = [('partner_id', 'in', partner_child_ids)]
 
         context = params['value']['context']
         if context == 'quotation':
-            domain += ["|","|",("state","=","draft"),("state","=","sent"),("state","=","cancel")]
+            domain += ["|", "|", ("state", "=", "draft"),
+                       ("state", "=", "sent"), ("state", "=", "cancel")]
         if not context:
-            domain += ["|",("state","=","sale"),("state","=","done")]
+            domain += ["|", ("state", "=", "sale"), ("state", "=", "done")]
 
         if params['value']['name']:
             name = params['value']['name'].replace(' ', '%')
             domain += [
                 '|',
-                ('name', 'ilike', '%'+ name +'%'),
-                ('partner_purchase_order_name', 'ilike', '%'+ name +'%')
+                ('name', 'ilike', '%' + name + '%'),
+                ('partner_purchase_order_name', 'ilike', '%' + name + '%')
             ]
-        sale_orders = request.env['sale.order'].search(domain, offset=offset, limit=limit)
+        sale_orders = request.env['sale.order'].search(
+            domain, offset=offset, limit=limit)
         data = {
             'sale_order_total': request.env['sale.order'].search_count(domain),
             'sale_orders': [request.env['sale.order'].api_v1_single_response(x) for x in sale_orders]
@@ -91,9 +94,10 @@ class SaleOrder(controller.Controller):
             'id': ['number']
         })
         if not params['valid']:
-            return self.response(code=400, description=params) 
-        
-        partner_child_ids = self.get_partner_child_ids(params['value']['partner_id'])
+            return self.response(code=400, description=params)
+
+        partner_child_ids = self.get_partner_child_ids(
+            params['value']['partner_id'])
         domain = [
             ('id', '=', params['value']['id']),
             ('partner_id', 'in', partner_child_ids)
@@ -101,8 +105,9 @@ class SaleOrder(controller.Controller):
         data = {}
         sale_order = request.env['sale.order'].search(domain)
         if sale_order:
-            data = request.env['sale.order'].api_v1_single_response(sale_order, context='with_detail')
-        
+            data = request.env['sale.order'].api_v1_single_response(
+                sale_order, context='with_detail')
+
         return self.response(data)
 
     @http.route(PREFIX_PARTNER + 'sale_order/<id>/checkout', auth='public', method=['POST', 'OPTIONS'], csrf=False)
@@ -113,9 +118,10 @@ class SaleOrder(controller.Controller):
             'id': ['number']
         })
         if not params['valid']:
-            return self.response(code=400, description=params) 
+            return self.response(code=400, description=params)
 
-        partner_child_ids = self.get_partner_child_ids(params['value']['partner_id'])
+        partner_child_ids = self.get_partner_child_ids(
+            params['value']['partner_id'])
         domain = [
             ('id', '=', params['value']['id']),
             ('partner_id', 'in', partner_child_ids)
@@ -124,13 +130,14 @@ class SaleOrder(controller.Controller):
         sale_order = request.env['sale.order'].search(domain)
         if sale_order:
             sale_order.approval_status = 'pengajuan1'
-            data = request.env['sale.order'].api_v1_single_response(sale_order, context='with_detail')
+            data = request.env['sale.order'].api_v1_single_response(
+                sale_order, context='with_detail')
 
         return self.response(data)
 
     @http.route(PREFIX_PARTNER + 'sale_order/<id>/upload_po', auth='public', method=['POST', 'OPTIONS'], csrf=False)
     def partner_upload_po_sale_order(self, **kw):
-        user_token = self.authenticate() 
+        user_token = self.authenticate()
         if not user_token:
             return self.unauthorized_response()
 
@@ -144,7 +151,8 @@ class SaleOrder(controller.Controller):
             return self.unauthorized_response()
         if not params['valid']:
             return self.response(code=400, description=params)
-        partner_child_ids = self.get_partner_child_ids(params['value']['partner_id'])
+        partner_child_ids = self.get_partner_child_ids(
+            params['value']['partner_id'])
         domain = [
             ('id', '=', params['value']['id']),
             ('partner_id', 'in', partner_child_ids)
@@ -166,9 +174,11 @@ class SaleOrder(controller.Controller):
         md5_valid = rest_api.md5_salt_valid(id, 'sale.order', token)
         if not md5_valid:
             return self.response('Unauthorized')
-        
-        sale_order = request.env['sale.order'].sudo().search_read([('id', '=', id)], ['partner_purchase_order_name'])
-        attachment = rest_api.get_single_attachment('sale.order', 'partner_purchase_order_file', id)
+
+        sale_order = request.env['sale.order'].sudo().search_read(
+            [('id', '=', id)], ['partner_purchase_order_name'])
+        attachment = rest_api.get_single_attachment(
+            'sale.order', 'partner_purchase_order_file', id)
         if attachment and len(sale_order) > 0:
             return rest_api.response_attachment({
                 'content': attachment['datas'],
@@ -186,9 +196,11 @@ class SaleOrder(controller.Controller):
         md5_valid = rest_api.md5_salt_valid(id, 'sale.order', token)
         if not md5_valid:
             return self.response('Unauthorized')
-        
-        sale_order = request.env['sale.order'].sudo().search_read([('id', '=', id)], ['name'])
-        pdf, type = request.env['ir.actions.report'].sudo().search([('report_name', '=', 'indoteknik_custom.report_saleorder_website')])._render_qweb_pdf([id])
+
+        sale_order = request.env['sale.order'].sudo(
+        ).search_read([('id', '=', id)], ['name'])
+        pdf, type = request.env['ir.actions.report'].sudo().search(
+            [('report_name', '=', 'indoteknik_custom.report_saleorder_website')])._render_qweb_pdf([id])
         if pdf and len(sale_order) > 0:
             return rest_api.response_attachment({
                 'content': pdf,
@@ -205,9 +217,10 @@ class SaleOrder(controller.Controller):
             'id': ['number']
         })
         if not params['valid']:
-            return self.response(code=400, description=params) 
+            return self.response(code=400, description=params)
 
-        partner_child_ids = self.get_partner_child_ids(params['value']['partner_id'])
+        partner_child_ids = self.get_partner_child_ids(
+            params['value']['partner_id'])
         domain = [
             ('id', '=', params['value']['id']),
             ('partner_id', 'in', partner_child_ids)
@@ -218,15 +231,19 @@ class SaleOrder(controller.Controller):
             sale_order.state = 'cancel'
             data = sale_order.id
         return self.response(data)
-        
+
     @http.route(PREFIX_PARTNER + 'sale_order/checkout', auth='public', method=['POST', 'OPTIONS'], csrf=False)
     @controller.Controller.must_authorized(private=True, private_key='partner_id')
     def create_partner_sale_order(self, **kw):
         config = request.env['ir.config_parameter']
-        product_pricelist_default_discount_id = int(config.get_param('product.pricelist.default_discount_id'))
-        product_pricelist_tier1 = int(config.get_param('product.pricelist.tier1'))
-        product_pricelist_tier2 = int(config.get_param('product.pricelist.tier2'))
-        product_pricelist_tier3 = int(config.get_param('product.pricelist.tier3'))
+        product_pricelist_default_discount_id = int(
+            config.get_param('product.pricelist.default_discount_id'))
+        product_pricelist_tier1 = int(
+            config.get_param('product.pricelist.tier1'))
+        product_pricelist_tier2 = int(
+            config.get_param('product.pricelist.tier2'))
+        product_pricelist_tier3 = int(
+            config.get_param('product.pricelist.tier3'))
 
         params = self.get_request_params(kw, {
             'partner_id': ['number'],
@@ -238,9 +255,10 @@ class SaleOrder(controller.Controller):
             'type': [],
             'delivery_amount': ['number', 'default:0'],
             'carrier_id': [],
-            'delivery_service_type': []
+            'delivery_service_type': [],
+            'voucher': []
         })
-        
+
         if not params['valid']:
             return self.response(code=400, description=params)
 
@@ -275,10 +293,12 @@ class SaleOrder(controller.Controller):
         order_line = json.loads(params['value']['order_line'])
         parameters = []
 
-        partner = request.env['res.partner'].browse(params['value']['partner_id']) 
+        partner = request.env['res.partner'].browse(
+            params['value']['partner_id'])
         partner_pricelist = partner.property_product_pricelist
         for line in order_line:
-            product = request.env['product.product'].search([('id', '=', line['product_id'])], limit=1)
+            product = request.env['product.product'].search(
+                [('id', '=', line['product_id'])], limit=1)
             discount = product._get_website_disc(0)
 
             price_tier = False
@@ -288,15 +308,19 @@ class SaleOrder(controller.Controller):
                 'tier3': product._get_pricelist_tier3,
             }
             partner_pricelist_id = partner_pricelist.id if partner_pricelist else False
-            if partner_pricelist_id == product_pricelist_tier1: price_tier = 'tier1'
-            if partner_pricelist_id == product_pricelist_tier2: price_tier = 'tier2'
-            if partner_pricelist_id == product_pricelist_tier3: price_tier = 'tier3'
+            if partner_pricelist_id == product_pricelist_tier1:
+                price_tier = 'tier1'
+            if partner_pricelist_id == product_pricelist_tier2:
+                price_tier = 'tier2'
+            if partner_pricelist_id == product_pricelist_tier3:
+                price_tier = 'tier3'
 
             if price_tier:
                 price = pricelist[price_tier]()
                 discount_key = 'discount_%s' % price_tier
-                if price[discount_key] > 0: discount = price[discount_key]
-            
+                if price[discount_key] > 0:
+                    discount = price[discount_key]
+
             flashsale = product._get_flashsale_price()
             flashsale_discount = flashsale['flashsale_discount']
             if flashsale_discount > 0 and flashsale_discount > discount:
@@ -310,13 +334,29 @@ class SaleOrder(controller.Controller):
                 'price_unit': product._get_website_price_exclude_tax(),
                 'discount': discount
             })
-            
+
         request.env['sale.order.line'].create(parameters)
+
+        amount_untaxed = sale_order.amount_untaxed
+        voucher = request.env['voucher'].search([
+            ('code', '=', params['value']['voucher'])
+        ])
+        if voucher:
+            sale_order.voucher_id = voucher.id
+            voucher_discount = voucher.calculate_discount(amount_untaxed)
+            total_qty = sum(line.product_uom_qty for line in sale_order.order_line)
+            voucher_discount_item = voucher_discount / total_qty
+            for line in sale_order.order_line:
+                discount_decimal = line.discount / 100
+                voucher_discount_line = voucher_discount_item / (1 - discount_decimal)
+                price = line.price_unit - voucher_discount_line
+                line.price_unit = price
+
         return self.response({
             'id': sale_order.id,
             'name': sale_order.name
         })
-        
+
     @http.route('/api/sale_order/invoiced', auth='public', methods=['GET'])
     def get_sale_order_invoiced_by_partner_id(self, **kw):
         if not self.authenticate():
@@ -328,8 +368,10 @@ class SaleOrder(controller.Controller):
 
         partner_id = int(partner_id)
         # Get company member by partner_id
-        parent_partner_id = request.env['res.partner'].search([('id', '=', partner_id)], limit=1).parent_id.id
-        partner_childs = request.env['res.partner'].search([('parent_id', '=', int(parent_partner_id))])
+        parent_partner_id = request.env['res.partner'].search(
+            [('id', '=', partner_id)], limit=1).parent_id.id
+        partner_childs = request.env['res.partner'].search(
+            [('parent_id', '=', int(parent_partner_id))])
         partner_child_ids = [v['id'] for v in partner_childs] + [partner_id]
 
         # Get sale order by company member and invoiced
@@ -378,9 +420,12 @@ class SaleOrder(controller.Controller):
             ('id', '=', id),
             ('state', 'in', ['sale', 'done'])
         ]
-        parent_partner_id = request.env['res.partner'].search([('id', '=', int(partner_id))], limit=1).parent_id.id
-        partner_childs = request.env['res.partner'].search([('parent_id', '=', int(parent_partner_id))])
-        partner_child_ids = [v['id'] for v in partner_childs] + [int(partner_id)]
+        parent_partner_id = request.env['res.partner'].search(
+            [('id', '=', int(partner_id))], limit=1).parent_id.id
+        partner_childs = request.env['res.partner'].search(
+            [('parent_id', '=', int(parent_partner_id))])
+        partner_child_ids = [v['id']
+                             for v in partner_childs] + [int(partner_id)]
         default_domain.append(('partner_id', 'in', partner_child_ids))
 
         sale_order = self.search_filter('sale.order', kw, default_domain)
diff --git a/indoteknik_custom/models/voucher.py b/indoteknik_custom/models/voucher.py
index 00192cec..b327df6d 100644
--- a/indoteknik_custom/models/voucher.py
+++ b/indoteknik_custom/models/voucher.py
@@ -4,7 +4,9 @@ from datetime import datetime, timedelta
 
 class Voucher(models.Model):
     _name = 'voucher'
+    _rec_name = 'display_name'
 
+    display_name = fields.Char(string='Display Name', compute='_compute_display_name')
     name = fields.Char(string='Name')
     image = fields.Binary(string='Image')
     code = fields.Char(string='Code', help='Kode voucher yang akan berlaku untuk pengguna')
@@ -36,6 +38,10 @@ class Voucher(models.Model):
     max_discount_amount = fields.Integer(string='Max. Discount Amount', help='Max nominal terhadap persentase diskon')
     order_ids = fields.One2many('sale.order', 'voucher_id', string='Order')
 
+    def _compute_display_name(self):
+        for voucher in self:
+            voucher.display_name = f'{voucher.name} ({voucher.code})'
+
     def res_format(self):
         datas = [voucher.format() for voucher in self]
         return datas
@@ -78,6 +84,20 @@ class Voucher(models.Model):
         calculate_time = self.end_time - datetime.now()
         return round(calculate_time.total_seconds())
     
+    def calculate_discount(self, price):
+        if price < self.min_purchase_amount:
+            return 0
+        
+        if self.discount_type == 'fixed_price':
+            return self.discount_amount
+        
+        if self.discount_type == 'percentage':
+            discount = price * self.discount_amount / 100
+            max_disc = self.max_discount_amount
+            return max_disc if discount > max_disc else discount
+        
+        return 0
+    
     def get_active_voucher(self, parameter):
         current_time = datetime.now().strftime('%Y-%m-%d %H:%M:%S')
         parameter += [
diff --git a/indoteknik_custom/views/sale_order.xml b/indoteknik_custom/views/sale_order.xml
index 70e3392f..b55fefff 100755
--- a/indoteknik_custom/views/sale_order.xml
+++ b/indoteknik_custom/views/sale_order.xml
@@ -26,6 +26,7 @@
                     <field name="delivery_amt"/>
                     <field name="fee_third_party"/>
                     <field name="total_percent_margin"/>
+                    <field name="voucher_id" />
                 </field>
                 <field name="analytic_account_id" position="after">
                     <field name="customer_type" attrs="{'required': ['|', ('create_date', '&gt;', '2023-06-28'), ('create_date', '=', False)]}"/>