def getJobForHost(src_hostname): result = [] conn = MongoClient(config.MONGO_SERVER)[config.MONGO_DB] timetable = conn["timetable"].find() hostlist = common.load_hostlist() dt = datetime.datetime.now() for row in timetable: if (src_hostname == row['src']): if (row['hour'] == dt.hour) and (row['minute'] == dt.minute): result.append({'type': row['type'], 'hostname': row['dest']}) return result
def gen_timetable(): timetable = list() hostlist = common.load_hostlist() dt = datetime.datetime.now() for host1 in hostlist: for host2 in hostlist: timetable.append({ 'src': host1, 'dest': host2, 'type': 'iperf', 'hour': dt.hour, 'minute': dt.minute }) timetable.append({ 'src': host1, 'dest': host2, 'type': 'ping', 'hour': dt.hour, 'minute': dt.minute }) dt += datetime.timedelta(minutes=1) return timetable