Mac/Tools/macfreeze/directives.py

import re
import os

# The regular expression for freeze directives. These are comments with the
# word macfreeze immedeately followed by a colon, followed by a directive,
# followed by argument(s)
#
# The directives supported are
# include - Include a module or file
# exclude - Exclude a module
# optional - Include a module if it is found, but don't complain if it isn't
# path - Add sys.path entries. Relative paths are relative to the source file.
#
# See the macfreeze.py main program for a real live example.
#
DIRECTIVE_RE=r'^\s*#\s*macfreeze:\s*(\S*)\s*(.*)\s*$'
REPROG=re.compile(DIRECTIVE_RE)

def findfreezedirectives(program):
	extra_modules = []
	exclude_modules = []
	optional_modules = []
	extra_path = []
	progdir, filename = os.path.split(program)
	fp = open(program)
	for line in fp.readlines():
		match = REPROG.match(line)
		if match:
			directive = match.group(1)
			argument = match.group(2)
			if directive == 'include':
				extra_modules.append(argument)
			elif directive == 'exclude':
				exclude_modules.append(argument)
			elif directive == 'optional':
				optional_modules.append(argument)
			elif directive == 'path':
				argument = os.path.join(progdir, argument)
				extra_path.append(argument)
			else:
				print '** Unknown directive', line
	return extra_modules, exclude_modules, optional_modules, extra_path