← Back to branch summary

~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to setup.py

  • Committer: Patch Queue Manager
  • Date: 2016-04-21 04:10:52 UTC
  • mfrom: (6616.1.1 fix-en-user-guide)
  • Revision ID: pqm@pqm.ubuntu.com-20160421041052-clcye7ns1qcl2n7w
(richard-wilbur) Ensure build of English use guide always uses English text
 even when user's locale specifies a different language. (Jelmer Vernooij)

Show diffs side-by-side

added added

removed removed

Lines of Context:
setup.py
9
9
import os
10
10
import os.path
11
11
import sys
 
12
import copy
 
13
import glob
12
14
 
13
 
if sys.version_info < (2, 4):
14
 
    sys.stderr.write("[ERROR] Not a supported Python version. Need 2.4+\n")
 
15
if sys.version_info < (2, 6):
 
16
    sys.stderr.write("[ERROR] Not a supported Python version. Need 2.6+\n")
15
17
    sys.exit(1)
16
18
 
17
19
# NOTE: The directory containing setup.py, whether run by 'python setup.py' or
37
39
    'version':      bzrlib.__version__,
38
40
    'author':       'Canonical Ltd',
39
41
    'author_email': 'bazaar@lists.canonical.com',
40
 
    'url':          'http://www.bazaar-vcs.org/',
 
42
    'url':          'http://bazaar.canonical.com/',
41
43
    'description':  'Friendly distributed version control system',
42
44
    'license':      'GNU GPL v2',
43
 
    'download_url': 'http://bazaar-vcs.org/Download',
 
45
    'download_url': 'https://launchpad.net/bzr/+download',
44
46
    'long_description': get_long_description(),
45
47
    'classifiers': [
46
48
        'Development Status :: 6 - Mature',
65
67
            'package_data': {'bzrlib': ['doc/api/*.txt',
66
68
                                        'tests/test_patches_data/*',
67
69
                                        'help_topics/en/*.txt',
 
70
                                        'tests/ssl_certs/ca.crt',
 
71
                                        'tests/ssl_certs/server_without_pass.key',
 
72
                                        'tests/ssl_certs/server_with_pass.key',
 
73
                                        'tests/ssl_certs/server.crt',
68
74
                                       ]},
69
75
           }
70
 
 
 
76
I18N_FILES = []
 
77
for filepath in glob.glob("bzrlib/locale/*/LC_MESSAGES/*.mo"):
 
78
    langfile = filepath[len("bzrlib/locale/"):]
 
79
    targetpath = os.path.dirname(os.path.join("share/locale", langfile))
 
80
    I18N_FILES.append((targetpath, [filepath]))
71
81
 
72
82
def get_bzrlib_packages():
73
83
    """Recurse through the bzrlib directory, and extract the package names"""
93
103
BZRLIB['packages'] = get_bzrlib_packages()
94
104
 
95
105
 
 
106
from distutils import log
96
107
from distutils.core import setup
 
108
from distutils.version import LooseVersion
97
109
from distutils.command.install_scripts import install_scripts
98
110
from distutils.command.install_data import install_data
99
111
from distutils.command.build import build
121
133
                f = file(batch_path, "w")
122
134
                f.write(batch_str)
123
135
                f.close()
124
 
                print "Created:", batch_path
125
 
            except Exception, e:
126
 
                print "ERROR: Unable to create %s: %s" % (batch_path, e)
 
136
                print("Created: %s" % batch_path)
 
137
            except Exception:
 
138
                e = sys.exc_info()[1]
 
139
                print("ERROR: Unable to create %s: %s" % (batch_path, e))
127
140
 
128
141
    def _quoted_path(self, path):
129
142
        if ' ' in path:
144
157
    """Customized build distutils action.
145
158
    Generate bzr.1.
146
159
    """
 
160
 
 
161
    sub_commands = build.sub_commands + [
 
162
            ('build_mo', lambda _: True),
 
163
            ]
 
164
 
147
165
    def run(self):
148
166
        build.run(self)
149
167
 
150
 
        import generate_docs
 
168
        from tools import generate_docs
151
169
        generate_docs.main(argv=["bzr", "man"])
152
170
 
153
171
 
155
173
## Setup
156
174
########################
157
175
 
 
176
from bzrlib.bzr_distutils import build_mo
 
177
 
158
178
command_classes = {'install_scripts': my_install_scripts,
159
 
                   'build': bzr_build}
 
