如何使 Wordpress save_post_($post-type) 钩子适用于多种自定义帖子类型

Question

我有一个与 save_post_($post-type) 挂钩和 ACF 库字段相关的问题。

我从某个地方得到了这个代码 sn-p。让我解释一下：我创建了一个 ACF 画廊字段（acf-gallery 是字段名称）显示在自定义帖子类型（cpt1 是 slug）然后使用这个 sn-p 将此画廊的第一张图片设置为特色图片当像 Woocommerce 一样保存时。

但是如果我希望它与另一种自定义帖子类型一起使用（假设 slug 是 cpt2）怎么办？我可以用array( 'cpt1', 'cpt2' ) 代替cpt1 吗？有没有办法包含多个自定义帖子类型？

/* Set the first image generated by ACF gallery field as featured image */

add_action( 'save_post_cpt1', 'set_featured_image_from_gallery' );

function set_featured_image_from_gallery() {

    global $post;
    $post_id = $post->ID;

    $images = get_field('acf_gallery', $post_id, false);
    $image_id = $images[0];

    if ( $image_id ) {
        set_post_thumbnail( $post_id, $image_id );
    }
}

我根据下面的 cmets 使用 save-post 挂钩编辑了这个 sn-p。但我不知道它是否有效。有人可以帮忙吗？

/* Set the first image generated by ACF gallery field as featured image */

add_action( 'save_post', 'set_featured_image_from_gallery' );

function set_featured_image_from_gallery($post_id) {

    if (get_post_type($post_id) != array( 'cpt1', 'cpt2')) {
        return;
    }

    $has_thumbnail = get_the_post_thumbnail($post_id);

      if ( !$has_thumbnail ) {

        $images = get_field('acf_gallery', $post_id, false);
        $image_id = $images[0];

        if ( $image_id ) {
          set_post_thumbnail( $post_id, $image_id );
        }
      }
}

Answer 1

我通常使用save_post钩子、$post_id变量、get_post_type和in_array函数。

function set_featured_image_from_gallery($post_id)
{
    $included_cpts = array('cpt1', 'cpt2', 'cpt3');

    if (in_array(get_post_type($post_id), $included_cpts)) {

        $has_thumbnail = get_the_post_thumbnail($post_id);

        if (!$has_thumbnail) {

            $images = get_field('acf_gallery', $post_id, false);

            $image_id = $images[0];

            if ($image_id) {

                set_post_thumbnail($post_id, $image_id);
                
            }
        }
    }
}

add_action('save_post', 'set_featured_image_from_gallery');