From 64800c3741b7e3c75d25c92ba06d2da9f3a2b426 Mon Sep 17 00:00:00 2001
From: Johannes Findeisen <you@hanez.org>
Date: Sat, 23 Nov 2013 20:18:38 +0100
Subject: [PATCH] added progess in percent while scheduling jobs... yay... :)

---
 bin/linspector | 26 +++++++++++++++++++++-----
 1 file changed, 21 insertions(+), 5 deletions(-)

diff --git a/bin/linspector b/bin/linspector
index da8b8ba..048ae0d 100755
--- a/bin/linspector
+++ b/bin/linspector
@@ -20,7 +20,7 @@ along with this program. If not, see <http://www.gnu.org/licenses/>.
 """
 
 
-__version__ = "0.14/AMNESIA"
+__version__ = "0.15/AMNESIA"
 
 import argparse
 import datetime
@@ -28,6 +28,7 @@ import logging
 import logging.handlers
 import os
 import os.path as path
+import sys
 
 from linspector.backends.jsonrpc import JsonrpcBackend
 from linspector.config.parser import FullConfigParser
@@ -116,21 +117,36 @@ def main():
         logger.error(msg)
         exit()
 
-    print("Scheduling jobs. Be patient...")
-
     scheduler = Scheduler({"apscheduler.threadpool.core_threads": args.corethreads,
                            "apscheduler.threadpool.max_threads": args.threads})
     scheduler.start()
     TaskExecutor.Instance()
 
+    job_count = 0
+    for layout in linConf.get_enabled_layouts():
+        for hostgroup in layout.get_hostgroups():
+            job_count += (hostgroup.get_services().__len__() * hostgroup.get_hosts().__len__())
+
     start_date = datetime.datetime.now()
     time_delta = 0
     jobs = []
+    count = 0
+    percent = 0
     for layout in linConf.get_enabled_layouts():
         for hostgroup in layout.get_hostgroups():
             for service in hostgroup.get_services():
                 for host in hostgroup.get_hosts():
                     for period in service.get_periods():
+                        count += 1
+                        if job_count >= 100:
+                            if count % (job_count/100) == 0:
+                                percent += 1
+                                sys.stdout.write("Scheduling jobs. Be patient... " + str(percent) + "%\r")
+                                sys.stdout.flush()
+                        else:
+                            sys.stdout.write("Scheduling jobs. Be patient...\r")
+                            sys.stdout.flush()
+
                         time_delta += float(args.delay)
                         new_start_date = start_date + datetime.timedelta(seconds=time_delta)
                         job = Job(service,
@@ -143,6 +159,8 @@ def main():
                             job.set_job(scheduler_job)
                             jobs.append(job)
 
+    print("\nScheduled " + str(job_count) + " jobs")
+
     interface = LinspectorInterface(jobs, scheduler, linConf)
 
     if "jsonrpc_backend" in core and core["jsonrpc_backend"]:
@@ -150,8 +168,6 @@ def main():
         jsonrpc.daemon = True
         jsonrpc.start()
 
-    print("Ready!")
-
     print("Type \"help\" to view available commands or \"help COMMAND\" to view the commands help")
 
     LishFrontend(interface)