179
                   'build': bzr_build,
 
180
                   'build_mo': build_mo,
 
181
                   }
160
182
from distutils import log
161
183
from distutils.errors import CCompilerError, DistutilsPlatformError
162
184
from distutils.extension import Extension
163
185
ext_modules = []
164
186
try:
165
 
    from Pyrex.Distutils import build_ext
 
187
    try:
 
188
        from Cython.Distutils import build_ext
 
189
        from Cython.Compiler.Version import version as pyrex_version
 
190
    except ImportError:
 
191
        print("No Cython, trying Pyrex...")
 
192
        from Pyrex.Distutils import build_ext
 
193
        from Pyrex.Compiler.Version import version as pyrex_version
166
194
except ImportError:
167
195
    have_pyrex = False
168
196
    # try to build the extension from the prior generated source.
169
 
    print
170
 
    print ("The python package 'Pyrex' is not available."
171
 
           " If the .c files are available,")
172
 
    print ("they will be built,"
173
 
           " but modifying the .pyx files will not rebuild them.")
174
 
    print
 
197
    print("")
 
198
    print("The python package 'Pyrex' is not available."
 
199
          " If the .c files are available,")
 
200
    print("they will be built,"
 
201
          " but modifying the .pyx files will not rebuild them.")
 
202
    print("")
175
203
    from distutils.command.build_ext import build_ext
176
204
else:
177
205
    have_pyrex = True
 
206
    pyrex_version_info = LooseVersion(pyrex_version)
178
207
 
179
208
 
180
209
class build_ext_if_possible(build_ext):
181
210
 
 
211
    user_options = build_ext.user_options + [
 
212
        ('allow-python-fallback', None,
 
213
         "When an extension cannot be built, allow falling"
 
214
         " back to the pure-python implementation.")
 
215
        ]
 
216
 
 
217
    def initialize_options(self):
 
218
        build_ext.initialize_options(self)
 
219
        self.allow_python_fallback = False
 
220
 
182
221
    def run(self):
183
222
        try:
184
223
            build_ext.run(self)
185
 
        except DistutilsPlatformError, e:
 
224
        except DistutilsPlatformError:
 
225
            e = sys.exc_info()[1]
 
226
            if not self.allow_python_fallback:
 
227
                log.warn('\n  Cannot build extensions.\n'
 
228
                         '  Use "build_ext --allow-python-fallback" to use'
 
229
                         ' slower python implementations instead.\n')
 
230
                raise
186
231
            log.warn(str(e))
187
 
            log.warn('Extensions cannot be built, '
188
 
                     'will use the Python versions instead')
 
232
            log.warn('\n  Extensions cannot be built.\n'
 
233
                     '  Using the slower Python implementations instead.\n')
189
234
 
190
235
    def build_extension(self, ext):
191
236
        try:
192
237
            build_ext.build_extension(self, ext)
193
238
        except CCompilerError:
194
 
            log.warn('Building of "%s" extension failed, '
195
 
                     'will use the Python version instead' % (ext.name,))
 
239
            if not self.allow_python_fallback:
 
240
                log.warn('\n  Cannot build extension "%s".\n'
 
241
                         '  Use "build_ext --allow-python-fallback" to use'
 
242
                         ' slower python implementations instead.\n'
 
243
                         % (ext.name,))
 
244
                raise
 
245
            log.warn('\n  Building of "%s" extension failed.\n'
 
246
                     '  Using the slower Python implementation instead.'
 
247
                     % (ext.name,))
196
248
 
197
249
 
198
250
# Override the build_ext if we have Pyrex available
200
252
unavailable_files = []
201
253
 
202
254
 
203
 
def add_pyrex_extension(module_name, **kwargs):
 
