devel_generate_batch.inc 2.0 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768
  1. <?php
  2. /**
  3. * Devel Generate batch handling functions using the BatchAPI.
  4. */
  5. /**
  6. * Form constructor generating in batches.
  7. */
  8. function devel_generate_batch_content($form_state) {
  9. $operations = array();
  10. // Setup the batch operations and save the variables.
  11. $operations[] = array('devel_generate_batch_content_pre_node', array($form_state['values']));
  12. // Add the kill operation.
  13. if ($form_state['values']['kill_content']) {
  14. $operations[] = array('devel_generate_batch_content_kill', array());
  15. }
  16. // Add the operations to create the nodes.
  17. for ($num = 0; $num < $form_state['values']['num_nodes']; $num ++) {
  18. $operations[] = array('devel_generate_batch_content_add_node', array());
  19. }
  20. // Start the batch.
  21. $batch = array(
  22. 'title' => t('Generating Content'),
  23. 'operations' => $operations,
  24. 'finished' => 'devel_generate_batch_finished',
  25. 'file' => drupal_get_path('module', 'devel_generate') . '/devel_generate_batch.inc',
  26. );
  27. batch_set($batch);
  28. }
  29. function devel_generate_batch_content_kill(&$context) {
  30. module_load_include('inc', 'devel_generate', 'devel_generate');
  31. devel_generate_content_kill($context['results']);
  32. }
  33. function devel_generate_batch_content_pre_node($vars, &$context) {
  34. $context['results'] = $vars;
  35. $context['results']['num_nids'] = 0;
  36. module_load_include('inc', 'devel_generate', 'devel_generate');
  37. devel_generate_content_pre_node($context['results']);
  38. }
  39. /**
  40. * Batch API callback: Generate nodes.
  41. */
  42. function devel_generate_batch_content_add_node(&$context) {
  43. module_load_include('inc', 'devel_generate', 'devel_generate');
  44. devel_generate_content_add_node($context['results']);
  45. $context['results']['num_nids'] ++;
  46. }
  47. /**
  48. * Display a message when a batch is complete.
  49. */
  50. function devel_generate_batch_finished($success, $results, $operations) {
  51. if ($success) {
  52. $message = t('Finished @num_nids nodes created successfully.', array('@num_nids' => $results['num_nids']));
  53. }
  54. else {
  55. $message = t('Finished with an error.');
  56. }
  57. drupal_set_message($message);
  58. }