trackhelper.py

#!/usr/bin/python2.7

import Data;
import Graphics;
import Filefunc;
import Service;

import svgwrite;

Filefunc.LoadData();

commands = "[<h>elp] [<?>], [<l>ink], [<u>nlink], [<i>nsert node], [<v>erify nodes and trains], [<r>eload], [<s>ave], [<e>xport to SVG], [<q>uit]";

print("Enter command:");
print(commands);

while True:
	command = raw_input();
	print(command);
	if command == "" or command[0] == "h" or command[0] == "?":
		print("Commands:");
		print(commands);
	elif command[0] == "l":
		print("Node 1:");
		begin = int(raw_input());
		print("Node 2:");
		end = int(raw_input());
		Data.AddLink(begin, end);
	elif command[0] == "u":
		print("Node 1:");
		begin = int(raw_input());
		print("Node 2:");
		end = int(raw_input());
		try:
			Data.links[begin].remove(end);
			Data.links[end].remove(begin);
		except KeyError:
			pass;
	elif command[0] == "i":
		print("Node 1:");
		begin = int(raw_input());
		print("Node 2:");
		end = int(raw_input());
		print("New node:");
		new = int(raw_input());
		try:
			Data.links[begin].remove(end);
			Data.links[end].remove(begin);
			Data.links[begin].append(new);
			Data.links[new].append(begin);
			Data.links[new].append(end);
			Data.links[end].append(new);
		except KeyError:
			pass;
	elif command[0] == "v":
		# find all unlinked nodes
		for node in Data.nodes:
			if node not in Data.links:
				print("Node {} ({}) has no links".format(node, Data.NodeName(node)));
		
		# find non-existent platforms
		for name in Service.services:
			for stop in Service.services[name].orders:
				if stop[1] not in Data.places:
					print("{} stops at non-existent station {}".format(service, stop[1]));
	elif command[0] == "r":
		print("Sure?");
		print("[<y>es], [<n>o]");
		command = raw_input();
		if command[0] == "y":
			Filefunc.LoadData();
	elif command[0] == "s":
		Filefunc.SaveData();
	elif command[0] == "e":
		drawing = svgwrite.Drawing("track.svg", profile="full");
		for begin in Data.links:
			for end in Data.links[begin]:
				if begin < end:
					path = drawing.path();
					path.push("M");
					path.push(Graphics.SVGPos(Data.nodes[begin].pos));
					path.push(Graphics.SVGPos(Data.nodes[end].pos));
					drawing.add(path.stroke('black', width=0.01));
		drawing.save();
		print("Done.");
	elif command[0] == "q":
		print("Sure?");
		print("[<y>es], [<n>o]");
		command = raw_input();
		if command[0] == "y":
			break;