255
def add_pyrex_extension(module_name, libraries=None, extra_source=[]):
204
256
    """Add a pyrex module to build.
205
257
 
206
258
    This will use Pyrex to auto-generate the .c file if it is available.
216
268
    path = module_name.replace('.', '/')
217
269
    pyrex_name = path + '.pyx'
218
270
    c_name = path + '.c'
 
271
    define_macros = []
 
272
    if sys.platform == 'win32':
 
273
        # pyrex uses the macro WIN32 to detect the platform, even though it
 
274
        # should be using something like _WIN32 or MS_WINDOWS, oh well, we can
 
275
        # give it the right value.
 
276
        define_macros.append(('WIN32', None))
219
277
    if have_pyrex:
220
 
        ext_modules.append(Extension(module_name, [pyrex_name], **kwargs))
 
278
        source = [pyrex_name]
221
279
    else:
222
280
        if not os.path.isfile(c_name):
223
281
            unavailable_files.append(c_name)
 
282
            return
224
283
        else:
225
 
            ext_modules.append(Extension(module_name, [c_name], **kwargs))
226
 
 
227
 
 
228
 
add_pyrex_extension('bzrlib._btree_serializer_c')
229
 
add_pyrex_extension('bzrlib._dirstate_helpers_c')
230
 
add_pyrex_extension('bzrlib._knit_load_data_c')
231
 
add_pyrex_extension('bzrlib._readdir_pyx')
 
284
            source = [c_name]
 
285
    source.extend(extra_source)
 
286
    ext_modules.append(Extension(module_name, source,
 
287
        define_macros=define_macros, libraries=libraries))
 
288
 
 
289
 
 
290
add_pyrex_extension('bzrlib._annotator_pyx')
 
291
add_pyrex_extension('bzrlib._bencode_pyx')
 
292
add_pyrex_extension('bzrlib._chunks_to_lines_pyx')
 
293
add_pyrex_extension('bzrlib._groupcompress_pyx',
 
294
                    extra_source=['bzrlib/diff-delta.c'])
 
295
add_pyrex_extension('bzrlib._knit_load_data_pyx')
 
296
add_pyrex_extension('bzrlib._known_graph_pyx')
 
297
add_pyrex_extension('bzrlib._rio_pyx')
232
298
if sys.platform == 'win32':
233
 
    # pyrex uses the macro WIN32 to detect the platform, even though it should
234
 
    # be using something like _WIN32 or MS_WINDOWS, oh well, we can give it the
235
 
    # right value.
236
 
    add_pyrex_extension('bzrlib._walkdirs_win32',
237
 
                        define_macros=[('WIN32', None)])
238
 
ext_modules.append(Extension('bzrlib._patiencediff_c', ['bzrlib/_patiencediff_c.c']))
 
299
    add_pyrex_extension('bzrlib._dirstate_helpers_pyx',
 
300
                        libraries=['Ws2_32'])
 
301
    add_pyrex_extension('bzrlib._walkdirs_win32')
 
302
else:
 
303
    if have_pyrex and pyrex_version_info == LooseVersion("0.9.4.1"):
 
304
        # Pyrex 0.9.4.1 fails to compile this extension correctly
 
305
        # The code it generates re-uses a "local" pointer and
 
306
        # calls "PY_DECREF" after having set it to NULL. (It mixes PY_XDECREF
 
307
        # which is NULL safe with PY_DECREF which is not.)
 
308
        # <https://bugs.launchpad.net/bzr/+bug/449372>
 
309
        # <https://bugs.launchpad.net/bzr/+bug/276868>
 
310
        print('Cannot build extension "bzrlib._dirstate_helpers_pyx" using')
 
311
        print('your version of pyrex "%s". Please upgrade your pyrex'
 
312
              % (pyrex_version,))
 
313
        print('install. For now, the non-compiled (python) version will')
 
314
        print('be used instead.')
 
315
    else:
 
316
        add_pyrex_extension('bzrlib._dirstate_helpers_pyx')
 
317
    add_pyrex_extension('bzrlib._readdir_pyx')
 
318
add_pyrex_extension('bzrlib._chk_map_pyx')
 
319
ext_modules.append(Extension('bzrlib._patiencediff_c',
 
320
                             ['bzrlib/_patiencediff_c.c']))
 
321
if have_pyrex and pyrex_version_info < LooseVersion("0.9.6.3"):
 
322
    print("")
 
323
    print('Your Pyrex/Cython version %s is too old to build the simple_set' % (
 
324
        pyrex_version))
 
325
    print('and static_tuple extensions.')
 
326
    print('Please upgrade to at least Pyrex 0.9.6.3')
 
327
    print("")
 
328
    # TODO: Should this be a fatal error?
 
329
else:
 
330
    # We only need 0.9.6.3 to build _simple_set_pyx, but static_tuple depends
 
331
    # on simple_set
 
332
    add_pyrex_extension('bzrlib._simple_set_pyx')
 
333
    ext_modules.append(Extension('bzrlib._static_tuple_c',
 
334
                                 ['bzrlib/_static_tuple_c.c']))
 
335
add_pyrex_extension('bzrlib._btree_serializer_pyx')
239
336
 
240
337
 
241
338
if unavailable_files:
242
 
    print 'C extension(s) not found:'
243
 
    print '   %s' % ('\n  '.join(unavailable_files),)
244
 
    print 'The python versions will be used instead.'
245
 
    print
 
339
    print('C extension(s) not found:')
 
340
    print('   %s' % ('\n  '.join(unavailable_files),))
 
341
    print('The python versions will be used instead.')
 
342
    print("")
246
343
 
247
344
 
248
345
def get_tbzr_py2exe_info(includes, excludes, packages, console_targets,
249
 
                         gui_targets):
 
346
                         gui_targets, data_files):
250
347
    packages.append('tbzrcommands')
251
348
 
252
349
    # ModuleFinder can't handle runtime changes to __path__, but
253
350
    # win32com uses them.  Hook this in so win32com.shell is found.
254
351
    import modulefinder
255
352
    import win32com
 
353
    import cPickle as pickle
256
354
    for p in win32com.__path__[1:]:
257
355
        modulefinder.AddPackagePath("win32com", p)
258
356
    for extra in ["win32com.shell"]:
267
365
    # Ensure tbzrlib itself is on sys.path
268
366
    sys.path.append(tbzr_root)
269
367
 
270
 
    # Ensure our COM "entry-point" is on sys.path
271
 
    sys.path.append(os.path.join(tbzr_root, "shellext", "python"))
272
 
 
273
368
    packages.append("tbzrlib")
 
369
 
 
370
    # collect up our icons.
 
371
    cwd = os.getcwd()
 
372
    ico_root = os.path.join(tbzr_root, 'tbzrlib', 'resources')
 
373
    icos = [] # list of (path_root, relative_ico_path)
 
374
    # First always bzr's icon and its in the root of the bzr tree.
 
375
    icos.append(('', 'bzr.ico'))
 
376
    for root, dirs, files in os.walk(ico_root):
 
377
        icos.extend([(ico_root, os.path.join(root, f)[len(ico_root)+1:])
 
378
                     for f in files if f.endswith('.ico')])
 
379
    # allocate an icon ID for each file and the full path to the ico
 
380
    icon_resources = [(rid, os.path.join(ico_dir, ico_name))
 
381
                      for rid, (ico_dir, ico_name) in enumerate(icos)]
 
382
    # create a string resource with the mapping.  Might as well save the
 
383
    # runtime some effort and write a pickle.
 
384
    # Runtime expects unicode objects with forward-slash seps.
 
385
    fse = sys.getfilesystemencoding()
 
386
    map_items = [(f.replace('\\', '/').decode(fse), rid)
 
387
                 for rid, (_, f) in enumerate(icos)]
 
388
    ico_map = dict(map_items)
 
389
    # Create a new resource type of 'ICON_MAP', and use ID=1
 
390
    other_resources = [ ("ICON_MAP", 1, pickle.dumps(ico_map))]
 
391
 
274
392
    excludes.extend("""pywin pywin.dialogs pywin.dialogs.list
