Импорт XML-файла WordPress из functions.php

Ваш вопрос довольно специфичен, если вы хотите "просто" автоматически импортировать некоторые записи/страницы. Существуют другие способы сделать это, помимо использования XML-файла экспорта.

Если у вас есть только текстовые записи, то вам следует использовать LOAD DATA INFILE. Сначала вам нужно экспортировать свои записи.

global $wpdb, $wp_filesystem;

$tables = array(
        'posts'    => array( 'posts', 'postmeta' ),
        'comments' => array( 'comments', 'commentmeta' ),
        'terms'    => array( 'terms', 'term_taxonomy', 'term_relationships' ),
        'users'    => array( 'user', 'usermeta' ),
        'links'    => array( 'links' ),
        'options'  => array( 'options' ),
        'other'    => array(),
        // для мультисайтов
        'multiside' => array( 'blogs', 'signups', 'site', 'sitemeta', 'sitecategories', 'registration_log', 'blog_versions' )

);

$exports = array( 'posts', 'comments', 'users' );

$exportdir = TEMPLATEPATH . '/export';

if ( ! is_dir( $exportdir ) {
    $mkdir = wp_mkdir_p( $exportdir );
    if ( false == $mkdir || ! is_dir( $exportdir ) )
        throw new Exception( 'Невозможно создать директорию для экспорта. Прерывание.' );
}

// очистка директории экспорта, иначе MySQL выдаст ошибки
$files = glob( $exportdir . '/*' );
if ( ! empty( $files ) ) {
    foreach( $files as $file )
        unlink( $file );
}

foreach ( $exports as $export ) {

    if ( ! isset( $tables[$export] ) )
        continue;

    if ( ! empty( $tables[$export] ) ) {
        foreach ( $tables[$export] as $table ) {

            $outfile =  sprintf( '%s/%s_dump.sql', $exportdir, $table );
            $sql = "SELECT * FROM {$wpdb->$table} INTO OUTFILE '%s'";
            $res = $wpdb->query( $wpdb->prepare( $sql, $outfile ) );

            if ( is_wp_error( $res ) )
                echo "<p>Не удалось экспортировать {$table} в {$outfile}</p>";
        }
    }
}

Это создаст директорию в папке вашей темы (убедитесь, что она доступна для записи!) и экспортирует записи и комментарии (с их метаданными) в файлы дампа. Используйте массив export для определения того, что вы хотите экспортировать. Я сгруппировал элементы более или менее логично (если вы хотите экспортировать записи, то вам следует также экспортировать postmeta и т.д.).

Преимущество этого решения в том, что с помощью оператора SELECT вы можете определить конкретные данные (например, только записи из определенной категории или только страницы, или только удаленные записи).

Теперь вы хотите импортировать эти данные в новый блог

global $wpdb;

$exportdir = TEMPLATEPATH . '/export';

$files = glob( $exportdir . '/*_dump.sql' );

foreach ( $files as $file ) {

    preg_match( '#/([^/]+)_dump.sql$#is', $file, $match );

    if ( ! isset( $match[1] ) )
        continue;

    $sql = "LOAD DATA LOCAL INFILE '%s' INTO TABLE {$wpdb->$match[1]};";

    $res = $wpdb->query( $wpdb->prepare( $sql, $file ) );

    if ( is_wp_error( $res ) )
        echo "<p>Не удалось импортировать данные из файла {$file} в таблицу {$wpdb->$match[1]}</p>";
}

Это решение хорошо, если записи не содержат вложений, таких как изображения. Другая проблема заключается в том, что пользователи и категории не будут импортированы. Убедитесь, что они созданы до начала импорта (или включите пользователей и категории в ваш экспорт). Это очень грубый метод импорта, он перезапишет существующие данные!

Если вы хотите экспортировать также вложения, вам придется проделать немного больше работы.

(Примечание: Пожалуйста, прочитайте полный ответ и "Очень последние слова" в конце! Эта тема не для новичков, и я не буду писать предупреждения для каждой рискованной строки кода)

Плагин WordPress Importer кажется хорошим способом импортировать все данные и автоматически загрузить вложения. Давайте посмотрим, что делает этот плагин.

Сначала плагин запрашивает XML-файл для загрузки. Затем он анализирует XML-файл и запрашивает сопоставление авторов и следует ли загружать вложения.

Для автоматического запуска плагина нам нужно изменить некоторые вещи. Сначала нам нужно пропустить процесс загрузки. Это довольно просто, потому что вы можете включить XML-файл в тему и знать, где он находится. Затем нам нужно пропустить вопросы, которые появляются после загрузки XML-файла. Мы можем предопределить наши собственные значения и передать их в процесс импорта.

Начните с копии плагина. Создайте директорию в вашей теме, например autoimport, и скопируйте файлы wordpress-importer.php и parsers.php в нее. Хорошая идея — переименовать файл wordpress-importer.php во что-то вроде autoimporter.php. В функции вашей темы добавьте вызов функции для запуска автоматического импорта

/**
 * Автоматический импорт XML-файла
 */
add_action( 'after_setup_theme', 'autoimport' );

function autoimport() {
    // получаем файл
    require_once TEMPLATEPATH . '/autoimport/autoimporter.php';

    if ( ! class_exists( 'Auto_Importer' ) )
        die( 'Auto_Importer не найден' );

    // вызываем функцию
    $args = array(
        'file'        => TEMPLATEPATH . '/autoimport/import.xml',
        'map_user_id' => 1
    );

    auto_import( $args );

}

Сначала мы устанавливаем некоторые аргументы. Первое — это полный путь к XML-файлу. Второе — ID существующего пользователя. Нам нужен этот пользователь для сопоставления авторов, это пользователь, к которому будут привязаны все записи, если не нужно создавать новых авторов.

Теперь нам нужно понять, как работает плагин. Откройте переименованный файл плагина и прокрутите до конца. Там есть функция wordpress_importer_init() и вызов действия. Удалите оба, они больше не нужны. Теперь перейдите к началу файла и удалите заголовок плагина (комментарий в начале файла). После этого переименуйте класс WP_Importer во что-то вроде Auto_Importer, не забудьте изменить оператор function_exists и первый метод WP_Importer (это конструктор в стиле PHP4).

Позже мы передадим XML-файл напрямую в конструктор класса, измените первый метод на этот

var $xmlfile = '';
var $map_user_id = 0;

function Auto_Importer( $args ) {

    if ( file_exists( $args['file'] ) ) {

        // для систем Windows
        $file = str_replace( '\\', '/', $args['file'] );

        $this->xmlfile = $file;
    }

    if ( isset( $args['map_user_id'] ) )
        $this->map_user_id = $args['map_user_id'];

}

Теперь нам нужно удалить и изменить некоторые методы внутри класса. Первый метод — это метод dispatch(). Этот метод показывает, как работает класс. Он выполняет три шага. Сначала загружает XML-файл, затем обрабатывает его и, наконец, импортирует данные.

Case zero — это первый шаг, приветствие. Это то, что вы видите при первом вызове импорта. Он запрашивает файл для загрузки. Case two обрабатывает загрузку и отображает форму для параметров импорта. Case three, наконец, выполняет импорт. Другими словами: первые два шага только запрашивают данные, которые мы можем предоставить сами. Нам нужен только шаг 3 (case 2), и мы должны предоставить данные, запрошенные на первом и втором шагах.

На втором шаге вы видите вызов функции wp_import_handle_upload(). Эта функция устанавливает некоторую информацию о XML-файле. Мы не можем больше использовать эту функцию, потому что мы не загружали файл. Поэтому нам нужно скопировать и изменить функцию. Создайте новый метод внутри класса

function import_handle_upload() {

    $url = get_template_directory_uri() . str_replace( TEMPLATEPATH, '', $this->xmlfile );
    $type = 'application/xml'; // мы знаем MIME-тип нашего файла
    $file = $this->xmlfile;
    $filename = basename( $this->xmlfile );

    // Создаем массив объекта
    $object = array( 'post_title' => $filename,
            'post_content' => $url,
            'post_mime_type' => $type,
            'guid' => $url,
            'context' => 'import',
            'post_status' => 'private'
    );

    // Сохраняем данные
    $id = wp_insert_attachment( $object, $file );

    // планируем очистку через день на случай неудачного импорта или отсутствия вызова wp_import_cleanup()
    wp_schedule_single_event( time() + DAY_IN_SECONDS, 'importer_scheduled_cleanup', array( $id ) );

    return array( 'file' => $file, 'id' => $id );
}

И замените вызов функции $file = wp_import_handle_upload(); в методе handle_upload() на наш новый метод $file = $this->import_handle_upload();

Теперь мы заменили процесс загрузки нашим собственным файлом (который уже должен существовать). Продолжаем и удаляем больше ненужных методов. Методы greet(), header() и footer() больше не нужны (header и footer только выводят текст) и могут быть удалены из класса. В методе dispatch() удалите вызовы этих методов ($this->header() и $this->footer()).

Первый шаг завершен, теперь нам нужно позаботиться о втором шаге, параметрах импорта. Параметры импорта спрашивают, следует ли разрешить загрузку вложений и сопоставление авторов.

Первая часть проста. Установите true, если вложения должны быть загружены, или false, если нет. Сопоставление авторов немного сложнее. Если разрешено создавать новых пользователей (авторов из файла импорта), создайте их. Если нет, назначьте записи существующему пользователю. Это делается в методе get_author_mapping(). Нам нужно заменить данные $_POST существующими данными. Здесь нам нужно простое решение, поэтому мы просто сопоставляем всех новых авторов с существующим, если не разрешено создавать новых пользователей. Или просто создаем всех новых пользователей. Во втором случае убедитесь, что все новые пользователи являются фиктивными. Если нет, каждый раз при их импорте они будут получать электронное письмо с логином и паролем для нового блога!! Я не объясняю каждую строку кода, вот полностью переписанный метод

function get_author_mapping( $map_users_id ) {
    if ( empty( $this->authors ) )
        return;

    $create_users = $this->allow_create_users();

    foreach ( (array) $this->authors as $i => $data ) {

        $old_login = $data['author_login'];

        // Multisite добавляет strtolower к sanitize_user. Нужно очистить здесь, чтобы избежать ошибок в process_posts.
        $santized_old_login = sanitize_user( $old_login, true );
        $old_id = isset( $this->authors[$old_login]['author_id'] ) ? intval($this->authors[$old_login]['author_id']) : false;

        if ( ! $create_users ) {
            $user = get_userdata( intval($map_users_id) );
            if ( isset( $user->ID ) ) {
                if ( $old_id )
                    $this->processed_authors[$old_id] = $user->ID;
                $this->author_mapping[$santized_old_login] = $user->ID;
            }
        } else if ( $create_users ) {
            if ( ! empty($this->authors[$i]) ) {
                $user_id = wp_create_user( $this->authors[$i]['author_login'], wp_generate_password() );
            } else if ( $this->version != '1.0' ) {
                $user_data = array(
                    'user_login' => $old_login,
                    'user_pass' => wp_generate_password(),
                    'user_email' => isset( $this->authors[$old_login]['author_email'] ) ? $this->authors[$old_login]['author_email'] : '',
                    'display_name' => $this->authors[$old_login]['author_display_name'],
                    'first_name' => isset( $this->authors[$old_login]['author_first_name'] ) ? $this->authors[$old_login]['author_first_name'] : '',
                    'last_name' => isset( $this->authors[$old_login]['author_last_name'] ) ? $this->authors[$old_login]['author_last_name'] : '',
                );
                $user_id = wp_insert_user( $user_data );
            }

            if ( ! is_wp_error( $user_id ) ) {
                if ( $old_id )
                    $this->processed_authors[$old_id] = $user_id;
                $this->author_mapping[$santized_old_login] = $user_id;
            } else {
                printf( __( 'Не удалось создать нового пользователя для %s. Их записи будут привязаны к текущему пользователю.', 'wordpress-importer' ), esc_html($this->authors[$old_login]['author_display_name']) );
                if ( defined('IMPORT_DEBUG') && IMPORT_DEBUG )
                    echo ' ' . $user_id->get_error_message();
                echo '<br />';
            }
        }

        // на всякий случай: если user_id был недействителен, используем текущего пользователя
        if ( ! isset( $this->author_mapping[$santized_old_login] ) ) {
            if ( $old_id )
                $this->processed_authors[$old_id] = (int) get_current_user_id();
            $this->author_mapping[$santized_old_login] = (int) get_current_user_id();
        }
    }
}

Осталось еще немного работы. Сначала добавим функцию auto_import()

function auto_import( $args ) {

    $defaults = array( 'file' => '', 'map_user_id' => 0);
    $args = wp_parse_args( $args, $defaults );

    $autoimport = new Auto_Importer( $args );
    $autoimport->do_import();

}

Поместите эту функцию после класса. В этой функции не хватает обработки ошибок и проверок (например, для пустого аргумента файла).

Если теперь запустить класс, вы получите много сообщений об ошибках. Первое — что класс отсутствует. Это связано с оператором if в начале.

if ( ! defined( 'WP_LOAD_IMPORTERS' ) )
    return;

Нам нужно удалить его, иначе файл не будет полностью проанализирован. Затем есть несколько функций, которые не загружены в этот момент. Нам нужно включить некоторые файлы.

$required = array(
    'post_exists'                     => ABSPATH . 'wp-admin/includes/post.php',
    'wp_generate_attachment_metadata' => ABSPATH . 'wp-admin/includes/image.php',
    'comment_exists'                  => ABSPATH . 'wp-admin/includes/comment.php'
);

foreach ( $required as $func => $req_file ) {
    if ( ! function_exists( $func ) )
        require_once $req_file;
}

В основном это все. Я протестировал это на локальной установке с тестовыми данными XML от WordPress. Это работает для меня, но это не идеальное решение для продакшена!

И несколько последних слов о настройке некоторых опций. Есть две опции, которые можно изменить с помощью фильтра:

add_filter( 'import_allow_create_users', function() { return false; } );
add_filter( 'import_allow_fetch_attachments', '__return_false' );

Думаю, мне не нужно это объяснять. Поместите эти фильтры в functions.php и установите true или false (первый в стиле PHP5.3, второй в стиле WP).

Очень последние слова

Я собрал все в этом гисте. Используйте на свой страх и риск! Я не несу ответственности за что-либо!. Пожалуйста, посмотрите файлы в гисте, я не объяснял здесь каждый маленький шаг.

Что я не сделал: Установил значение, например, в (теме) опциях после импорта. Иначе импорт будет запускаться каждый раз при активации темы.

Возможно, я поработаю над этим в будущем, очищу некоторые вещи и проведу больше тестов.