users.inc in Data export import 6

Enables users to be exported and imported.

File

includes/profiles/users.inc

<?php

/**
 * @file
 * Enables users to be exported and imported.
 */

/**
 * Function to add items to the main menu.
 */
function data_export_import_users_add_items_to_menu(&$items) {
  $items['admin/content/data_export_import/users'] = array(
    'title' => 'Users',
    'page callback' => 'data_export_import_callback_export_users',
    'access arguments' => array(
      'access data export import',
    ),
    'weight' => 1,
    'description' => 'Export Users',
    'type' => MENU_LOCAL_TASK,
  );
  $items['admin/content/data_export_import/users/export'] = array(
    'title' => 'Export',
    'page callback' => 'data_export_import_callback_export_users',
    'access arguments' => array(
      'access data export import',
    ),
    'weight' => 2,
    'description' => 'Export Users',
    'type' => MENU_DEFAULT_LOCAL_TASK,
  );
  $items['admin/content/data_export_import/users/import'] = array(
    'title' => 'Import',
    'page callback' => 'data_export_import_callback_import_users',
    'access arguments' => array(
      'access data export import',
    ),
    'weight' => 3,
    'description' => 'Import Users',
    'type' => MENU_LOCAL_TASK,
  );
}

/**
 * Callback function to export users.
 */
function data_export_import_callback_export_users() {
  return drupal_get_form('data_export_import_export_users_form');
}

/**
 * Callback function to import users.
 */
function data_export_import_callback_import_users() {
  return drupal_get_form('data_export_import_import_users_form');
}

/**
 * Function to create form to export users.
 */
function data_export_import_export_users_form($form_state) {
  $form = array();

  // This stops the values being posted from being flattened into a
  // single level array.  It might be easier to extract the
  // information from a non-flattened array.
  $form['#tree'] = TRUE;
  $form['users'] = array(
    '#type' => 'fieldset',
    '#title' => t('Options'),
    '#collapsible' => TRUE,
    '#collapsed' => FALSE,
  );
  $form['users']['users'] = array(
    '#type' => 'checkbox',
    '#title' => 'Export all Users',
  );
  $form['submit'] = array(
    '#type' => 'submit',
    '#value' => t('Create dataset file'),
  );
  return $form;
}

/**
 * Function to create form to import users.
 */
function data_export_import_import_users_form($form_state) {
  $form = array();
  $form['users'] = array(
    '#type' => 'fieldset',
    '#title' => t('Users'),
    '#collapsible' => TRUE,
    '#collapsed' => FALSE,
  );

  // Get the contents of the dataset directory and create a list of
  // links to dataset files.
  $directory = file_directory_path() . "/data_export_import/users";
  $mask = '.dataset';
  $files = file_scan_directory($directory, $mask);

  // Sort them by the filename which is used as the key.  Since the
  // files are named using datetime stamps they will be listed in
  // date/time order.
  ksort($files);
  $options = array();
  $options['none'] = "None";
  foreach ($files as $file) {
    $options[$file->basename] = check_plain($file->basename);
  }
  $form['users']['users'] = array(
    '#type' => 'radios',
    '#title' => t('Please select file to import - NB You might want to switch off account activation email notification under the user settings'),
    '#default_value' => 'none',
    '#options' => $options,
  );
  $form['submit'] = array(
    '#type' => 'submit',
    '#value' => 'Import dataset files',
  );
  return $form;
}

/**
 * Function to process form to export users.
 */
function data_export_import_export_users_form_submit($form, &$form_state) {

  // Check that the check box has been ticked.
  if ($form_state['clicked_button']['#post']['users']['users'] != 1) {
    drupal_set_message(t("No option was selected."));
    return TRUE;
  }

  // This is the function which will carry out the exporting of the
  // users to a dataset file.
  $dataset_file_created = data_export_import_export_users_to_file();
  drupal_set_message(t("The following dataset file was created:"));
  drupal_set_message(check_plain($dataset_file_created));
  return TRUE;
}

/**
 * Function to process form to import users.
 */