275
393
                       win32ui crawler.Crawler""".split())
276
394
 
277
 
    tbzr = dict(
278
 
        modules=["tbzr"],
279
 
        create_exe = False, # we only want a .dll
280
 
    )
281
 
    com_targets.append(tbzr)
282
 
 
283
395
    # tbzrcache executables - a "console" version for debugging and a
284
396
    # GUI version that is generally used.
285
397
    tbzrcache = dict(
286
 
        script = os.path.join(tbzr_root, "Scripts", "tbzrcache.py"),
287
 
        icon_resources = [(0,'bzr.ico')],
 
398
        script = os.path.join(tbzr_root, "scripts", "tbzrcache.py"),
 
399
        icon_resources = icon_resources,
 
400
        other_resources = other_resources,
288
401
    )
289
402
    console_targets.append(tbzrcache)
290
403
 
295
408
 
296
409
    # ditto for the tbzrcommand tool
297
410
    tbzrcommand = dict(
298
 
        script = os.path.join(tbzr_root, "Scripts", "tbzrcommand.py"),
299
 
        icon_resources = [(0,'bzr.ico')],
 
411
        script = os.path.join(tbzr_root, "scripts", "tbzrcommand.py"),
 
412
        icon_resources = icon_resources,
 
413
        other_resources = other_resources,
300
414
    )
301
415
    console_targets.append(tbzrcommand)
302
416
    tbzrcommandw = tbzrcommand.copy()
303
417
    tbzrcommandw["dest_base"]="tbzrcommandw"
304
418
    gui_targets.append(tbzrcommandw)
305
419
    
306
 
    # tbzr tests
307
 
    tbzrtest = dict(
308
 
        script = os.path.join(tbzr_root, "Scripts", "tbzrtest.py"),
309
 
    )
310
 
    console_targets.append(tbzrtest)
311
 
 
312
 
    # A utility to see python output from the shell extension - this will
313
 
    # die when we get a c++ extension
314
 
    # any .py file from pywin32's win32 lib will do (other than
315
 
    # win32traceutil itself that is)
316
 
    import winerror
317
 
    win32_lib_dir = os.path.dirname(winerror.__file__)
318
 
    tracer = dict(script = os.path.join(win32_lib_dir, "win32traceutil.py"),
319
 
                  dest_base="tbzr_tracer")
 
420
    # A utility to see python output from both C++ and Python based shell
 
421
    # extensions
 
422
    tracer = dict(script=os.path.join(tbzr_root, "scripts", "tbzrtrace.py"))
320
423
    console_targets.append(tracer)
321
424
 
322
 
 
323
 
def get_qbzr_py2exe_info(includes, excludes, packages):
 
425
    # The C++ implemented shell extensions.
 
426
    dist_dir = os.path.join(tbzr_root, "shellext", "build")
 
427
    data_files.append(('', [os.path.join(dist_dir, 'tbzrshellext_x86.dll')]))
 
428
    data_files.append(('', [os.path.join(dist_dir, 'tbzrshellext_x64.dll')]))
 
429
 
 
430
 
 
431
def get_qbzr_py2exe_info(includes, excludes, packages, data_files):
324
432
    # PyQt4 itself still escapes the plugin detection code for some reason...
325
 
    packages.append('PyQt4')
326
 
    excludes.append('PyQt4.elementtree.ElementTree')
 
433
    includes.append('PyQt4.QtCore')
 
434
    includes.append('PyQt4.QtGui')
 
435
    includes.append('PyQt4.QtTest')
327
436
    includes.append('sip') # extension module required for Qt.
328
437
    packages.append('pygments') # colorizer for qbzr
329
 
    # but we can avoid many Qt4 Dlls.
330
 
    dll_excludes.extend(
331
 
        """QtAssistantClient4.dll QtCLucene4.dll QtDesigner4.dll
