Added get_nb_commands_to_run() to command_manager

Author: mgermain

scripts/smart_dispatch.py

@@ -59,6 +59,7 @@ def main():
             command_manager.set_commands_to_run(commands)
         else:
             command_manager.reset_running_commands()
+            nb_commands = command_manager.get_nb_commands_to_run()
 
         worker_command = 'smart_worker.py "{0}" "{1}"'.format(command_manager._commands_filename, path_job_logs)
         # Replace commands with `args.pool` workers

smartdispatch/command_manager.py

@@ -36,6 +36,10 @@ def get_command_to_run(self):
                 self._move_line_between_files(commands_file, running_commands_file, command)
         return command[:-1]
 
+    def get_nb_commands_to_run(self):
+        with utils.open_with_lock(self._commands_filename, 'r') as commands_file:
+            return len(commands_file.readline())
+
     def set_running_command_as_finished(self, command):
         with utils.open_with_lock(self._running_commands_filename, 'r+') as running_commands_file:
             with utils.open_with_lock(self._finished_commands_filename, 'a') as finished_commands_file:

smartdispatch/tests/test_command_manager.py

@@ -55,6 +55,9 @@ def test_get_command_to_run(self):
 
         assert_true(not os.path.isfile(self.command_manager._finished_commands_filename))
 
+    def test_get_nb_commands_to_run(self):
+        assert_equal(self.command_manager.get_nb_commands_to_run(), 3)
+
     def test_set_running_command_as_finished(self):
         # SetUp
         command = self.command_manager.get_command_to_run()