Mac/Tools/macfreeze/directives.py

   1 import re
   2 import os
   3
   4 # The regular expression for freeze directives. These are comments with the
   5 # word macfreeze immedeately followed by a colon, followed by a directive,
   6 # followed by argument(s)
   7 #
   8 # The directives supported are
   9 # include - Include a module or file
  10 # exclude - Exclude a module
  11 # path - Add sys.path entries. Relative paths are relative to the source file.
  12 #
  13 # See the macfreeze.py main program for a real live example.
  14 #
  15 DIRECTIVE_RE=r'^\s*#\s*macfreeze:\s*(\S*)\s*(.*)\s*$'
  16 REPROG=re.compile(DIRECTIVE_RE)
  17
  18 def findfreezedirectives(program):
  19         extra_modules = []
  20         exclude_modules = []
  21         extra_path = []
  22         progdir, filename = os.path.split(program)
  23         fp = open(program)
  24         for line in fp.readlines():
  25                 match = REPROG.match(line)
  26                 if match:
  27                         directive = match.group(1)
  28                         argument = match.group(2)
  29                         if directive == 'include':
  30                                 extra_modules.append(argument)
  31                         elif directive == 'exclude':
  32                                 exclude_modules.append(argument)
  33                         elif directive == 'path':
  34                                 argument = os.path.join(progdir, argument)
  35                                 extra_path.append(argument)
  36                         else:
  37                                 print '** Unknown directive', line
  38         return extra_modules, exclude_modules, extra_path
  39