woocommerce /登录后重定向用户

Question

我要在用户登录my-account页面后将其重定向到wesbite的主页，但仅当他们从帐户页面登录时才保留，并在他们从checkout页面登录时保留初始重定向。例子。

这是我到目前为止编写的代码：

add_filter('woocommerce_login_redirect', 'login_redirect');

function login_redirect($redirect_to) {

  $redirect_to = site_url();

    return $redirect_to;

}

使用此代码，无论用户来自何处，都将对其进行重定向。

如何编写条件以检查用户是否来自登录页面？

感谢您的帮助