332
 
        QtHelp4.dll QtNetwork4.dll QtOpenGL4.dll QtScript4.dll
333
 
        QtSql4.dll QtTest4.dll QtWebKit4.dll QtXml4.dll
334
 
        qscintilla2.dll""".split())
 
438
    packages.append('docutils') # html formatting
 
439
    includes.append('win32event')  # for qsubprocess stuff
335
440
    # the qt binaries might not be on PATH...
336
 
    qt_dir = os.path.join(sys.prefix, "PyQt4", "bin")
337
 
    path = os.environ.get("PATH","")
338
 
    if qt_dir.lower() not in [p.lower() for p in path.split(os.pathsep)]:
339
 
        os.environ["PATH"] = path + os.pathsep + qt_dir
 
441
    # They seem to install to a place like C:\Python25\PyQt4\*
 
442
    # Which is not the same as C:\Python25\Lib\site-packages\PyQt4
 
443
    pyqt_dir = os.path.join(sys.prefix, "PyQt4")
 
444
    pyqt_bin_dir = os.path.join(pyqt_dir, "bin")
 
445
    if os.path.isdir(pyqt_bin_dir):
 
446
        path = os.environ.get("PATH", "")
 
447
        if pyqt_bin_dir.lower() not in [p.lower() for p in path.split(os.pathsep)]:
 
448
            os.environ["PATH"] = path + os.pathsep + pyqt_bin_dir
 
449
    # also add all imageformat plugins to distribution
 
450
    # We will look in 2 places, dirname(PyQt4.__file__) and pyqt_dir
 
451
    base_dirs_to_check = []
 
452
    if os.path.isdir(pyqt_dir):
 
453
        base_dirs_to_check.append(pyqt_dir)
 
454
    try:
 
455
        import PyQt4
 
456
    except ImportError:
 
457
        pass
 
458
    else:
 
459
        pyqt4_base_dir = os.path.dirname(PyQt4.__file__)
 
460
        if pyqt4_base_dir != pyqt_dir:
 
461
            base_dirs_to_check.append(pyqt4_base_dir)
 
462
    if not base_dirs_to_check:
 
463
        log.warn("Can't find PyQt4 installation -> not including imageformat"
 
464
                 " plugins")
 
465
    else:
 
466
        files = []
 
467
        for base_dir in base_dirs_to_check:
 
468
            plug_dir = os.path.join(base_dir, 'plugins', 'imageformats')
 
469
            if os.path.isdir(plug_dir):
 
470
                for fname in os.listdir(plug_dir):
 
471
                    # Include plugin dlls, but not debugging dlls
 
472
                    fullpath = os.path.join(plug_dir, fname)
 
473
                    if fname.endswith('.dll') and not fname.endswith('d4.dll'):
 
474
                        files.append(fullpath)
 
475
        if files:
 
476
            data_files.append(('imageformats', files))
 
477
        else:
 
478
            log.warn('PyQt4 was found, but we could not find any imageformat'
 
479
                     ' plugins. Are you sure your configuration is correct?')
 
480
 
 
481
 
 
482
def get_svn_py2exe_info(includes, excludes, packages):
 
483
    packages.append('subvertpy')
 
484
    packages.append('sqlite3')
 
485
 
 
486
 
 
487
def get_git_py2exe_info(includes, excludes, packages):
 
488
    packages.append('dulwich')
 
489
 
 
490
 
 
491
def get_fastimport_py2exe_info(includes, excludes, packages):
 
492
    # This is the python-fastimport package, not to be confused with the
 
493
    # bzr-fastimport plugin.
 
494
    packages.append('fastimport')
340
495
 
341
496
 
342
497
if 'bdist_wininst' in sys.argv:
363
518
            # help pages
364
519
            'data_files': find_docs(),
365
520
            # for building pyrex extensions
366
 
            'cmdclass': {'build_ext': build_ext_if_possible},
 
521
            'cmdclass': command_classes,
367
522
           }
368
523
 
369
524
    ARGS.update(META_INFO)
370
525
    ARGS.update(BZRLIB)
 
526
    PKG_DATA['package_data']['bzrlib'].append('locale/*/LC_MESSAGES/*.mo')
371
527
    ARGS.update(PKG_DATA)
372
 
    
 
528
 
373
529
    setup(**ARGS)
374
530
 
375
531
elif 'py2exe' in sys.argv:
376
 
    import glob
377
532
    # py2exe setup
378
533
    import py2exe
379
534
 
400
555
            install_data.run(self)
401
556
 
402
557
            py2exe = self.distribution.get_command_obj('py2exe', False)
403
 
            optimize = py2exe.optimize
 
558
            # GZ 2010-04-19: Setup has py2exe.optimize as 2, but give plugins
 
559
            #                time before living with docstring stripping
 
560
            optimize = 1
404
561
            compile_names = [f for f in self.outfiles if f.endswith('.py')]
 
562
            # Round mtime to nearest even second so that installing on a FAT
 
563
            # filesystem bytecode internal and script timestamps will match
 
564
            for f in compile_names:
 
565
                mtime = os.stat(f).st_mtime
 
566
                remainder = mtime % 2
 
567
                if remainder:
 
568
                    mtime -= remainder
 
569
                    os.utime(f, (mtime, mtime))
405
570
            byte_compile(compile_names,
406
571
                         optimize=optimize,
407
572
                         force=self.force, prefix=self.install_dir,
408
573
                         dry_run=self.dry_run)
409
 
            if optimize:
410
 
                suffix = 'o'
411
 
            else:
412
 
                suffix = 'c'
413
 
            self.outfiles.extend([f + suffix for f in compile_names])
 
574
            self.outfiles.extend([f + 'o' for f in compile_names])
414
575
    # end of class install_data_with_bytecompile
415
576
 
416
577
    target = py2exe.build_exe.Target(script = "bzr",
420
581
                                     version = version_str,
421
582
                                     description = META_INFO['description'],
422
583
                                     author = META_INFO['author'],
423
 
                                     copyright = "(c) Canonical Ltd, 2005-2007",
 
584
                                     copyright = "(c) Canonical Ltd, 2005-2010",
424
585
                                     company_name = "Canonical Ltd.",
425
586
                                     comments = META_INFO['description'],
426
587
                                    )
 
588
    gui_target = copy.copy(target)
 
589
    gui_target.dest_base = "bzrw"
427
590
 
428
591
    packages = BZRLIB['packages']
429
592
    packages.remove('bzrlib')
439
602
    if sys.version.startswith('2.4'):
440
603
        # adding elementtree package
441
604
        additional_packages.add('elementtree')
442
 
    elif sys.version.startswith('2.5'):
 
605
    elif sys.version.startswith('2.6') or sys.version.startswith('2.5'):
443
606
        additional_packages.add('xml.etree')
444
607
    else:
445
608
        import warnings
453
616
                  ImaginaryModule cElementTree elementtree.ElementTree
454
617
                  Crypto.PublicKey._fastmath
455
618
                  medusa medusa.filesys medusa.ftp_server
456
 
                  tools tools.doc_generate
 
619
                  tools
457
620
                  resource validate""".split()
