p2q2_main.py

# p2q2_main.py v1.0

from utility import *
from p2q2 import *
import copy

# do not submit this file. You may modify this file, but your final solution in p2q1.py should run with this original main file

# replace these parameters with different csv file locations
flags_csv = "./data/flags_m4.csv" # <-- change!!!!
p = 1500                           # <-- change!!!!
v = 1                             # <-- change!!!! 1 (non-cycle) or 2 (cycle)
n = 2                             # <-- change!!!!

flags = list_reader(flags_csv)
flags_dict = generate_flags_dict(flags)

# call your function
your_routes = get_routes(p, v, flags, n) 
print(your_routes) 

# print results or error msg (if any)
err_msg, dist, points = get_dist_and_points_q2(your_routes, flags_dict, v, n)

if err_msg !=None:
  print("Error : " + err_msg)
else:  
  print("Points for all routes :" + str(points) + ", p :" + str(p))
  
  if points < p:
    print("Error : points is lesser than p!")
  else:
    print("Quality score (dist for all routes) for q2 is : " + str(dist)) # smaller score is better