addmint.py

#!/usr/bin/env python
# encoding: utf-8
"""
addmint.py

Adds Mint analytics to a static HTML page; requires BeautifulSoup.

(I use this for Aperture exports.)

Created by Will Benton on 2008-07-15.
Copyright (c) 2008 Aition Technologies, LLC and Will Benton. All rights reserved.
"""

import sys
import getopt
from BeautifulSoup import BeautifulSoup, Tag, Comment

help_message = '''
The help message goes here.
'''
# edit this next line or else pass the --url option
minturl = "http://YOURSERVER.com/mint/"
files = []

class Usage(Exception):
	def __init__(self, msg):
		self.msg = msg

def process(filename):
	global minturl
	
	f = open(filename)
	lines = f.read()
	
	soup = BeautifulSoup(''.join(lines))
	
	head = soup.head
	mtag = Tag(soup, "script")
	mtag["src"] = "%s/?js" % minturl
	mtag["type"] = "text/javascript"
	
	head.insert(0, mtag)
	
	f.close()
	
	f = open(filename, "w")
	f.write(soup.prettify())
	f.close()

def main(argv=None):
	global minturl, files
	
	if argv is None:
		argv = sys.argv
	try:
		try:
			opts, args = getopt.getopt(argv[1:], "hu:v", ["help", "url="])
		except getopt.error, msg:
			raise Usage(msg)
	
		# option processing
		for option, value in opts:
			if option == "-v":
				verbose = True
			if option in ("-h", "--help"):
				raise Usage(help_message)
			if option in ("-u", "--url"):
				minturl = value
		
		for f in args:
			process(f)
	
	except Usage, err:
		print >> sys.stderr, sys.argv[0].split("/")[-1] + ": " + str(err.msg)
		print >> sys.stderr, "\t for help use --help"
		return 2


if __name__ == "__main__":
	sys.exit(main())