~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to setup.py

Replace remaining to unittest.TestResult methods with super

Show diffs side-by-side

added added

removed removed

Lines of Context:
37
37
    'version':      bzrlib.__version__,
38
38
    'author':       'Canonical Ltd',
39
39
    'author_email': 'bazaar@lists.canonical.com',
40
 
    'url':          'http://www.bazaar-vcs.org/',
 
40
    'url':          'http://bazaar.canonical.com/',
41
41
    'description':  'Friendly distributed version control system',
42
42
    'license':      'GNU GPL v2',
43
 
    'download_url': 'http://bazaar-vcs.org/Download',
 
43
    'download_url': 'https://launchpad.net/bzr/+download',
44
44
    'long_description': get_long_description(),
45
45
    'classifiers': [
46
46
        'Development Status :: 6 - Mature',
65
65
            'package_data': {'bzrlib': ['doc/api/*.txt',
66
66
                                        'tests/test_patches_data/*',
67
67
                                        'help_topics/en/*.txt',
 
68
                                        'tests/ssl_certs/server_without_pass.key',
 
69
                                        'tests/ssl_certs/server_with_pass.key',
 
70
                                        'tests/ssl_certs/server.crt'
68
71
                                       ]},
69
72
           }
70
73
 
93
96
BZRLIB['packages'] = get_bzrlib_packages()
94
97
 
95
98
 
 
99
from distutils import log
96
100
from distutils.core import setup
97
101
from distutils.command.install_scripts import install_scripts
98
102
from distutils.command.install_data import install_data
121
125
                f = file(batch_path, "w")
122
126
                f.write(batch_str)
123
127
                f.close()
124
 
                print "Created:", batch_path
125
 
            except Exception, e:
126
 
                print "ERROR: Unable to create %s: %s" % (batch_path, e)
 
128
                print("Created: %s" % batch_path)
 
129
            except Exception:
 
130
                e = sys.exc_info()[1]
 
131
                print("ERROR: Unable to create %s: %s" % (batch_path, e))
127
132
 
128
133
    def _quoted_path(self, path):
129
134
        if ' ' in path:
148
153
    def run(self):
149
154
        build.run(self)
150
155
 
151
 
        import generate_docs
 
156
        from tools import generate_docs
152
157
        generate_docs.main(argv=["bzr", "man"])
153
158
 
154
159
 
163
168
from distutils.extension import Extension
164
169
ext_modules = []
165
170
try:
166
 
    from Pyrex.Distutils import build_ext
 
171
    try:
 
172
        from Pyrex.Distutils import build_ext
 
173
        from Pyrex.Compiler.Version import version as pyrex_version
 
174
    except ImportError:
 
175
        print("No Pyrex, trying Cython...")
 
176
        from Cython.Distutils import build_ext
 
177
        from Cython.Compiler.Version import version as pyrex_version
167
178
except ImportError:
168
179
    have_pyrex = False
169
180
    # 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
 
181
    print("")
 
182
    print("The python package 'Pyrex' is not available."
 
183
          " If the .c files are available,")
 
184
    print("they will be built,"
 
185
          " but modifying the .pyx files will not rebuild them.")
 
186
    print("")
176
187
    from distutils.command.build_ext import build_ext
177
188
else:
178
189
    have_pyrex = True
179
 
    from Pyrex.Compiler.Version import version as pyrex_version
 
190
    pyrex_version_info = tuple(map(int, pyrex_version.split('.')))
180
191
 
181
192
 
182
193
class build_ext_if_possible(build_ext):
194
205
    def run(self):
195
206
        try:
196
207
            build_ext.run(self)
197
 
        except DistutilsPlatformError, e:
 
208
        except DistutilsPlatformError:
 
209
            e = sys.exc_info()[1]
198
210
            if not self.allow_python_fallback:
199
211
                log.warn('\n  Cannot build extensions.\n'
200
212
                         '  Use "build_ext --allow-python-fallback" to use'
209
221
            build_ext.build_extension(self, ext)
210
222
        except CCompilerError:
211
223
            if not self.allow_python_fallback:
212
 
                log.warn('\n  Cannot build extensions.\n'
 
224
                log.warn('\n  Cannot build extension "%s".\n'
213
225
                         '  Use "build_ext --allow-python-fallback" to use'
214
226
                         ' slower python implementations instead.\n'
215
227
                         % (ext.name,))
224
236
unavailable_files = []
225
237
 
226
238
 
227
 
def add_pyrex_extension(module_name, libraries=None):
 
239
def add_pyrex_extension(module_name, libraries=None, extra_source=[]):
228
240
    """Add a pyrex module to build.
229
241
 
230
242
    This will use Pyrex to auto-generate the .c file if it is available.
242
254
    c_name = path + '.c'
243
255
    define_macros = []
244
256
    if sys.platform == 'win32':
245
 
        # pyrex uses the macro WIN32 to detect the platform, even though it should
246
 
        # be using something like _WIN32 or MS_WINDOWS, oh well, we can give it the
247
 
        # right value.
 
257
        # pyrex uses the macro WIN32 to detect the platform, even though it
 
258
        # should be using something like _WIN32 or MS_WINDOWS, oh well, we can
 
259
        # give it the right value.
248
260
        define_macros.append(('WIN32', None))
249
261
    if have_pyrex:
250
 
        ext_modules.append(Extension(module_name, [pyrex_name],
251
 
            define_macros=define_macros, libraries=libraries))
 
262
        source = [pyrex_name]
252
263
    else:
253
264
        if not os.path.isfile(c_name):
254
265
            unavailable_files.append(c_name)
 
266
            return
255
267
        else:
256
 
            ext_modules.append(Extension(module_name, [c_name],
257
 
                define_macros=define_macros, libraries=libraries))
258
 
 
259
 
 
260
 
add_pyrex_extension('bzrlib._btree_serializer_c')
 
268
            source = [c_name]
 
269
    source.extend(extra_source)
 
270
    ext_modules.append(Extension(module_name, source,
 
271
        define_macros=define_macros, libraries=libraries))
 
272
 
 
273
 
 
274
add_pyrex_extension('bzrlib._annotator_pyx')
 
275
add_pyrex_extension('bzrlib._bencode_pyx')
261
276
add_pyrex_extension('bzrlib._chunks_to_lines_pyx')
262
 
add_pyrex_extension('bzrlib._knit_load_data_c')
 
277
add_pyrex_extension('bzrlib._groupcompress_pyx',
 
278
                    extra_source=['bzrlib/diff-delta.c'])
 
279
add_pyrex_extension('bzrlib._knit_load_data_pyx')
 
280
add_pyrex_extension('bzrlib._known_graph_pyx')
 
281
add_pyrex_extension('bzrlib._rio_pyx')
263
282
if sys.platform == 'win32':
264
 
    add_pyrex_extension('bzrlib._dirstate_helpers_c',
 
283
    add_pyrex_extension('bzrlib._dirstate_helpers_pyx',
265
284
                        libraries=['Ws2_32'])
266
285
    add_pyrex_extension('bzrlib._walkdirs_win32')
267
286
else:
268
 
    if have_pyrex and pyrex_version == '0.9.4.1':
 
287
    if have_pyrex and pyrex_version_info[:3] == (0,9,4):
269
288
        # Pyrex 0.9.4.1 fails to compile this extension correctly
270
289
        # The code it generates re-uses a "local" pointer and
271
290
        # calls "PY_DECREF" after having set it to NULL. (It mixes PY_XDECREF
272
291
        # which is NULL safe with PY_DECREF which is not.)
273
 
        print 'Cannot build extension "bzrlib._dirstate_helpers_c" using'
274
 
        print 'your version of pyrex "%s". Please upgrade your pyrex' % (
275
 
            pyrex_version,)
276
 
        print 'install. For now, the non-compiled (python) version will'
277
 
        print 'be used instead.'
 
292
        # <https://bugs.edge.launchpad.net/bzr/+bug/449372>
 
293
        # <https://bugs.edge.launchpad.net/bzr/+bug/276868>
 
294
        print('Cannot build extension "bzrlib._dirstate_helpers_pyx" using')
 
295
        print('your version of pyrex "%s". Please upgrade your pyrex' % (
 
296
            pyrex_version,))
 
297
        print('install. For now, the non-compiled (python) version will')
 
298
        print('be used instead.')
278
299
    else:
279
 
        add_pyrex_extension('bzrlib._dirstate_helpers_c')
 
300
        add_pyrex_extension('bzrlib._dirstate_helpers_pyx')
280
301
    add_pyrex_extension('bzrlib._readdir_pyx')
281
 
ext_modules.append(Extension('bzrlib._patiencediff_c', ['bzrlib/_patiencediff_c.c']))
 
302
add_pyrex_extension('bzrlib._chk_map_pyx')
 
303
ext_modules.append(Extension('bzrlib._patiencediff_c',
 
304
                             ['bzrlib/_patiencediff_c.c']))
 
305
if have_pyrex and pyrex_version_info < (0, 9, 6, 3):
 
306
    print("")
 
307
    print('Your Pyrex/Cython version %s is too old to build the simple_set' % (
 
308
        pyrex_version))
 
309
    print('and static_tuple extensions.')
 
310
    print('Please upgrade to at least Pyrex 0.9.6.3')
 
311
    print("")
 
312
    # TODO: Should this be a fatal error?
 
313
else:
 
314
    # We only need 0.9.6.3 to build _simple_set_pyx, but static_tuple depends
 
315
    # on simple_set
 
316
    add_pyrex_extension('bzrlib._simple_set_pyx')
 
317
    ext_modules.append(Extension('bzrlib._static_tuple_c',
 
318
                                 ['bzrlib/_static_tuple_c.c']))
 
319
add_pyrex_extension('bzrlib._btree_serializer_pyx')
282
320
 
283
321
 
284
322
if unavailable_files:
285
 
    print 'C extension(s) not found:'
286
 
    print '   %s' % ('\n  '.join(unavailable_files),)
287
 
    print 'The python versions will be used instead.'
288
 
    print
 
323
    print('C extension(s) not found:')
 
324
    print('   %s' % ('\n  '.join(unavailable_files),))
 
325
    print('The python versions will be used instead.')
 
326
    print("")
289
327
 
290
328
 
291
329
def get_tbzr_py2exe_info(includes, excludes, packages, console_targets,
311
349
    # Ensure tbzrlib itself is on sys.path
312
350
    sys.path.append(tbzr_root)
313
351
 
314
 
    # Ensure our COM "entry-point" is on sys.path
315
 
    sys.path.append(os.path.join(tbzr_root, "shellext", "python"))
316
 
 
317
352
    packages.append("tbzrlib")
318
353
 
319
354
    # collect up our icons.
341
376
    excludes.extend("""pywin pywin.dialogs pywin.dialogs.list
342
377
                       win32ui crawler.Crawler""".split())
343
378
 
344
 
    # NOTE: We still create a DLL version of the Python implemented shell
345
 
    # extension for testing purposes - but it is *not* registered by
346
 
    # default - our C++ one is instead.  To discourage people thinking
347
 
    # this DLL is still necessary, its called 'tbzr_old.dll'
348
 
    tbzr = dict(
349
 
        modules=["tbzr"],
350
 
        create_exe = False, # we only want a .dll
351
 
        dest_base = 'tbzr_old',
352
 
    )
353
 
    com_targets.append(tbzr)
354
 
 
355
379
    # tbzrcache executables - a "console" version for debugging and a
356
380
    # GUI version that is generally used.
357
381
    tbzrcache = dict(
382
406
    console_targets.append(tracer)
383
407
 
384
408
    # The C++ implemented shell extensions.
385
 
    dist_dir = os.path.join(tbzr_root, "shellext", "cpp", "tbzrshellext",
386
 
                            "build", "dist")
 
409
    dist_dir = os.path.join(tbzr_root, "shellext", "build")
387
410
    data_files.append(('', [os.path.join(dist_dir, 'tbzrshellext_x86.dll')]))
388
411
    data_files.append(('', [os.path.join(dist_dir, 'tbzrshellext_x64.dll')]))
389
412
 
390
413
 
391
 
def get_qbzr_py2exe_info(includes, excludes, packages):
 
414
def get_qbzr_py2exe_info(includes, excludes, packages, data_files):
392
415
    # PyQt4 itself still escapes the plugin detection code for some reason...
393
 
    packages.append('PyQt4')
394
 
    excludes.append('PyQt4.elementtree.ElementTree')
 
416
    includes.append('PyQt4.QtCore')
 
417
    includes.append('PyQt4.QtGui')
395
418
    includes.append('sip') # extension module required for Qt.
396
419
    packages.append('pygments') # colorizer for qbzr
397
420
    packages.append('docutils') # html formatting
398
 
    # but we can avoid many Qt4 Dlls.
399
 
    dll_excludes.extend(
400
 
        """QtAssistantClient4.dll QtCLucene4.dll QtDesigner4.dll
401
 
        QtHelp4.dll QtNetwork4.dll QtOpenGL4.dll QtScript4.dll
402
 
        QtSql4.dll QtTest4.dll QtWebKit4.dll QtXml4.dll
403
 
        qscintilla2.dll""".split())
 
421
    includes.append('win32event')  # for qsubprocess stuff
404
422
    # the qt binaries might not be on PATH...
405
 
    qt_dir = os.path.join(sys.prefix, "PyQt4", "bin")
406
 
    path = os.environ.get("PATH","")
407
 
    if qt_dir.lower() not in [p.lower() for p in path.split(os.pathsep)]:
408
 
        os.environ["PATH"] = path + os.pathsep + qt_dir
 
423
    # They seem to install to a place like C:\Python25\PyQt4\*
 
424
    # Which is not the same as C:\Python25\Lib\site-packages\PyQt4
 
425
    pyqt_dir = os.path.join(sys.prefix, "PyQt4")
 
426
    pyqt_bin_dir = os.path.join(pyqt_dir, "bin")
 
427
    if os.path.isdir(pyqt_bin_dir):
 
428
        path = os.environ.get("PATH", "")
 
429
        if pyqt_bin_dir.lower() not in [p.lower() for p in path.split(os.pathsep)]:
 
430
            os.environ["PATH"] = path + os.pathsep + pyqt_bin_dir
 
431
    # also add all imageformat plugins to distribution
 
432
    # We will look in 2 places, dirname(PyQt4.__file__) and pyqt_dir
 
433
    base_dirs_to_check = []
 
434
    if os.path.isdir(pyqt_dir):
 
435
        base_dirs_to_check.append(pyqt_dir)
 
436
    try:
 
437
        import PyQt4
 
438
    except ImportError:
 
439
        pass
 
440
    else:
 
441
        pyqt4_base_dir = os.path.dirname(PyQt4.__file__)
 
442
        if pyqt4_base_dir != pyqt_dir:
 
443
            base_dirs_to_check.append(pyqt4_base_dir)
 
444
    if not base_dirs_to_check:
 
445
        log.warn("Can't find PyQt4 installation -> not including imageformat"
 
446
                 " plugins")
 
447
    else:
 
448
        files = []
 
449
        for base_dir in base_dirs_to_check:
 
450
            plug_dir = os.path.join(base_dir, 'plugins', 'imageformats')
 
451
            if os.path.isdir(plug_dir):
 
452
                for fname in os.listdir(plug_dir):
 
453
                    # Include plugin dlls, but not debugging dlls
 
454
                    fullpath = os.path.join(plug_dir, fname)
 
455
                    if fname.endswith('.dll') and not fname.endswith('d4.dll'):
 
456
                        files.append(fullpath)
 
457
        if files:
 
458
            data_files.append(('imageformats', files))
 
459
        else:
 
460
            log.warn('PyQt4 was found, but we could not find any imageformat'
 
461
                     ' plugins. Are you sure your configuration is correct?')
409
462
 
410
463
 
411
464
def get_svn_py2exe_info(includes, excludes, packages):
412
465
    packages.append('subvertpy')
 
466
    packages.append('sqlite3')
413
467
 
414
468
 
415
469
if 'bdist_wininst' in sys.argv:
473
527
            install_data.run(self)
474
528
 
475
529
            py2exe = self.distribution.get_command_obj('py2exe', False)
476
 
            optimize = py2exe.optimize
 
530
            # GZ 2010-04-19: Setup has py2exe.optimize as 2, but give plugins
 
531
            #                time before living with docstring stripping
 
532
            optimize = 1
477
533
            compile_names = [f for f in self.outfiles if f.endswith('.py')]
 
534
            # Round mtime to nearest even second so that installing on a FAT
 
535
            # filesystem bytecode internal and script timestamps will match
 
536
            for f in compile_names:
 
537
                mtime = os.stat(f).st_mtime
 
538
                remainder = mtime % 2
 
539
                if remainder:
 
540
                    mtime -= remainder
 
541
                    os.utime(f, (mtime, mtime))
478
542
            byte_compile(compile_names,
479
543
                         optimize=optimize,
480
544
                         force=self.force, prefix=self.install_dir,
481
545
                         dry_run=self.dry_run)
482
 
            if optimize:
483
 
                suffix = 'o'
484
 
            else:
485
 
                suffix = 'c'
486
 
            self.outfiles.extend([f + suffix for f in compile_names])
 
546
            self.outfiles.extend([f + 'o' for f in compile_names])
487
547
    # end of class install_data_with_bytecompile
488
548
 
489
549
    target = py2exe.build_exe.Target(script = "bzr",
493
553
                                     version = version_str,
494
554
                                     description = META_INFO['description'],
495
555
                                     author = META_INFO['author'],
496
 
                                     copyright = "(c) Canonical Ltd, 2005-2007",
 
556
                                     copyright = "(c) Canonical Ltd, 2005-2010",
497
557
                                     company_name = "Canonical Ltd.",
498
558
                                     comments = META_INFO['description'],
499
559
                                    )
512
572
    if sys.version.startswith('2.4'):
513
573
        # adding elementtree package
514
574
        additional_packages.add('elementtree')
515
 
    elif sys.version.startswith('2.5'):
 
575
    elif sys.version.startswith('2.6') or sys.version.startswith('2.5'):
516
576
        additional_packages.add('xml.etree')
517
577
    else:
518
578
        import warnings
526
586
                  ImaginaryModule cElementTree elementtree.ElementTree
527
587
                  Crypto.PublicKey._fastmath
528
588
                  medusa medusa.filesys medusa.ftp_server
529
 
                  tools tools.doc_generate
 
589
                  tools
530
590
                  resource validate""".split()
531
591
    dll_excludes = []
532
592
 
564
624
            excludes.extend(["bzrlib.plugins." + d for d in dirs])
565
625
        x = []
566
626
        for i in files:
567
 
            if os.path.splitext(i)[1] not in [".py", ".pyd", ".dll", ".mo"]:
 
627
            # Throw away files we don't want packaged. Note that plugins may
 
628
            # have data files with all sorts of extensions so we need to
 
629
            # be conservative here about what we ditch.
 
630
            ext = os.path.splitext(i)[1]
 
631
            if ext.endswith('~') or ext in [".pyc", ".swp"]:
568
632
                continue
569
633
            if i == '__init__.py' and root == 'bzrlib/plugins':
570
634
                continue
584
648
                       'tools/win32/bzr_postinstall.py',
585
649
                       ]
586
650
    gui_targets = []
587
 
    com_targets = []
588
651
    data_files = topics_files + plugins_files
589
652
 
590
653
    if 'qbzr' in plugins:
591
 
        get_qbzr_py2exe_info(includes, excludes, packages)
 
654
        get_qbzr_py2exe_info(includes, excludes, packages, data_files)
592
655
 
593
656
    if 'svn' in plugins:
594
657
        get_svn_py2exe_info(includes, excludes, packages)
597
660
        # TORTOISE_OVERLAYS_MSI_WIN32 must be set to the location of the
598
661
        # TortoiseOverlays MSI installer file. It is in the TSVN svn repo and
599
662
        # can be downloaded from (username=guest, blank password):
600
 
        # http://tortoisesvn.tigris.org/svn/tortoisesvn/TortoiseOverlays/version-1.0.4/bin/TortoiseOverlays-1.0.4.11886-win32.msi
 
663
        # http://tortoisesvn.tigris.org/svn/tortoisesvn/TortoiseOverlays
 
664
        # look for: version-1.0.4/bin/TortoiseOverlays-1.0.4.11886-win32.msi
601
665
        # Ditto for TORTOISE_OVERLAYS_MSI_X64, pointing at *-x64.msi.
602
666
        for needed in ('TORTOISE_OVERLAYS_MSI_WIN32',
603
667
                       'TORTOISE_OVERLAYS_MSI_X64'):
 
668
            url = ('http://guest:@tortoisesvn.tigris.org/svn/tortoisesvn'
 
669
                   '/TortoiseOverlays')
604
670
            if not os.path.isfile(os.environ.get(needed, '<nofile>')):
605
 
                raise RuntimeError("Please set %s to the"
606
 
                                   " location of the relevant TortoiseOverlays"
607
 
                                   " .msi installer file" % needed)
 
671
                raise RuntimeError(
 
672
                    "\nPlease set %s to the location of the relevant"
 
673
                    "\nTortoiseOverlays .msi installer file."
 
674
                    " The installers can be found at"
 
675
                    "\n  %s"
 
676
                    "\ncheck in the version-X.Y.Z/bin/ subdir" % (needed, url))
608
677
        get_tbzr_py2exe_info(includes, excludes, packages, console_targets,
609
678
                             gui_targets, data_files)
610
679
    else:
611
680
        # print this warning to stderr as output is redirected, so it is seen
612
681
        # at build time.  Also to stdout so it appears in the log
613
682
        for f in (sys.stderr, sys.stdout):
614
 
            print >> f, \
615
 
                "Skipping TBZR binaries - please set TBZR to a directory to enable"
 
683
            f.write("Skipping TBZR binaries - "
 
684
                "please set TBZR to a directory to enable\n")
616
685
 
617
686
    # MSWSOCK.dll is a system-specific library, which py2exe accidentally pulls
618
687
    # in on Vista.
619
 
    dll_excludes.append("MSWSOCK.dll")
 
688
    dll_excludes.extend(["MSWSOCK.dll", "MSVCP60.dll", "powrprof.dll"])
620
689
    options_list = {"py2exe": {"packages": packages + list(additional_packages),
621
690
                               "includes": includes,
622
691
                               "excludes": excludes,
623
692
                               "dll_excludes": dll_excludes,
624
693
                               "dist_dir": "win32_bzr.exe",
625
 
                               "optimize": 1,
 
694
                               "optimize": 2,
626
695
                              },
627
696
                   }
628
 
 
629
 
    setup(options=options_list,
630
 
          console=console_targets,
631
 
          windows=gui_targets,
632
 
          com_server=com_targets,
633
 
          zipfile='lib/library.zip',
634
 
          data_files=data_files,
635
 
          cmdclass={'install_data': install_data_with_bytecompile},
636
 
          )
 
697
    if __name__ == '__main__':
 
698
        setup(options=options_list,
 
699
              console=console_targets,
 
700
              windows=gui_targets,
 
701
              zipfile='lib/library.zip',
 
702
              data_files=data_files,
 
703
              cmdclass={'install_data': install_data_with_bytecompile},
 
704
              )
637
705
 
638
706
else:
639
707
    # ad-hoc for easy_install
640
708
    DATA_FILES = []
641
709
    if not 'bdist_egg' in sys.argv:
642
 
        # generate and install bzr.1 only with plain install, not easy_install one
 
710
        # generate and install bzr.1 only with plain install, not the
 
711
        # easy_install one
643
712
        DATA_FILES = [('man/man1', ['bzr.1'])]
644
713
 
 
714
    if sys.platform != 'win32':
 
715
        # see https://wiki.kubuntu.org/Apport/DeveloperHowTo
 
716
        #
 
717
        # checking the paths and hardcoding the check for root is a bit gross,
 
718
        # but I don't see a cleaner way to find out the locations in a way
 
719
        # that's going to align with the hardcoded paths in apport.
 
720
        if os.geteuid() == 0:
 
721
            DATA_FILES += [
 
722
                ('/usr/share/apport/package-hooks',
 
723
                    ['apport/source_bzr.py']),
 
724
                ('/etc/apport/crashdb.conf.d/',
 
725
                    ['apport/bzr-crashdb.conf']),]
 
726
 
645
727
    # std setup
646
728
    ARGS = {'scripts': ['bzr'],
647
729
            'data_files': DATA_FILES,
653
735
    ARGS.update(BZRLIB)
654
736
    ARGS.update(PKG_DATA)
655
737
 
656
 
    setup(**ARGS)
 
738
    if __name__ == '__main__':
 
739
        setup(**ARGS)