script.py

# collect the common function
# import script

def get_traffic_volume(traffic_file):
    # only support "cross" and "synthetic"
    if "cross" in traffic_file:
        sta = traffic_file.find("equal_") + len("equal_")
        end = traffic_file.find(".xml")
        return int(traffic_file[sta:end])
    # elif "synthetic" in traffic_file:
    #     traffic_file_list = traffic_file.split("-")
    #     volume_list = []
    #     for i in range(2, 6):
    #         volume_list.append(int(traffic_file_list[i][2:]))
    #
    #     vol = min(max(volume_list[0:2]), max(volume_list[2:]))
    #
    #     return int(vol/100)*100
    elif "anon" in traffic_file:
        return int(traffic_file.split('_')[3].split('.')[0])
        # sta = traffic_file.find("flow_1_1_") + len("flow_1_1_")
        # end = traffic_file.find(".json")
        # return int(traffic_file[sta:end])