stage3.py

#!/usr/bin/env python
import sys
import re
#from PIL import Image, ImageDraw


def get_station_pixel_points(station):
	try:
		pixels = open("station_pixel_location.txt", "r")
	except IOError as e:
		print("ERROR: 'station_pixel_location.txt' file not found.")
		del e
		sys.exit()

	points = []

	for l in pixels:
		if not l:  # skip blank lines
			continue
		l = l.lower()  # convert to lowercase
		l = re.findall('"([^"]*)"', l)  # extract/separate info

		if not l:  # skip any blank lines (line 1)
			continue

		# If station number found:
		if l[0] == station:
			for i in range(1, 5):
				# Save array of pixel points for text line:
				points.append(int(l[i]))
			break

	pixels.close()
	return points


def draw_pic(station):
	# Load metro image to modify:
	im = Image.open('assets/metro.png')

	# Read pixel point coordinates from text file for drawing:
	p = get_station_pixel_points(station)

	if not p:
		# If p is empty, station not in pixel point file was picked:
		print('ERROR: Unsupported station selected.\nOnly Eltham, Sunbury and Werribee implemented.')
	else:
		# Draw box around station:
		dr = ImageDraw.Draw(im)
		dr.rectangle(((p[0], p[1]), (p[2], p[3])), outline="red")
		del dr

	# Save new image:
	im.save('assets/metroModified.png', "PNG")