458
621
    dll_excludes = []
459
622
 
482
645
    for root, dirs, files in os.walk('bzrlib/plugins'):
483
646
        if root == 'bzrlib/plugins':
484
647
            plugins = set(dirs)
 
648
            # We ship plugins as normal files on the file-system - however,
 
649
            # the build process can cause *some* of these plugin files to end
 
650
            # up in library.zip. Thus, we saw (eg) "plugins/svn/test" in
 
651
            # library.zip, and then saw import errors related to that as the
 
652
            # rest of the svn plugin wasn't. So we tell py2exe to leave the
 
653
            # plugins out of the .zip file
 
654
            excludes.extend(["bzrlib.plugins." + d for d in dirs])
485
655
        x = []
486
656
        for i in files:
487
 
            if os.path.splitext(i)[1] not in [".py", ".pyd", ".dll"]:
 
657
            # Throw away files we don't want packaged. Note that plugins may
 
658
            # have data files with all sorts of extensions so we need to
 
659
            # be conservative here about what we ditch.
 
660
            ext = os.path.splitext(i)[1]
 
661
            if ext.endswith('~') or ext in [".pyc", ".swp"]:
488
662
                continue
489
663
            if i == '__init__.py' and root == 'bzrlib/plugins':
490
664
                continue
503
677
    console_targets = [target,
504
678
                       'tools/win32/bzr_postinstall.py',
505
679
                       ]
