Index: devel_generate.inc
===================================================================
RCS file: /cvs/drupal-contrib/contributions/modules/devel/devel_generate.inc,v
retrieving revision 1.1.2.13
diff -u -p -r1.1.2.13 devel_generate.inc
--- devel_generate.inc	15 Jun 2008 20:26:16 -0000	1.1.2.13
+++ devel_generate.inc	27 Apr 2009 15:51:10 -0000
@@ -47,7 +47,7 @@ function devel_generate_content($num_nod
   devel_create_nodes($num_nodes, $users, $title_length, $node_types);
   drupal_set_message(t('%num nodes created.', array('%num' => $num_nodes)));
 
-  $nodes = devel_get_nodes($next_nid);
+  $nodes = devel_get_nodes($next_nid, $node_types);
 
   $comments = devel_get_comments();
 
@@ -395,11 +395,13 @@ function devel_get_users() {
   return $users;
 }
 
-function devel_get_nodes($nid = 0) {
+function devel_get_nodes($nid = 0, $node_types = array()) {
   $nodes = array();
-  $result = db_query("SELECT nid, type FROM {node} WHERE type IN ('story', 'blog', 'forum', 'page') AND comment = 2 AND nid > %d", $nid);
-  while($node = db_fetch_object($result)){
-    $nodes[$node->nid] = $node->type ;
+  if (is_array($node_types)) {
+    $result = db_query("SELECT nid, type FROM {node} WHERE type IN ('" . implode('\', \'', $node_types)  . "') AND comment = 2 AND nid > %d", $nid);
+    while($node = db_fetch_object($result)){
+      $nodes[$node->nid] = $node->type ;
+    }
   }
   return $nodes;
 }