def delete_all_directories(step): x = rglob.rglob(world.root, "*") x.sort(reverse=True) for d in x: if os.path.isdir(d): os.rmdir(d) else: os.remove(d) del world.dirs world.dirs = [] del world.known_sizes world.known_sizes = {}
def parseFiles(): # (bufsz, protocol, src, dest, flow_id, time, throughput) res = [] # (bufsz, time, fairness) res2 = [] for file_name in rglob.rglob(".", "*.txt"): with open(file_name) as fp: lines = fp.readlines() bufsz = int(lines[0].split()[-2]) lines = lines[3:] while len(lines) > 0: while(len(lines[0].split()) == 0): lines = lines[1:] time = int(lines[0].split()[-2]) lines = lines[3:] if time >= 35: limit = 10 else: limit = 8 for i in range(limit): vals = lines[i].split() flowid = int(vals[0]) protocol = vals[1] src = vals[2] dest = vals[3] throughput = float(vals[4]) res.append((bufsz, protocol, src, dest, flowid, time, throughput)) lines = lines[limit + 1:] fairness = float(lines[0].split()[-1]) res2.append((bufsz, time, fairness)) lines = lines[3:] return res, res2
def find_files(step, expected_num_of_files, file_type): x = rglob.rglob(world.root, "*{}".format(file_type)) assert len(x) == int(expected_num_of_files), \ "Found {} files".format(len(x))
def find_directories(step, expected_num_of_dirs): x = rglob.rglob(world.root, "*_rglob") assert len(x) == int(expected_num_of_dirs), \ "Found {} directories".format(len(x))