Odoo18-Base/addons/website_sale_loyalty/controllers/main.py
2025-01-06 10:57:38 +07:00

150 lines
6.3 KiB
Python

# Part of Odoo. See LICENSE file for full copyright and licensing details.
from werkzeug.urls import url_encode, url_parse
from odoo import _
from odoo.exceptions import UserError
from odoo.http import request, route
from odoo.addons.website_sale.controllers import main
class WebsiteSale(main.WebsiteSale):
@route()
def pricelist(self, promo, **post):
order = request.website.sale_get_order()
if not order:
return request.redirect('/shop')
coupon_status = order._try_apply_code(promo)
if coupon_status.get('not_found'):
return super().pricelist(promo, **post)
elif coupon_status.get('error'):
request.session['error_promo_code'] = coupon_status['error']
elif 'error' not in coupon_status:
reward_successfully_applied = True
if len(coupon_status) == 1:
coupon, rewards = next(iter(coupon_status.items()))
if request.env.context.get('product_id') or (len(rewards) == 1 and not rewards.multi_product):
reward_successfully_applied = self._apply_reward(order, rewards, coupon)
if reward_successfully_applied:
request.session['successful_code'] = promo
return request.redirect(post.get('r', '/shop/cart'))
@route()
def shop_payment(self, **post):
order = request.website.sale_get_order()
if order:
order._update_programs_and_rewards()
order._auto_apply_rewards()
return super().shop_payment(**post)
@route()
def cart(self, **post):
order = request.website.sale_get_order()
if order and order.state != 'draft':
request.session['sale_order_id'] = None
order = request.website.sale_get_order()
if order:
order._update_programs_and_rewards()
order._auto_apply_rewards()
return super().cart(**post)
@route(['/coupon/<string:code>'], type='http', auth='public', website=True, sitemap=False)
def activate_coupon(self, code, r='/shop', **kw):
url_parts = url_parse(r)
url_query = url_parts.decode_query()
url_query.pop('coupon_error', False) # trust only Odoo error message
url_query.pop('coupon_error_type', False)
code = code.strip()
request.session['pending_coupon_code'] = code
order = request.website.sale_get_order()
if order:
result = order._try_pending_coupon()
if isinstance(result, dict) and 'error' in result:
url_query['coupon_error'] = result['error']
else:
url_query['notify_coupon'] = code
else:
url_query['coupon_error'] = _("The coupon will be automatically applied when you add something in your cart.")
url_query['coupon_error_type'] = 'warning'
redirect = url_parts.replace(query=url_encode(url_query))
return request.redirect(redirect.to_url())
@route('/shop/claimreward', type='http', auth='public', website=True, sitemap=False)
def claim_reward(self, reward_id, code=None, **post):
order_sudo = request.website.sale_get_order()
redirect = post.get('r', '/shop/cart')
if not order_sudo:
return request.redirect(redirect)
try:
reward_id = int(reward_id)
except ValueError:
reward_id = None
reward_sudo = request.env['loyalty.reward'].sudo().browse(reward_id).exists()
if not reward_sudo:
return request.redirect(redirect)
if reward_sudo.multi_product and 'product_id' in post:
request.update_context(product_id=int(post['product_id']))
else:
request.redirect(redirect)
program_sudo = reward_sudo.program_id
claimable_rewards = order_sudo._get_claimable_and_showable_rewards()
coupon = request.env['loyalty.card']
for coupon_, rewards in claimable_rewards.items():
if reward_sudo in rewards:
coupon = coupon_
if code == coupon.code and (
(program_sudo.trigger == 'with_code' and program_sudo.program_type != 'promo_code')
or (program_sudo.trigger == 'auto'
and program_sudo.applies_on == 'future'
and program_sudo.program_type not in ('ewallet', 'loyalty'))
):
return self.pricelist(code)
if coupon:
self._apply_reward(order_sudo, reward_sudo, coupon)
return request.redirect(redirect)
def _apply_reward(self, order, reward, coupon):
"""Try to apply the given program reward
:returns: whether the reward was successfully applied
:rtype: bool
"""
product_id = request.env.context.get('product_id')
product = product_id and request.env['product.product'].sudo().browse(product_id)
try:
reward_status = order._apply_program_reward(reward, coupon, product=product)
except UserError as e:
request.session['error_promo_code'] = str(e)
return False
if 'error' in reward_status:
request.session['error_promo_code'] = reward_status['error']
return False
order._update_programs_and_rewards()
if order.carrier_id.free_over and not reward.program_id.is_payment_program:
# update shiping cost if it's `free_over` and reward isn't eWallet or gift card
# will call `_update_programs_and_rewards` again, updating applied eWallet/gift cards
res = order.carrier_id.rate_shipment(order)
if res.get('success'):
order.set_delivery_line(order.carrier_id, res['price'])
else:
order._remove_delivery_line()
return True
@route()
def cart_update_json(self, *args, set_qty=None, **kwargs):
# When a reward line is deleted we remove it from the auto claimable rewards
if set_qty == 0:
request.update_context(website_sale_loyalty_delete=True)
# We need to update the website since `get_sale_order` is called on the website
# and does not follow the request's context
request.website = request.website.with_context(website_sale_loyalty_delete=True)
return super().cart_update_json(*args, set_qty=set_qty, **kwargs)