function data_export_import_import_users_form_submit($form, &$form_state) {
  if ($form_state['values']['users'] != 'none') {
    $result = data_export_import_import_users($form_state['clicked_button']['#post']['users']);
    if ($result) {
      drupal_set_message(t('The Users dataset file %dataset_file was imported.', array(
        '%dataset_file' => $form_state['clicked_button']['#post']['users'],
      )));
      drupal_set_message(t('Any required role settings for the users will now have to be set by using the users administration screen.'));
    }
    else {
      drupal_set_message(t('The Users dataset file %dataset_file was not imported.', array(
        '%dataset_file' => $form_state['clicked_button']['#post']['users'],
      )), 'error');
    }
  }
}

/**
 * Export the required dataset file.
 */
function data_export_import_export_users_to_file() {

  // This will be the main array which will hold the data which will
  // be output to the dataset file.
  $dataset = array();

  // Here will add the list of roles to the dataset - this is so we can check
  // that the receiving instance has a matching set of roles.
  $dataset['roles'] = user_roles();
  $result = db_query("SELECT uid FROM {users}");
  while ($user = db_fetch_object($result)) {

    // We will not put the anonymous or admin users into the file.  These are
    // expected to exist as standard.
    if ($user->uid > 1) {

      // Attach the data for a user to the array.
      $dataset['users'][$user->uid] = user_load($user->uid);
    }
  }

  // Here we will serialize the array to convert it to a string which
  // can then be output to a file.
  $dataset_serialized = serialize($dataset);

  // Create the default directory to hold the datasets.
  $dataset_directory_parent_directory = file_directory_path() . "/data_export_import";
  file_check_directory($dataset_directory_parent_directory, $mode = FILE_CREATE_DIRECTORY);
  $dataset_directory = file_directory_path() . "/data_export_import/users/";
  file_check_directory($dataset_directory, $mode = FILE_CREATE_DIRECTORY);

  // Save the string as a file. This is the dataset data file.
  $file_name = format_date(time(), 'custom', 'Ymd_His') . "_users.dataset";
  $file_path_and_name = $dataset_directory . "/" . $file_name;
  file_save_data($dataset_serialized, $file_path_and_name, FILE_EXISTS_REPLACE);
  return $file_name;
}

/**
 * Make users list match the users from a dataset file.
 *
 * This function is to effectively 'import'  all the users in the dataset file.
 *
 * We will check that the roles on the receiving instance match the roles from
 * the dataset.
 *
 * Then we will bring in the users from the dataset file.
 *
 * NB - When this import has finished the users in the receiving
 * instance should be an exact match with the users in the imported
 * dataset file.  Think in terms of rsync with the '--delete'
 * option. This means that as well as importing new users and updating
 * existing users we need to delete users from the receiving instance
 * which are not in the imported dataset.
 *
 * This synchronisation will be carried out by two passes.
 *
 * First we will loop through the users in the receiving instance  and
 * check against the the imported dataset. If the user exists in the
 * dataset then it will be updated in the receiving instance.  If it
 * doesn't exist in the dataset then it will be deleted from the
 * receiving Drupal instance.
 *
 * The second pass will be a loop through the dataset - any users
 * which are in the dataset but are not in the receiving Drupal
 * instance will be created.
 *
 * This will effectively mean that the users have been sychronised
 * completely.
 *
 * NB - This function will create new users with the same ID as they had
 * originally when they were exported to the dataset file.
 *
 * @param string $file
 *   The dataset file which is being imported.
 *
 * @return bool
 *   TRUE if the import process ran without errors.
 */
