function drush_nodequeue_generate in Nodequeue 7.2

Same name and namespace in other branches

1. 7.3 nodequeue_generate.drush.inc \drush_nodequeue_generate()

Re-populates specified nodequeues with random nodes.

File

./nodequeue_generate.drush.inc, line 58

Nodequeue generate drush integration.

Code

function drush_nodequeue_generate() {
  $args = func_get_args();

  // At least one queue must be specified.
  if (count($args) < 1) {
    drush_set_error('error', dt('At least one queue must be specified.'));
  }

  // Get qids from machine names.
  $qids = array();
  foreach ($args as $queue) {
    $qid = db_select('nodequeue_queue', 'nq')
      ->fields('nq', array(
      'qid',
    ))
      ->condition('name', $queue)
      ->execute()
      ->fetchField();
    if ($qid) {
      $qids[] = $qid;
    }
    else {
      drush_set_error('error', dt('Queue @queue was not found.', array(
        '@queue' => $queue,
      )));
    }
  }
  nodequeue_generate_rehash();
  nodequeue_generate_repopulate_queues($qids);
}