blob: ddde909fbd893aea61fb191beda1693452c488d8 [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
Greg Wardfe6462c2000-04-04 01:40:52 +00007__revision__ = "$Id$"
8
Neal Norwitz9d72bb42007-04-17 08:48:32 +00009import sys, os, re
Greg Wardfe6462c2000-04-04 01:40:52 +000010from copy import copy
Andrew M. Kuchlingff4ad9a2002-10-31 13:22:41 +000011
12try:
13 import warnings
Andrew M. Kuchlingccf4e422002-10-31 13:39:33 +000014except ImportError:
Andrew M. Kuchlingff4ad9a2002-10-31 13:22:41 +000015 warnings = None
16
Greg Wardfe6462c2000-04-04 01:40:52 +000017from distutils.errors import *
Greg Ward2f2b6c62000-09-25 01:58:07 +000018from distutils.fancy_getopt import FancyGetopt, translate_longopt
Andrew M. Kuchlinga7210ed2001-03-22 03:06:52 +000019from distutils.util import check_environ, strtobool, rfc822_escape
Jeremy Hyltoncd8a1142002-06-04 20:14:43 +000020from distutils import log
Jeremy Hyltonfcd73532002-09-11 16:31:53 +000021from distutils.debug import DEBUG
Greg Wardfe6462c2000-04-04 01:40:52 +000022
23# Regex to define acceptable Distutils command names. This is not *quite*
24# the same as a Python NAME -- I don't allow leading underscores. The fact
25# that they're very similar is no coincidence; the default naming scheme is
26# to look for a Python module named after the command.
27command_re = re.compile (r'^[a-zA-Z]([a-zA-Z0-9_]*)$')
28
29
30class Distribution:
Greg Ward8ff5a3f2000-06-02 00:44:53 +000031 """The core of the Distutils. Most of the work hiding behind 'setup'
32 is really done within a Distribution instance, which farms the work out
33 to the Distutils commands specified on the command line.
Greg Wardfe6462c2000-04-04 01:40:52 +000034
Greg Ward8ff5a3f2000-06-02 00:44:53 +000035 Setup scripts will almost never instantiate Distribution directly,
36 unless the 'setup()' function is totally inadequate to their needs.
37 However, it is conceivable that a setup script might wish to subclass
38 Distribution for some specialized purpose, and then pass the subclass
39 to 'setup()' as the 'distclass' keyword argument. If so, it is
40 necessary to respect the expectations that 'setup' has of Distribution.
41 See the code for 'setup()', in core.py, for details.
42 """
Greg Wardfe6462c2000-04-04 01:40:52 +000043
44
45 # 'global_options' describes the command-line options that may be
Greg Ward82715e12000-04-21 02:28:14 +000046 # supplied to the setup script prior to any actual commands.
47 # Eg. "./setup.py -n" or "./setup.py --quiet" both take advantage of
Greg Wardfe6462c2000-04-04 01:40:52 +000048 # these global options. This list should be kept to a bare minimum,
49 # since every global option is also valid as a command option -- and we
50 # don't want to pollute the commands with too many options that they
51 # have minimal control over.
Jeremy Hyltoncd8a1142002-06-04 20:14:43 +000052 # The fourth entry for verbose means that it can be repeated.
53 global_options = [('verbose', 'v', "run verbosely (default)", 1),
Greg Wardd5d8a992000-05-23 01:42:17 +000054 ('quiet', 'q', "run quietly (turns verbosity off)"),
55 ('dry-run', 'n', "don't actually do anything"),
56 ('help', 'h', "show detailed help message"),
Greg Wardfe6462c2000-04-04 01:40:52 +000057 ]
Greg Ward82715e12000-04-21 02:28:14 +000058
Martin v. Löwis8ed338a2005-03-03 08:12:27 +000059 # 'common_usage' is a short (2-3 line) string describing the common
60 # usage of the setup script.
61 common_usage = """\
62Common commands: (see '--help-commands' for more)
63
64 setup.py build will build the package underneath 'build/'
65 setup.py install will install the package
66"""
67
Greg Ward82715e12000-04-21 02:28:14 +000068 # options that are not propagated to the commands
69 display_options = [
70 ('help-commands', None,
71 "list all available commands"),
72 ('name', None,
73 "print package name"),
74 ('version', 'V',
75 "print package version"),
76 ('fullname', None,
77 "print <package name>-<version>"),
78 ('author', None,
79 "print the author's name"),
80 ('author-email', None,
81 "print the author's email address"),
82 ('maintainer', None,
83 "print the maintainer's name"),
84 ('maintainer-email', None,
85 "print the maintainer's email address"),
86 ('contact', None,
Greg Wardd5d8a992000-05-23 01:42:17 +000087 "print the maintainer's name if known, else the author's"),
Greg Ward82715e12000-04-21 02:28:14 +000088 ('contact-email', None,
Greg Wardd5d8a992000-05-23 01:42:17 +000089 "print the maintainer's email address if known, else the author's"),
Greg Ward82715e12000-04-21 02:28:14 +000090 ('url', None,
91 "print the URL for this package"),
Greg Ward82715e12000-04-21 02:28:14 +000092 ('license', None,
Andrew M. Kuchlingfa7dc572001-08-10 18:49:23 +000093 "print the license of the package"),
94 ('licence', None,
95 "alias for --license"),
Greg Ward82715e12000-04-21 02:28:14 +000096 ('description', None,
97 "print the package description"),
Greg Warde5a584e2000-04-26 02:26:55 +000098 ('long-description', None,
99 "print the long package description"),
Andrew M. Kuchlinga7210ed2001-03-22 03:06:52 +0000100 ('platforms', None,
101 "print the list of platforms"),
Andrew M. Kuchling282e2c32003-01-03 15:24:36 +0000102 ('classifiers', None,
103 "print the list of classifiers"),
Andrew M. Kuchlinga7210ed2001-03-22 03:06:52 +0000104 ('keywords', None,
105 "print the list of keywords"),
Fred Drakedb7b0022005-03-20 22:19:47 +0000106 ('provides', None,
107 "print the list of packages/modules provided"),
108 ('requires', None,
109 "print the list of packages/modules required"),
110 ('obsoletes', None,
111 "print the list of packages/modules made obsolete")
Greg Ward82715e12000-04-21 02:28:14 +0000112 ]
Guido van Rossumc1f779c2007-07-03 08:25:58 +0000113 display_option_names = [translate_longopt(x[0]) for x in display_options]
Greg Ward82715e12000-04-21 02:28:14 +0000114
115 # negative options are options that exclude other options
Greg Wardfe6462c2000-04-04 01:40:52 +0000116 negative_opt = {'quiet': 'verbose'}
117
118
119 # -- Creation/initialization methods -------------------------------
Fred Drakeb94b8492001-12-06 20:51:35 +0000120
Greg Wardfe6462c2000-04-04 01:40:52 +0000121 def __init__ (self, attrs=None):
122 """Construct a new Distribution instance: initialize all the
Greg Ward8ff5a3f2000-06-02 00:44:53 +0000123 attributes of a Distribution, and then use 'attrs' (a dictionary
124 mapping attribute names to values) to assign some of those
125 attributes their "real" values. (Any attributes not mentioned in
126 'attrs' will be assigned to some null value: 0, None, an empty list
127 or dictionary, etc.) Most importantly, initialize the
128 'command_obj' attribute to the empty dictionary; this will be
129 filled in with real command objects by 'parse_command_line()'.
130 """
Greg Wardfe6462c2000-04-04 01:40:52 +0000131
132 # Default values for our command-line options
133 self.verbose = 1
134 self.dry_run = 0
Greg Wardfe6462c2000-04-04 01:40:52 +0000135 self.help = 0
Greg Ward82715e12000-04-21 02:28:14 +0000136 for attr in self.display_option_names:
137 setattr(self, attr, 0)
Greg Wardfe6462c2000-04-04 01:40:52 +0000138
Greg Ward82715e12000-04-21 02:28:14 +0000139 # Store the distribution meta-data (name, version, author, and so
140 # forth) in a separate object -- we're getting to have enough
141 # information here (and enough command-line options) that it's
142 # worth it. Also delegate 'get_XXX()' methods to the 'metadata'
143 # object in a sneaky and underhanded (but efficient!) way.
Greg Wardfd7b91e2000-09-26 01:52:25 +0000144 self.metadata = DistributionMetadata()
Neil Schemenauera8aefe52001-09-03 15:47:21 +0000145 for basename in self.metadata._METHOD_BASENAMES:
Greg Ward4982f982000-04-22 02:52:44 +0000146 method_name = "get_" + basename
147 setattr(self, method_name, getattr(self.metadata, method_name))
Greg Wardfe6462c2000-04-04 01:40:52 +0000148
149 # 'cmdclass' maps command names to class objects, so we
150 # can 1) quickly figure out which class to instantiate when
151 # we need to create a new command object, and 2) have a way
Greg Ward82715e12000-04-21 02:28:14 +0000152 # for the setup script to override command classes
Greg Wardfe6462c2000-04-04 01:40:52 +0000153 self.cmdclass = {}
154
Fred Draked04573f2004-08-03 16:37:40 +0000155 # 'command_packages' is a list of packages in which commands
156 # are searched for. The factory for command 'foo' is expected
157 # to be named 'foo' in the module 'foo' in one of the packages
158 # named here. This list is searched from the left; an error
159 # is raised if no named package provides the command being
160 # searched for. (Always access using get_command_packages().)
161 self.command_packages = None
162
Greg Ward9821bf42000-08-29 01:15:18 +0000163 # 'script_name' and 'script_args' are usually set to sys.argv[0]
164 # and sys.argv[1:], but they can be overridden when the caller is
165 # not necessarily a setup script run from the command-line.
166 self.script_name = None
167 self.script_args = None
168
Greg Wardd5d8a992000-05-23 01:42:17 +0000169 # 'command_options' is where we store command options between
170 # parsing them (from config files, the command-line, etc.) and when
171 # they are actually needed -- ie. when the command in question is
172 # instantiated. It is a dictionary of dictionaries of 2-tuples:
173 # command_options = { command_name : { option : (source, value) } }
Gregory P. Smith14263542000-05-12 00:41:33 +0000174 self.command_options = {}
175
Martin v. Löwis98da5622005-03-23 18:54:36 +0000176 # 'dist_files' is the list of (command, pyversion, file) that
177 # have been created by any dist commands run so far. This is
178 # filled regardless of whether the run is dry or not. pyversion
179 # gives sysconfig.get_python_version() if the dist file is
180 # specific to a Python version, 'any' if it is good for all
181 # Python versions on the target platform, and '' for a source
182 # file. pyversion should not be used to specify minimum or
183 # maximum required Python versions; use the metainfo for that
184 # instead.
Martin v. Löwis55f1bb82005-03-21 20:56:35 +0000185 self.dist_files = []
186
Greg Wardfe6462c2000-04-04 01:40:52 +0000187 # These options are really the business of various commands, rather
188 # than of the Distribution itself. We provide aliases for them in
189 # Distribution as a convenience to the developer.
Greg Wardfe6462c2000-04-04 01:40:52 +0000190 self.packages = None
Fred Drake0eb32a62004-06-11 21:50:33 +0000191 self.package_data = {}
Greg Wardfe6462c2000-04-04 01:40:52 +0000192 self.package_dir = None
193 self.py_modules = None
194 self.libraries = None
Greg Ward51def7d2000-05-27 01:36:14 +0000195 self.headers = None
Greg Wardfe6462c2000-04-04 01:40:52 +0000196 self.ext_modules = None
197 self.ext_package = None
198 self.include_dirs = None
199 self.extra_path = None
Gregory P. Smithb2e3bb32000-05-12 00:52:23 +0000200 self.scripts = None
Gregory P. Smith6a901dd2000-05-13 03:09:50 +0000201 self.data_files = None
Greg Wardfe6462c2000-04-04 01:40:52 +0000202
203 # And now initialize bookkeeping stuff that can't be supplied by
204 # the caller at all. 'command_obj' maps command names to
205 # Command instances -- that's how we enforce that every command
206 # class is a singleton.
207 self.command_obj = {}
208
209 # 'have_run' maps command names to boolean values; it keeps track
210 # of whether we have actually run a particular command, to make it
211 # cheap to "run" a command whenever we think we might need to -- if
212 # it's already been done, no need for expensive filesystem
213 # operations, we just check the 'have_run' dictionary and carry on.
214 # It's only safe to query 'have_run' for a command class that has
215 # been instantiated -- a false value will be inserted when the
216 # command object is created, and replaced with a true value when
Greg Ward612eb9f2000-07-27 02:13:20 +0000217 # the command is successfully run. Thus it's probably best to use
Greg Wardfe6462c2000-04-04 01:40:52 +0000218 # '.get()' rather than a straight lookup.
219 self.have_run = {}
220
221 # Now we'll use the attrs dictionary (ultimately, keyword args from
Greg Ward82715e12000-04-21 02:28:14 +0000222 # the setup script) to possibly override any or all of these
223 # distribution options.
224
Greg Wardfe6462c2000-04-04 01:40:52 +0000225 if attrs:
Greg Wardfe6462c2000-04-04 01:40:52 +0000226 # Pull out the set of command options and work on them
227 # specifically. Note that this order guarantees that aliased
228 # command options will override any supplied redundantly
229 # through the general options dictionary.
Greg Wardfd7b91e2000-09-26 01:52:25 +0000230 options = attrs.get('options')
Greg Wardfe6462c2000-04-04 01:40:52 +0000231 if options:
232 del attrs['options']
233 for (command, cmd_options) in options.items():
Greg Ward0e48cfd2000-05-26 01:00:15 +0000234 opt_dict = self.get_option_dict(command)
235 for (opt, val) in cmd_options.items():
236 opt_dict[opt] = ("setup script", val)
Greg Wardfe6462c2000-04-04 01:40:52 +0000237
Guido van Rossume2b70bc2006-08-18 22:13:04 +0000238 if 'licence' in attrs:
Andrew M. Kuchlinga52b8522003-03-03 20:07:27 +0000239 attrs['license'] = attrs['licence']
240 del attrs['licence']
241 msg = "'licence' distribution option is deprecated; use 'license'"
242 if warnings is not None:
243 warnings.warn(msg)
244 else:
245 sys.stderr.write(msg + "\n")
246
Greg Wardfe6462c2000-04-04 01:40:52 +0000247 # Now work on the rest of the attributes. Any attribute that's
248 # not already defined is invalid!
249 for (key,val) in attrs.items():
Fred Drakedb7b0022005-03-20 22:19:47 +0000250 if hasattr(self.metadata, "set_" + key):
251 getattr(self.metadata, "set_" + key)(val)
252 elif hasattr(self.metadata, key):
Greg Wardfd7b91e2000-09-26 01:52:25 +0000253 setattr(self.metadata, key, val)
254 elif hasattr(self, key):
255 setattr(self, key, val)
Anthony Baxter73cc8472004-10-13 13:22:34 +0000256 else:
Andrew M. Kuchlingff4ad9a2002-10-31 13:22:41 +0000257 msg = "Unknown distribution option: %s" % repr(key)
258 if warnings is not None:
259 warnings.warn(msg)
260 else:
261 sys.stderr.write(msg + "\n")
Greg Wardfe6462c2000-04-04 01:40:52 +0000262
Andrew M. Kuchlinga7210ed2001-03-22 03:06:52 +0000263 self.finalize_options()
Fred Drakeb94b8492001-12-06 20:51:35 +0000264
Greg Wardfe6462c2000-04-04 01:40:52 +0000265
Greg Ward0e48cfd2000-05-26 01:00:15 +0000266 def get_option_dict (self, command):
267 """Get the option dictionary for a given command. If that
268 command's option dictionary hasn't been created yet, then create it
269 and return the new dictionary; otherwise, return the existing
270 option dictionary.
271 """
272
273 dict = self.command_options.get(command)
274 if dict is None:
275 dict = self.command_options[command] = {}
276 return dict
277
278
Greg Wardc32d9a62000-05-28 23:53:06 +0000279 def dump_option_dicts (self, header=None, commands=None, indent=""):
280 from pprint import pformat
281
282 if commands is None: # dump all command option dicts
Guido van Rossumd4ee1672007-10-15 01:27:53 +0000283 commands = sorted(self.command_options.keys())
Greg Wardc32d9a62000-05-28 23:53:06 +0000284
285 if header is not None:
Guido van Rossumbe19ed72007-02-09 05:37:30 +0000286 print(indent + header)
Greg Wardc32d9a62000-05-28 23:53:06 +0000287 indent = indent + " "
288
289 if not commands:
Guido van Rossumbe19ed72007-02-09 05:37:30 +0000290 print(indent + "no commands known yet")
Greg Wardc32d9a62000-05-28 23:53:06 +0000291 return
292
293 for cmd_name in commands:
294 opt_dict = self.command_options.get(cmd_name)
295 if opt_dict is None:
Guido van Rossumbe19ed72007-02-09 05:37:30 +0000296 print(indent + "no option dict for '%s' command" % cmd_name)
Greg Wardc32d9a62000-05-28 23:53:06 +0000297 else:
Guido van Rossumbe19ed72007-02-09 05:37:30 +0000298 print(indent + "option dict for '%s' command:" % cmd_name)
Greg Wardc32d9a62000-05-28 23:53:06 +0000299 out = pformat(opt_dict)
Neal Norwitz9d72bb42007-04-17 08:48:32 +0000300 for line in out.split("\n"):
Guido van Rossumbe19ed72007-02-09 05:37:30 +0000301 print(indent + " " + line)
Greg Wardc32d9a62000-05-28 23:53:06 +0000302
Greg Wardc32d9a62000-05-28 23:53:06 +0000303
304
Greg Wardd5d8a992000-05-23 01:42:17 +0000305 # -- Config file finding/parsing methods ---------------------------
306
Gregory P. Smith14263542000-05-12 00:41:33 +0000307 def find_config_files (self):
308 """Find as many configuration files as should be processed for this
309 platform, and return a list of filenames in the order in which they
310 should be parsed. The filenames returned are guaranteed to exist
311 (modulo nasty race conditions).
312
Andrew M. Kuchlingd303b612001-12-06 16:32:05 +0000313 There are three possible config files: distutils.cfg in the
314 Distutils installation directory (ie. where the top-level
315 Distutils __inst__.py file lives), a file in the user's home
316 directory named .pydistutils.cfg on Unix and pydistutils.cfg
317 on Windows/Mac, and setup.cfg in the current directory.
Greg Wardd5d8a992000-05-23 01:42:17 +0000318 """
Gregory P. Smith14263542000-05-12 00:41:33 +0000319 files = []
Greg Wardacf3f6a2000-06-07 02:26:19 +0000320 check_environ()
Gregory P. Smith14263542000-05-12 00:41:33 +0000321
Greg Ward11696872000-06-07 02:29:03 +0000322 # Where to look for the system-wide Distutils config file
323 sys_dir = os.path.dirname(sys.modules['distutils'].__file__)
324
325 # Look for the system config file
326 sys_file = os.path.join(sys_dir, "distutils.cfg")
Greg Wardacf3f6a2000-06-07 02:26:19 +0000327 if os.path.isfile(sys_file):
328 files.append(sys_file)
Gregory P. Smith14263542000-05-12 00:41:33 +0000329
Greg Ward11696872000-06-07 02:29:03 +0000330 # What to call the per-user config file
331 if os.name == 'posix':
Jeremy Hylton65d6edb2000-07-07 20:45:21 +0000332 user_filename = ".pydistutils.cfg"
333 else:
334 user_filename = "pydistutils.cfg"
Greg Wardfa9ff762000-10-14 04:06:40 +0000335
Greg Ward11696872000-06-07 02:29:03 +0000336 # And look for the user config file
Alexandre Vassalotti5f8ced22008-05-16 00:03:33 +0000337 user_file = os.path.join(os.path.expanduser('~'), user_filename)
338 if os.path.isfile(user_file):
339 files.append(user_file)
Gregory P. Smith14263542000-05-12 00:41:33 +0000340
Gregory P. Smith14263542000-05-12 00:41:33 +0000341 # All platforms support local setup.cfg
342 local_file = "setup.cfg"
343 if os.path.isfile(local_file):
344 files.append(local_file)
345
346 return files
347
Gregory P. Smith14263542000-05-12 00:41:33 +0000348
349 def parse_config_files (self, filenames=None):
350
Alexandre Vassalotti1d1eaa42008-05-14 22:59:42 +0000351 from configparser import ConfigParser
Gregory P. Smith14263542000-05-12 00:41:33 +0000352
353 if filenames is None:
354 filenames = self.find_config_files()
355
Guido van Rossumbe19ed72007-02-09 05:37:30 +0000356 if DEBUG: print("Distribution.parse_config_files():")
Greg Ward47460772000-05-23 03:47:35 +0000357
Gregory P. Smith14263542000-05-12 00:41:33 +0000358 parser = ConfigParser()
Greg Wardd5d8a992000-05-23 01:42:17 +0000359 for filename in filenames:
Guido van Rossumbe19ed72007-02-09 05:37:30 +0000360 if DEBUG: print(" reading", filename)
Greg Wardd5d8a992000-05-23 01:42:17 +0000361 parser.read(filename)
362 for section in parser.sections():
363 options = parser.options(section)
Greg Ward0e48cfd2000-05-26 01:00:15 +0000364 opt_dict = self.get_option_dict(section)
Gregory P. Smith14263542000-05-12 00:41:33 +0000365
Greg Wardd5d8a992000-05-23 01:42:17 +0000366 for opt in options:
367 if opt != '__name__':
Greg Wardceb9e222000-09-25 01:23:52 +0000368 val = parser.get(section,opt)
Neal Norwitz9d72bb42007-04-17 08:48:32 +0000369 opt = opt.replace('-', '_')
Greg Wardceb9e222000-09-25 01:23:52 +0000370 opt_dict[opt] = (filename, val)
Gregory P. Smith14263542000-05-12 00:41:33 +0000371
Greg Ward47460772000-05-23 03:47:35 +0000372 # Make the ConfigParser forget everything (so we retain
Fred Drakef06116d2004-02-17 22:35:19 +0000373 # the original filenames that options come from)
Greg Ward47460772000-05-23 03:47:35 +0000374 parser.__init__()
Gregory P. Smith14263542000-05-12 00:41:33 +0000375
Greg Wardceb9e222000-09-25 01:23:52 +0000376 # If there was a "global" section in the config file, use it
377 # to set Distribution options.
378
Guido van Rossume2b70bc2006-08-18 22:13:04 +0000379 if 'global' in self.command_options:
Greg Wardceb9e222000-09-25 01:23:52 +0000380 for (opt, (src, val)) in self.command_options['global'].items():
381 alias = self.negative_opt.get(opt)
382 try:
383 if alias:
384 setattr(self, alias, not strtobool(val))
385 elif opt in ('verbose', 'dry_run'): # ugh!
386 setattr(self, opt, strtobool(val))
Fred Draked04573f2004-08-03 16:37:40 +0000387 else:
388 setattr(self, opt, val)
Guido van Rossumb940e112007-01-10 16:19:56 +0000389 except ValueError as msg:
Collin Winter5b7e9d72007-08-30 03:52:21 +0000390 raise DistutilsOptionError(msg)
Greg Wardceb9e222000-09-25 01:23:52 +0000391
Gregory P. Smith14263542000-05-12 00:41:33 +0000392
Greg Wardd5d8a992000-05-23 01:42:17 +0000393 # -- Command-line parsing methods ----------------------------------
394
Greg Ward9821bf42000-08-29 01:15:18 +0000395 def parse_command_line (self):
396 """Parse the setup script's command line, taken from the
397 'script_args' instance attribute (which defaults to 'sys.argv[1:]'
398 -- see 'setup()' in core.py). This list is first processed for
399 "global options" -- options that set attributes of the Distribution
400 instance. Then, it is alternately scanned for Distutils commands
401 and options for that command. Each new command terminates the
402 options for the previous command. The allowed options for a
403 command are determined by the 'user_options' attribute of the
404 command class -- thus, we have to be able to load command classes
405 in order to parse the command line. Any error in that 'options'
406 attribute raises DistutilsGetoptError; any error on the
407 command-line raises DistutilsArgError. If no Distutils commands
408 were found on the command line, raises DistutilsArgError. Return
Greg Wardceb9e222000-09-25 01:23:52 +0000409 true if command-line was successfully parsed and we should carry
Greg Ward9821bf42000-08-29 01:15:18 +0000410 on with executing commands; false if no errors but we shouldn't
411 execute commands (currently, this only happens if user asks for
412 help).
Greg Wardd5d8a992000-05-23 01:42:17 +0000413 """
Andrew M. Kuchling3f819ec2001-01-15 16:09:35 +0000414 #
Fred Drake981a1782001-08-10 18:59:30 +0000415 # We now have enough information to show the Macintosh dialog
416 # that allows the user to interactively specify the "command line".
Andrew M. Kuchling3f819ec2001-01-15 16:09:35 +0000417 #
Fred Draked04573f2004-08-03 16:37:40 +0000418 toplevel_options = self._get_toplevel_options()
Andrew M. Kuchling3f819ec2001-01-15 16:09:35 +0000419 if sys.platform == 'mac':
420 import EasyDialogs
421 cmdlist = self.get_command_list()
422 self.script_args = EasyDialogs.GetArgv(
Fred Draked04573f2004-08-03 16:37:40 +0000423 toplevel_options + self.display_options, cmdlist)
Fred Drakeb94b8492001-12-06 20:51:35 +0000424
Greg Wardfe6462c2000-04-04 01:40:52 +0000425 # We have to parse the command line a bit at a time -- global
426 # options, then the first command, then its options, and so on --
427 # because each command will be handled by a different class, and
Greg Wardd5d8a992000-05-23 01:42:17 +0000428 # the options that are valid for a particular class aren't known
429 # until we have loaded the command class, which doesn't happen
430 # until we know what the command is.
Greg Wardfe6462c2000-04-04 01:40:52 +0000431
432 self.commands = []
Fred Draked04573f2004-08-03 16:37:40 +0000433 parser = FancyGetopt(toplevel_options + self.display_options)
Greg Wardfd7b91e2000-09-26 01:52:25 +0000434 parser.set_negative_aliases(self.negative_opt)
Andrew M. Kuchlingfa7dc572001-08-10 18:49:23 +0000435 parser.set_aliases({'licence': 'license'})
Greg Wardfd7b91e2000-09-26 01:52:25 +0000436 args = parser.getopt(args=self.script_args, object=self)
Greg Ward82715e12000-04-21 02:28:14 +0000437 option_order = parser.get_option_order()
Jeremy Hyltoncd8a1142002-06-04 20:14:43 +0000438 log.set_verbosity(self.verbose)
Greg Wardfe6462c2000-04-04 01:40:52 +0000439
Greg Ward82715e12000-04-21 02:28:14 +0000440 # for display options we return immediately
441 if self.handle_display_options(option_order):
Greg Wardfe6462c2000-04-04 01:40:52 +0000442 return
Fred Drakeb94b8492001-12-06 20:51:35 +0000443
Greg Wardfe6462c2000-04-04 01:40:52 +0000444 while args:
Greg Wardd5d8a992000-05-23 01:42:17 +0000445 args = self._parse_command_opts(parser, args)
446 if args is None: # user asked for help (and got it)
Greg Wardfe6462c2000-04-04 01:40:52 +0000447 return
Greg Wardfe6462c2000-04-04 01:40:52 +0000448
Greg Wardd5d8a992000-05-23 01:42:17 +0000449 # Handle the cases of --help as a "global" option, ie.
450 # "setup.py --help" and "setup.py --help command ...". For the
451 # former, we show global options (--verbose, --dry-run, etc.)
452 # and display-only options (--name, --version, etc.); for the
453 # latter, we omit the display-only options and show help for
454 # each command listed on the command line.
Greg Wardfe6462c2000-04-04 01:40:52 +0000455 if self.help:
Greg Wardd5d8a992000-05-23 01:42:17 +0000456 self._show_help(parser,
457 display_options=len(self.commands) == 0,
458 commands=self.commands)
Greg Wardfe6462c2000-04-04 01:40:52 +0000459 return
460
461 # Oops, no commands found -- an end-user error
462 if not self.commands:
Collin Winter5b7e9d72007-08-30 03:52:21 +0000463 raise DistutilsArgError("no commands supplied")
Greg Wardfe6462c2000-04-04 01:40:52 +0000464
465 # All is well: return true
Collin Winter5b7e9d72007-08-30 03:52:21 +0000466 return True
Greg Wardfe6462c2000-04-04 01:40:52 +0000467
Fred Draked04573f2004-08-03 16:37:40 +0000468 def _get_toplevel_options (self):
469 """Return the non-display options recognized at the top level.
470
471 This includes options that are recognized *only* at the top
472 level as well as options recognized for commands.
473 """
474 return self.global_options + [
475 ("command-packages=", None,
476 "list of packages that provide distutils commands"),
477 ]
478
Greg Wardd5d8a992000-05-23 01:42:17 +0000479 def _parse_command_opts (self, parser, args):
Greg Wardd5d8a992000-05-23 01:42:17 +0000480 """Parse the command-line options for a single command.
481 'parser' must be a FancyGetopt instance; 'args' must be the list
482 of arguments, starting with the current command (whose options
483 we are about to parse). Returns a new version of 'args' with
484 the next command at the front of the list; will be the empty
485 list if there are no more commands on the command line. Returns
486 None if the user asked for help on this command.
487 """
488 # late import because of mutual dependence between these modules
489 from distutils.cmd import Command
490
491 # Pull the current command from the head of the command line
492 command = args[0]
Greg Wardfd7b91e2000-09-26 01:52:25 +0000493 if not command_re.match(command):
Collin Winter5b7e9d72007-08-30 03:52:21 +0000494 raise SystemExit("invalid command name '%s'" % command)
Greg Wardfd7b91e2000-09-26 01:52:25 +0000495 self.commands.append(command)
Greg Wardd5d8a992000-05-23 01:42:17 +0000496
497 # Dig up the command class that implements this command, so we
498 # 1) know that it's a valid command, and 2) know which options
499 # it takes.
500 try:
Greg Wardfd7b91e2000-09-26 01:52:25 +0000501 cmd_class = self.get_command_class(command)
Guido van Rossumb940e112007-01-10 16:19:56 +0000502 except DistutilsModuleError as msg:
Collin Winter5b7e9d72007-08-30 03:52:21 +0000503 raise DistutilsArgError(msg)
Greg Wardd5d8a992000-05-23 01:42:17 +0000504
505 # Require that the command class be derived from Command -- want
506 # to be sure that the basic "command" interface is implemented.
Greg Wardfd7b91e2000-09-26 01:52:25 +0000507 if not issubclass(cmd_class, Command):
Collin Winter5b7e9d72007-08-30 03:52:21 +0000508 raise DistutilsClassError(
509 "command class %s must subclass Command" % cmd_class)
Greg Wardd5d8a992000-05-23 01:42:17 +0000510
511 # Also make sure that the command object provides a list of its
512 # known options.
Greg Wardfd7b91e2000-09-26 01:52:25 +0000513 if not (hasattr(cmd_class, 'user_options') and
Guido van Rossum13257902007-06-07 23:15:56 +0000514 isinstance(cmd_class.user_options, list)):
Collin Winter5b7e9d72007-08-30 03:52:21 +0000515 raise DistutilsClassError(("command class %s must provide " +
Greg Wardd5d8a992000-05-23 01:42:17 +0000516 "'user_options' attribute (a list of tuples)") % \
Collin Winter5b7e9d72007-08-30 03:52:21 +0000517 cmd_class)
Greg Wardd5d8a992000-05-23 01:42:17 +0000518
519 # If the command class has a list of negative alias options,
520 # merge it in with the global negative aliases.
521 negative_opt = self.negative_opt
Greg Wardfd7b91e2000-09-26 01:52:25 +0000522 if hasattr(cmd_class, 'negative_opt'):
523 negative_opt = copy(negative_opt)
524 negative_opt.update(cmd_class.negative_opt)
Greg Wardd5d8a992000-05-23 01:42:17 +0000525
Greg Wardfa9ff762000-10-14 04:06:40 +0000526 # Check for help_options in command class. They have a different
527 # format (tuple of four) so we need to preprocess them here.
Jeremy Hylton65d6edb2000-07-07 20:45:21 +0000528 if (hasattr(cmd_class, 'help_options') and
Guido van Rossum13257902007-06-07 23:15:56 +0000529 isinstance(cmd_class.help_options, list)):
Greg Ward2ff78872000-06-24 00:23:20 +0000530 help_options = fix_help_options(cmd_class.help_options)
531 else:
Greg Ward55fced32000-06-24 01:22:41 +0000532 help_options = []
Greg Ward2ff78872000-06-24 00:23:20 +0000533
Greg Ward9d17a7a2000-06-07 03:00:06 +0000534
Greg Wardd5d8a992000-05-23 01:42:17 +0000535 # All commands support the global options too, just by adding
536 # in 'global_options'.
Greg Wardfd7b91e2000-09-26 01:52:25 +0000537 parser.set_option_table(self.global_options +
538 cmd_class.user_options +
539 help_options)
540 parser.set_negative_aliases(negative_opt)
541 (args, opts) = parser.getopt(args[1:])
Greg Ward47460772000-05-23 03:47:35 +0000542 if hasattr(opts, 'help') and opts.help:
Greg Wardd5d8a992000-05-23 01:42:17 +0000543 self._show_help(parser, display_options=0, commands=[cmd_class])
544 return
545
Jeremy Hylton65d6edb2000-07-07 20:45:21 +0000546 if (hasattr(cmd_class, 'help_options') and
Guido van Rossum13257902007-06-07 23:15:56 +0000547 isinstance(cmd_class.help_options, list)):
Jeremy Hylton65d6edb2000-07-07 20:45:21 +0000548 help_option_found=0
549 for (help_option, short, desc, func) in cmd_class.help_options:
550 if hasattr(opts, parser.get_attr_name(help_option)):
551 help_option_found=1
Greg Wardfa9ff762000-10-14 04:06:40 +0000552 #print "showing help for option %s of command %s" % \
Greg Ward2ff78872000-06-24 00:23:20 +0000553 # (help_option[0],cmd_class)
Greg Ward55fced32000-06-24 01:22:41 +0000554
Guido van Rossumd59da4b2007-05-22 18:11:13 +0000555 if hasattr(func, '__call__'):
Jeremy Hylton65d6edb2000-07-07 20:45:21 +0000556 func()
Greg Ward55fced32000-06-24 01:22:41 +0000557 else:
Fred Drake981a1782001-08-10 18:59:30 +0000558 raise DistutilsClassError(
Walter Dörwald70a6b492004-02-12 17:35:32 +0000559 "invalid help function %r for help option '%s': "
Fred Drake981a1782001-08-10 18:59:30 +0000560 "must be a callable object (function, etc.)"
Walter Dörwald70a6b492004-02-12 17:35:32 +0000561 % (func, help_option))
Greg Ward55fced32000-06-24 01:22:41 +0000562
Fred Drakeb94b8492001-12-06 20:51:35 +0000563 if help_option_found:
Jeremy Hylton65d6edb2000-07-07 20:45:21 +0000564 return
Greg Ward9d17a7a2000-06-07 03:00:06 +0000565
Greg Wardd5d8a992000-05-23 01:42:17 +0000566 # Put the options from the command-line into their official
567 # holding pen, the 'command_options' dictionary.
Greg Ward0e48cfd2000-05-26 01:00:15 +0000568 opt_dict = self.get_option_dict(command)
Greg Wardd5d8a992000-05-23 01:42:17 +0000569 for (name, value) in vars(opts).items():
Greg Ward0e48cfd2000-05-26 01:00:15 +0000570 opt_dict[name] = ("command line", value)
Greg Wardd5d8a992000-05-23 01:42:17 +0000571
572 return args
573
Andrew M. Kuchlinga7210ed2001-03-22 03:06:52 +0000574 def finalize_options (self):
575 """Set final values for all the options on the Distribution
576 instance, analogous to the .finalize_options() method of Command
577 objects.
578 """
579
Andrew M. Kuchlinga7210ed2001-03-22 03:06:52 +0000580 keywords = self.metadata.keywords
581 if keywords is not None:
Guido van Rossum3172c5d2007-10-16 18:12:55 +0000582 if isinstance(keywords, str):
Neal Norwitz9d72bb42007-04-17 08:48:32 +0000583 keywordlist = keywords.split(',')
584 self.metadata.keywords = [x.strip() for x in keywordlist]
Andrew M. Kuchlinga7210ed2001-03-22 03:06:52 +0000585
586 platforms = self.metadata.platforms
587 if platforms is not None:
Guido van Rossum3172c5d2007-10-16 18:12:55 +0000588 if isinstance(platforms, str):
Neal Norwitz9d72bb42007-04-17 08:48:32 +0000589 platformlist = platforms.split(',')
590 self.metadata.platforms = [x.strip() for x in platformlist]
Andrew M. Kuchlinga7210ed2001-03-22 03:06:52 +0000591
Greg Wardd5d8a992000-05-23 01:42:17 +0000592 def _show_help (self,
593 parser,
594 global_options=1,
595 display_options=1,
596 commands=[]):
597 """Show help for the setup script command-line in the form of
598 several lists of command-line options. 'parser' should be a
599 FancyGetopt instance; do not expect it to be returned in the
600 same state, as its option table will be reset to make it
601 generate the correct help text.
602
603 If 'global_options' is true, lists the global options:
604 --verbose, --dry-run, etc. If 'display_options' is true, lists
605 the "display-only" options: --name, --version, etc. Finally,
606 lists per-command help for every command name or command class
607 in 'commands'.
608 """
609 # late import because of mutual dependence between these modules
Greg Ward9821bf42000-08-29 01:15:18 +0000610 from distutils.core import gen_usage
Greg Wardd5d8a992000-05-23 01:42:17 +0000611 from distutils.cmd import Command
612
613 if global_options:
Fred Draked04573f2004-08-03 16:37:40 +0000614 if display_options:
615 options = self._get_toplevel_options()
616 else:
617 options = self.global_options
618 parser.set_option_table(options)
Martin v. Löwis8ed338a2005-03-03 08:12:27 +0000619 parser.print_help(self.common_usage + "\nGlobal options:")
Guido van Rossumbe19ed72007-02-09 05:37:30 +0000620 print()
Greg Wardd5d8a992000-05-23 01:42:17 +0000621
622 if display_options:
Greg Wardfd7b91e2000-09-26 01:52:25 +0000623 parser.set_option_table(self.display_options)
624 parser.print_help(
Greg Wardd5d8a992000-05-23 01:42:17 +0000625 "Information display options (just display " +
626 "information, ignore any commands)")
Guido van Rossumbe19ed72007-02-09 05:37:30 +0000627 print()
Greg Wardd5d8a992000-05-23 01:42:17 +0000628
629 for command in self.commands:
Guido van Rossum13257902007-06-07 23:15:56 +0000630 if isinstance(command, type) and issubclass(command, Command):
Greg Wardd5d8a992000-05-23 01:42:17 +0000631 klass = command
632 else:
Greg Wardfd7b91e2000-09-26 01:52:25 +0000633 klass = self.get_command_class(command)
Jeremy Hylton65d6edb2000-07-07 20:45:21 +0000634 if (hasattr(klass, 'help_options') and
Guido van Rossum13257902007-06-07 23:15:56 +0000635 isinstance(klass.help_options, list)):
Greg Wardfd7b91e2000-09-26 01:52:25 +0000636 parser.set_option_table(klass.user_options +
637 fix_help_options(klass.help_options))
Jeremy Hylton65d6edb2000-07-07 20:45:21 +0000638 else:
Greg Wardfd7b91e2000-09-26 01:52:25 +0000639 parser.set_option_table(klass.user_options)
640 parser.print_help("Options for '%s' command:" % klass.__name__)
Guido van Rossumbe19ed72007-02-09 05:37:30 +0000641 print()
Greg Wardd5d8a992000-05-23 01:42:17 +0000642
Guido van Rossumbe19ed72007-02-09 05:37:30 +0000643 print(gen_usage(self.script_name))
Greg Wardd5d8a992000-05-23 01:42:17 +0000644 return
645
Greg Wardd5d8a992000-05-23 01:42:17 +0000646
Greg Ward82715e12000-04-21 02:28:14 +0000647 def handle_display_options (self, option_order):
648 """If there were any non-global "display-only" options
Greg Wardd5d8a992000-05-23 01:42:17 +0000649 (--help-commands or the metadata display options) on the command
650 line, display the requested info and return true; else return
651 false.
652 """
Greg Ward9821bf42000-08-29 01:15:18 +0000653 from distutils.core import gen_usage
Greg Ward82715e12000-04-21 02:28:14 +0000654
655 # User just wants a list of commands -- we'll print it out and stop
656 # processing now (ie. if they ran "setup --help-commands foo bar",
657 # we ignore "foo bar").
658 if self.help_commands:
Greg Wardfd7b91e2000-09-26 01:52:25 +0000659 self.print_commands()
Guido van Rossumbe19ed72007-02-09 05:37:30 +0000660 print()
661 print(gen_usage(self.script_name))
Greg Ward82715e12000-04-21 02:28:14 +0000662 return 1
663
664 # If user supplied any of the "display metadata" options, then
665 # display that metadata in the order in which the user supplied the
666 # metadata options.
667 any_display_options = 0
668 is_display_option = {}
669 for option in self.display_options:
670 is_display_option[option[0]] = 1
671
672 for (opt, val) in option_order:
673 if val and is_display_option.get(opt):
Greg Ward2f2b6c62000-09-25 01:58:07 +0000674 opt = translate_longopt(opt)
Andrew M. Kuchlinga7210ed2001-03-22 03:06:52 +0000675 value = getattr(self.metadata, "get_"+opt)()
676 if opt in ['keywords', 'platforms']:
Neal Norwitz9d72bb42007-04-17 08:48:32 +0000677 print(','.join(value))
Fred Drakedb7b0022005-03-20 22:19:47 +0000678 elif opt in ('classifiers', 'provides', 'requires',
679 'obsoletes'):
Neal Norwitz9d72bb42007-04-17 08:48:32 +0000680 print('\n'.join(value))
Andrew M. Kuchlinga7210ed2001-03-22 03:06:52 +0000681 else:
Guido van Rossumbe19ed72007-02-09 05:37:30 +0000682 print(value)
Greg Ward82715e12000-04-21 02:28:14 +0000683 any_display_options = 1
684
685 return any_display_options
686
Greg Wardfe6462c2000-04-04 01:40:52 +0000687 def print_command_list (self, commands, header, max_length):
688 """Print a subset of the list of all commands -- used by
Greg Wardd5d8a992000-05-23 01:42:17 +0000689 'print_commands()'.
690 """
Guido van Rossumbe19ed72007-02-09 05:37:30 +0000691 print(header + ":")
Greg Wardfe6462c2000-04-04 01:40:52 +0000692
693 for cmd in commands:
Greg Wardfd7b91e2000-09-26 01:52:25 +0000694 klass = self.cmdclass.get(cmd)
Greg Wardfe6462c2000-04-04 01:40:52 +0000695 if not klass:
Greg Wardfd7b91e2000-09-26 01:52:25 +0000696 klass = self.get_command_class(cmd)
Greg Wardfe6462c2000-04-04 01:40:52 +0000697 try:
698 description = klass.description
699 except AttributeError:
700 description = "(no description available)"
701
Guido van Rossumbe19ed72007-02-09 05:37:30 +0000702 print(" %-*s %s" % (max_length, cmd, description))
Greg Wardfe6462c2000-04-04 01:40:52 +0000703
Greg Wardfe6462c2000-04-04 01:40:52 +0000704
705 def print_commands (self):
Greg Wardd5d8a992000-05-23 01:42:17 +0000706 """Print out a help message listing all available commands with a
707 description of each. The list is divided into "standard commands"
708 (listed in distutils.command.__all__) and "extra commands"
709 (mentioned in self.cmdclass, but not a standard command). The
710 descriptions come from the command class attribute
711 'description'.
712 """
Greg Wardfe6462c2000-04-04 01:40:52 +0000713 import distutils.command
714 std_commands = distutils.command.__all__
715 is_std = {}
716 for cmd in std_commands:
717 is_std[cmd] = 1
718
719 extra_commands = []
720 for cmd in self.cmdclass.keys():
721 if not is_std.get(cmd):
Greg Wardfd7b91e2000-09-26 01:52:25 +0000722 extra_commands.append(cmd)
Greg Wardfe6462c2000-04-04 01:40:52 +0000723
724 max_length = 0
725 for cmd in (std_commands + extra_commands):
Greg Wardfd7b91e2000-09-26 01:52:25 +0000726 if len(cmd) > max_length:
727 max_length = len(cmd)
Greg Wardfe6462c2000-04-04 01:40:52 +0000728
Greg Wardfd7b91e2000-09-26 01:52:25 +0000729 self.print_command_list(std_commands,
730 "Standard commands",
731 max_length)
Greg Wardfe6462c2000-04-04 01:40:52 +0000732 if extra_commands:
Guido van Rossumbe19ed72007-02-09 05:37:30 +0000733 print()
Greg Wardfd7b91e2000-09-26 01:52:25 +0000734 self.print_command_list(extra_commands,
735 "Extra commands",
736 max_length)
Greg Wardfe6462c2000-04-04 01:40:52 +0000737
Greg Wardf6fc8752000-11-11 02:47:11 +0000738 def get_command_list (self):
739 """Get a list of (command, description) tuples.
740 The list is divided into "standard commands" (listed in
741 distutils.command.__all__) and "extra commands" (mentioned in
742 self.cmdclass, but not a standard command). The descriptions come
743 from the command class attribute 'description'.
744 """
745 # Currently this is only used on Mac OS, for the Mac-only GUI
746 # Distutils interface (by Jack Jansen)
Greg Wardf6fc8752000-11-11 02:47:11 +0000747 import distutils.command
748 std_commands = distutils.command.__all__
749 is_std = {}
750 for cmd in std_commands:
751 is_std[cmd] = 1
752
753 extra_commands = []
754 for cmd in self.cmdclass.keys():
755 if not is_std.get(cmd):
756 extra_commands.append(cmd)
757
758 rv = []
759 for cmd in (std_commands + extra_commands):
760 klass = self.cmdclass.get(cmd)
761 if not klass:
762 klass = self.get_command_class(cmd)
763 try:
764 description = klass.description
765 except AttributeError:
766 description = "(no description available)"
767 rv.append((cmd, description))
768 return rv
Greg Wardfe6462c2000-04-04 01:40:52 +0000769
770 # -- Command class/object methods ----------------------------------
771
Fred Draked04573f2004-08-03 16:37:40 +0000772 def get_command_packages (self):
773 """Return a list of packages from which commands are loaded."""
774 pkgs = self.command_packages
775 if not isinstance(pkgs, type([])):
Neal Norwitz9d72bb42007-04-17 08:48:32 +0000776 pkgs = (pkgs or "").split(",")
Fred Draked04573f2004-08-03 16:37:40 +0000777 for i in range(len(pkgs)):
Neal Norwitz9d72bb42007-04-17 08:48:32 +0000778 pkgs[i] = pkgs[i].strip()
Guido van Rossumc1f779c2007-07-03 08:25:58 +0000779 pkgs = [p for p in pkgs if p]
Fred Draked04573f2004-08-03 16:37:40 +0000780 if "distutils.command" not in pkgs:
781 pkgs.insert(0, "distutils.command")
782 self.command_packages = pkgs
783 return pkgs
784
Greg Wardd5d8a992000-05-23 01:42:17 +0000785 def get_command_class (self, command):
786 """Return the class that implements the Distutils command named by
787 'command'. First we check the 'cmdclass' dictionary; if the
788 command is mentioned there, we fetch the class object from the
789 dictionary and return it. Otherwise we load the command module
790 ("distutils.command." + command) and fetch the command class from
791 the module. The loaded class is also stored in 'cmdclass'
792 to speed future calls to 'get_command_class()'.
Greg Wardfe6462c2000-04-04 01:40:52 +0000793
Gregory P. Smith14263542000-05-12 00:41:33 +0000794 Raises DistutilsModuleError if the expected module could not be
Greg Wardd5d8a992000-05-23 01:42:17 +0000795 found, or if that module does not define the expected class.
796 """
797 klass = self.cmdclass.get(command)
798 if klass:
799 return klass
Greg Wardfe6462c2000-04-04 01:40:52 +0000800
Fred Draked04573f2004-08-03 16:37:40 +0000801 for pkgname in self.get_command_packages():
802 module_name = "%s.%s" % (pkgname, command)
803 klass_name = command
Greg Wardfe6462c2000-04-04 01:40:52 +0000804
Fred Draked04573f2004-08-03 16:37:40 +0000805 try:
806 __import__ (module_name)
807 module = sys.modules[module_name]
808 except ImportError:
809 continue
Greg Wardfe6462c2000-04-04 01:40:52 +0000810
Fred Draked04573f2004-08-03 16:37:40 +0000811 try:
812 klass = getattr(module, klass_name)
813 except AttributeError:
Collin Winter5b7e9d72007-08-30 03:52:21 +0000814 raise DistutilsModuleError(
815 "invalid command '%s' (no class '%s' in module '%s')"
816 % (command, klass_name, module_name))
Greg Wardfe6462c2000-04-04 01:40:52 +0000817
Fred Draked04573f2004-08-03 16:37:40 +0000818 self.cmdclass[command] = klass
819 return klass
820
821 raise DistutilsModuleError("invalid command '%s'" % command)
822
Greg Wardd5d8a992000-05-23 01:42:17 +0000823 def get_command_obj (self, command, create=1):
824 """Return the command object for 'command'. Normally this object
Greg Ward612eb9f2000-07-27 02:13:20 +0000825 is cached on a previous call to 'get_command_obj()'; if no command
Greg Wardd5d8a992000-05-23 01:42:17 +0000826 object for 'command' is in the cache, then we either create and
827 return it (if 'create' is true) or return None.
828 """
829 cmd_obj = self.command_obj.get(command)
Greg Wardfe6462c2000-04-04 01:40:52 +0000830 if not cmd_obj and create:
Greg Ward2bd3f422000-06-02 01:59:33 +0000831 if DEBUG:
Guido van Rossumbe19ed72007-02-09 05:37:30 +0000832 print("Distribution.get_command_obj(): " \
833 "creating '%s' command object" % command)
Greg Ward47460772000-05-23 03:47:35 +0000834
Greg Wardd5d8a992000-05-23 01:42:17 +0000835 klass = self.get_command_class(command)
Greg Ward47460772000-05-23 03:47:35 +0000836 cmd_obj = self.command_obj[command] = klass(self)
837 self.have_run[command] = 0
838
839 # Set any options that were supplied in config files
840 # or on the command line. (NB. support for error
841 # reporting is lame here: any errors aren't reported
842 # until 'finalize_options()' is called, which means
843 # we won't report the source of the error.)
844 options = self.command_options.get(command)
845 if options:
Greg Wardc32d9a62000-05-28 23:53:06 +0000846 self._set_command_options(cmd_obj, options)
Greg Wardfe6462c2000-04-04 01:40:52 +0000847
848 return cmd_obj
849
Greg Wardc32d9a62000-05-28 23:53:06 +0000850 def _set_command_options (self, command_obj, option_dict=None):
Greg Wardc32d9a62000-05-28 23:53:06 +0000851 """Set the options for 'command_obj' from 'option_dict'. Basically
852 this means copying elements of a dictionary ('option_dict') to
853 attributes of an instance ('command').
854
Greg Wardceb9e222000-09-25 01:23:52 +0000855 'command_obj' must be a Command instance. If 'option_dict' is not
Greg Wardc32d9a62000-05-28 23:53:06 +0000856 supplied, uses the standard option dictionary for this command
857 (from 'self.command_options').
858 """
Greg Wardc32d9a62000-05-28 23:53:06 +0000859 command_name = command_obj.get_command_name()
860 if option_dict is None:
861 option_dict = self.get_option_dict(command_name)
862
Guido van Rossumbe19ed72007-02-09 05:37:30 +0000863 if DEBUG: print(" setting options for '%s' command:" % command_name)
Greg Wardc32d9a62000-05-28 23:53:06 +0000864 for (option, (source, value)) in option_dict.items():
Guido van Rossumbe19ed72007-02-09 05:37:30 +0000865 if DEBUG: print(" %s = %s (from %s)" % (option, value, source))
Greg Wardceb9e222000-09-25 01:23:52 +0000866 try:
Greg Ward2f2b6c62000-09-25 01:58:07 +0000867 bool_opts = map(translate_longopt, command_obj.boolean_options)
Greg Wardceb9e222000-09-25 01:23:52 +0000868 except AttributeError:
869 bool_opts = []
870 try:
871 neg_opt = command_obj.negative_opt
872 except AttributeError:
873 neg_opt = {}
874
875 try:
Guido van Rossum3172c5d2007-10-16 18:12:55 +0000876 is_string = isinstance(value, str)
Guido van Rossume2b70bc2006-08-18 22:13:04 +0000877 if option in neg_opt and is_string:
Greg Wardceb9e222000-09-25 01:23:52 +0000878 setattr(command_obj, neg_opt[option], not strtobool(value))
Greg Ward2c08cf02000-09-27 00:15:37 +0000879 elif option in bool_opts and is_string:
Greg Wardceb9e222000-09-25 01:23:52 +0000880 setattr(command_obj, option, strtobool(value))
881 elif hasattr(command_obj, option):
882 setattr(command_obj, option, value)
883 else:
Collin Winter5b7e9d72007-08-30 03:52:21 +0000884 raise DistutilsOptionError(
885 "error in %s: command '%s' has no such option '%s'"
886 % (source, command_name, option))
Guido van Rossumb940e112007-01-10 16:19:56 +0000887 except ValueError as msg:
Collin Winter5b7e9d72007-08-30 03:52:21 +0000888 raise DistutilsOptionError(msg)
Greg Wardc32d9a62000-05-28 23:53:06 +0000889
Greg Wardf449ea52000-09-16 15:23:28 +0000890 def reinitialize_command (self, command, reinit_subcommands=0):
Greg Wardc32d9a62000-05-28 23:53:06 +0000891 """Reinitializes a command to the state it was in when first
892 returned by 'get_command_obj()': ie., initialized but not yet
Greg Ward7d9c7052000-06-28 01:25:27 +0000893 finalized. This provides the opportunity to sneak option
Greg Wardc32d9a62000-05-28 23:53:06 +0000894 values in programmatically, overriding or supplementing
895 user-supplied values from the config files and command line.
896 You'll have to re-finalize the command object (by calling
897 'finalize_options()' or 'ensure_finalized()') before using it for
Fred Drakeb94b8492001-12-06 20:51:35 +0000898 real.
Greg Wardc32d9a62000-05-28 23:53:06 +0000899
Greg Wardf449ea52000-09-16 15:23:28 +0000900 'command' should be a command name (string) or command object. If
901 'reinit_subcommands' is true, also reinitializes the command's
902 sub-commands, as declared by the 'sub_commands' class attribute (if
903 it has one). See the "install" command for an example. Only
904 reinitializes the sub-commands that actually matter, ie. those
905 whose test predicates return true.
906
Greg Wardc32d9a62000-05-28 23:53:06 +0000907 Returns the reinitialized command object.
908 """
909 from distutils.cmd import Command
910 if not isinstance(command, Command):
911 command_name = command
912 command = self.get_command_obj(command_name)
913 else:
914 command_name = command.get_command_name()
915
916 if not command.finalized:
Greg Ward282c7a02000-06-01 01:09:47 +0000917 return command
Greg Wardc32d9a62000-05-28 23:53:06 +0000918 command.initialize_options()
919 command.finalized = 0
Greg Ward43955c92000-06-06 02:52:36 +0000920 self.have_run[command_name] = 0
Greg Wardc32d9a62000-05-28 23:53:06 +0000921 self._set_command_options(command)
Greg Wardf449ea52000-09-16 15:23:28 +0000922
Greg Wardf449ea52000-09-16 15:23:28 +0000923 if reinit_subcommands:
Greg Wardf449ea52000-09-16 15:23:28 +0000924 for sub in command.get_sub_commands():
Fred Drakeb94b8492001-12-06 20:51:35 +0000925 self.reinitialize_command(sub, reinit_subcommands)
Greg Wardf449ea52000-09-16 15:23:28 +0000926
Greg Wardc32d9a62000-05-28 23:53:06 +0000927 return command
928
Fred Drakeb94b8492001-12-06 20:51:35 +0000929
Greg Wardfe6462c2000-04-04 01:40:52 +0000930 # -- Methods that operate on the Distribution ----------------------
931
932 def announce (self, msg, level=1):
Jeremy Hyltoncd8a1142002-06-04 20:14:43 +0000933 log.debug(msg)
Greg Wardfe6462c2000-04-04 01:40:52 +0000934
935 def run_commands (self):
Greg Ward82715e12000-04-21 02:28:14 +0000936 """Run each command that was seen on the setup script command line.
Greg Wardd5d8a992000-05-23 01:42:17 +0000937 Uses the list of commands found and cache of command objects
Greg Wardfd7b91e2000-09-26 01:52:25 +0000938 created by 'get_command_obj()'.
939 """
Greg Wardfe6462c2000-04-04 01:40:52 +0000940 for cmd in self.commands:
Greg Wardfd7b91e2000-09-26 01:52:25 +0000941 self.run_command(cmd)
Greg Wardfe6462c2000-04-04 01:40:52 +0000942
943
Greg Wardfe6462c2000-04-04 01:40:52 +0000944 # -- Methods that operate on its Commands --------------------------
945
946 def run_command (self, command):
Greg Wardfe6462c2000-04-04 01:40:52 +0000947 """Do whatever it takes to run a command (including nothing at all,
Greg Wardd5d8a992000-05-23 01:42:17 +0000948 if the command has already been run). Specifically: if we have
949 already created and run the command named by 'command', return
950 silently without doing anything. If the command named by 'command'
951 doesn't even have a command object yet, create one. Then invoke
952 'run()' on that command object (or an existing one).
953 """
Greg Wardfe6462c2000-04-04 01:40:52 +0000954 # Already been here, done that? then return silently.
Greg Wardfd7b91e2000-09-26 01:52:25 +0000955 if self.have_run.get(command):
Greg Wardfe6462c2000-04-04 01:40:52 +0000956 return
957
Jeremy Hyltoncd8a1142002-06-04 20:14:43 +0000958 log.info("running %s", command)
Greg Wardfd7b91e2000-09-26 01:52:25 +0000959 cmd_obj = self.get_command_obj(command)
960 cmd_obj.ensure_finalized()
961 cmd_obj.run()
Greg Wardfe6462c2000-04-04 01:40:52 +0000962 self.have_run[command] = 1
963
964
Greg Wardfe6462c2000-04-04 01:40:52 +0000965 # -- Distribution query methods ------------------------------------
966
967 def has_pure_modules (self):
Greg Wardfd7b91e2000-09-26 01:52:25 +0000968 return len(self.packages or self.py_modules or []) > 0
Greg Wardfe6462c2000-04-04 01:40:52 +0000969
970 def has_ext_modules (self):
Greg Wardfd7b91e2000-09-26 01:52:25 +0000971 return self.ext_modules and len(self.ext_modules) > 0
Greg Wardfe6462c2000-04-04 01:40:52 +0000972
973 def has_c_libraries (self):
Greg Wardfd7b91e2000-09-26 01:52:25 +0000974 return self.libraries and len(self.libraries) > 0
Greg Wardfe6462c2000-04-04 01:40:52 +0000975
976 def has_modules (self):
977 return self.has_pure_modules() or self.has_ext_modules()
978
Greg Ward51def7d2000-05-27 01:36:14 +0000979 def has_headers (self):
980 return self.headers and len(self.headers) > 0
981
Greg Ward44a61bb2000-05-20 15:06:48 +0000982 def has_scripts (self):
983 return self.scripts and len(self.scripts) > 0
984
985 def has_data_files (self):
986 return self.data_files and len(self.data_files) > 0
987
Greg Wardfe6462c2000-04-04 01:40:52 +0000988 def is_pure (self):
989 return (self.has_pure_modules() and
990 not self.has_ext_modules() and
991 not self.has_c_libraries())
992
Greg Ward82715e12000-04-21 02:28:14 +0000993 # -- Metadata query methods ----------------------------------------
994
995 # If you're looking for 'get_name()', 'get_version()', and so forth,
996 # they are defined in a sneaky way: the constructor binds self.get_XXX
997 # to self.metadata.get_XXX. The actual code is in the
998 # DistributionMetadata class, below.
999
1000# class Distribution
1001
1002
1003class DistributionMetadata:
1004 """Dummy class to hold the distribution meta-data: name, version,
Greg Wardfd7b91e2000-09-26 01:52:25 +00001005 author, and so forth.
1006 """
Greg Ward82715e12000-04-21 02:28:14 +00001007
Neil Schemenauera8aefe52001-09-03 15:47:21 +00001008 _METHOD_BASENAMES = ("name", "version", "author", "author_email",
1009 "maintainer", "maintainer_email", "url",
1010 "license", "description", "long_description",
1011 "keywords", "platforms", "fullname", "contact",
Andrew M. Kuchlinga52b8522003-03-03 20:07:27 +00001012 "contact_email", "license", "classifiers",
Fred Drakedb7b0022005-03-20 22:19:47 +00001013 "download_url",
1014 # PEP 314
1015 "provides", "requires", "obsoletes",
1016 )
Neil Schemenauera8aefe52001-09-03 15:47:21 +00001017
Greg Ward82715e12000-04-21 02:28:14 +00001018 def __init__ (self):
1019 self.name = None
1020 self.version = None
1021 self.author = None
1022 self.author_email = None
1023 self.maintainer = None
1024 self.maintainer_email = None
1025 self.url = None
Andrew M. Kuchlingfa7dc572001-08-10 18:49:23 +00001026 self.license = None
Greg Ward82715e12000-04-21 02:28:14 +00001027 self.description = None
Greg Warde5a584e2000-04-26 02:26:55 +00001028 self.long_description = None
Andrew M. Kuchlinga7210ed2001-03-22 03:06:52 +00001029 self.keywords = None
1030 self.platforms = None
Andrew M. Kuchling282e2c32003-01-03 15:24:36 +00001031 self.classifiers = None
Andrew M. Kuchling188d85f2003-02-19 14:16:01 +00001032 self.download_url = None
Fred Drakedb7b0022005-03-20 22:19:47 +00001033 # PEP 314
1034 self.provides = None
1035 self.requires = None
1036 self.obsoletes = None
Fred Drakeb94b8492001-12-06 20:51:35 +00001037
Andrew M. Kuchlinga7210ed2001-03-22 03:06:52 +00001038 def write_pkg_info (self, base_dir):
1039 """Write the PKG-INFO file into the release tree.
1040 """
Andrew M. Kuchlinga7210ed2001-03-22 03:06:52 +00001041 pkg_info = open( os.path.join(base_dir, 'PKG-INFO'), 'w')
1042
Fred Drakedb7b0022005-03-20 22:19:47 +00001043 self.write_pkg_file(pkg_info)
Andrew M. Kuchling282e2c32003-01-03 15:24:36 +00001044
Andrew M. Kuchlinga7210ed2001-03-22 03:06:52 +00001045 pkg_info.close()
Fred Drakeb94b8492001-12-06 20:51:35 +00001046
Fred Drakedb7b0022005-03-20 22:19:47 +00001047 def write_pkg_file (self, file):
1048 """Write the PKG-INFO format data to a file object.
1049 """
1050 version = '1.0'
1051 if self.provides or self.requires or self.obsoletes:
1052 version = '1.1'
1053
1054 file.write('Metadata-Version: %s\n' % version)
1055 file.write('Name: %s\n' % self.get_name() )
1056 file.write('Version: %s\n' % self.get_version() )
1057 file.write('Summary: %s\n' % self.get_description() )
1058 file.write('Home-page: %s\n' % self.get_url() )
1059 file.write('Author: %s\n' % self.get_contact() )
1060 file.write('Author-email: %s\n' % self.get_contact_email() )
1061 file.write('License: %s\n' % self.get_license() )
1062 if self.download_url:
1063 file.write('Download-URL: %s\n' % self.download_url)
1064
1065 long_desc = rfc822_escape( self.get_long_description() )
1066 file.write('Description: %s\n' % long_desc)
1067
Neal Norwitz9d72bb42007-04-17 08:48:32 +00001068 keywords = ','.join(self.get_keywords())
Fred Drakedb7b0022005-03-20 22:19:47 +00001069 if keywords:
1070 file.write('Keywords: %s\n' % keywords )
1071
1072 self._write_list(file, 'Platform', self.get_platforms())
1073 self._write_list(file, 'Classifier', self.get_classifiers())
1074
1075 # PEP 314
1076 self._write_list(file, 'Requires', self.get_requires())
1077 self._write_list(file, 'Provides', self.get_provides())
1078 self._write_list(file, 'Obsoletes', self.get_obsoletes())
1079
1080 def _write_list (self, file, name, values):
1081 for value in values:
1082 file.write('%s: %s\n' % (name, value))
1083
Greg Ward82715e12000-04-21 02:28:14 +00001084 # -- Metadata query methods ----------------------------------------
1085
Greg Wardfe6462c2000-04-04 01:40:52 +00001086 def get_name (self):
1087 return self.name or "UNKNOWN"
1088
Greg Ward82715e12000-04-21 02:28:14 +00001089 def get_version(self):
Thomas Hellerbcd89752001-12-06 20:44:19 +00001090 return self.version or "0.0.0"
Greg Wardfe6462c2000-04-04 01:40:52 +00001091
Greg Ward82715e12000-04-21 02:28:14 +00001092 def get_fullname (self):
1093 return "%s-%s" % (self.get_name(), self.get_version())
1094
1095 def get_author(self):
1096 return self.author or "UNKNOWN"
1097
1098 def get_author_email(self):
1099 return self.author_email or "UNKNOWN"
1100
1101 def get_maintainer(self):
1102 return self.maintainer or "UNKNOWN"
1103
1104 def get_maintainer_email(self):
1105 return self.maintainer_email or "UNKNOWN"
1106
1107 def get_contact(self):
1108 return (self.maintainer or
1109 self.author or
1110 "UNKNOWN")
1111
1112 def get_contact_email(self):
1113 return (self.maintainer_email or
1114 self.author_email or
1115 "UNKNOWN")
1116
1117 def get_url(self):
1118 return self.url or "UNKNOWN"
1119
Andrew M. Kuchlingfa7dc572001-08-10 18:49:23 +00001120 def get_license(self):
1121 return self.license or "UNKNOWN"
1122 get_licence = get_license
Fred Drakeb94b8492001-12-06 20:51:35 +00001123
Greg Ward82715e12000-04-21 02:28:14 +00001124 def get_description(self):
1125 return self.description or "UNKNOWN"
Greg Warde5a584e2000-04-26 02:26:55 +00001126
1127 def get_long_description(self):
1128 return self.long_description or "UNKNOWN"
1129
Andrew M. Kuchlinga7210ed2001-03-22 03:06:52 +00001130 def get_keywords(self):
1131 return self.keywords or []
1132
1133 def get_platforms(self):
1134 return self.platforms or ["UNKNOWN"]
1135
Andrew M. Kuchling282e2c32003-01-03 15:24:36 +00001136 def get_classifiers(self):
1137 return self.classifiers or []
1138
Andrew M. Kuchling188d85f2003-02-19 14:16:01 +00001139 def get_download_url(self):
1140 return self.download_url or "UNKNOWN"
1141
Fred Drakedb7b0022005-03-20 22:19:47 +00001142 # PEP 314
1143
1144 def get_requires(self):
1145 return self.requires or []
1146
1147 def set_requires(self, value):
1148 import distutils.versionpredicate
1149 for v in value:
1150 distutils.versionpredicate.VersionPredicate(v)
1151 self.requires = value
1152
1153 def get_provides(self):
1154 return self.provides or []
1155
1156 def set_provides(self, value):
1157 value = [v.strip() for v in value]
1158 for v in value:
1159 import distutils.versionpredicate
Fred Drake227e8ff2005-03-21 06:36:32 +00001160 distutils.versionpredicate.split_provision(v)
Fred Drakedb7b0022005-03-20 22:19:47 +00001161 self.provides = value
1162
1163 def get_obsoletes(self):
1164 return self.obsoletes or []
1165
1166 def set_obsoletes(self, value):
1167 import distutils.versionpredicate
1168 for v in value:
1169 distutils.versionpredicate.VersionPredicate(v)
1170 self.obsoletes = value
1171
Greg Ward2ff78872000-06-24 00:23:20 +00001172
1173def fix_help_options (options):
1174 """Convert a 4-tuple 'help_options' list as found in various command
1175 classes to the 3-tuple form required by FancyGetopt.
1176 """
1177 new_options = []
1178 for help_tuple in options:
1179 new_options.append(help_tuple[0:3])
1180 return new_options
1181
1182
Greg Wardfe6462c2000-04-04 01:40:52 +00001183if __name__ == "__main__":
Greg Wardfd7b91e2000-09-26 01:52:25 +00001184 dist = Distribution()
Guido van Rossumbe19ed72007-02-09 05:37:30 +00001185 print("ok")