How to set WP locale based on Weglot current language

Below is an example of how to set the WordPress locale based on the current Weglot language, using the WordPress locale filter:

add_filter( 'locale', 'weglot_set_locale' );
function weglot_set_locale( $lang ) {
    if ( function_exists( 'weglot_get_current_language' ) ) {
        $current_language = weglot_get_current_language();
        switch ( $current_language ) {
            case 'sv':
                return 'sv_SE';
            case 'en':
                return 'en_US';
            case 'no':
                return 'nb_NO';
            default:
                return $lang;
        }
    }
    return $lang;
}