如何根据类别向wordpress页面添加脚本

function add_scripts_based_on_category($query) {
    if ($query->is_category() && $query->is_main_query()) {
        $category_id = $query->get_queried_object_id();
        switch ($category_id) {
            case 1:
                wp_enqueue_script('script-for-category-1', get_template_directory_uri() . '/js/category-1.js', array('jquery'), '1.0', true);
                break;
            case 2:
                wp_enqueue_script('script-for-category-2', get_template_directory_uri() . '/js/category-2.js', array('jquery'), '1.0', true);
                break;
            // 添加更多类别处理
        }
    }
}

add_action('pre_get_posts', 'add_scripts_based_on_category');