Mercurial > p > roundup > code
comparison roundup/dist/command/build_scripts.py @ 4068:e233d7a66343
Refactor setup.py.
| author | Stefan Seefeld <stefan@seefeld.name> |
|---|---|
| date | Mon, 23 Feb 2009 14:30:32 +0000 |
| parents | |
| children | 9250620c7219 |
comparison
equal
deleted
inserted
replaced
| 4067:7309ac3b6e24 | 4068:e233d7a66343 |
|---|---|
| 1 # | |
| 2 # Copyright (C) 2009 Stefan Seefeld | |
| 3 # All rights reserved. | |
| 4 # For license terms see the file COPYING.txt. | |
| 5 # | |
| 6 from distutils.command.build_scripts import build_scripts as base | |
| 7 import sys, os, string | |
| 8 | |
| 9 class build_scripts(base): | |
| 10 """ Overload the build_scripts command and create the scripts | |
| 11 from scratch, depending on the target platform. | |
| 12 | |
| 13 You have to define the name of your package in an inherited | |
| 14 class (due to the delayed instantiation of command classes | |
| 15 in distutils, this cannot be passed to __init__). | |
| 16 | |
| 17 The scripts are created in an uniform scheme: they start the | |
| 18 run() function in the module | |
| 19 | |
| 20 <packagename>.scripts.<mangled_scriptname> | |
| 21 | |
| 22 The mangling of script names replaces '-' and '/' characters | |
| 23 with '-' and '.', so that they are valid module paths. | |
| 24 | |
| 25 If the target platform is win32, create .bat files instead of | |
| 26 *nix shell scripts. Target platform is set to "win32" if main | |
| 27 command is 'bdist_wininst' or if the command is 'bdist' and | |
| 28 it has the list of formats (from command line or config file) | |
| 29 and the first item on that list is wininst. Otherwise | |
| 30 target platform is set to current (build) platform. | |
| 31 """ | |
| 32 package_name = 'roundup' | |
| 33 | |
| 34 def initialize_options(self): | |
| 35 base.initialize_options(self) | |
| 36 self.script_preamble = None | |
| 37 self.target_platform = None | |
| 38 self.python_executable = None | |
| 39 | |
| 40 def finalize_options(self): | |
| 41 base.finalize_options(self) | |
| 42 cmdopt=self.distribution.command_options | |
| 43 | |
| 44 # find the target platform | |
| 45 if self.target_platform: | |
| 46 # TODO? allow explicit setting from command line | |
| 47 target = self.target_platform | |
| 48 if cmdopt.has_key("bdist_wininst"): | |
| 49 target = "win32" | |
| 50 elif cmdopt.get("bdist", {}).has_key("formats"): | |
| 51 formats = cmdopt["bdist"]["formats"][1].split(",") | |
| 52 if formats[0] == "wininst": | |
| 53 target = "win32" | |
| 54 else: | |
| 55 target = sys.platform | |
| 56 if len(formats) > 1: | |
| 57 self.warn( | |
| 58 "Scripts are built for %s only (requested formats: %s)" | |
| 59 % (target, ",".join(formats))) | |
| 60 else: | |
| 61 # default to current platform | |
| 62 target = sys.platform | |
| 63 self.target_platfom = target | |
| 64 | |
| 65 # for native builds, use current python executable path; | |
| 66 # for cross-platform builds, use default executable name | |
| 67 if self.python_executable: | |
| 68 # TODO? allow command-line option | |
| 69 pass | |
| 70 if target == sys.platform: | |
| 71 self.python_executable = os.path.normpath(sys.executable) | |
| 72 else: | |
| 73 self.python_executable = "python" | |
| 74 | |
| 75 # for windows builds, add ".bat" extension | |
| 76 if target == "win32": | |
| 77 # *nix-like scripts may be useful also on win32 (cygwin) | |
| 78 # to build both script versions, use: | |
| 79 #self.scripts = list(self.scripts) + [script + ".bat" | |
| 80 # for script in self.scripts] | |
| 81 self.scripts = [script + ".bat" for script in self.scripts] | |
| 82 | |
| 83 # tweak python path for installations outside main python library | |
| 84 if cmdopt.get("install", {}).has_key("prefix"): | |
| 85 prefix = os.path.expanduser(cmdopt['install']['prefix'][1]) | |
| 86 version = '%d.%d'%sys.version_info[:2] | |
| 87 self.script_preamble = """ | |
| 88 import sys | |
| 89 sys.path.insert(1, "%s/lib/python%s/site-packages") | |
| 90 """%(prefix, version) | |
| 91 else: | |
| 92 self.script_preamble = '' | |
| 93 | |
| 94 def copy_scripts(self): | |
| 95 """ Create each script listed in 'self.scripts' | |
| 96 """ | |
| 97 | |
| 98 to_module = string.maketrans('-/', '_.') | |
| 99 | |
| 100 self.mkpath(self.build_dir) | |
| 101 for script in self.scripts: | |
| 102 outfile = os.path.join(self.build_dir, os.path.basename(script)) | |
| 103 | |
| 104 #if not self.force and not newer(script, outfile): | |
| 105 # self.announce("not copying %s (up-to-date)" % script) | |
| 106 # continue | |
| 107 | |
| 108 if self.dry_run: | |
| 109 self.announce("would create %s" % outfile) | |
| 110 continue | |
| 111 | |
| 112 module = os.path.splitext(os.path.basename(script))[0] | |
| 113 module = string.translate(module, to_module) | |
| 114 script_vars = { | |
| 115 'python': self.python_executable, | |
| 116 'package': self.package_name, | |
| 117 'module': module, | |
| 118 'prefix': self.script_preamble, | |
| 119 } | |
| 120 | |
| 121 self.announce("creating %s" % outfile) | |
| 122 file = open(outfile, 'w') | |
| 123 | |
| 124 try: | |
| 125 # could just check self.target_platform, | |
| 126 # but looking at the script extension | |
| 127 # makes it possible to build both *nix-like | |
| 128 # and windows-like scripts on win32. | |
| 129 # may be useful for cygwin. | |
| 130 if os.path.splitext(outfile)[1] == ".bat": | |
| 131 file.write('@echo off\n' | |
| 132 'if NOT "%%_4ver%%" == "" "%(python)s" -c "from %(package)s.scripts.%(module)s import run; run()" %%$\n' | |
| 133 'if "%%_4ver%%" == "" "%(python)s" -c "from %(package)s.scripts.%(module)s import run; run()" %%*\n' | |
| 134 % script_vars) | |
| 135 else: | |
| 136 file.write('#! %(python)s\n%(prefix)s' | |
| 137 'from %(package)s.scripts.%(module)s import run\n' | |
| 138 'run()\n' | |
| 139 % script_vars) | |
| 140 finally: | |
| 141 file.close() | |
| 142 os.chmod(outfile, 0755) |