填写您的邮件地址,订阅我们的精彩内容:

WordPress抓取文章中的第一张远程图片作为特色图片并保存到本地

739

在某些项目中,网站允许用户使用外链图片发布文章,或者处于某些特殊原因,网站希望抓取文中的第一张远程图片作为特色图片,并像QQ空间一样,将这张图片保存到本地,并插入到数据库中。总之,完成一系列动作之后,网站希望发现这张原本是远程外链的图片被放在本地,并成为了文章的特色图片。

这个动作我们使用一个hook来实现:

add_action('publish_post', 'fetch_images',999);

再创建一个fetch_images函数,来实现本文所说的所有功能。

function fetch_images( $post_ID ){
??? if ( defined('DOING_AUTOSAVE') && DOING_AUTOSAVE ) return;
??? if ( !current_user_can('edit_post', $post_ID) ) return;
??? $post = get_post($post_ID);
    ....

接下来,我们要获取文章内容中的第一张图片:

    $first_image = '';
?? ?preg_match('/<img.+src=['"]([^'"]+)['"].* />/i',$post->post_content,$images);
?? ?if(!empty($images))foreach($images as $image){
?? ??? ?if(strpos($image,'http') === 0){
?? ??? ??? ?$first_image = $images[1];
?? ??? ??? ?break;
?? ??? ?}
?? ?}

但实际上,通过上述的代码获得的图片src可能也是有问题的,或者根本没有抓取到数据。不过我们先不考虑这些问题,我们先实现本文的目标。

接下来就是关键代码一,它要实现“抓取-保存到本地”两个功能:

    $get = wp_remote_get( $get_image_src );
?? ?$type = wp_remote_retrieve_header( $get, 'content-type' );
?? ?$file_name = basename($get_image_src);
?? ?$file_content = wp_remote_retrieve_body($get);
?? ?$mirror = wp_upload_bits($file_name,null,$file_content);

这个地方有一个变化,上面我们获得了$first_image,这个地方的$get_image_src是上述得到的要抓取的远程图片地址。

在wordpress中,提供了wp_remote_getwp_remote_retrieve_body等原创抓取和信息获取函数,你可以查看官方文档以了解和remote相关的函数。wp_upload_bits则将抓取到的图片的二进制内容保存到本地,根据其文件类型,最终成为本地保存的图片,并将保存完后获得的本地图片信息保存在$mirror中。

既然已经保存到本地了,接下来就是将图片信息保存到数据库中。

$attachment = array(
??  'post_title'=> $file_name,
??  'post_mime_type' => $type
);
$attach_id = wp_insert_attachment( $attachment, $mirror['file'], $post_ID );
$attach_data = wp_generate_attachment_metadata( $attach_id, $mirror['file'] );
wp_update_attachment_metadata( $attach_id, $attach_data );
set_post_thumbnail( $post_ID, $attach_id );

这里所使用到的各种变量都是经过上一步获得的。这段代码将保存到本地的信息通过attachment相关函数插入到数据库中。经过这些函数的处理,同时还会产生于该附件相关的各个尺寸的实际图片,例如你在后台设置的small、large等尺寸的图片,想要了解wordpress缩略图,请详细阅读《WordPress文章特色图片功能详解》。而想要了解这些函数的具体实现过程,你还可以阅读《整合kindeditor到wordpress,特别是图片上传部分》一文,里面对这几个附件函数进行了详细的介绍。

最后,通过set_post_thumbnail函数将这个附件作为文章的特色图片。

当然,如果你需要将这个图片作为本地保存的特色图片的话,最好将文章内部的这个图片也换为本地图片。于是,使用一段替换代码,把文章内部的对应的图片地址替换为本地图片即可。

$updated = str_replace($get_image_src, $mirror['url'], $post->post_content);
wp_update_post(array('ID' => $post_ID, 'post_content' => $updated));

好了,这个功能基本上就靠这样的一个思路实现了。不过在最后,仍然提醒你函数还没有关闭呢,记得最后的}

下面我简单说几句