Exclude restricted products from WooCommerce queries

//Alters the output of the query that displays the Category Archive (including Shop) pages, filtering the restricted products from the output
add_action( 'woocommerce_product_query', 'pmsc_exclude_restricted_products_from_woocoommerce_category_queries' );
function pmsc_exclude_restricted_products_from_woocoommerce_category_queries( $query ) {
	//if you add $query->is_tax() to the if, it will affect only the category pages; else it affects all queries including the shop page
    if( !is_admin() && function_exists('pms_is_post_restricted') ) {
		$args = $query->query_vars;
		$args['suppress_filters'] = true;
		$args['posts_per_page'] = -1;
 
		$products = wc_get_products($args);
 
		$product_ids = array();
 
		foreach ($products as $product) {
			$product_ids[] = $product->get_id();
		}
 
		$restricted_ids = array_filter($product_ids, 'pms_is_post_restricted');
 
		$query->set( 'post__not_in', $restricted_ids );
    }
}

//Alters the output of the [products] shortcode from WooCommerce, filtering restricted products from the output
add_action( 'woocommerce_shortcode_products_query', 'pmsc_exclude_restricted_products_from_woocoommerce_products_shortcode_queries' );
function pmsc_exclude_restricted_products_from_woocoommerce_products_shortcode_queries( $query_args ) {
    if( !is_admin() && function_exists('pms_is_post_restricted') ) {
		$posts_per_page = $query_args['posts_per_page'];
 
		$query_args['suppress_filters'] = true;
		$query_args['posts_per_page'] = '-1';
 
		$products = wc_get_products($query_args);
 
		$query_args['posts_per_page'] = $posts_per_page;
 
		$product_ids = array();
 
		foreach ($products as $product) {
			$product_ids[] = $product->get_id();
		}
 
		$restricted_ids = array_filter($product_ids, 'pms_is_post_restricted');
 
		$query_args['post__not_in'] = $restricted_ids;
    }
 
	return $query_args;
}