delete_all.module in Delete all 7

Delete all 7.x module

File

delete_all.module

<?php

/**
 * @var DELETE_ALL_MAX_EXECUTION
 *
 * Total seconds to allow the delete script to run
 */
define('DELETE_ALL_MAX_EXECUTION', 600);

/**
 * @file
 * Delete all 7.x module
 */
function delete_all_menu() {
  $items = array();
  $items['admin/content/delete_content'] = array(
    'title' => 'Delete all content',
    'description' => 'Delete all nodes and comments on this site. This is useful for development sites, or prior to launching the site. Otherwise it destroys all data on a site.',
    'page callback' => 'drupal_get_form',
    'page arguments' => array(
      'delete_all_content',
    ),
    'access arguments' => array(
      'administer nodes',
    ),
    'type' => MENU_NORMAL_ITEM,
  );
  $items['admin/content/delete_content/confirm'] = array(
    'title' => 'Confirm deletion of all content',
    'page callback' => 'drupal_get_form',
    'page arguments' => array(
      'delete_all_content_confirm',
    ),
    'access arguments' => array(
      'administer nodes',
    ),
    'type' => MENU_CALLBACK,
  );
  $items['admin/content/delete_users'] = array(
    'title' => 'Delete all users',
    'description' => 'Delete all users on this site. This is useful for development sites, or prior to launching the site. Otherwise it destroys all data on a site.',
    'page callback' => 'drupal_get_form',
    'page arguments' => array(
      'delete_all_users',
    ),
    'access arguments' => array(
      'administer users',
    ),
    'type' => MENU_NORMAL_ITEM,
  );
  return $items;
}
function delete_all_content() {

  // count how many nodes we have of each type
  $result = db_query("SELECT type, COUNT(*) AS num FROM {node} GROUP BY type");
  $count = array();
  foreach ($result as $data) {
    $count[$data->type] = $data->num;
  }

  // Add the types to the form. If there are no eligible types to delete,
  // we don't need to render the form.
  $types = array();
  foreach (node_type_get_types() as $type => $info) {
    if (array_key_exists($type, $count)) {
      $types[$type] = $info->name . ' (' . $count[$type] . ')';
    }
  }
  asort($types);
  if (module_exists('taxonomy')) {

    // count how many taxonomy terms we have for each vocabulary
    $result = db_query("SELECT taxonomy_vocabulary.machine_name AS vocabulary, COUNT(*) AS num FROM {taxonomy_vocabulary} INNER JOIN {taxonomy_term_data} USING (vid) GROUP BY vocabulary");
    $taxocount = array();
    foreach ($result as $data) {
      $taxocount[$data->vocabulary] = $data->num;
    }

    // Add the taxonomy vocabularies to the form. If there are no eligible types to delete,
    // we don't need to render the form.
    $vocabularies = array();
    foreach (taxonomy_get_vocabularies() as $vocabulary) {
      $name = $vocabulary->machine_name;
      if (array_key_exists($name, $taxocount)) {
        $vocabularies[$name] = $vocabulary->name . ' (' . $taxocount[$name] . ')';
      }
    }
    asort($vocabularies);
  }
  if (empty($types) && empty($vocabularies)) {
    $form = array();
    $form['no_content_types'] = array(
      '#prefix' => '<p>',
      '#suffix' => '</p>',
      '#value' => t('There are no content types with content available to delete. You must <a href="@node-add">create some content</a> in order to delete it.', array(
        '@node-add' => url('node/add'),
      )),
    );
    if (module_exists('devel')) {
      $form['generate_content_suggestion'] = array(
        '#prefix' => '<p>',
        '#suffix' => '</p>',
        '#value' => t('You can generate content quickly at the <a href="@generate-content-page">generate content page</a>.', array(
          '@generate-content-page' => url('admin/generate/content'),
        )),
      );
    }
    return $form;
  }
  drupal_add_js(drupal_get_path('module', 'delete_all') . '/delete_all.js');
  $form = array();
  $form['all'] = array(
    '#type' => 'checkbox',
    '#default_value' => FALSE,
    '#title' => t('Delete All Content'),
    '#description' => t('Select to delete all content'),
    '#attributes' => array(
      'class' => array(
        'delete-all',
      ),
    ),
  );
  if (module_exists('taxonomy')) {
    $form['taxonomy'] = array(
      '#type' => 'checkbox',
      '#default_value' => FALSE,
      '#title' => t('Delete all Taxonomy'),
      '#description' => t('Select to delete all taxonomy terms'),
      '#attributes' => array(
        'class' => array(
          'delete-all',
        ),
      ),
    );
  }
  $form['reset'] = array(
    '#type' => 'checkbox',
    '#default_value' => FALSE,
    '#title' => t('Reset node count'),
    '#description' => t('Select to reset the node count'),
    '#attributes' => array(
      'class' => array(
        'delete-reset',
      ),
    ),
  );
  $form['type-fieldset'] = array(
    '#type' => 'fieldset',
    '#title' => t('Node types'),
    '#collapsible' => TRUE,
    '#collapsed' => FALSE,
    'types' => array(
      '#type' => 'checkboxes',
      '#options' => $types,
      '#description' => t('Select the types of node content to delete'),
    ),
  );
  if (module_exists('taxonomy')) {
    $form['taxonomy-fieldset'] = array(
      '#type' => 'fieldset',
      '#title' => t('Taxonomy'),
      '#collapsible' => TRUE,
      '#collapsed' => FALSE,
      'vocabularies' => array(
        '#type' => 'checkboxes',
        '#options' => $vocabularies,
        '#description' => t('Select the taxonomy vocabularies to delete'),
      ),
    );
  }
  $form['method-fieldset'] = array(
    '#type' => 'fieldset',
    '#title' => t('Method'),
    '#collapsible' => TRUE,
    '#collapsed' => FALSE,
    'method' => array(
      '#type' => 'radios',
      '#title' => t('Method'),
      '#options' => array(
        'normal' => t('Normal'),
        'quick' => t('Quick'),
      ),
      '#default_value' => 'normal',
      '#description' => t('Normal node delete calls node_delete() on every node in the database.  If you have only a few hundred nodes, this can take a very long time.  Use the quick node delete method to get around this problem.  This method deletes directly from the database, skipping the extra php processing.  The downside is that it can miss related tables that are normally handled by module hook_delete\'s.'),
    ),
  );
  $form['submit'] = array(
    '#type' => 'submit',
    '#value' => t('Delete'),
  );
  $form['#action'] = url('admin/content/delete_content/confirm');
  return $form;
}
function delete_all_content_confirm() {
  $results = $_POST;
  $form = array();
  $form['method'] = array(
    '#type' => 'hidden',
    '#value' => isset($results['method']) ? $results['method'] : '',
  );
  if (!isset($results['all'])) {
    $results['all'] = '';
  }
  if (!isset($results['reset'])) {
    $results['reset'] = '';
  }
  $form['all'] = array(
    '#type' => 'hidden',
    '#value' => isset($results['all']) ? $results['all'] : '',
  );
  if (isset($results['all']) && !isset($results['types'])) {
    if (!empty($results['all'])) {

      // Only carry reset value through if delete all nodes is set
      $form['reset'] = array(
        '#type' => 'hidden',
        '#value' => $results['reset'],
      );
      $form['all_warning'] = array(
        '#markup' => '<p>' . t('All content in all content types will be deleted. Be sure to have a backup of any important data!') . '</p>',
      );
    }
    if (!empty($results['reset'])) {
      if (empty($results['all'])) {
        $form['all_reset'] = array(
          '#markup' => '<p>' . t('Sorry, we can\'t reset the counters because you are not deleting all nodes.') . '</p>',
        );
      }
      else {
        $form['all_reset'] = array(
          '#markup' => '<p>' . t('Node, revision and comment counts will be reset.') . '</p>',
        );
      }
    }
  }
  if (isset($results['types']) && is_array($results['types'])) {
    foreach ($results['types'] as $key_type => $type) {
      $form['type_' . $key_type] = array(
        '#type' => 'hidden',
        '#value' => $type,
      );
      $info = node_type_get_types('type', $type);
      $info = $info[$type];
      $form[$type . '_warning'] = array(
        '#markup' => '<p>' . t('All node content of type %type will be deleted. Be sure to have a backup of any important data!', array(
          '%type' => $info->name,
        )) . '</p>',
      );
    }
  }
  if (module_exists('taxonomy')) {
    $vocabulary_info = array();
    foreach (taxonomy_get_vocabularies() as $vocabulary) {
      $vocabulary_info[$vocabulary->machine_name] = $vocabulary->name;
    }
    if (isset($results['vocabularies'])) {
      if (is_array($results['vocabularies'])) {
        foreach ($results['vocabularies'] as $key_vocabulary => $vocabulary) {
          $form['vocabulary_' . $key_vocabulary] = array(
            '#type' => 'hidden',
            '#value' => $vocabulary,
          );
          $info = $vocabulary_info[$vocabulary];
          $form[$vocabulary . '_warning'] = array(
            '#markup' => '<p>' . t('All taxonomy terms of vocabulary %vocabulary will be deleted. Be sure to have a backup of any important data!', array(
              '%vocabulary' => $info,
            )) . '</p>',
          );
        }
      }
    }
  }
  $keys = array_filter(array_keys($results), "_delete_all_type_keys");
  foreach ($keys as $key) {
    $form[$key] = array(
      '#type' => 'hidden',
      '#value' => $results[$key],
    );
  }
  $vocabulary_keys = array_filter(array_keys($results), "_delete_all_vocabulary_keys");
  foreach ($vocabulary_keys as $vkey) {
    $form[$vkey] = array(
      '#type' => 'hidden',
      '#value' => $results[$vkey],
    );
  }
  return confirm_form($form, t('Are you sure you wish to delete content?'), 'admin/content/delete_content', NULL, t('Delete all content now'), t('Cancel delete of all content'));
}
function delete_all_content_confirm_submit($form, &$form_state) {
  $types = array();
  $vocabularies = array();
  $keys = array_filter(array_keys($form_state['values']), "_delete_all_type_keys");
  foreach ($keys as $key) {
    $types[] = $form_state['values'][$key];
  }
  if (isset($form_state['values']['taxonomy'])) {
    $vocabulary_info = array();
    foreach (taxonomy_get_vocabularies() as $vocabulary) {
      $vocabularies[$vocabulary->machine_name] = $vocabulary->vid;
    }
  }
  $vkeys = array_filter(array_keys($form_state['values']), "_delete_all_vocabulary_keys");
  foreach ($vkeys as $key) {
    $vocabularies[] = $form_state['values'][$key];
  }
  switch ($form_state['values']['method']) {
    case 'normal':
      $count = _delete_all_normal($form_state['values']['all'], $types, $vocabularies);
      break;
    case 'quick':

      // the quick method doesn't support an all option,
      // so just get all the content types and delete all of those
      if ($form_state['values']['all']) {
        $result = db_query("SELECT DISTINCT type FROM {node}");
        foreach ($result as $data) {

          // while ($data = db_fetch_object($result)) {
          $types[] = $type;
        }
        if ($form_state['values']['taxonomy']) {
          $result = db_query("SELECT DISTINCT machine_name FROM {taxonomy_vocabulary}");
          foreach ($result as $data) {
            $vocabularies[] = $type;
          }
        }
      }
      $count = _delete_all_quick($types, $vocabularies);
      break;
  }
  if (!$types) {

    // Delete the URL aliases
    db_query("DELETE FROM {url_alias} WHERE source LIKE 'node/%%'");
    drupal_set_message(t('All nodes, comments and URL aliases have been deleted. Number of nodes deleted: !count_nodes.', array(
      '!count' => check_plain($count['nodes']),
    )));
    if ($count['terms']) {
      drupal_set_message(t('All taxonomy terms deleted. Number of terms deleted: !count_terms.', array(
        '!count' => check_plain($count['terms']),
      )));
    }
    if ($form_state['values']['reset']) {
      db_query("ALTER TABLE node AUTO_INCREMENT=1");
      db_query("ALTER TABLE node_revision AUTO_INCREMENT=1");
      if (module_exists('comment')) {
        db_query("ALTER TABLE comment AUTO_INCREMENT=1");
        drupal_set_message(t('All node, revision and comment counts have been reset.'));
      }
      else {
        drupal_set_message(t('All node and revision counts have been reset.'));
      }
    }
  }
  else {
    drupal_set_message(t('Nodes and comments of type @type have been deleted. Number of nodes deleted: !count.', array(
      '!count' => check_plain($count),
      '@type' => implode(', ', $types),
    )));
  }
  drupal_goto('admin');
}
function _delete_all_normal($all, $types, $vocabularies = NULL) {
  $deleted = array(
    'nodes' => 0,
    'terms' => 0,
  );
  if (is_array($types) && count($types) > 0) {
    foreach ($types as $type) {
      $result = db_query('SELECT nid FROM {node} WHERE type = :type', array(
        ':type' => $type,
      ));
    }
  }
  else {
    $result = db_query('SELECT nid FROM {node}');
  }
  if ($result) {
    foreach ($result as $data) {
      set_time_limit(30);
      node_delete($data->nid);
      $deleted['nodes']++;
    }
  }
  if ($vocabularies) {
    $vocabulary_info = array();
    foreach (taxonomy_get_vocabularies() as $vocabulary) {
      $vocabulary_info[$vocabulary->machine_name] = $vocabulary->vid;
    }
    foreach ($vocabularies as $vocabulary) {
      foreach (taxonomy_get_tree($vocabulary_info[$vocabulary]) as $term) {
        taxonomy_term_delete($term->tid);
        $deleted['terms']++;
      }
    }
  }
  return $deleted;
}
function _delete_all_quick($types, $vocabularies) {
  $deleted = array(
    'nodes' => 0,
    'terms' => 0,
  );
  foreach ($types as $type) {

    // keep this alive
    set_time_limit(240);

    // determine how many items will be deleted
    $count = db_select('node')
      ->condition('type', $type)
      ->countQuery()
      ->execute()
      ->fetchField();
    if ($count) {

      // should always be positive

      /**
       * build a list of tables that need to be deleted from
       *
       * The tables array is of the format table_name => array('col1', 'col2', ...)
       * where "col1, col2" are using "nid, vid", but could simply be "nid".
       */
      $nid_vid = array(
        'nid',
        'vid',
      );
      $nid = array(
        'nid',
      );
      $tables = array(
        'node_revisions' => $nid_vid,
        'comments' => $nid,
      );
      $tables[_content_tablename($type, CONTENT_DB_STORAGE_PER_CONTENT_TYPE)] = $nid_vid;
      $content = content_types($type);
      if (count($content['fields'])) {
        foreach ($content['fields'] as $field) {
          $field_info = content_database_info($field);
          $tables[$field_info['table']] = $nid_vid;
        }
      }

      // find all other tables that might be related
      switch ($GLOBALS['db_type']) {
        case 'mysql':
        case 'mysqli':
          $result_tables = db_query("SHOW TABLES");
          while ($data = db_fetch_array($result_tables)) {
            $table = array_pop($data);
            if (isset($tables[$table]) || substr($table, 0, 8) == 'content_') {
              continue;
            }
            $result_cols = db_query("SHOW COLUMNS FROM %s", $table);
            $cols = array();
            while ($data = db_fetch_array($result_cols)) {
              $cols[$data['Field']] = $data;
            }
            if (isset($cols['nid'])) {
              $tables[$table] = isset($cols['vid']) ? $nid_vid : $nid;
            }
          }
          break;
        case 'pgsql':

          // @TODO: inspect the database and look for nid fields
          break;
      }

      // @todo: update all node related nid references
      // delete from all of the content tables in one sql statement
      $sql = array(
        'delete' => array(),
        'from' => array(),
        'where' => array(),
      );
      $index = 0;
      foreach ($tables as $table => $cols) {
        $table = '{' . $table . '}';
        $sql['cols'][] = "t{$index}.*";

        // build the ON clause
        $on = array();
        foreach ($cols as $col) {
          $on[] = "t{$index}.{$col} = n.{$col}";
        }

        // now that we have the ON clause, build the join clause
        $sql['join'][] = " LEFT JOIN {$table} t{$index} ON " . implode(' AND ', $on);
        $index++;
      }
      $delete_sql = "DELETE n.*, " . implode(', ', $sql['cols']) . " FROM {node} n " . implode(' ', $sql['join']);
      db_query($delete_sql . " WHERE n.type = '%s'", $type);
      $deleted['nodes'] += $count;
    }
  }
  $vocabulary_info = array();
  foreach (taxonomy_get_vocabularies() as $vocabulary) {
    $vocabulary_info[$vocabulary->machine_name] = $vocabulary->vid;
  }
  foreach ($vocabularies as $vocabulary) {
    foreach (taxonomy_get_tree($vocabulary_info[$vocabulary]) as $term) {
      taxonomy_term_delete($term->tid);
      $deleted['terms']++;
    }
  }
  return $deleted;
}
function delete_all_users() {
  $form = array();
  return confirm_form($form, t('Are you sure you want to delete all users (uid > 1)?'), 'admin', t('This will delete all users except for User 1 (the administrative user). This action cannot be undone'), t('Delete'), t('Cancel'), 'delete_all_content');
}
function delete_all_users_submit($form, &$form_state) {
  ini_set('max_execution_time', DELETE_ALL_MAX_EXECUTION);
  $count = delete_all_users_delete();
  drupal_set_message(t('All users have been deleted. Number of users deleted: !count.', array(
    '!count' => check_plain($count),
  )));
  drupal_goto('admin');
}
function delete_all_users_delete($roles = NULL) {
  $count = 0;
  if (!$roles) {
    $result = db_query('SELECT uid FROM {users} WHERE uid > 1');
    foreach ($result as $data) {
      user_delete($data->uid);
      $count++;
    }

    // Delete the URL aliases
    db_query("DELETE FROM {url_alias} WHERE source LIKE 'user/%%'");
  }
  else {
    if (is_array($roles)) {
      $result = array();
      foreach ($roles as $role) {
        $rid = db_select('role', 'r')
          ->fields('r', array(
          'rid',
        ))
          ->condition('name', $role, '=')
          ->execute()
          ->fetchField();
        $result = array_merge($result, db_select('users_roles', 'ur')
          ->fields('ur', array(
          'uid',
        ))
          ->condition('rid', $rid, '=')
          ->execute()
          ->fetchCol('uid'));
      }
    }
    else {
      $rid = db_select('role', 'r')
        ->fields('r', array(
        'rid',
      ))
        ->condition('name', $roles, '=')
        ->execute()
        ->fetchField();
      $result = db_select('users_roles', 'ur')
        ->fields('ur', array(
        'uid',
      ))
        ->condition('rid', $rid, '=')
        ->execute()
        ->fetchCol('uid');
    }
    foreach ($result as $data) {
      user_delete($data);
      $count++;
    }

    // @TODO Delete individual aliases
  }
  return $count;
}

/**
 * Private callback to determine if a variable starts with 'type_'.
 * @param $var
 *   The string to test against.
 * @return bool
 *  TRUE if $var begins with 'type_'
 */
function _delete_all_type_keys($var) {
  return strpos($var, 'type_') === 0 ? TRUE : FALSE;
}

/**
 * Private callback to determine if a variable starts with 'type_'.
 * @param $var
 *   The string to test against.
 * @return bool
 *  TRUE if $var begins with 'type_'
 */
function _delete_all_vocabulary_keys($var) {
  return strpos($var, 'vocabulary_') === 0 ? TRUE : FALSE;
}

Functions

Name	Description
delete_all_content
delete_all_content_confirm
delete_all_content_confirm_submit
delete_all_menu	@file Delete all 7.x module
delete_all_users
delete_all_users_delete
delete_all_users_submit
_delete_all_normal
_delete_all_quick
_delete_all_type_keys	Private callback to determine if a variable starts with 'type_'.
_delete_all_vocabulary_keys	Private callback to determine if a variable starts with 'type_'.

Constants

Name	Description
DELETE_ALL_MAX_EXECUTION	Total seconds to allow the delete script to run