php [将图像上传到媒体库]将图像上传到媒体库并将其设置为$ post_id（如果已指定）的特色图像。 #wordpress

Posted 2021-05-05

/**
 * Upload the image into the media library and set it as featured image for $post_id (if specified).
 *
 * @param string $url the image url/path.
 * @param int    $post_id the post it to whom assign the featured image.
 *
 * @return int the attachment id.
 */
function nine3_upload_image( $url, $post_id = false ) {
	// Use file_get_contents for file saved locally and wp_remote_get for the remote ones.
	if ( preg_match( '/http.?:\/\//', $url ) ) {
		$content = wp_remote_get( $url );
	} else {
		$content = file_get_contents( $url );
	}

	// Set the upload date as the post date.
	$post_date   = $post_id ? get_post_field( 'post_date', $post_id ) : null;
	$filename    = basename( $url );
	$upload_file = wp_upload_bits( $filename, '', $content, $post_date );

	if ( ! $upload_file['error'] ) {
		$wp_filetype = wp_check_filetype( $filename, null );

		$attachment = array(
			'post_mime_type' => $wp_filetype['type'],
			'post_parent'    => (int) $post_id,
			'post_title'     => preg_replace( '/\.[^.]+$/', '', $filename ),
			'post_content'   => '',
			'post_status'    => 'inherit',
		);

		if ( ! empty( $post_date ) ) {
			$attachment['post_date'] = $post_date;
		}

		$attachment_id = wp_insert_attachment( $attachment, $upload_file['file'], 0 );

		if ( ! is_wp_error( $attachment_id ) ) {
			require_once( ABSPATH . 'wp-admin/includes/image.php' );

			$attachment_data = wp_generate_attachment_metadata( $attachment_id, $upload_file['file'] );
			wp_update_attachment_metadata( $attachment_id, $attachment_data );

			return $attachment_id;
		}
	}

	return null;
}