Clean up the task initialization for rebuild thumbs and resizes.

gallery · Sep 28, 2009 · ab5e680 · ab5e680
1 parent c97a885
commit ab5e680
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/modules/gallery/helpers/gallery_task.php b/modules/gallery/helpers/gallery_task.php
@@ -48,11 +48,13 @@ static function rebuild_dirty_images($task) {
     $errors = array();
     try {
       $result = graphics::find_dirty_images_query();
-      $total_count = $task->get("total_count", -1);
-      if ($total_count < 0) {
-        $total_count = $result->count();
+      $total_count = $task->get("total_count", $result->count());
+      $mode = $task->get("mode", "init");
+      if ($mode == "init") {
         $task->set("total_count", $total_count);
+        $task->get("mode", "process");
       }
+
       $completed = $task->get("completed", 0);
       $ignored = $task->get("ignored", array());