wordpress_migrate.drush.inc in WordPress Migrate 7

Drush support for the wordpress_migrate module

File

wordpress_migrate.drush.inc

<?php

/**
 * @file
 * Drush support for the wordpress_migrate module
 */

/**
 * Implementation of hook_drush_help().
 */
function wordpress_migrate_drush_help($section) {
  switch ($section) {
    case 'drush:wordpress-migrate-import':
      return dt('Import a WordPress blog');
    case 'drush:wordpress-migrate-rollback':
      return dt('Rollback a WordPress blog');
  }
}

/**
 * Implementation of hook_drush_command().
 */
function wordpress_migrate_drush_command() {
  static $commands = FALSE;
  $items['wordpress-migrate-import'] = array(
    'description' => 'Import a WordPress blog',
    'arguments' => array(
      'filename' => 'Filename of blog export to import',
    ),
    'examples' => array(
      'wordpress-migrate-import public://wordpress/blog.xml' => 'Import blog data',
    ),
    'drupal dependencies' => array(
      'migrate',
    ),
  );
  $items['wordpress-migrate-rollback'] = array(
    'description' => 'Rollback a WordPress blog',
    'arguments' => array(
      'filename' => 'Filename of blog export to rollback',
    ),
    'examples' => array(
      'wordpress-migrate-rollback public://wordpress/blog.xml' => 'Rollback blog data',
    ),
    'drupal dependencies' => array(
      'migrate',
    ),
  );
  return $items;
}
function drush_wordpress_migrate_import($filename) {

  // Capture non-informational output for mailing
  ob_start();
  try {
    $blog = wordpress_migrate_blog($filename);
    $mail_key = 'import_complete';
    $migrations = $blog
      ->migrations();
    foreach ($migrations as $migration) {
      $result = $migration
        ->processImport();
      if ($result != MigrationBase::RESULT_COMPLETED) {
        $mail_key = 'import_failure';
        break;
      }
    }
  } catch (Exception $e) {
    drush_log($e
      ->getMessage(), 'error');
    $mail_key = 'import_failure';
  }

  // Notify user
  $account = user_load($blog
    ->getUid());
  $params['account'] = $account;
  $params['output'] = ob_get_contents();
  ob_end_flush();
  drupal_mail('wordpress_migrate', $mail_key, $account->mail, user_preferred_language($account), $params);
}
function drush_wordpress_migrate_rollback($filename) {
  try {
    $blog = wordpress_migrate_blog($filename);
    $migrations = array_reverse($blog
      ->migrations());
    $incomplete = FALSE;
    foreach ($migrations as $migration) {
      $result = $migration
        ->processRollback();
      if ($result != MigrationBase::RESULT_COMPLETED) {
        $incomplete = TRUE;
        drush_log(dt('!title rollback failed to complete', array(
          '!title' => $blog
            ->getTitle(),
        )));
        break;
      }

      // Remove map/message tables, and migrate_status table entry
      Migration::deregisterMigration($migration
        ->getMachineName());
    }
    if (!$incomplete) {

      // Clear wordpress_migrate table entry
      db_delete('wordpress_migrate')
        ->condition('filename', $filename)
        ->execute();

      // Delete WXR file
      file_unmanaged_delete($filename);

      // Delete photo gallery
      if (module_exists('media_gallery')) {
        $account = user_load($blog
          ->getUid());
        $blog_title = t("@name's blog", array(
          '@name' => format_username($account),
        ));
        $gallery_nid = db_select('node', 'n')
          ->fields('n', array(
          'nid',
        ))
          ->condition('type', 'media_gallery')
          ->condition('title', $blog
          ->getTitle())
          ->execute()
          ->fetchField();
        if ($gallery_nid) {
          node_delete($gallery_nid);
        }
      }
    }
  } catch (Exception $e) {
    drush_log($e, 'error');
  }
}

Functions

Name	Description
drush_wordpress_migrate_import
drush_wordpress_migrate_rollback
wordpress_migrate_drush_command	Implementation of hook_drush_command().
wordpress_migrate_drush_help	Implementation of hook_drush_help().