system/Completer.py

   1 #!/usr/bin/env python
   2 import sys, time
   3 from datetime import datetime, timedelta
   4
   5 import utils
   6
   7 ### more generic code for waiting for any number of things
   8 ### within a given timeframe i.e. given some timeout/silent/period
   9 ### takes in argument a list of tasks that are instances
  10 ### of a CompleterTask subclass
  11 class Completer:
  12     def __init__ (self, tasks, verbose=True):
  13         self.tasks=tasks
  14         self.verbose=verbose
  15     def run (self, timeout_timedelta, silent_timedelta, period=None):
  16         begin = datetime.now()
  17         timeout = begin+timeout_timedelta
  18         timeout_minutes = timeout_timedelta.total_seconds()/60
  19         graceout = datetime.now()+silent_timedelta
  20         silent_minutes = silent_timedelta.total_seconds()/60
  21         period_seconds=int(period.total_seconds())
  22         if self.verbose:
  23             utils.header("max timeout is %d minutes, silent for %d minutes (period is %s s)"%\
  24                              (timeout_minutes,silent_minutes,period_seconds))
  25         tasks=self.tasks
  26         while tasks:
  27             fine=[]
  28             for task in tasks:
  29                 success=task.run (silent=datetime.now() <= graceout)
  30                 if success: fine.append(task)
  31             for task in fine: tasks.remove(task)
  32             if not tasks:
  33                 print "Completer duration = {}".format(int(datetime.now()-begin))
  34                 return True
  35             if datetime.now() > timeout:
  36                 for task in tasks:
  37                     task.failure_epilogue()
  38                 return False
  39             if self.verbose:
  40                 print '%ds..'%period_seconds,
  41             time.sleep(period_seconds)
  42         # in case we're empty
  43         return True
  44
  45
  46 #################### CompleterTask
  47 ### . run(silent)  (return True or False)
  48 ###   silent is an input boolean indicating if we're within the silent period
  49 ### . failure()    (print a message)
  50
  51 ########## expectations (+ first arg self)
  52 # failure()     (to describe which went wrong once it's over)
  53 # -- and --
  54 # run (silent)
  55 # -- or --
  56 # actual_run()
  57 # message()
  58
  59 class CompleterTask:
  60     def run (self, silent):
  61         result=self.actual_run()
  62         if silent:
  63             print '+' if result else '.',
  64             sys.stdout.flush()
  65         else:
  66             print self.message(),"->","OK" if result else "KO"
  67         return result
  68     def message (self): return "you-need-to-redefine-message"
  69     def failure_epilogue (self): print "you-need-to-redefine-failure_epilogue"
  70
  71 # random result
  72 class TaskTest (CompleterTask):
  73     counter=1
  74     def __init__ (self,max):
  75         import random
  76         self.counter=TaskTest.counter
  77         TaskTest.counter+=1
  78         self.delay=random.random()*max
  79         self.fire=datetime.now()+timedelta(seconds=self.delay)
  80     def actual_run(self):
  81         return datetime.now()>=self.fire
  82     def message (self):
  83         return "Task %d - delay was %d s"%(self.counter,self.delay)
  84
  85     def failure_epilogue (self): print "BOTTOM LINE: FAILURE with task (%s)"%self.counter
  86
  87 def main ():
  88     import sys
  89     if len(sys.argv)!=6:
  90         print "Usage: <command> number_tasks max_random timeout_s silent_s period_s"
  91         sys.exit(1)
  92     [number,max,timeout,silent,period]= [ int(x) for x in sys.argv[1:]]
  93     tasks = [ TaskTest(max) for i in range(number)]
  94     success=Completer(tasks,verbose=True).run(timedelta(seconds=timeout),
  95                                               timedelta(seconds=silent),
  96                                               timedelta(seconds=period))
  97     print "OVERALL",success
  98
  99 if __name__ == '__main__':
 100     main()