blob: b648f24eb7e09c5595634e19115ee6863b2d0a92 [file] [log] [blame]
Greg Wardfe6462c2000-04-04 01:40:52 +00001"""distutils.dist
2
3Provides the Distribution class, which represents the module distribution
Greg Ward8ff5a3f2000-06-02 00:44:53 +00004being built/installed/distributed.
5"""
Greg Wardfe6462c2000-04-04 01:40:52 +00006
7# created 2000/04/03, Greg Ward
8# (extricated from core.py; actually dates back to the beginning)
9
10__revision__ = "$Id$"
11
Gregory P. Smith14263542000-05-12 00:41:33 +000012import sys, os, string, re
Greg Wardfe6462c2000-04-04 01:40:52 +000013from types import *
14from copy import copy
15from distutils.errors import *
Greg Ward2f2b6c62000-09-25 01:58:07 +000016from distutils.fancy_getopt import FancyGetopt, translate_longopt
Andrew M. Kuchlinga7210ed2001-03-22 03:06:52 +000017from distutils.util import check_environ, strtobool, rfc822_escape
Greg Wardfe6462c2000-04-04 01:40:52 +000018
19
20# Regex to define acceptable Distutils command names. This is not *quite*
21# the same as a Python NAME -- I don't allow leading underscores. The fact
22# that they're very similar is no coincidence; the default naming scheme is
23# to look for a Python module named after the command.
24command_re = re.compile (r'^[a-zA-Z]([a-zA-Z0-9_]*)$')
25
26
27class Distribution:
Greg Ward8ff5a3f2000-06-02 00:44:53 +000028 """The core of the Distutils. Most of the work hiding behind 'setup'
29 is really done within a Distribution instance, which farms the work out
30 to the Distutils commands specified on the command line.
Greg Wardfe6462c2000-04-04 01:40:52 +000031
Greg Ward8ff5a3f2000-06-02 00:44:53 +000032 Setup scripts will almost never instantiate Distribution directly,
33 unless the 'setup()' function is totally inadequate to their needs.
34 However, it is conceivable that a setup script might wish to subclass
35 Distribution for some specialized purpose, and then pass the subclass
36 to 'setup()' as the 'distclass' keyword argument. If so, it is
37 necessary to respect the expectations that 'setup' has of Distribution.
38 See the code for 'setup()', in core.py, for details.
39 """
Greg Wardfe6462c2000-04-04 01:40:52 +000040
41
42 # 'global_options' describes the command-line options that may be
Greg Ward82715e12000-04-21 02:28:14 +000043 # supplied to the setup script prior to any actual commands.
44 # Eg. "./setup.py -n" or "./setup.py --quiet" both take advantage of
Greg Wardfe6462c2000-04-04 01:40:52 +000045 # these global options. This list should be kept to a bare minimum,
46 # since every global option is also valid as a command option -- and we
47 # don't want to pollute the commands with too many options that they
48 # have minimal control over.
Greg Wardd5d8a992000-05-23 01:42:17 +000049 global_options = [('verbose', 'v', "run verbosely (default)"),
50 ('quiet', 'q', "run quietly (turns verbosity off)"),
51 ('dry-run', 'n', "don't actually do anything"),
52 ('help', 'h', "show detailed help message"),
Greg Wardfe6462c2000-04-04 01:40:52 +000053 ]
Greg Ward82715e12000-04-21 02:28:14 +000054
55 # options that are not propagated to the commands
56 display_options = [
57 ('help-commands', None,
58 "list all available commands"),
59 ('name', None,
60 "print package name"),
61 ('version', 'V',
62 "print package version"),
63 ('fullname', None,
64 "print <package name>-<version>"),
65 ('author', None,
66 "print the author's name"),
67 ('author-email', None,
68 "print the author's email address"),
69 ('maintainer', None,
70 "print the maintainer's name"),
71 ('maintainer-email', None,
72 "print the maintainer's email address"),
73 ('contact', None,
Greg Wardd5d8a992000-05-23 01:42:17 +000074 "print the maintainer's name if known, else the author's"),
Greg Ward82715e12000-04-21 02:28:14 +000075 ('contact-email', None,
Greg Wardd5d8a992000-05-23 01:42:17 +000076 "print the maintainer's email address if known, else the author's"),
Greg Ward82715e12000-04-21 02:28:14 +000077 ('url', None,
78 "print the URL for this package"),
Greg Ward82715e12000-04-21 02:28:14 +000079 ('license', None,
Andrew M. Kuchlingfa7dc572001-08-10 18:49:23 +000080 "print the license of the package"),
81 ('licence', None,
82 "alias for --license"),
Greg Ward82715e12000-04-21 02:28:14 +000083 ('description', None,
84 "print the package description"),
Greg Warde5a584e2000-04-26 02:26:55 +000085 ('long-description', None,
86 "print the long package description"),
Andrew M. Kuchlinga7210ed2001-03-22 03:06:52 +000087 ('platforms', None,
88 "print the list of platforms"),
89 ('keywords', None,
90 "print the list of keywords"),
Greg Ward82715e12000-04-21 02:28:14 +000091 ]
Greg Ward2f2b6c62000-09-25 01:58:07 +000092 display_option_names = map(lambda x: translate_longopt(x[0]),
93 display_options)
Greg Ward82715e12000-04-21 02:28:14 +000094
95 # negative options are options that exclude other options
Greg Wardfe6462c2000-04-04 01:40:52 +000096 negative_opt = {'quiet': 'verbose'}
97
98
99 # -- Creation/initialization methods -------------------------------
Fred Drakeb94b8492001-12-06 20:51:35 +0000100
Greg Wardfe6462c2000-04-04 01:40:52 +0000101 def __init__ (self, attrs=None):
102 """Construct a new Distribution instance: initialize all the
Greg Ward8ff5a3f2000-06-02 00:44:53 +0000103 attributes of a Distribution, and then use 'attrs' (a dictionary
104 mapping attribute names to values) to assign some of those
105 attributes their "real" values. (Any attributes not mentioned in
106 'attrs' will be assigned to some null value: 0, None, an empty list
107 or dictionary, etc.) Most importantly, initialize the
108 'command_obj' attribute to the empty dictionary; this will be
109 filled in with real command objects by 'parse_command_line()'.
110 """
Greg Wardfe6462c2000-04-04 01:40:52 +0000111
112 # Default values for our command-line options
113 self.verbose = 1
114 self.dry_run = 0
Greg Wardfe6462c2000-04-04 01:40:52 +0000115 self.help = 0
Greg Ward82715e12000-04-21 02:28:14 +0000116 for attr in self.display_option_names:
117 setattr(self, attr, 0)
Greg Wardfe6462c2000-04-04 01:40:52 +0000118
Greg Ward82715e12000-04-21 02:28:14 +0000119 # Store the distribution meta-data (name, version, author, and so
120 # forth) in a separate object -- we're getting to have enough
121 # information here (and enough command-line options) that it's
122 # worth it. Also delegate 'get_XXX()' methods to the 'metadata'
123 # object in a sneaky and underhanded (but efficient!) way.
Greg Wardfd7b91e2000-09-26 01:52:25 +0000124 self.metadata = DistributionMetadata()
Neil Schemenauera8aefe52001-09-03 15:47:21 +0000125 for basename in self.metadata._METHOD_BASENAMES:
Greg Ward4982f982000-04-22 02:52:44 +0000126 method_name = "get_" + basename
127 setattr(self, method_name, getattr(self.metadata, method_name))
Greg Wardfe6462c2000-04-04 01:40:52 +0000128
129 # 'cmdclass' maps command names to class objects, so we
130 # can 1) quickly figure out which class to instantiate when
131 # we need to create a new command object, and 2) have a way
Greg Ward82715e12000-04-21 02:28:14 +0000132 # for the setup script to override command classes
Greg Wardfe6462c2000-04-04 01:40:52 +0000133 self.cmdclass = {}
134
Greg Ward9821bf42000-08-29 01:15:18 +0000135 # 'script_name' and 'script_args' are usually set to sys.argv[0]
136 # and sys.argv[1:], but they can be overridden when the caller is
137 # not necessarily a setup script run from the command-line.
138 self.script_name = None
139 self.script_args = None
140
Greg Wardd5d8a992000-05-23 01:42:17 +0000141 # 'command_options' is where we store command options between
142 # parsing them (from config files, the command-line, etc.) and when
143 # they are actually needed -- ie. when the command in question is
144 # instantiated. It is a dictionary of dictionaries of 2-tuples:
145 # command_options = { command_name : { option : (source, value) } }
Gregory P. Smith14263542000-05-12 00:41:33 +0000146 self.command_options = {}
147
Greg Wardfe6462c2000-04-04 01:40:52 +0000148 # These options are really the business of various commands, rather
149 # than of the Distribution itself. We provide aliases for them in
150 # Distribution as a convenience to the developer.
Greg Wardfe6462c2000-04-04 01:40:52 +0000151 self.packages = None
152 self.package_dir = None
153 self.py_modules = None
154 self.libraries = None
Greg Ward51def7d2000-05-27 01:36:14 +0000155 self.headers = None
Greg Wardfe6462c2000-04-04 01:40:52 +0000156 self.ext_modules = None
157 self.ext_package = None
158 self.include_dirs = None
159 self.extra_path = None
Gregory P. Smithb2e3bb32000-05-12 00:52:23 +0000160 self.scripts = None
Gregory P. Smith6a901dd2000-05-13 03:09:50 +0000161 self.data_files = None
Greg Wardfe6462c2000-04-04 01:40:52 +0000162
163 # And now initialize bookkeeping stuff that can't be supplied by
164 # the caller at all. 'command_obj' maps command names to
165 # Command instances -- that's how we enforce that every command
166 # class is a singleton.
167 self.command_obj = {}
168
169 # 'have_run' maps command names to boolean values; it keeps track
170 # of whether we have actually run a particular command, to make it
171 # cheap to "run" a command whenever we think we might need to -- if
172 # it's already been done, no need for expensive filesystem
173 # operations, we just check the 'have_run' dictionary and carry on.
174 # It's only safe to query 'have_run' for a command class that has
175 # been instantiated -- a false value will be inserted when the
176 # command object is created, and replaced with a true value when
Greg Ward612eb9f2000-07-27 02:13:20 +0000177 # the command is successfully run. Thus it's probably best to use
Greg Wardfe6462c2000-04-04 01:40:52 +0000178 # '.get()' rather than a straight lookup.
179 self.have_run = {}
180
181 # Now we'll use the attrs dictionary (ultimately, keyword args from
Greg Ward82715e12000-04-21 02:28:14 +0000182 # the setup script) to possibly override any or all of these
183 # distribution options.
184
Greg Wardfe6462c2000-04-04 01:40:52 +0000185 if attrs:
186
187 # Pull out the set of command options and work on them
188 # specifically. Note that this order guarantees that aliased
189 # command options will override any supplied redundantly
190 # through the general options dictionary.
Greg Wardfd7b91e2000-09-26 01:52:25 +0000191 options = attrs.get('options')
Greg Wardfe6462c2000-04-04 01:40:52 +0000192 if options:
193 del attrs['options']
194 for (command, cmd_options) in options.items():
Greg Ward0e48cfd2000-05-26 01:00:15 +0000195 opt_dict = self.get_option_dict(command)
196 for (opt, val) in cmd_options.items():
197 opt_dict[opt] = ("setup script", val)
Greg Wardfe6462c2000-04-04 01:40:52 +0000198
199 # Now work on the rest of the attributes. Any attribute that's
200 # not already defined is invalid!
201 for (key,val) in attrs.items():
Greg Wardfd7b91e2000-09-26 01:52:25 +0000202 if hasattr(self.metadata, key):
203 setattr(self.metadata, key, val)
204 elif hasattr(self, key):
205 setattr(self, key, val)
Greg Wardfe6462c2000-04-04 01:40:52 +0000206 else:
Greg Ward02a1a2b2000-04-15 22:15:07 +0000207 raise DistutilsSetupError, \
Greg Wardfe6462c2000-04-04 01:40:52 +0000208 "invalid distribution option '%s'" % key
209
Andrew M. Kuchlinga7210ed2001-03-22 03:06:52 +0000210 self.finalize_options()
Fred Drakeb94b8492001-12-06 20:51:35 +0000211
Greg Wardfe6462c2000-04-04 01:40:52 +0000212 # __init__ ()
213
214
Greg Ward0e48cfd2000-05-26 01:00:15 +0000215 def get_option_dict (self, command):
216 """Get the option dictionary for a given command. If that
217 command's option dictionary hasn't been created yet, then create it
218 and return the new dictionary; otherwise, return the existing
219 option dictionary.
220 """
221
222 dict = self.command_options.get(command)
223 if dict is None:
224 dict = self.command_options[command] = {}
225 return dict
226
227
Greg Wardc32d9a62000-05-28 23:53:06 +0000228 def dump_option_dicts (self, header=None, commands=None, indent=""):
229 from pprint import pformat
230
231 if commands is None: # dump all command option dicts
232 commands = self.command_options.keys()
233 commands.sort()
234
235 if header is not None:
236 print indent + header
237 indent = indent + " "
238
239 if not commands:
240 print indent + "no commands known yet"
241 return
242
243 for cmd_name in commands:
244 opt_dict = self.command_options.get(cmd_name)
245 if opt_dict is None:
246 print indent + "no option dict for '%s' command" % cmd_name
247 else:
248 print indent + "option dict for '%s' command:" % cmd_name
249 out = pformat(opt_dict)
250 for line in string.split(out, "\n"):
251 print indent + " " + line
252
253 # dump_option_dicts ()
Fred Drakeb94b8492001-12-06 20:51:35 +0000254
Greg Wardc32d9a62000-05-28 23:53:06 +0000255
256
Greg Wardd5d8a992000-05-23 01:42:17 +0000257 # -- Config file finding/parsing methods ---------------------------
258
Gregory P. Smith14263542000-05-12 00:41:33 +0000259 def find_config_files (self):
260 """Find as many configuration files as should be processed for this
261 platform, and return a list of filenames in the order in which they
262 should be parsed. The filenames returned are guaranteed to exist
263 (modulo nasty race conditions).
264
Andrew M. Kuchlingd303b612001-12-06 16:32:05 +0000265 There are three possible config files: distutils.cfg in the
266 Distutils installation directory (ie. where the top-level
267 Distutils __inst__.py file lives), a file in the user's home
268 directory named .pydistutils.cfg on Unix and pydistutils.cfg
269 on Windows/Mac, and setup.cfg in the current directory.
Greg Wardd5d8a992000-05-23 01:42:17 +0000270 """
Gregory P. Smith14263542000-05-12 00:41:33 +0000271 files = []
Greg Wardacf3f6a2000-06-07 02:26:19 +0000272 check_environ()
Gregory P. Smith14263542000-05-12 00:41:33 +0000273
Greg Ward11696872000-06-07 02:29:03 +0000274 # Where to look for the system-wide Distutils config file
275 sys_dir = os.path.dirname(sys.modules['distutils'].__file__)
276
277 # Look for the system config file
278 sys_file = os.path.join(sys_dir, "distutils.cfg")
Greg Wardacf3f6a2000-06-07 02:26:19 +0000279 if os.path.isfile(sys_file):
280 files.append(sys_file)
Gregory P. Smith14263542000-05-12 00:41:33 +0000281
Greg Ward11696872000-06-07 02:29:03 +0000282 # What to call the per-user config file
283 if os.name == 'posix':
Jeremy Hylton65d6edb2000-07-07 20:45:21 +0000284 user_filename = ".pydistutils.cfg"
285 else:
286 user_filename = "pydistutils.cfg"
Greg Wardfa9ff762000-10-14 04:06:40 +0000287
Greg Ward11696872000-06-07 02:29:03 +0000288 # And look for the user config file
Jeremy Hylton65d6edb2000-07-07 20:45:21 +0000289 if os.environ.has_key('HOME'):
290 user_file = os.path.join(os.environ.get('HOME'), user_filename)
Gregory P. Smith14263542000-05-12 00:41:33 +0000291 if os.path.isfile(user_file):
292 files.append(user_file)
293
Gregory P. Smith14263542000-05-12 00:41:33 +0000294 # All platforms support local setup.cfg
295 local_file = "setup.cfg"
296 if os.path.isfile(local_file):
297 files.append(local_file)
298
299 return files
300
301 # find_config_files ()
302
303
304 def parse_config_files (self, filenames=None):
305
306 from ConfigParser import ConfigParser
Greg Ward2bd3f422000-06-02 01:59:33 +0000307 from distutils.core import DEBUG
Gregory P. Smith14263542000-05-12 00:41:33 +0000308
309 if filenames is None:
310 filenames = self.find_config_files()
311
Greg Ward2bd3f422000-06-02 01:59:33 +0000312 if DEBUG: print "Distribution.parse_config_files():"
Greg Ward47460772000-05-23 03:47:35 +0000313
Gregory P. Smith14263542000-05-12 00:41:33 +0000314 parser = ConfigParser()
Greg Wardd5d8a992000-05-23 01:42:17 +0000315 for filename in filenames:
Greg Ward2bd3f422000-06-02 01:59:33 +0000316 if DEBUG: print " reading", filename
Greg Wardd5d8a992000-05-23 01:42:17 +0000317 parser.read(filename)
318 for section in parser.sections():
319 options = parser.options(section)
Greg Ward0e48cfd2000-05-26 01:00:15 +0000320 opt_dict = self.get_option_dict(section)
Gregory P. Smith14263542000-05-12 00:41:33 +0000321
Greg Wardd5d8a992000-05-23 01:42:17 +0000322 for opt in options:
323 if opt != '__name__':
Greg Wardceb9e222000-09-25 01:23:52 +0000324 val = parser.get(section,opt)
325 opt = string.replace(opt, '-', '_')
326 opt_dict[opt] = (filename, val)
Gregory P. Smith14263542000-05-12 00:41:33 +0000327
Greg Ward47460772000-05-23 03:47:35 +0000328 # Make the ConfigParser forget everything (so we retain
329 # the original filenames that options come from) -- gag,
330 # retch, puke -- another good reason for a distutils-
331 # specific config parser (sigh...)
332 parser.__init__()
Gregory P. Smith14263542000-05-12 00:41:33 +0000333
Greg Wardceb9e222000-09-25 01:23:52 +0000334 # If there was a "global" section in the config file, use it
335 # to set Distribution options.
336
337 if self.command_options.has_key('global'):
338 for (opt, (src, val)) in self.command_options['global'].items():
339 alias = self.negative_opt.get(opt)
340 try:
341 if alias:
342 setattr(self, alias, not strtobool(val))
343 elif opt in ('verbose', 'dry_run'): # ugh!
344 setattr(self, opt, strtobool(val))
345 except ValueError, msg:
346 raise DistutilsOptionError, msg
347
348 # parse_config_files ()
349
Gregory P. Smith14263542000-05-12 00:41:33 +0000350
Greg Wardd5d8a992000-05-23 01:42:17 +0000351 # -- Command-line parsing methods ----------------------------------
352
Greg Ward9821bf42000-08-29 01:15:18 +0000353 def parse_command_line (self):
354 """Parse the setup script's command line, taken from the
355 'script_args' instance attribute (which defaults to 'sys.argv[1:]'
356 -- see 'setup()' in core.py). This list is first processed for
357 "global options" -- options that set attributes of the Distribution
358 instance. Then, it is alternately scanned for Distutils commands
359 and options for that command. Each new command terminates the
360 options for the previous command. The allowed options for a
361 command are determined by the 'user_options' attribute of the
362 command class -- thus, we have to be able to load command classes
363 in order to parse the command line. Any error in that 'options'
364 attribute raises DistutilsGetoptError; any error on the
365 command-line raises DistutilsArgError. If no Distutils commands
366 were found on the command line, raises DistutilsArgError. Return
Greg Wardceb9e222000-09-25 01:23:52 +0000367 true if command-line was successfully parsed and we should carry
Greg Ward9821bf42000-08-29 01:15:18 +0000368 on with executing commands; false if no errors but we shouldn't
369 execute commands (currently, this only happens if user asks for
370 help).
Greg Wardd5d8a992000-05-23 01:42:17 +0000371 """
Andrew M. Kuchling3f819ec2001-01-15 16:09:35 +0000372 #
Fred Drake981a1782001-08-10 18:59:30 +0000373 # We now have enough information to show the Macintosh dialog
374 # that allows the user to interactively specify the "command line".
Andrew M. Kuchling3f819ec2001-01-15 16:09:35 +0000375 #
376 if sys.platform == 'mac':
377 import EasyDialogs
378 cmdlist = self.get_command_list()
379 self.script_args = EasyDialogs.GetArgv(
380 self.global_options + self.display_options, cmdlist)
Fred Drakeb94b8492001-12-06 20:51:35 +0000381
Greg Wardfe6462c2000-04-04 01:40:52 +0000382 # We have to parse the command line a bit at a time -- global
383 # options, then the first command, then its options, and so on --
384 # because each command will be handled by a different class, and
Greg Wardd5d8a992000-05-23 01:42:17 +0000385 # the options that are valid for a particular class aren't known
386 # until we have loaded the command class, which doesn't happen
387 # until we know what the command is.
Greg Wardfe6462c2000-04-04 01:40:52 +0000388
389 self.commands = []
Greg Wardfd7b91e2000-09-26 01:52:25 +0000390 parser = FancyGetopt(self.global_options + self.display_options)
391 parser.set_negative_aliases(self.negative_opt)
Andrew M. Kuchlingfa7dc572001-08-10 18:49:23 +0000392 parser.set_aliases({'licence': 'license'})
Greg Wardfd7b91e2000-09-26 01:52:25 +0000393 args = parser.getopt(args=self.script_args, object=self)
Greg Ward82715e12000-04-21 02:28:14 +0000394 option_order = parser.get_option_order()
Greg Wardfe6462c2000-04-04 01:40:52 +0000395
Greg Ward82715e12000-04-21 02:28:14 +0000396 # for display options we return immediately
397 if self.handle_display_options(option_order):
Greg Wardfe6462c2000-04-04 01:40:52 +0000398 return
Fred Drakeb94b8492001-12-06 20:51:35 +0000399
Greg Wardfe6462c2000-04-04 01:40:52 +0000400 while args:
Greg Wardd5d8a992000-05-23 01:42:17 +0000401 args = self._parse_command_opts(parser, args)
402 if args is None: # user asked for help (and got it)
Greg Wardfe6462c2000-04-04 01:40:52 +0000403 return
Greg Wardfe6462c2000-04-04 01:40:52 +0000404
Greg Wardd5d8a992000-05-23 01:42:17 +0000405 # Handle the cases of --help as a "global" option, ie.
406 # "setup.py --help" and "setup.py --help command ...". For the
407 # former, we show global options (--verbose, --dry-run, etc.)
408 # and display-only options (--name, --version, etc.); for the
409 # latter, we omit the display-only options and show help for
410 # each command listed on the command line.
Greg Wardfe6462c2000-04-04 01:40:52 +0000411 if self.help:
Greg Wardd5d8a992000-05-23 01:42:17 +0000412 self._show_help(parser,
413 display_options=len(self.commands) == 0,
414 commands=self.commands)
Greg Wardfe6462c2000-04-04 01:40:52 +0000415 return
416
417 # Oops, no commands found -- an end-user error
418 if not self.commands:
419 raise DistutilsArgError, "no commands supplied"
420
421 # All is well: return true
422 return 1
423
424 # parse_command_line()
425
Greg Wardd5d8a992000-05-23 01:42:17 +0000426 def _parse_command_opts (self, parser, args):
Greg Wardd5d8a992000-05-23 01:42:17 +0000427 """Parse the command-line options for a single command.
428 'parser' must be a FancyGetopt instance; 'args' must be the list
429 of arguments, starting with the current command (whose options
430 we are about to parse). Returns a new version of 'args' with
431 the next command at the front of the list; will be the empty
432 list if there are no more commands on the command line. Returns
433 None if the user asked for help on this command.
434 """
435 # late import because of mutual dependence between these modules
436 from distutils.cmd import Command
437
438 # Pull the current command from the head of the command line
439 command = args[0]
Greg Wardfd7b91e2000-09-26 01:52:25 +0000440 if not command_re.match(command):
Greg Wardd5d8a992000-05-23 01:42:17 +0000441 raise SystemExit, "invalid command name '%s'" % command
Greg Wardfd7b91e2000-09-26 01:52:25 +0000442 self.commands.append(command)
Greg Wardd5d8a992000-05-23 01:42:17 +0000443
444 # Dig up the command class that implements this command, so we
445 # 1) know that it's a valid command, and 2) know which options
446 # it takes.
447 try:
Greg Wardfd7b91e2000-09-26 01:52:25 +0000448 cmd_class = self.get_command_class(command)
Greg Wardd5d8a992000-05-23 01:42:17 +0000449 except DistutilsModuleError, msg:
450 raise DistutilsArgError, msg
451
452 # Require that the command class be derived from Command -- want
453 # to be sure that the basic "command" interface is implemented.
Greg Wardfd7b91e2000-09-26 01:52:25 +0000454 if not issubclass(cmd_class, Command):
Greg Wardd5d8a992000-05-23 01:42:17 +0000455 raise DistutilsClassError, \
456 "command class %s must subclass Command" % cmd_class
457
458 # Also make sure that the command object provides a list of its
459 # known options.
Greg Wardfd7b91e2000-09-26 01:52:25 +0000460 if not (hasattr(cmd_class, 'user_options') and
461 type(cmd_class.user_options) is ListType):
Greg Wardd5d8a992000-05-23 01:42:17 +0000462 raise DistutilsClassError, \
463 ("command class %s must provide " +
464 "'user_options' attribute (a list of tuples)") % \
465 cmd_class
466
467 # If the command class has a list of negative alias options,
468 # merge it in with the global negative aliases.
469 negative_opt = self.negative_opt
Greg Wardfd7b91e2000-09-26 01:52:25 +0000470 if hasattr(cmd_class, 'negative_opt'):
471 negative_opt = copy(negative_opt)
472 negative_opt.update(cmd_class.negative_opt)
Greg Wardd5d8a992000-05-23 01:42:17 +0000473
Greg Wardfa9ff762000-10-14 04:06:40 +0000474 # Check for help_options in command class. They have a different
475 # format (tuple of four) so we need to preprocess them here.
Jeremy Hylton65d6edb2000-07-07 20:45:21 +0000476 if (hasattr(cmd_class, 'help_options') and
Greg Wardfd7b91e2000-09-26 01:52:25 +0000477 type(cmd_class.help_options) is ListType):
Greg Ward2ff78872000-06-24 00:23:20 +0000478 help_options = fix_help_options(cmd_class.help_options)
479 else:
Greg Ward55fced32000-06-24 01:22:41 +0000480 help_options = []
Greg Ward2ff78872000-06-24 00:23:20 +0000481
Greg Ward9d17a7a2000-06-07 03:00:06 +0000482
Greg Wardd5d8a992000-05-23 01:42:17 +0000483 # All commands support the global options too, just by adding
484 # in 'global_options'.
Greg Wardfd7b91e2000-09-26 01:52:25 +0000485 parser.set_option_table(self.global_options +
486 cmd_class.user_options +
487 help_options)
488 parser.set_negative_aliases(negative_opt)
489 (args, opts) = parser.getopt(args[1:])
Greg Ward47460772000-05-23 03:47:35 +0000490 if hasattr(opts, 'help') and opts.help:
Greg Wardd5d8a992000-05-23 01:42:17 +0000491 self._show_help(parser, display_options=0, commands=[cmd_class])
492 return
493
Jeremy Hylton65d6edb2000-07-07 20:45:21 +0000494 if (hasattr(cmd_class, 'help_options') and
Greg Wardfd7b91e2000-09-26 01:52:25 +0000495 type(cmd_class.help_options) is ListType):
Jeremy Hylton65d6edb2000-07-07 20:45:21 +0000496 help_option_found=0
497 for (help_option, short, desc, func) in cmd_class.help_options:
498 if hasattr(opts, parser.get_attr_name(help_option)):
499 help_option_found=1
Greg Wardfa9ff762000-10-14 04:06:40 +0000500 #print "showing help for option %s of command %s" % \
Greg Ward2ff78872000-06-24 00:23:20 +0000501 # (help_option[0],cmd_class)
Greg Ward55fced32000-06-24 01:22:41 +0000502
503 if callable(func):
Jeremy Hylton65d6edb2000-07-07 20:45:21 +0000504 func()
Greg Ward55fced32000-06-24 01:22:41 +0000505 else:
Fred Drake981a1782001-08-10 18:59:30 +0000506 raise DistutilsClassError(
507 "invalid help function %s for help option '%s': "
508 "must be a callable object (function, etc.)"
509 % (`func`, help_option))
Greg Ward55fced32000-06-24 01:22:41 +0000510
Fred Drakeb94b8492001-12-06 20:51:35 +0000511 if help_option_found:
Jeremy Hylton65d6edb2000-07-07 20:45:21 +0000512 return
Greg Ward9d17a7a2000-06-07 03:00:06 +0000513
Greg Wardd5d8a992000-05-23 01:42:17 +0000514 # Put the options from the command-line into their official
515 # holding pen, the 'command_options' dictionary.
Greg Ward0e48cfd2000-05-26 01:00:15 +0000516 opt_dict = self.get_option_dict(command)
Greg Wardd5d8a992000-05-23 01:42:17 +0000517 for (name, value) in vars(opts).items():
Greg Ward0e48cfd2000-05-26 01:00:15 +0000518 opt_dict[name] = ("command line", value)
Greg Wardd5d8a992000-05-23 01:42:17 +0000519
520 return args
521
522 # _parse_command_opts ()
523
524
Andrew M. Kuchlinga7210ed2001-03-22 03:06:52 +0000525 def finalize_options (self):
526 """Set final values for all the options on the Distribution
527 instance, analogous to the .finalize_options() method of Command
528 objects.
529 """
530
Andrew M. Kuchlinga7210ed2001-03-22 03:06:52 +0000531 keywords = self.metadata.keywords
532 if keywords is not None:
533 if type(keywords) is StringType:
534 keywordlist = string.split(keywords, ',')
535 self.metadata.keywords = map(string.strip, keywordlist)
536
537 platforms = self.metadata.platforms
538 if platforms is not None:
539 if type(platforms) is StringType:
540 platformlist = string.split(platforms, ',')
541 self.metadata.platforms = map(string.strip, platformlist)
542
Greg Wardd5d8a992000-05-23 01:42:17 +0000543 def _show_help (self,
544 parser,
545 global_options=1,
546 display_options=1,
547 commands=[]):
548 """Show help for the setup script command-line in the form of
549 several lists of command-line options. 'parser' should be a
550 FancyGetopt instance; do not expect it to be returned in the
551 same state, as its option table will be reset to make it
552 generate the correct help text.
553
554 If 'global_options' is true, lists the global options:
555 --verbose, --dry-run, etc. If 'display_options' is true, lists
556 the "display-only" options: --name, --version, etc. Finally,
557 lists per-command help for every command name or command class
558 in 'commands'.
559 """
560 # late import because of mutual dependence between these modules
Greg Ward9821bf42000-08-29 01:15:18 +0000561 from distutils.core import gen_usage
Greg Wardd5d8a992000-05-23 01:42:17 +0000562 from distutils.cmd import Command
563
564 if global_options:
Greg Wardfd7b91e2000-09-26 01:52:25 +0000565 parser.set_option_table(self.global_options)
566 parser.print_help("Global options:")
Greg Wardd5d8a992000-05-23 01:42:17 +0000567 print
568
569 if display_options:
Greg Wardfd7b91e2000-09-26 01:52:25 +0000570 parser.set_option_table(self.display_options)
571 parser.print_help(
Greg Wardd5d8a992000-05-23 01:42:17 +0000572 "Information display options (just display " +
573 "information, ignore any commands)")
574 print
575
576 for command in self.commands:
Andrew M. Kuchlingfa7dc572001-08-10 18:49:23 +0000577 if type(command) is ClassType and issubclass(command, Command):
Greg Wardd5d8a992000-05-23 01:42:17 +0000578 klass = command
579 else:
Greg Wardfd7b91e2000-09-26 01:52:25 +0000580 klass = self.get_command_class(command)
Jeremy Hylton65d6edb2000-07-07 20:45:21 +0000581 if (hasattr(klass, 'help_options') and
Greg Wardfd7b91e2000-09-26 01:52:25 +0000582 type(klass.help_options) is ListType):
583 parser.set_option_table(klass.user_options +
584 fix_help_options(klass.help_options))
Jeremy Hylton65d6edb2000-07-07 20:45:21 +0000585 else:
Greg Wardfd7b91e2000-09-26 01:52:25 +0000586 parser.set_option_table(klass.user_options)
587 parser.print_help("Options for '%s' command:" % klass.__name__)
Greg Wardd5d8a992000-05-23 01:42:17 +0000588 print
589
Greg Ward9821bf42000-08-29 01:15:18 +0000590 print gen_usage(self.script_name)
Greg Wardd5d8a992000-05-23 01:42:17 +0000591 return
592
593 # _show_help ()
Greg Wardfa9ff762000-10-14 04:06:40 +0000594
Greg Wardd5d8a992000-05-23 01:42:17 +0000595
Greg Ward82715e12000-04-21 02:28:14 +0000596 def handle_display_options (self, option_order):
597 """If there were any non-global "display-only" options
Greg Wardd5d8a992000-05-23 01:42:17 +0000598 (--help-commands or the metadata display options) on the command
599 line, display the requested info and return true; else return
600 false.
601 """
Greg Ward9821bf42000-08-29 01:15:18 +0000602 from distutils.core import gen_usage
Greg Ward82715e12000-04-21 02:28:14 +0000603
604 # User just wants a list of commands -- we'll print it out and stop
605 # processing now (ie. if they ran "setup --help-commands foo bar",
606 # we ignore "foo bar").
607 if self.help_commands:
Greg Wardfd7b91e2000-09-26 01:52:25 +0000608 self.print_commands()
Greg Ward82715e12000-04-21 02:28:14 +0000609 print
Greg Ward9821bf42000-08-29 01:15:18 +0000610 print gen_usage(self.script_name)
Greg Ward82715e12000-04-21 02:28:14 +0000611 return 1
612
613 # If user supplied any of the "display metadata" options, then
614 # display that metadata in the order in which the user supplied the
615 # metadata options.
616 any_display_options = 0
617 is_display_option = {}
618 for option in self.display_options:
619 is_display_option[option[0]] = 1
620
621 for (opt, val) in option_order:
622 if val and is_display_option.get(opt):
Greg Ward2f2b6c62000-09-25 01:58:07 +0000623 opt = translate_longopt(opt)
Andrew M. Kuchlinga7210ed2001-03-22 03:06:52 +0000624 value = getattr(self.metadata, "get_"+opt)()
625 if opt in ['keywords', 'platforms']:
626 print string.join(value, ',')
627 else:
628 print value
Greg Ward82715e12000-04-21 02:28:14 +0000629 any_display_options = 1
630
631 return any_display_options
632
633 # handle_display_options()
Greg Wardfe6462c2000-04-04 01:40:52 +0000634
635 def print_command_list (self, commands, header, max_length):
636 """Print a subset of the list of all commands -- used by
Greg Wardd5d8a992000-05-23 01:42:17 +0000637 'print_commands()'.
638 """
Greg Wardfe6462c2000-04-04 01:40:52 +0000639
640 print header + ":"
641
642 for cmd in commands:
Greg Wardfd7b91e2000-09-26 01:52:25 +0000643 klass = self.cmdclass.get(cmd)
Greg Wardfe6462c2000-04-04 01:40:52 +0000644 if not klass:
Greg Wardfd7b91e2000-09-26 01:52:25 +0000645 klass = self.get_command_class(cmd)
Greg Wardfe6462c2000-04-04 01:40:52 +0000646 try:
647 description = klass.description
648 except AttributeError:
649 description = "(no description available)"
650
651 print " %-*s %s" % (max_length, cmd, description)
652
653 # print_command_list ()
654
655
656 def print_commands (self):
Greg Wardd5d8a992000-05-23 01:42:17 +0000657 """Print out a help message listing all available commands with a
658 description of each. The list is divided into "standard commands"
659 (listed in distutils.command.__all__) and "extra commands"
660 (mentioned in self.cmdclass, but not a standard command). The
661 descriptions come from the command class attribute
662 'description'.
663 """
Greg Wardfe6462c2000-04-04 01:40:52 +0000664
665 import distutils.command
666 std_commands = distutils.command.__all__
667 is_std = {}
668 for cmd in std_commands:
669 is_std[cmd] = 1
670
671 extra_commands = []
672 for cmd in self.cmdclass.keys():
673 if not is_std.get(cmd):
Greg Wardfd7b91e2000-09-26 01:52:25 +0000674 extra_commands.append(cmd)
Greg Wardfe6462c2000-04-04 01:40:52 +0000675
676 max_length = 0
677 for cmd in (std_commands + extra_commands):
Greg Wardfd7b91e2000-09-26 01:52:25 +0000678 if len(cmd) > max_length:
679 max_length = len(cmd)
Greg Wardfe6462c2000-04-04 01:40:52 +0000680
Greg Wardfd7b91e2000-09-26 01:52:25 +0000681 self.print_command_list(std_commands,
682 "Standard commands",
683 max_length)
Greg Wardfe6462c2000-04-04 01:40:52 +0000684 if extra_commands:
685 print
Greg Wardfd7b91e2000-09-26 01:52:25 +0000686 self.print_command_list(extra_commands,
687 "Extra commands",
688 max_length)
Greg Wardfe6462c2000-04-04 01:40:52 +0000689
690 # print_commands ()
Greg Wardfe6462c2000-04-04 01:40:52 +0000691
Greg Wardf6fc8752000-11-11 02:47:11 +0000692 def get_command_list (self):
693 """Get a list of (command, description) tuples.
694 The list is divided into "standard commands" (listed in
695 distutils.command.__all__) and "extra commands" (mentioned in
696 self.cmdclass, but not a standard command). The descriptions come
697 from the command class attribute 'description'.
698 """
699 # Currently this is only used on Mac OS, for the Mac-only GUI
700 # Distutils interface (by Jack Jansen)
701
702 import distutils.command
703 std_commands = distutils.command.__all__
704 is_std = {}
705 for cmd in std_commands:
706 is_std[cmd] = 1
707
708 extra_commands = []
709 for cmd in self.cmdclass.keys():
710 if not is_std.get(cmd):
711 extra_commands.append(cmd)
712
713 rv = []
714 for cmd in (std_commands + extra_commands):
715 klass = self.cmdclass.get(cmd)
716 if not klass:
717 klass = self.get_command_class(cmd)
718 try:
719 description = klass.description
720 except AttributeError:
721 description = "(no description available)"
722 rv.append((cmd, description))
723 return rv
Greg Wardfe6462c2000-04-04 01:40:52 +0000724
725 # -- Command class/object methods ----------------------------------
726
Greg Wardd5d8a992000-05-23 01:42:17 +0000727 def get_command_class (self, command):
728 """Return the class that implements the Distutils command named by
729 'command'. First we check the 'cmdclass' dictionary; if the
730 command is mentioned there, we fetch the class object from the
731 dictionary and return it. Otherwise we load the command module
732 ("distutils.command." + command) and fetch the command class from
733 the module. The loaded class is also stored in 'cmdclass'
734 to speed future calls to 'get_command_class()'.
Greg Wardfe6462c2000-04-04 01:40:52 +0000735
Gregory P. Smith14263542000-05-12 00:41:33 +0000736 Raises DistutilsModuleError if the expected module could not be
Greg Wardd5d8a992000-05-23 01:42:17 +0000737 found, or if that module does not define the expected class.
738 """
739 klass = self.cmdclass.get(command)
740 if klass:
741 return klass
Greg Wardfe6462c2000-04-04 01:40:52 +0000742
743 module_name = 'distutils.command.' + command
744 klass_name = command
745
746 try:
747 __import__ (module_name)
748 module = sys.modules[module_name]
749 except ImportError:
750 raise DistutilsModuleError, \
751 "invalid command '%s' (no module named '%s')" % \
752 (command, module_name)
753
754 try:
Greg Wardd5d8a992000-05-23 01:42:17 +0000755 klass = getattr(module, klass_name)
756 except AttributeError:
Greg Wardfe6462c2000-04-04 01:40:52 +0000757 raise DistutilsModuleError, \
758 "invalid command '%s' (no class '%s' in module '%s')" \
759 % (command, klass_name, module_name)
760
Greg Wardd5d8a992000-05-23 01:42:17 +0000761 self.cmdclass[command] = klass
Greg Wardfe6462c2000-04-04 01:40:52 +0000762 return klass
763
Greg Wardd5d8a992000-05-23 01:42:17 +0000764 # get_command_class ()
Greg Wardfe6462c2000-04-04 01:40:52 +0000765
Greg Wardd5d8a992000-05-23 01:42:17 +0000766 def get_command_obj (self, command, create=1):
767 """Return the command object for 'command'. Normally this object
Greg Ward612eb9f2000-07-27 02:13:20 +0000768 is cached on a previous call to 'get_command_obj()'; if no command
Greg Wardd5d8a992000-05-23 01:42:17 +0000769 object for 'command' is in the cache, then we either create and
770 return it (if 'create' is true) or return None.
771 """
Greg Ward2bd3f422000-06-02 01:59:33 +0000772 from distutils.core import DEBUG
Greg Wardd5d8a992000-05-23 01:42:17 +0000773 cmd_obj = self.command_obj.get(command)
Greg Wardfe6462c2000-04-04 01:40:52 +0000774 if not cmd_obj and create:
Greg Ward2bd3f422000-06-02 01:59:33 +0000775 if DEBUG:
776 print "Distribution.get_command_obj(): " \
777 "creating '%s' command object" % command
Greg Ward47460772000-05-23 03:47:35 +0000778
Greg Wardd5d8a992000-05-23 01:42:17 +0000779 klass = self.get_command_class(command)
Greg Ward47460772000-05-23 03:47:35 +0000780 cmd_obj = self.command_obj[command] = klass(self)
781 self.have_run[command] = 0
782
783 # Set any options that were supplied in config files
784 # or on the command line. (NB. support for error
785 # reporting is lame here: any errors aren't reported
786 # until 'finalize_options()' is called, which means
787 # we won't report the source of the error.)
788 options = self.command_options.get(command)
789 if options:
Greg Wardc32d9a62000-05-28 23:53:06 +0000790 self._set_command_options(cmd_obj, options)
Greg Wardfe6462c2000-04-04 01:40:52 +0000791
792 return cmd_obj
793
Greg Wardc32d9a62000-05-28 23:53:06 +0000794 def _set_command_options (self, command_obj, option_dict=None):
Greg Wardc32d9a62000-05-28 23:53:06 +0000795 """Set the options for 'command_obj' from 'option_dict'. Basically
796 this means copying elements of a dictionary ('option_dict') to
797 attributes of an instance ('command').
798
Greg Wardceb9e222000-09-25 01:23:52 +0000799 'command_obj' must be a Command instance. If 'option_dict' is not
Greg Wardc32d9a62000-05-28 23:53:06 +0000800 supplied, uses the standard option dictionary for this command
801 (from 'self.command_options').
802 """
803 from distutils.core import DEBUG
Fred Drakeb94b8492001-12-06 20:51:35 +0000804
Greg Wardc32d9a62000-05-28 23:53:06 +0000805 command_name = command_obj.get_command_name()
806 if option_dict is None:
807 option_dict = self.get_option_dict(command_name)
808
809 if DEBUG: print " setting options for '%s' command:" % command_name
810 for (option, (source, value)) in option_dict.items():
811 if DEBUG: print " %s = %s (from %s)" % (option, value, source)
Greg Wardceb9e222000-09-25 01:23:52 +0000812 try:
Greg Ward2f2b6c62000-09-25 01:58:07 +0000813 bool_opts = map(translate_longopt, command_obj.boolean_options)
Greg Wardceb9e222000-09-25 01:23:52 +0000814 except AttributeError:
815 bool_opts = []
816 try:
817 neg_opt = command_obj.negative_opt
818 except AttributeError:
819 neg_opt = {}
820
821 try:
Greg Ward2c08cf02000-09-27 00:15:37 +0000822 is_string = type(value) is StringType
823 if neg_opt.has_key(option) and is_string:
Greg Wardceb9e222000-09-25 01:23:52 +0000824 setattr(command_obj, neg_opt[option], not strtobool(value))
Greg Ward2c08cf02000-09-27 00:15:37 +0000825 elif option in bool_opts and is_string:
Greg Wardceb9e222000-09-25 01:23:52 +0000826 setattr(command_obj, option, strtobool(value))
827 elif hasattr(command_obj, option):
828 setattr(command_obj, option, value)
829 else:
830 raise DistutilsOptionError, \
831 ("error in %s: command '%s' has no such option '%s'"
832 % (source, command_name, option))
833 except ValueError, msg:
834 raise DistutilsOptionError, msg
Greg Wardc32d9a62000-05-28 23:53:06 +0000835
Greg Wardf449ea52000-09-16 15:23:28 +0000836 def reinitialize_command (self, command, reinit_subcommands=0):
Greg Wardc32d9a62000-05-28 23:53:06 +0000837 """Reinitializes a command to the state it was in when first
838 returned by 'get_command_obj()': ie., initialized but not yet
Greg Ward7d9c7052000-06-28 01:25:27 +0000839 finalized. This provides the opportunity to sneak option
Greg Wardc32d9a62000-05-28 23:53:06 +0000840 values in programmatically, overriding or supplementing
841 user-supplied values from the config files and command line.
842 You'll have to re-finalize the command object (by calling
843 'finalize_options()' or 'ensure_finalized()') before using it for
Fred Drakeb94b8492001-12-06 20:51:35 +0000844 real.
Greg Wardc32d9a62000-05-28 23:53:06 +0000845
Greg Wardf449ea52000-09-16 15:23:28 +0000846 'command' should be a command name (string) or command object. If
847 'reinit_subcommands' is true, also reinitializes the command's
848 sub-commands, as declared by the 'sub_commands' class attribute (if
849 it has one). See the "install" command for an example. Only
850 reinitializes the sub-commands that actually matter, ie. those
851 whose test predicates return true.
852
Greg Wardc32d9a62000-05-28 23:53:06 +0000853 Returns the reinitialized command object.
854 """
855 from distutils.cmd import Command
856 if not isinstance(command, Command):
857 command_name = command
858 command = self.get_command_obj(command_name)
859 else:
860 command_name = command.get_command_name()
861
862 if not command.finalized:
Greg Ward282c7a02000-06-01 01:09:47 +0000863 return command
Greg Wardc32d9a62000-05-28 23:53:06 +0000864 command.initialize_options()
865 command.finalized = 0
Greg Ward43955c92000-06-06 02:52:36 +0000866 self.have_run[command_name] = 0
Greg Wardc32d9a62000-05-28 23:53:06 +0000867 self._set_command_options(command)
Greg Wardf449ea52000-09-16 15:23:28 +0000868
Greg Wardf449ea52000-09-16 15:23:28 +0000869 if reinit_subcommands:
Greg Wardf449ea52000-09-16 15:23:28 +0000870 for sub in command.get_sub_commands():
Fred Drakeb94b8492001-12-06 20:51:35 +0000871 self.reinitialize_command(sub, reinit_subcommands)
Greg Wardf449ea52000-09-16 15:23:28 +0000872
Greg Wardc32d9a62000-05-28 23:53:06 +0000873 return command
874
Fred Drakeb94b8492001-12-06 20:51:35 +0000875
Greg Wardfe6462c2000-04-04 01:40:52 +0000876 # -- Methods that operate on the Distribution ----------------------
877
878 def announce (self, msg, level=1):
879 """Print 'msg' if 'level' is greater than or equal to the verbosity
Greg Wardd5d8a992000-05-23 01:42:17 +0000880 level recorded in the 'verbose' attribute (which, currently, can be
881 only 0 or 1).
882 """
Greg Wardfe6462c2000-04-04 01:40:52 +0000883 if self.verbose >= level:
884 print msg
885
886
887 def run_commands (self):
Greg Ward82715e12000-04-21 02:28:14 +0000888 """Run each command that was seen on the setup script command line.
Greg Wardd5d8a992000-05-23 01:42:17 +0000889 Uses the list of commands found and cache of command objects
Greg Wardfd7b91e2000-09-26 01:52:25 +0000890 created by 'get_command_obj()'.
891 """
Greg Wardfe6462c2000-04-04 01:40:52 +0000892 for cmd in self.commands:
Greg Wardfd7b91e2000-09-26 01:52:25 +0000893 self.run_command(cmd)
Greg Wardfe6462c2000-04-04 01:40:52 +0000894
895
Greg Wardfe6462c2000-04-04 01:40:52 +0000896 # -- Methods that operate on its Commands --------------------------
897
898 def run_command (self, command):
Greg Wardfe6462c2000-04-04 01:40:52 +0000899 """Do whatever it takes to run a command (including nothing at all,
Greg Wardd5d8a992000-05-23 01:42:17 +0000900 if the command has already been run). Specifically: if we have
901 already created and run the command named by 'command', return
902 silently without doing anything. If the command named by 'command'
903 doesn't even have a command object yet, create one. Then invoke
904 'run()' on that command object (or an existing one).
905 """
Greg Wardfe6462c2000-04-04 01:40:52 +0000906 # Already been here, done that? then return silently.
Greg Wardfd7b91e2000-09-26 01:52:25 +0000907 if self.have_run.get(command):
Greg Wardfe6462c2000-04-04 01:40:52 +0000908 return
909
Greg Wardfd7b91e2000-09-26 01:52:25 +0000910 self.announce("running " + command)
911 cmd_obj = self.get_command_obj(command)
912 cmd_obj.ensure_finalized()
913 cmd_obj.run()
Greg Wardfe6462c2000-04-04 01:40:52 +0000914 self.have_run[command] = 1
915
916
Greg Wardfe6462c2000-04-04 01:40:52 +0000917 # -- Distribution query methods ------------------------------------
918
919 def has_pure_modules (self):
Greg Wardfd7b91e2000-09-26 01:52:25 +0000920 return len(self.packages or self.py_modules or []) > 0
Greg Wardfe6462c2000-04-04 01:40:52 +0000921
922 def has_ext_modules (self):
Greg Wardfd7b91e2000-09-26 01:52:25 +0000923 return self.ext_modules and len(self.ext_modules) > 0
Greg Wardfe6462c2000-04-04 01:40:52 +0000924
925 def has_c_libraries (self):
Greg Wardfd7b91e2000-09-26 01:52:25 +0000926 return self.libraries and len(self.libraries) > 0
Greg Wardfe6462c2000-04-04 01:40:52 +0000927
928 def has_modules (self):
929 return self.has_pure_modules() or self.has_ext_modules()
930
Greg Ward51def7d2000-05-27 01:36:14 +0000931 def has_headers (self):
932 return self.headers and len(self.headers) > 0
933
Greg Ward44a61bb2000-05-20 15:06:48 +0000934 def has_scripts (self):
935 return self.scripts and len(self.scripts) > 0
936
937 def has_data_files (self):
938 return self.data_files and len(self.data_files) > 0
939
Greg Wardfe6462c2000-04-04 01:40:52 +0000940 def is_pure (self):
941 return (self.has_pure_modules() and
942 not self.has_ext_modules() and
943 not self.has_c_libraries())
944
Greg Ward82715e12000-04-21 02:28:14 +0000945 # -- Metadata query methods ----------------------------------------
946
947 # If you're looking for 'get_name()', 'get_version()', and so forth,
948 # they are defined in a sneaky way: the constructor binds self.get_XXX
949 # to self.metadata.get_XXX. The actual code is in the
950 # DistributionMetadata class, below.
951
952# class Distribution
953
954
955class DistributionMetadata:
956 """Dummy class to hold the distribution meta-data: name, version,
Greg Wardfd7b91e2000-09-26 01:52:25 +0000957 author, and so forth.
958 """
Greg Ward82715e12000-04-21 02:28:14 +0000959
Neil Schemenauera8aefe52001-09-03 15:47:21 +0000960 _METHOD_BASENAMES = ("name", "version", "author", "author_email",
961 "maintainer", "maintainer_email", "url",
962 "license", "description", "long_description",
963 "keywords", "platforms", "fullname", "contact",
964 "contact_email", "licence")
965
Greg Ward82715e12000-04-21 02:28:14 +0000966 def __init__ (self):
967 self.name = None
968 self.version = None
969 self.author = None
970 self.author_email = None
971 self.maintainer = None
972 self.maintainer_email = None
973 self.url = None
Andrew M. Kuchlingfa7dc572001-08-10 18:49:23 +0000974 self.license = None
Greg Ward82715e12000-04-21 02:28:14 +0000975 self.description = None
Greg Warde5a584e2000-04-26 02:26:55 +0000976 self.long_description = None
Andrew M. Kuchlinga7210ed2001-03-22 03:06:52 +0000977 self.keywords = None
978 self.platforms = None
Fred Drakeb94b8492001-12-06 20:51:35 +0000979
Andrew M. Kuchlinga7210ed2001-03-22 03:06:52 +0000980 def write_pkg_info (self, base_dir):
981 """Write the PKG-INFO file into the release tree.
982 """
983
984 pkg_info = open( os.path.join(base_dir, 'PKG-INFO'), 'w')
985
986 pkg_info.write('Metadata-Version: 1.0\n')
987 pkg_info.write('Name: %s\n' % self.get_name() )
988 pkg_info.write('Version: %s\n' % self.get_version() )
989 pkg_info.write('Summary: %s\n' % self.get_description() )
990 pkg_info.write('Home-page: %s\n' % self.get_url() )
Andrew M. Kuchlingffb963c2001-03-22 15:32:23 +0000991 pkg_info.write('Author: %s\n' % self.get_contact() )
992 pkg_info.write('Author-email: %s\n' % self.get_contact_email() )
Andrew M. Kuchlingfa7dc572001-08-10 18:49:23 +0000993 pkg_info.write('License: %s\n' % self.get_license() )
Andrew M. Kuchlinga7210ed2001-03-22 03:06:52 +0000994
995 long_desc = rfc822_escape( self.get_long_description() )
996 pkg_info.write('Description: %s\n' % long_desc)
997
998 keywords = string.join( self.get_keywords(), ',')
999 if keywords:
1000 pkg_info.write('Keywords: %s\n' % keywords )
1001
1002 for platform in self.get_platforms():
1003 pkg_info.write('Platform: %s\n' % platform )
1004
1005 pkg_info.close()
Fred Drakeb94b8492001-12-06 20:51:35 +00001006
Andrew M. Kuchlinga7210ed2001-03-22 03:06:52 +00001007 # write_pkg_info ()
Fred Drakeb94b8492001-12-06 20:51:35 +00001008
Greg Ward82715e12000-04-21 02:28:14 +00001009 # -- Metadata query methods ----------------------------------------
1010
Greg Wardfe6462c2000-04-04 01:40:52 +00001011 def get_name (self):
1012 return self.name or "UNKNOWN"
1013
Greg Ward82715e12000-04-21 02:28:14 +00001014 def get_version(self):
Thomas Hellerbcd89752001-12-06 20:44:19 +00001015 return self.version or "0.0.0"
Greg Wardfe6462c2000-04-04 01:40:52 +00001016
Greg Ward82715e12000-04-21 02:28:14 +00001017 def get_fullname (self):
1018 return "%s-%s" % (self.get_name(), self.get_version())
1019
1020 def get_author(self):
1021 return self.author or "UNKNOWN"
1022
1023 def get_author_email(self):
1024 return self.author_email or "UNKNOWN"
1025
1026 def get_maintainer(self):
1027 return self.maintainer or "UNKNOWN"
1028
1029 def get_maintainer_email(self):
1030 return self.maintainer_email or "UNKNOWN"
1031
1032 def get_contact(self):
1033 return (self.maintainer or
1034 self.author or
1035 "UNKNOWN")
1036
1037 def get_contact_email(self):
1038 return (self.maintainer_email or
1039 self.author_email or
1040 "UNKNOWN")
1041
1042 def get_url(self):
1043 return self.url or "UNKNOWN"
1044
Andrew M. Kuchlingfa7dc572001-08-10 18:49:23 +00001045 def get_license(self):
1046 return self.license or "UNKNOWN"
1047 get_licence = get_license
Fred Drakeb94b8492001-12-06 20:51:35 +00001048
Greg Ward82715e12000-04-21 02:28:14 +00001049 def get_description(self):
1050 return self.description or "UNKNOWN"
Greg Warde5a584e2000-04-26 02:26:55 +00001051
1052 def get_long_description(self):
1053 return self.long_description or "UNKNOWN"
1054
Andrew M. Kuchlinga7210ed2001-03-22 03:06:52 +00001055 def get_keywords(self):
1056 return self.keywords or []
1057
1058 def get_platforms(self):
1059 return self.platforms or ["UNKNOWN"]
1060
Greg Ward82715e12000-04-21 02:28:14 +00001061# class DistributionMetadata
Greg Wardfe6462c2000-04-04 01:40:52 +00001062
Greg Ward2ff78872000-06-24 00:23:20 +00001063
1064def fix_help_options (options):
1065 """Convert a 4-tuple 'help_options' list as found in various command
1066 classes to the 3-tuple form required by FancyGetopt.
1067 """
1068 new_options = []
1069 for help_tuple in options:
1070 new_options.append(help_tuple[0:3])
1071 return new_options
1072
1073
Greg Wardfe6462c2000-04-04 01:40:52 +00001074if __name__ == "__main__":
Greg Wardfd7b91e2000-09-26 01:52:25 +00001075 dist = Distribution()
Greg Wardfe6462c2000-04-04 01:40:52 +00001076 print "ok"