from src.stop import *
import sys
from datetime import datetime
from gmplot import gmplot

if __name__ == "__main__":
    for word in sys.argv[1:]:
        for area in search_stop_by_name(word):
            for stop in show_stops_from_ref(area["ref"])["stop_points"]:
                for route in stop["routes"]:
                    if "Tram" in route["line_human"]:
                        sr = StopRoute(stop["id"], route["line_id"])
                        line = sr.get_line()
                        v = line.get_vehicle(0)
                        if v.is_realtime:
                            print(
                                str(v.wait_time_text)
                                + " ("
                                + datetime.fromtimestamp(v.arrival).strftime("%H:%M")
                                + ") → "
                                + v.destination
                                + " "
                                + str(v.location)
                            )
                        gmap = gmplot.GoogleMapPlotter(v.location[0], v.location[1], 13)
                        gmap.marker(v.location[0], v.location[1], "cornflowerblue")
                        gmap.draw("map.html")