php - Wordpress 和 woocommerce 插件免税

Question

我是 wordpress 和编写插件的新手。我非常接近完成我的插件，但我被困在最后一步。我有两个功能。一个用于设置免税，另一个用于重新计算总额。如果第一个函数将用户设置为免税，我需要第二个函数来取消税收。关于如何让这个工作的任何想法？

/* update the order total set exempt */
add_action( 'woocommerce_checkout_update_order_review', 'taxexempt_checkout_update_order_review' );
function taxexempt_checkout_update_order_review() {
    global $woocommerce;

    $woocommerce->customer->set_is_vat_exempt(FALSE);

    if (!empty($_POST['tax_exempt_number']))  { $woocommerce->customer->set_is_vat_exempt(true); }    

}

add_action( 'woocommerce_calculate_totals', 'remove_tax_for_exempt' );
function remove_tax_for_exempt( $cart ) {
    global $woocommerce;

    if ($woocommerce->customer->is_vat_exempt()){ 
        $cart->remove_taxes();
        $woocommerce->add_error(__('removing taxes'));
    }
    return $cart;
}