function data_export_import_import_users($file) {

  // Load the dataset file into a variable.
  $file_content = file_get_contents(file_directory_path() . "/data_export_import/users/" . $file);

  // Decode the serialized data and store it in an array of objects.
  $file_content_as_array = unserialize($file_content);

  // Here we will pick up if the file contains no users.
  if (empty($file_content_as_array['users'])) {
    drupal_set_message(t("There are no users in the dataset file"), 'error');
    return TRUE;
  }

  // This will check for differences between the roles which were in
  // the exporting instance and the roles in the receiving instance.
  // Currently the import will not set roles for the users imported
  // and the roles will have to be manually set after the import.  An
  // enhancement will be to put a tick box in the options box to say
  // to set the roles to be the same as the roles which were
  // exported. This would require a strict check to be in place RE
  // the roles. This check would have to check that the roles needed
  // exist on the receiving instance and that they have the same ID
  // numbers. If the ID numbers do not match then importing the users
  // will mean that users would be assigned to the wrong roles.
  $current_roles = user_roles();

  // Loop through the roles from the dataset and check there is a corresponding
  // role in the receiving instance.
  foreach ($file_content_as_array['roles'] as $dataset_role_key => $dataset_role_value) {
    if (!isset($current_roles[$dataset_role_key])) {
      drupal_set_message(t('The receiving instance does not have a role with ID of %id and name of %name.', array(
        '%id' => $dataset_role_key,
        '%name' => $dataset_role_value,
      )), 'warning');
    }
    elseif ($current_roles[$dataset_role_key] != $dataset_role_value) {
      drupal_set_message(t('The dataset contains a role with ID of %id and name of %name but the current role with that ID has a name of %current_role_name.', array(
        '%id' => $dataset_role_key,
        '%name' => $dataset_role_value,
        '%current_role_name' => $current_roles[$dataset_role_key],
      )), 'warning');
    }
  }

  // Here we will loop through the users in the receiving instance.
  // For each user we will look into the dataset file being imported.
  // If the user exists in the dataset file then we will check to see
  // if it matches and update it if it doesn't - if it does not exist
  // then we will delete it.
  $result = db_query("SELECT uid FROM {users}");
  while ($user = db_fetch_object($result)) {

    // Ignore the anonymous and default admin users.
    if ($user->uid == 0 || $user->uid == 1) {

      // Go to the start of the while loop.
      continue;
    }

    // See if a user with this ID exists in the dataset file.
    if (!isset($file_content_as_array['users'][$user->uid])) {

      // No user with this ID exists in the dataset file.  Therefore we will
      // delete the current user as we want the current user list to exactly
      // match the list in the dataset.  The current user could well be a test
      // user which was created during development.
      user_delete(array(), $user->uid);
    }
    else {

      // Else if it does see if it is the same and if not then update
      // the current user with the data from the dataset file.
      $current_user = user_load($user->uid);
      $user_from_dataset_file = $file_content_as_array['users'][$user->uid];

      // Here we are going to set the roles back to the basic value of
      // having the authenticated user role only.  This allows for the
      // receiving instance to have different roles and these may be
      // used for different purposes.
      $current_user->roles = array(
        DRUPAL_AUTHENTICATED_RID => 'authenticated user',
      );
      $user_from_dataset_file->roles = array(
        DRUPAL_AUTHENTICATED_RID => 'authenticated user',
      );
      if ($current_user != $user_from_dataset_file) {

        // Current user needs to be updated from the dataset file.
        user_save((object) array(
          'uid' => $user_from_dataset_file->uid,
        ), (array) $user_from_dataset_file);

        // Here we will update the password to match what the was
        // originally output.
        $sql_statement = "UPDATE {users} SET pass= '" . $user_from_dataset_file->pass . "' WHERE uid = " . $user_from_dataset_file->uid;
        update_sql($sql_statement);
      }
    }
  }

  // Here we will loop through all of the users in the dataset file
  // and create any users which are in the dataset file but not in the
  // current users.
  foreach ($file_content_as_array['users'] as $dataset_file_user_value) {
    if (!user_load($dataset_file_user_value->uid)) {
      user_save(NULL, (array) $dataset_file_user_value);

      // Here we will update the password to match what the was
      // originally output.
      $sql_statement = "UPDATE {users} SET pass= '" . $dataset_file_user_value->pass . "' WHERE uid = " . $dataset_file_user_value->uid;
      update_sql($sql_statement);
    }
  }
  return TRUE;
}

Functions

Name	Description
data_export_import_callback_export_users	Callback function to export users.
data_export_import_callback_import_users	Callback function to import users.
data_export_import_export_users_form	Function to create form to export users.
data_export_import_export_users_form_submit	Function to process form to export users.
data_export_import_export_users_to_file	Export the required dataset file.
data_export_import_import_users	Make users list match the users from a dataset file.
data_export_import_import_users_form	Function to create form to import users.
data_export_import_import_users_form_submit	Function to process form to import users.
data_export_import_users_add_items_to_menu	Function to add items to the main menu.