make.py

#!/usr/bin/env python

import re
import sys

from os import unlink
from json import loads, dumps
from glob import glob
from shutil import rmtree
from os.path import join, isdir, expanduser, exists
from collections import defaultdict

if not exists('./configure.py'):
		sys.stderr.write('Error: configure.py does not exist, did you forget to create it from the sample (configure.py.sample)?\n')
		sys.exit(1)
elif exists('./configure.pyc'):
		unlink('./configure.pyc')

from configure import config
from lib.utils import copy_tree

config["path"] = expanduser(config["path"])

def clean():
	if isdir("build"):
		rmtree("build")

	for f in glob("build/*.html"): unlink(f)

def build():
	#copy the toner4tilemill tree to a build dir
	copy_tree("toner4tilemill", "build")

	#remove the mml templates
	for f in glob("build/*.mml"):
		unlink(f)

	#load the project template
	templatefile = open(join('toner4tilemill', 'project.mml'))
	template = loads(templatefile.read())

	#fill in the project template
	for layer in template["Layer"]:
		if layer["id"] in ("land"):
			layer["Datasource"]["file"] = config["processed_p"]
		else:
			# Assume all other layers are PostGIS layers
			for opt, val in config["postgis"].iteritems():
				if (val == ""):
					if (opt in layer["Datasource"]):
						del layer["Datasource"][opt]
				else:
					layer["Datasource"][opt] = val

	template["name"] = config["name"]

	#dump the filled-in project template to the build dir
	with open(join('build', 'project.mml'), 'w') as output:
		output.write(dumps(template, sort_keys=True, indent=2))

def install():
	assert isdir(config["path"]), "Config.path does not point to your mapbox projects directory; please fix and re-run"
	sanitized_name = re.sub("[^\w]", "", config["name"])
	output_dir = join(config["path"], sanitized_name)
	print "installing to %s" % output_dir
	copy_tree("build", output_dir)

def pull():
	#copy the project from mapbox to toner4tilemill
	sanitized_name = re.sub("[^\w]", "", config["name"])
	output_dir = join(config["path"], sanitized_name)
	copy_tree(output_dir, "toner4tilemill", ("layers", ".thumb.png"))

	#load the project file
	project = loads(open(join("toner4tilemill", "project.mml")).read())

	#Make sure we reset postgis data in the project file back to its default values
	defaultconfig = defaultdict(defaultdict)
	defaultconfig["postgis"]["host"]     = ""
	defaultconfig["postgis"]["port"]     = ""
	defaultconfig["postgis"]["dbname"]   = "osm"
	defaultconfig["postgis"]["user"]     = ""
	defaultconfig["postgis"]["password"] = ""
	defaultconfig["postgis"]["extent"] = "-20037508.34 -20037508.34 20037508.34 20037508.34"
	defaultconfig["name"] = "Toner for Tilemill"
	defaultconfig["processed_p"] = "http://tile.openstreetmap.org/processed_p.tar.bz2"

	project["name"] = defaultconfig["name"]
	for layer in project["Layer"]:
		if layer["id"] in ("land"):
			layer["Datasource"]["file"] = defaultconfig["processed_p"]
		else:
			# Assume all other layers are PostGIS layers
			for opt, val in defaultconfig["postgis"].iteritems():
				if val and opt in layer["Datasource"]:
					layer["Datasource"][opt] = val
				elif opt in layer["Datasource"]:
					del layer["Datasource"][opt]

	project_template = open(join('toner4tilemill', 'project.mml'), 'w')
	project_template.write(dumps(project, sort_keys=True, indent=2))

	#now delete project.mml
	unlink(join("toner4tilemill", "project.mml"))

if __name__ == "__main__":
	if sys.argv[-1] == "clean":
		clean()
	elif sys.argv[-1] == "build":
		build()
	elif sys.argv[-1] == "install":
		install()
	elif sys.argv[-1] == "pull":
		pull()
	else:
		clean()
		build()
		install()