506
 
    gui_targets = []
507
 
    com_targets = []
 
680
    gui_targets = [gui_target]
 
681
    data_files = topics_files + plugins_files + I18N_FILES
508
682
 
509
683
    if 'qbzr' in plugins:
510
 
        get_qbzr_py2exe_info(includes, excludes, packages)
 
684
        get_qbzr_py2exe_info(includes, excludes, packages, data_files)
 
685
 
 
686
    if 'svn' in plugins:
 
687
        get_svn_py2exe_info(includes, excludes, packages)
 
688
 
 
689
    if 'git' in plugins:
 
690
        get_git_py2exe_info(includes, excludes, packages)
 
691
 
 
692
    if 'fastimport' in plugins:
 
693
        get_fastimport_py2exe_info(includes, excludes, packages)
511
694
 
512
695
    if "TBZR" in os.environ:
513
696
        # TORTOISE_OVERLAYS_MSI_WIN32 must be set to the location of the
514
697
        # TortoiseOverlays MSI installer file. It is in the TSVN svn repo and
515
698
        # can be downloaded from (username=guest, blank password):
516
 
        # http://tortoisesvn.tigris.org/svn/tortoisesvn/TortoiseOverlays/version-1.0.4/bin/TortoiseOverlays-1.0.4.11886-win32.msi
517
 
        if not os.path.isfile(os.environ.get('TORTOISE_OVERLAYS_MSI_WIN32',
518
 
                                             '<nofile>')):
519
 
            raise RuntimeError("Please set TORTOISE_OVERLAYS_MSI_WIN32 to the"
520
 
                               " location of the Win32 TortoiseOverlays .msi"
521
 
                               " installer file")
 
699
        # http://tortoisesvn.tigris.org/svn/tortoisesvn/TortoiseOverlays
 
700
        # look for: version-1.0.4/bin/TortoiseOverlays-1.0.4.11886-win32.msi
 
701
        # Ditto for TORTOISE_OVERLAYS_MSI_X64, pointing at *-x64.msi.
 
702
        for needed in ('TORTOISE_OVERLAYS_MSI_WIN32',
 
703
                       'TORTOISE_OVERLAYS_MSI_X64'):
 
