Mac/Tools/macfreeze/macmodulefinder.py

   1 """macmodulefinder - Find modules used in a script. Only slightly
   2 mac-specific, really."""
   3
   4 import sys
   5 import os
   6
   7 import directives
   8
   9 try:
  10         # This will work if we are frozen ourselves
  11         import modulefinder
  12 except ImportError:
  13         # And this will work otherwise
  14         _FREEZEDIR=os.path.join(sys.prefix, ':Tools:freeze')
  15         sys.path.insert(0, _FREEZEDIR)
  16         import modulefinder
  17
  18 #
  19 # Modules that must be included, and modules that need not be included
  20 # (but are if they are found)
  21 #
  22 MAC_INCLUDE_MODULES=['site', 'exceptions']
  23 MAC_MAYMISS_MODULES=['posix', 'os2', 'nt', 'ntpath', 'dos', 'dospath',
  24                 'win32api',
  25                 'nturl2path', 'pwd', 'sitecustomize']
  26
  27 # An exception:
  28 Missing="macmodulefinder.Missing"
  29
  30 class Module(modulefinder.Module):
  31
  32         def gettype(self):
  33                 """Return type of module"""
  34                 if self.__path__:
  35                         return 'package'
  36                 if self.__code__:
  37                         return 'module'
  38                 if self.__file__:
  39                         return 'dynamic'
  40                 return 'builtin'
  41
  42 class ModuleFinder(modulefinder.ModuleFinder):
  43
  44     def add_module(self, fqname):
  45         if self.modules.has_key(fqname):
  46             return self.modules[fqname]
  47         self.modules[fqname] = m = Module(fqname)
  48         return m
  49
  50 def process(program, modules=[], module_files = [], debug=0):
  51         error = []
  52         #
  53         # Add the standard modules needed for startup
  54         #
  55         modules = modules + MAC_INCLUDE_MODULES
  56         #
  57         # search the main source for directives
  58         #
  59         extra_modules, exclude_modules, extra_path = \
  60                         directives.findfreezedirectives(program)
  61         for m in extra_modules:
  62                 if os.sep in m:
  63                         # It is a file
  64                         module_files.append(m)
  65                 else:
  66                         modules.append(m)
  67     # collect all modules of the program
  68         path = sys.path[:]
  69         dir = os.path.dirname(program)
  70         path[0] = dir   # "current dir"
  71         path = extra_path + path
  72         #
  73         # Create the module finder and let it do its work
  74         #
  75         modfinder = ModuleFinder(path,
  76                         excludes=exclude_modules, debug=debug)
  77         for m in modules:
  78                 modfinder.import_hook(m)
  79         for m in module_files:
  80                 modfinder.load_file(m)
  81         modfinder.run_script(program)
  82         module_dict = modfinder.modules
  83         #
  84         # Tell the user about missing modules
  85         #
  86         maymiss = exclude_modules + MAC_MAYMISS_MODULES
  87         for m in modfinder.badmodules.keys():
  88                 if not m in maymiss:
  89                         if debug > 0:
  90                                 print 'Missing', m
  91                         error.append(m)
  92         #
  93         # Warn the user about unused builtins
  94         #
  95         for m in sys.builtin_module_names:
  96                 if m in ('__main__', '__builtin__'):
  97                         pass
  98                 elif not module_dict.has_key(m):
  99                         if debug > 0:
 100                                 print 'Unused', m
 101                 elif module_dict[m].gettype() != 'builtin':
 102                         # XXXX Can this happen?
 103                         if debug > 0:
 104                                 print 'Conflict', m
 105         if error:
 106                 raise Missing, error
 107         return module_dict