← Back to branch summary

~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to setup.py

  • Committer: Vincent Ladeuil
  • Date: 2011-08-20 09:28:27 UTC
  • mfrom: (5050.78.2 2.2)
  • mto: (5609.48.8 2.3)
  • mto: This revision was merged to the branch mainline in revision 6090.
  • Revision ID: v.ladeuil+lp@free.fr-20110820092827-9dyakfslp0r3hb1k
Merge 2.2 into 2.3 (including fix for #614713, #609187 and #812928)

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
12
13
 
13
14
if sys.version_info < (2, 4):
14
15
    sys.stderr.write("[ERROR] Not a supported Python version. Need 2.4+\n")
37
38
    'version':      bzrlib.__version__,
38
39
    'author':       'Canonical Ltd',
39
40
    'author_email': 'bazaar@lists.canonical.com',
40
 
    'url':          'http://www.bazaar-vcs.org/',
 
41
    'url':          'http://bazaar.canonical.com/',
41
42
    'description':  'Friendly distributed version control system',
42
43
    'license':      'GNU GPL v2',
43
 
    'download_url': 'http://bazaar-vcs.org/Download',
 
44
    'download_url': 'https://launchpad.net/bzr/+download',
44
45
    'long_description': get_long_description(),
45
46
    'classifiers': [
46
47
        'Development Status :: 6 - Mature',
65
66
            'package_data': {'bzrlib': ['doc/api/*.txt',
66
67
                                        'tests/test_patches_data/*',
67
68
                                        'help_topics/en/*.txt',
 
69
                                        'tests/ssl_certs/ca.crt',
 
70
                                        'tests/ssl_certs/server_without_pass.key',
 
71
                                        'tests/ssl_certs/server_with_pass.key',
 
72
                                        'tests/ssl_certs/server.crt'
68
73
                                       ]},
69
74
           }
70
75
 
93
98
BZRLIB['packages'] = get_bzrlib_packages()
94
99
 
95
100
 
 
101
from distutils import log
96
102
from distutils.core import setup
97
103
from distutils.command.install_scripts import install_scripts
98
104
from distutils.command.install_data import install_data
121
127
                f = file(batch_path, "w")
122
128
                f.write(batch_str)
123
129
                f.close()
124
 
                print "Created:", batch_path
125
 
            except Exception, e:
126
 
                print "ERROR: Unable to create %s: %s" % (batch_path, e)
 
130
                print("Created: %s" % batch_path)
 
131
            except Exception:
 
132
                e = sys.exc_info()[1]
 
133
                print("ERROR: Unable to create %s: %s" % (batch_path, e))
127
134
 
128
135
    def _quoted_path(self, path):
129
136
        if ' ' in path:
163
170
from distutils.extension import Extension
164
171
ext_modules = []
165
172
try:
166
 
    from Pyrex.Distutils import build_ext
 
173
    try:
 
174
        from Pyrex.Distutils import build_ext
 
175
        from Pyrex.Compiler.Version import version as pyrex_version
 
176
    except ImportError:
 
177
        print("No Pyrex, trying Cython...")
 
178
        from Cython.Distutils import build_ext
 
179
        from Cython.Compiler.Version import version as pyrex_version
167
180
except ImportError:
168
181
    have_pyrex = False
169
182
    # try to build the extension from the prior generated source.
170
 
    print
171
 
    print ("The python package 'Pyrex' is not available."
172
 
           " If the .c files are available,")
173
 
    print ("they will be built,"
174
 
           " but modifying the .pyx files will not rebuild them.")
175
 
    print
 
183
    print("")
 
184
    print("The python package 'Pyrex' is not available."
 
185
          " If the .c files are available,")
 
186
    print("they will be built,"
 
187
          " but modifying the .pyx files will not rebuild them.")
 
188
    print("")
176
189
    from distutils.command.build_ext import build_ext
177
190
else:
178
191
    have_pyrex = True
179
 
    from Pyrex.Compiler.Version import version as pyrex_version
 
192
    pyrex_version_info = tuple(map(int, pyrex_version.split('.')))
180
193
 
181
194
 
182
195
class build_ext_if_possible(build_ext):
194
207
    def run(self):
195
208
        try:
196
209
            build_ext.run(self)
197
 
        except DistutilsPlatformError, e:
 
210
        except DistutilsPlatformError:
 
211
            e = sys.exc_info()[1]
198
212
            if not self.allow_python_fallback:
199
213
                log.warn('\n  Cannot build extensions.\n'
200
214
                         '  Use "build_ext --allow-python-fallback" to use'
261
275
 
262
276
add_pyrex_extension('bzrlib._annotator_pyx')
263
277
add_pyrex_extension('bzrlib._bencode_pyx')
264
 
add_pyrex_extension('bzrlib._btree_serializer_pyx')
265
278
add_pyrex_extension('bzrlib._chunks_to_lines_pyx')
266
279
add_pyrex_extension('bzrlib._groupcompress_pyx',
267
280
                    extra_source=['bzrlib/diff-delta.c'])
272
285
    add_pyrex_extension('bzrlib._dirstate_helpers_pyx',
273
286
                        libraries=['Ws2_32'])
274
287
    add_pyrex_extension('bzrlib._walkdirs_win32')
275
 
    z_lib = 'zdll'
276
288
else:
277
 
    if have_pyrex and pyrex_version == '0.9.4.1':
 
289
    if have_pyrex and pyrex_version_info[:3] == (0,9,4):
278
290
        # Pyrex 0.9.4.1 fails to compile this extension correctly
279
291
        # The code it generates re-uses a "local" pointer and
280
292
        # calls "PY_DECREF" after having set it to NULL. (It mixes PY_XDECREF
281
293
        # which is NULL safe with PY_DECREF which is not.)
282
 
        print 'Cannot build extension "bzrlib._dirstate_helpers_pyx" using'
283
 
        print 'your version of pyrex "%s". Please upgrade your pyrex' % (
284
 
            pyrex_version,)
285
 
        print 'install. For now, the non-compiled (python) version will'
286
 
        print 'be used instead.'
 
294
        # <https://bugs.launchpad.net/bzr/+bug/449372>
 
295
        # <https://bugs.launchpad.net/bzr/+bug/276868>
 
296
        print('Cannot build extension "bzrlib._dirstate_helpers_pyx" using')
 
297
        print('your version of pyrex "%s". Please upgrade your pyrex'
 
298
              % (pyrex_version,))
 
299
        print('install. For now, the non-compiled (python) version will')
 
300
        print('be used instead.')
287
301
    else:
288
302
        add_pyrex_extension('bzrlib._dirstate_helpers_pyx')
289
303
    add_pyrex_extension('bzrlib._readdir_pyx')
290
 
    z_lib = 'z'
291
 
add_pyrex_extension('bzrlib._chk_map_pyx', libraries=[z_lib])
 
304
add_pyrex_extension('bzrlib._chk_map_pyx')
292
305
ext_modules.append(Extension('bzrlib._patiencediff_c',
293
306
                             ['bzrlib/_patiencediff_c.c']))
 
307
if have_pyrex and pyrex_version_info < (0, 9, 6, 3):
 
308
    print("")
 
309
    print('Your Pyrex/Cython version %s is too old to build the simple_set' % (
 
310
        pyrex_version))
 
311
    print('and static_tuple extensions.')
 
312
    print('Please upgrade to at least Pyrex 0.9.6.3')
 
313
    print("")
 
314
    # TODO: Should this be a fatal error?
 
315
else:
 
316
    # We only need 0.9.6.3 to build _simple_set_pyx, but static_tuple depends
 
317
    # on simple_set
 
318
    add_pyrex_extension('bzrlib._simple_set_pyx')
 
319
    ext_modules.append(Extension('bzrlib._static_tuple_c',
 
320
                                 ['bzrlib/_static_tuple_c.c']))
 
321
add_pyrex_extension('bzrlib._btree_serializer_pyx')
294
322
 
295
323
 
296
324
if unavailable_files:
297
 
    print 'C extension(s) not found:'
298
 
    print '   %s' % ('\n  '.join(unavailable_files),)
299
 
    print 'The python versions will be used instead.'
300
 
    print
 
325
    print('C extension(s) not found:')
 
326
    print('   %s' % ('\n  '.join(unavailable_files),))
 
327
    print('The python versions will be used instead.')
 
328
    print("")
301
329
 
302
330
 
303
331
def get_tbzr_py2exe_info(includes, excludes, packages, console_targets,
323
351
    # Ensure tbzrlib itself is on sys.path
324
352
    sys.path.append(tbzr_root)
325
353
 
326
 
    # Ensure our COM "entry-point" is on sys.path
327
 
    sys.path.append(os.path.join(tbzr_root, "shellext", "python"))
328
 
 
329
354
    packages.append("tbzrlib")
330
355
 
331
356
    # collect up our icons.
353
378
    excludes.extend("""pywin pywin.dialogs pywin.dialogs.list
354
379
                       win32ui crawler.Crawler""".split())
355
380
 
356
 
    # NOTE: We still create a DLL version of the Python implemented shell
357
 
    # extension for testing purposes - but it is *not* registered by
358
 
    # default - our C++ one is instead.  To discourage people thinking
359
 
    # this DLL is still necessary, its called 'tbzr_old.dll'
360
 
    tbzr = dict(
361
 
        modules=["tbzr"],
362
 
        create_exe = False, # we only want a .dll
363
 
        dest_base = 'tbzr_old',
364
 
    )
365
 
    com_targets.append(tbzr)
366
 
 
367
381
    # tbzrcache executables - a "console" version for debugging and a
368
382
    # GUI version that is generally used.
369
383
    tbzrcache = dict(
381
395
    # ditto for the tbzrcommand tool
382
396
    tbzrcommand = dict(
383
397
        script = os.path.join(tbzr_root, "scripts", "tbzrcommand.py"),
384
 
        icon_resources = [(0,'bzr.ico')],
 
398
        icon_resources = icon_resources,
 
399
        other_resources = other_resources,
385
400
    )
386
401
    console_targets.append(tbzrcommand)
387
402
    tbzrcommandw = tbzrcommand.copy()
394
409
    console_targets.append(tracer)
395
410
 
396
411
    # The C++ implemented shell extensions.
397
 
    dist_dir = os.path.join(tbzr_root, "shellext", "cpp", "tbzrshellext",
398
 
                            "build", "dist")
 
412
    dist_dir = os.path.join(tbzr_root, "shellext", "build")
399
413
    data_files.append(('', [os.path.join(dist_dir, 'tbzrshellext_x86.dll')]))
400
414
    data_files.append(('', [os.path.join(dist_dir, 'tbzrshellext_x64.dll')]))
401
415
 
402
416
 
403
 
def get_qbzr_py2exe_info(includes, excludes, packages):
 
417
def get_qbzr_py2exe_info(includes, excludes, packages, data_files):
404
418
    # PyQt4 itself still escapes the plugin detection code for some reason...
405
 
    packages.append('PyQt4')
406
 
    excludes.append('PyQt4.elementtree.ElementTree')
 
419
    includes.append('PyQt4.QtCore')
 
420
    includes.append('PyQt4.QtGui')
407
421
    includes.append('sip') # extension module required for Qt.
408
422
    packages.append('pygments') # colorizer for qbzr
409
423
    packages.append('docutils') # html formatting
410
 
    # but we can avoid many Qt4 Dlls.
411
 
    dll_excludes.extend(
412
 
        """QtAssistantClient4.dll QtCLucene4.dll QtDesigner4.dll
413
 
        QtHelp4.dll QtNetwork4.dll QtOpenGL4.dll QtScript4.dll
414
 
        QtSql4.dll QtTest4.dll QtWebKit4.dll QtXml4.dll
415
 
        qscintilla2.dll""".split())
 
424
    includes.append('win32event')  # for qsubprocess stuff
416
425
    # the qt binaries might not be on PATH...
417
 
    qt_dir = os.path.join(sys.prefix, "PyQt4", "bin")
418
 
    path = os.environ.get("PATH","")
419
 
    if qt_dir.lower() not in [p.lower() for p in path.split(os.pathsep)]:
420
 
        os.environ["PATH"] = path + os.pathsep + qt_dir
 
426
    # They seem to install to a place like C:\Python25\PyQt4\*
 
427
    # Which is not the same as C:\Python25\Lib\site-packages\PyQt4
 
428
    pyqt_dir = os.path.join(sys.prefix, "PyQt4")
 
429
    pyqt_bin_dir = os.path.join(pyqt_dir, "bin")
 
430
    if os.path.isdir(pyqt_bin_dir):
 
431
        path = os.environ.get("PATH", "")
 
432
        if pyqt_bin_dir.lower() not in [p.lower() for p in path.split(os.pathsep)]:
 
433
            os.environ["PATH"] = path + os.pathsep + pyqt_bin_dir
 
434
    # also add all imageformat plugins to distribution
 
435
    # We will look in 2 places, dirname(PyQt4.__file__) and pyqt_dir
 
436
    base_dirs_to_check = []
 
437
    if os.path.isdir(pyqt_dir):
 
438
        base_dirs_to_check.append(pyqt_dir)
 
439
    try:
 
440
        import PyQt4
 
441
    except ImportError:
 
442
        pass
 
443
    else:
 
444
        pyqt4_base_dir = os.path.dirname(PyQt4.__file__)
 
445
        if pyqt4_base_dir != pyqt_dir:
 
446
            base_dirs_to_check.append(pyqt4_base_dir)
 
447
    if not base_dirs_to_check:
 
448
        log.warn("Can't find PyQt4 installation -> not including imageformat"
 
449
                 " plugins")
 
450
    else:
 
451
        files = []
 
452
        for base_dir in base_dirs_to_check:
 
453
            plug_dir = os.path.join(base_dir, 'plugins', 'imageformats')
 
454
            if os.path.isdir(plug_dir):
 
455
                for fname in os.listdir(plug_dir):
 
456
                    # Include plugin dlls, but not debugging dlls
 
457
                    fullpath = os.path.join(plug_dir, fname)
 
458
                    if fname.endswith('.dll') and not fname.endswith('d4.dll'):
 
459
                        files.append(fullpath)
 
460
        if files:
 
461
            data_files.append(('imageformats', files))
 
462
        else:
 
463
            log.warn('PyQt4 was found, but we could not find any imageformat'
 
464
                     ' plugins. Are you sure your configuration is correct?')
421
465
 
422
466
 
423
467
def get_svn_py2exe_info(includes, excludes, packages):
424
468
    packages.append('subvertpy')
 
469
    packages.append('sqlite3')
 
470
 
 
471
 
 
472
def get_fastimport_py2exe_info(includes, excludes, packages):
 
473
    # This is the python-fastimport package, not to be confused with the
 
474
    # bzr-fastimport plugin.
 
475
    packages.append('fastimport')
425
476
 
426
477
 
427
478
if 'bdist_wininst' in sys.argv:
485
536
            install_data.run(self)
486
537
 
487
538
            py2exe = self.distribution.get_command_obj('py2exe', False)
488
 
            optimize = py2exe.optimize
 
539
            # GZ 2010-04-19: Setup has py2exe.optimize as 2, but give plugins
 
540
            #                time before living with docstring stripping
 
541
            optimize = 1
489
542
            compile_names = [f for f in self.outfiles if f.endswith('.py')]
 
543
            # Round mtime to nearest even second so that installing on a FAT
 
544
            # filesystem bytecode internal and script timestamps will match
 
545
            for f in compile_names:
 
546
                mtime = os.stat(f).st_mtime
 
547
                remainder = mtime % 2
 
548
                if remainder:
 
549
                    mtime -= remainder
 
550
                    os.utime(f, (mtime, mtime))
490
551
            byte_compile(compile_names,
491
552
                         optimize=optimize,
492
553
                         force=self.force, prefix=self.install_dir,
493
554
                         dry_run=self.dry_run)
494
 
            if optimize:
495
 
                suffix = 'o'
496
 
            else:
497
 
                suffix = 'c'
498
 
            self.outfiles.extend([f + suffix for f in compile_names])
 
555
            self.outfiles.extend([f + 'o' for f in compile_names])
499
556
    # end of class install_data_with_bytecompile
500
557
 
501
558
    target = py2exe.build_exe.Target(script = "bzr",
505
562
                                     version = version_str,
506
563
                                     description = META_INFO['description'],
507
564
                                     author = META_INFO['author'],
508
 
                                     copyright = "(c) Canonical Ltd, 2005-2007",
 
565
                                     copyright = "(c) Canonical Ltd, 2005-2010",
509
566
                                     company_name = "Canonical Ltd.",
510
567
                                     comments = META_INFO['description'],
511
568
                                    )
 
569
    gui_target = copy.copy(target)
 
570
    gui_target.dest_base = "bzrw"
512
571
 
513
572
    packages = BZRLIB['packages']
514
573
    packages.remove('bzrlib')
524
583
    if sys.version.startswith('2.4'):
525
584
        # adding elementtree package
526
585
        additional_packages.add('elementtree')
527
 
    elif sys.version.startswith('2.5'):
 
586
    elif sys.version.startswith('2.6') or sys.version.startswith('2.5'):
528
587
        additional_packages.add('xml.etree')
529
588
    else:
530
589
        import warnings
576
635
            excludes.extend(["bzrlib.plugins." + d for d in dirs])
577
636
        x = []
578
637
        for i in files:
579
 
            if os.path.splitext(i)[1] not in [".py", ".pyd", ".dll", ".mo"]:
 
638
            # Throw away files we don't want packaged. Note that plugins may
 
639
            # have data files with all sorts of extensions so we need to
 
640
            # be conservative here about what we ditch.
 
641
            ext = os.path.splitext(i)[1]
 
642
            if ext.endswith('~') or ext in [".pyc", ".swp"]:
580
643
                continue
581
644
            if i == '__init__.py' and root == 'bzrlib/plugins':
582
645
                continue
595
658
    console_targets = [target,
596
659
                       'tools/win32/bzr_postinstall.py',
597
660
                       ]
598
 
    gui_targets = []
599
 
    com_targets = []
 
661
    gui_targets = [gui_target]
600
662
    data_files = topics_files + plugins_files
601
663
 
602
664
    if 'qbzr' in plugins:
603
 
        get_qbzr_py2exe_info(includes, excludes, packages)
 
665
        get_qbzr_py2exe_info(includes, excludes, packages, data_files)
604
666
 
605
667
    if 'svn' in plugins:
606
668
        get_svn_py2exe_info(includes, excludes, packages)
607
669
 
 
670
    if 'fastimport' in plugins:
 
671
        get_fastimport_py2exe_info(includes, excludes, packages)
 
672
 
608
673
    if "TBZR" in os.environ:
609
674
        # TORTOISE_OVERLAYS_MSI_WIN32 must be set to the location of the
610
675
        # TortoiseOverlays MSI installer file. It is in the TSVN svn repo and
629
694
        # print this warning to stderr as output is redirected, so it is seen
630
695
        # at build time.  Also to stdout so it appears in the log
631
696
        for f in (sys.stderr, sys.stdout):
632
 
            print >> f, \
633
 
                "Skipping TBZR binaries - please set TBZR to a directory to enable"
 
697
            f.write("Skipping TBZR binaries - "
 
698
                "please set TBZR to a directory to enable\n")
634
699
 
635
700
    # MSWSOCK.dll is a system-specific library, which py2exe accidentally pulls
636
701
    # in on Vista.
637
 
    dll_excludes.extend(["MSWSOCK.dll", "MSVCP60.dll", "powrprof.dll"])
 
702
    dll_excludes.extend(["MSWSOCK.dll",
 
703
                         "MSVCP60.dll",
 
704
                         "MSVCP90.dll",
 
705
                         "powrprof.dll",
 
706
                         "SHFOLDER.dll"])
638
707
    options_list = {"py2exe": {"packages": packages + list(additional_packages),
639
708
                               "includes": includes,
640
709
                               "excludes": excludes,
641
710
                               "dll_excludes": dll_excludes,
642
711
                               "dist_dir": "win32_bzr.exe",
643
 
                               "optimize": 1,
 
712
                               "optimize": 2,
 
713
                               "custom_boot_script":
 
714
                                        "tools/win32/py2exe_boot_common.py",
644
715
                              },
645
716
                   }
646
717
 
647
 
    setup(options=options_list,
648
 
          console=console_targets,
649
 
          windows=gui_targets,
650
 
          com_server=com_targets,
651
 
          zipfile='lib/library.zip',
652
 
          data_files=data_files,
653
 
          cmdclass={'install_data': install_data_with_bytecompile},
654
 
          )
 
718
    # We want the libaray.zip to have optimize = 2, but the exe to have
 
719
    # optimize = 1, so that .py files that get compilied at run time
 
720
    # (e.g. user installed plugins) dont have their doc strings removed.
 
721
    class py2exe_no_oo_exe(py2exe.build_exe.py2exe):
 
722
        def build_executable(self, *args, **kwargs):
 
723
            self.optimize = 1
 
724
            py2exe.build_exe.py2exe.build_executable(self, *args, **kwargs)
 
725
            self.optimize = 2
 
726
 
 
727
    if __name__ == '__main__':
 
728
        setup(options=options_list,
 
729
              console=console_targets,
 
730
              windows=gui_targets,
 
731
              zipfile='lib/library.zip',
 
732
              data_files=data_files,
 
733
              cmdclass={'install_data': install_data_with_bytecompile,
 
734
                        'py2exe': py2exe_no_oo_exe},
 
735
              )
655
736
 
656
737
else:
657
738
    # ad-hoc for easy_install
672
753
    ARGS.update(BZRLIB)
673
754
    ARGS.update(PKG_DATA)
674
755
 
675
 
    setup(**ARGS)
 
756
    if __name__ == '__main__':
 
757
        setup(**ARGS)