from odoo import http
from odoo.addons.website_sale.controllers import main
from odoo.addons.portal.controllers import portal
from odoo.http import request


class WebsiteSale(main.WebsiteSale):

    @http.route("/shop/ups_check_service_type", type='json', auth="public", website=True, sitemap=False)
    def ups_check_service_type_is_available(self, **post):
        return request.env['sale.order'].sudo().check_ups_service_type(post)

    @http.route("/shop/property_ups_carrier_account/set", type='http', auth="public", website=True, sitemap=False)
    def set_property_ups_carrier_account(self, **post):
        order = request.website.sale_get_order()

        # set ups bill my account data in sale order
        if order.carrier_id.ups_bill_my_account and post.get('property_ups_carrier_account'):
            # Update Quotation property_ups_carrier_account
            order.write({
                'partner_ups_carrier_account': post['property_ups_carrier_account'],
            })
        return request.redirect("/shop/payment")

    @http.route("/shop/property_ups_carrier_account/unset", type='http', auth="public", website=True, sitemap=False)
    def reset_property_ups_carrier_account(self, **post):
        order = request.website.sale_get_order()
        # remove ups bill my account data in sale order
        if order.partner_ups_carrier_account:
            order.write({
                'partner_ups_carrier_account': False,
            })
        return request.redirect("/shop/payment")

class CustomerPortal(portal.CustomerPortal):
    OPTIONAL_BILLING_FIELDS = [*portal.CustomerPortal.OPTIONAL_BILLING_FIELDS, 'property_ups_carrier_account']