某些项目中,网站允许用户使用外链图片发布文章,或者处于某些特殊原因,网站希望抓取文中的第一张远程图片作为特色图片,并像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_get、wp_remote_retrieve_body等原创抓取和信息获取函数,你可以查看官方文档以了解和remote相关的函数。wp_upload_bits则将抓取到的图片的二进制内容保存到本地,根据其文件类型,最终成为本地保存的图片,并将保存完后获得的本地图片信息保存在$mirror中。
既然已经保存到本地了,接下来就是将图片信息保存到数据库中。