704
            url = ('http://guest:@tortoisesvn.tigris.org/svn/tortoisesvn'
 
705
                   '/TortoiseOverlays')
 
706
            if not os.path.isfile(os.environ.get(needed, '<nofile>')):
 
707
                raise RuntimeError(
 
708
                    "\nPlease set %s to the location of the relevant"
 
709
                    "\nTortoiseOverlays .msi installer file."
 
710
                    " The installers can be found at"
 
711
                    "\n  %s"
 
712
                    "\ncheck in the version-X.Y.Z/bin/ subdir" % (needed, url))
522
713
        get_tbzr_py2exe_info(includes, excludes, packages, console_targets,
523
 
                             gui_targets)
 
714
                             gui_targets, data_files)
524
715
    else:
525
716
        # print this warning to stderr as output is redirected, so it is seen
526
717
        # at build time.  Also to stdout so it appears in the log
527
718
        for f in (sys.stderr, sys.stdout):
528
 
            print >> f, \
529
 
                "Skipping TBZR binaries - please set TBZR to a directory to enable"
 
719
            f.write("Skipping TBZR binaries - "
 
720
                "please set TBZR to a directory to enable\n")
530
721
 
531
722
    # MSWSOCK.dll is a system-specific library, which py2exe accidentally pulls
532
723
    # in on Vista.
533
 
    dll_excludes.append("MSWSOCK.dll")
 
724
    dll_excludes.extend(["MSWSOCK.dll",
 
725
                         "MSVCP60.dll",
 
726
                         "MSVCP90.dll",
 
727
                         "powrprof.dll",
 
728
                         "SHFOLDER.dll"])
534
729
    options_list = {"py2exe": {"packages": packages + list(additional_packages),
535
730
                               "includes": includes,
536
731
                               "excludes": excludes,
537
732
                               "dll_excludes": dll_excludes,
538
733
                               "dist_dir": "win32_bzr.exe",
539
 
                               "optimize": 1,
 
734
                               "optimize": 2,
 
735
                               "custom_boot_script":
 
736
                                        "tools/win32/py2exe_boot_common.py",
540
737
                              },
541
738
                   }
542
739
 
543
 
    setup(options=options_list,
544
 
          console=console_targets,
545
 
          windows=gui_targets,
546
 
          com_server=com_targets,
547
 
          zipfile='lib/library.zip',
548
 
          data_files=topics_files + plugins_files,
549
 
          cmdclass={'install_data': install_data_with_bytecompile},
550
 
          )
 
740
    # We want the libaray.zip to have optimize = 2, but the exe to have
 
741
    # optimize = 1, so that .py files that get compilied at run time
 
742
    # (e.g. user installed plugins) dont have their doc strings removed.
 
743
    class py2exe_no_oo_exe(py2exe.build_exe.py2exe):
 
744
        def build_executable(self, *args, **kwargs):
 
745
            self.optimize = 1
 
746
            py2exe.build_exe.py2exe.build_executable(self, *args, **kwargs)
 
747
            self.optimize = 2
 
748
 
 
749
    if __name__ == '__main__':
 
750
        command_classes['install_data'] = install_data_with_bytecompile
 
751
        command_classes['py2exe'] = py2exe_no_oo_exe
 
752
        setup(options=options_list,
 
753
              console=console_targets,
 
754
              windows=gui_targets,
 
755
              zipfile='lib/library.zip',
 
756
              data_files=data_files,
 
757
              cmdclass=command_classes,
 
758
              )
551
759
 
552
760
else:
553
761
    # ad-hoc for easy_install
554
762
    DATA_FILES = []
555
763
    if not 'bdist_egg' in sys.argv:
556
 
        # generate and install bzr.1 only with plain install, not easy_install one
 
764
        # generate and install bzr.1 only with plain install, not the
 
765
        # easy_install one
557
766
        DATA_FILES = [('man/man1', ['bzr.1'])]
558
767
 
 
768
    DATA_FILES = DATA_FILES + I18N_FILES
559
769
    # std setup
560
770
    ARGS = {'scripts': ['bzr'],
561
771
            'data_files': DATA_FILES,
567
777
    ARGS.update(BZRLIB)
568
778
    ARGS.update(PKG_DATA)
569
779
 
570
 
    setup(**ARGS)
 
780
    if __name__ == '__main__':
 
781
        setup(**ARGS)