#!/usr/bin/env python
def read_file(filename):
with open(filename, 'r') as f:
for line in f:
yield line.strip()
def main():
graph = dict()
for line in read_file("connections.txt"):
tokens = line.split(" ")
tokens[0] = int(tokens[0])
tokens[1] = int(tokens[1])
# add vertex 1
if tokens[0] not in graph:
graph[tokens[0]] = set()
graph[tokens[0]].add(tokens[2])
# add vertex 2
if tokens[1] not in graph:
graph[tokens[1]] = set()
graph[tokens[1]].add(tokens[2])
for (k, v) in graph.items():
# svg base data
color_circle_up="yellow"
color_circle_down="yellow"
color_rectangle="white"
color_number="black"
if 'bus' in v:
color_circle_down="green"
if 'underground' in v:
color_rectangle="red"
color_number="white"
with open(f"../data/{k}.svg", "w") as f:
f.write("')
if __name__ == "__main__":
main()
'''
'''