← Back to branch summary

~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to setup.py

  • Committer: Andrew Bennetts
  • Date: 2009-08-07 04:17:51 UTC
  • mto: This revision was merged to the branch mainline in revision 4608.
  • Revision ID: andrew.bennetts@canonical.com-20090807041751-0vhb0y0g7k49hr45
Review comments from John.

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
12
 
14
 
if sys.version_info < (2, 6):
15
 
    sys.stderr.write("[ERROR] Not a supported Python version. Need 2.6+\n")
 
13
if sys.version_info < (2, 4):
 
14
    sys.stderr.write("[ERROR] Not a supported Python version. Need 2.4+\n")
16
15
    sys.exit(1)
17
16
 
18
17
# NOTE: The directory containing setup.py, whether run by 'python setup.py' or
38
37
    'version':      bzrlib.__version__,
39
38
    'author':       'Canonical Ltd',
40
39
    'author_email': 'bazaar@lists.canonical.com',
41
 
    'url':          'http://bazaar.canonical.com/',
 
40
    'url':          'http://www.bazaar-vcs.org/',
42
41
    'description':  'Friendly distributed version control system',
43
42
    'license':      'GNU GPL v2',
44
 
    'download_url': 'https://launchpad.net/bzr/+download',
 
43
    'download_url': 'http://bazaar-vcs.org/Download',
45
44
    'long_description': get_long_description(),
46
45
    'classifiers': [
47
46
        'Development Status :: 6 - Mature',
66
65
            'package_data': {'bzrlib': ['doc/api/*.txt',
67
66
                                        'tests/test_patches_data/*',
68
67
                                        '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',
73
 
                                        'locale/*/LC_MESSAGES/*.mo',
74
68
                                       ]},
75
69
           }
76
70
 
128
122
                f = file(batch_path, "w")
129
123
                f.write(batch_str)
130
124
                f.close()
131
 
                print("Created: %s" % batch_path)
132
 
            except Exception:
133
 
                e = sys.exc_info()[1]
134
 
                print("ERROR: Unable to create %s: %s" % (batch_path, e))
 
125
                print "Created:", batch_path
 
126
            except Exception, e:
 
127
                print "ERROR: Unable to create %s: %s" % (batch_path, e)
135
128
 
136
129
    def _quoted_path(self, path):
137
130
        if ' ' in path:
153
146
    Generate bzr.1.
154
147
    """
155
148
 
156
 
    sub_commands = build.sub_commands + [
157
 
            ('build_mo', lambda _: True),
158
 
            ]
159
 
 
160
149
    def run(self):
161
150
        build.run(self)
162
151
 
168
157
## Setup
169
158
########################
170
159
 
171
 
from tools.build_mo import build_mo
172
 
 
173
160
command_classes = {'install_scripts': my_install_scripts,
174
 
                   'build': bzr_build,
175
 
                   'build_mo': build_mo,
176
 
                   }
 
161
                   'build': bzr_build}
177
162
from distutils import log
178
163
from distutils.errors import CCompilerError, DistutilsPlatformError
179
164
from distutils.extension import Extension
180
165
ext_modules = []
181
166
try:
182
 
    try:
183
 
        from Cython.Distutils import build_ext
184
 
        from Cython.Compiler.Version import version as pyrex_version
185
 
    except ImportError:
186
 
        print("No Cython, trying Pyrex...")
187
 
        from Pyrex.Distutils import build_ext
188
 
        from Pyrex.Compiler.Version import version as pyrex_version
 
167
    from Pyrex.Distutils import build_ext
189
168
except ImportError:
190
169
    have_pyrex = False
191
170
    # try to build the extension from the prior generated source.
192
 
    print("")
193
 
    print("The python package 'Pyrex' is not available."
194
 
          " If the .c files are available,")
195
 
    print("they will be built,"
196
 
          " but modifying the .pyx files will not rebuild them.")
197
 
    print("")
 
171
    print
 
172
    print ("The python package 'Pyrex' is not available."
 
173
           " If the .c files are available,")
 
174
    print ("they will be built,"
 
175
           " but modifying the .pyx files will not rebuild them.")
 
176
    print
198
177
    from distutils.command.build_ext import build_ext
199
178
else:
200
179
    have_pyrex = True
201
 
    pyrex_version_info = tuple(map(int, pyrex_version.split('.')))
 
180
    from Pyrex.Compiler.Version import version as pyrex_version
202
181
 
203
182
 
204
183
class build_ext_if_possible(build_ext):
216
195
    def run(self):
217
196
        try:
218
197
            build_ext.run(self)
219
 
        except DistutilsPlatformError:
220
 
            e = sys.exc_info()[1]
 
198
        except DistutilsPlatformError, e:
221
199
            if not self.allow_python_fallback:
222
200
                log.warn('\n  Cannot build extensions.\n'
223
201
                         '  Use "build_ext --allow-python-fallback" to use'
284
262
 
285
263
add_pyrex_extension('bzrlib._annotator_pyx')
286
264
add_pyrex_extension('bzrlib._bencode_pyx')
 
265
add_pyrex_extension('bzrlib._btree_serializer_pyx')
287
266
add_pyrex_extension('bzrlib._chunks_to_lines_pyx')
288
267
add_pyrex_extension('bzrlib._groupcompress_pyx',
289
268
                    extra_source=['bzrlib/diff-delta.c'])
294
273
    add_pyrex_extension('bzrlib._dirstate_helpers_pyx',
295
274
                        libraries=['Ws2_32'])
296
275
    add_pyrex_extension('bzrlib._walkdirs_win32')
 
276
    z_lib = 'zdll'
297
277
else:
298
 
    if have_pyrex and pyrex_version_info[:3] == (0,9,4):
 
278
    if have_pyrex and pyrex_version == '0.9.4.1':
299
279
        # Pyrex 0.9.4.1 fails to compile this extension correctly
300
280
        # The code it generates re-uses a "local" pointer and
301
281
        # calls "PY_DECREF" after having set it to NULL. (It mixes PY_XDECREF
302
282
        # which is NULL safe with PY_DECREF which is not.)
303
 
        # <https://bugs.launchpad.net/bzr/+bug/449372>
304
 
        # <https://bugs.launchpad.net/bzr/+bug/276868>
305
 
        print('Cannot build extension "bzrlib._dirstate_helpers_pyx" using')
306
 
        print('your version of pyrex "%s". Please upgrade your pyrex'
307
 
              % (pyrex_version,))
308
 
        print('install. For now, the non-compiled (python) version will')
309
 
        print('be used instead.')
 
283
        print 'Cannot build extension "bzrlib._dirstate_helpers_pyx" using'
 
284
        print 'your version of pyrex "%s". Please upgrade your pyrex' % (
 
285
            pyrex_version,)
 
286
        print 'install. For now, the non-compiled (python) version will'
 
287
        print 'be used instead.'
310
288
    else:
311
289
        add_pyrex_extension('bzrlib._dirstate_helpers_pyx')
312
290
    add_pyrex_extension('bzrlib._readdir_pyx')
313
 
add_pyrex_extension('bzrlib._chk_map_pyx')
 
291
    z_lib = 'z'
 
292
add_pyrex_extension('bzrlib._chk_map_pyx', libraries=[z_lib])
314
293
ext_modules.append(Extension('bzrlib._patiencediff_c',
315
294
                             ['bzrlib/_patiencediff_c.c']))
316
 
if have_pyrex and pyrex_version_info < (0, 9, 6, 3):
317
 
    print("")
318
 
    print('Your Pyrex/Cython version %s is too old to build the simple_set' % (
319
 
        pyrex_version))
320
 
    print('and static_tuple extensions.')
321
 
    print('Please upgrade to at least Pyrex 0.9.6.3')
322
 
    print("")
323
 
    # TODO: Should this be a fatal error?
324
 
else:
325
 
    # We only need 0.9.6.3 to build _simple_set_pyx, but static_tuple depends
326
 
    # on simple_set
327
 
    add_pyrex_extension('bzrlib._simple_set_pyx')
328
 
    ext_modules.append(Extension('bzrlib._static_tuple_c',
329
 
                                 ['bzrlib/_static_tuple_c.c']))
330
 
add_pyrex_extension('bzrlib._btree_serializer_pyx')
331
295
 
332
296
 
333
297
if unavailable_files:
334
 
    print('C extension(s) not found:')
335
 
    print('   %s' % ('\n  '.join(unavailable_files),))
336
 
    print('The python versions will be used instead.')
337
 
    print("")
 
298
    print 'C extension(s) not found:'
 
299
    print '   %s' % ('\n  '.join(unavailable_files),)
 
300
    print 'The python versions will be used instead.'
 
301
    print
338
302
 
339
303
 
340
304
def get_tbzr_py2exe_info(includes, excludes, packages, console_targets,
360
324
    # Ensure tbzrlib itself is on sys.path
361
325
    sys.path.append(tbzr_root)
362
326
 
 
327
    # Ensure our COM "entry-point" is on sys.path
 
328
    sys.path.append(os.path.join(tbzr_root, "shellext", "python"))
 
329
 
363
330
    packages.append("tbzrlib")
364
331
 
365
332
    # collect up our icons.
387
354
    excludes.extend("""pywin pywin.dialogs pywin.dialogs.list
388
355
                       win32ui crawler.Crawler""".split())
389
356
 
 
357
    # NOTE: We still create a DLL version of the Python implemented shell
 
358
    # extension for testing purposes - but it is *not* registered by
 
359
    # default - our C++ one is instead.  To discourage people thinking
 
360
    # this DLL is still necessary, its called 'tbzr_old.dll'
 
361
    tbzr = dict(
 
362
        modules=["tbzr"],
 
363
        create_exe = False, # we only want a .dll
 
364
        dest_base = 'tbzr_old',
 
365
    )
 
366
    com_targets.append(tbzr)
 
367
 
390
368
    # tbzrcache executables - a "console" version for debugging and a
391
369
    # GUI version that is generally used.
392
370
    tbzrcache = dict(
404
382
    # ditto for the tbzrcommand tool
405
383
    tbzrcommand = dict(
406
384
        script = os.path.join(tbzr_root, "scripts", "tbzrcommand.py"),
407
 
        icon_resources = icon_resources,
408
 
        other_resources = other_resources,
 
385
        icon_resources = [(0,'bzr.ico')],
409
386
    )
410
387
    console_targets.append(tbzrcommand)
411
388
    tbzrcommandw = tbzrcommand.copy()
418
395
    console_targets.append(tracer)
419
396
 
420
397
    # The C++ implemented shell extensions.
421
 
    dist_dir = os.path.join(tbzr_root, "shellext", "build")
 
398
    dist_dir = os.path.join(tbzr_root, "shellext", "cpp", "tbzrshellext",
 
399
                            "build", "dist")
422
400
    data_files.append(('', [os.path.join(dist_dir, 'tbzrshellext_x86.dll')]))
423
401
    data_files.append(('', [os.path.join(dist_dir, 'tbzrshellext_x64.dll')]))
424
402
 
425
403
 
426
404
def get_qbzr_py2exe_info(includes, excludes, packages, data_files):
427
405
    # PyQt4 itself still escapes the plugin detection code for some reason...
428
 
    includes.append('PyQt4.QtCore')
429
 
    includes.append('PyQt4.QtGui')
 
406
    packages.append('PyQt4')
 
407
    excludes.append('PyQt4.elementtree.ElementTree')
430
408
    includes.append('sip') # extension module required for Qt.
431
409
    packages.append('pygments') # colorizer for qbzr
432
410
    packages.append('docutils') # html formatting
433
 
    includes.append('win32event')  # for qsubprocess stuff
 
411
    # but we can avoid many Qt4 Dlls.
 
412
    dll_excludes.extend(
 
413
        """QtAssistantClient4.dll QtCLucene4.dll QtDesigner4.dll
 
414
        QtHelp4.dll QtNetwork4.dll QtOpenGL4.dll QtScript4.dll
 
415
        QtSql4.dll QtTest4.dll QtWebKit4.dll QtXml4.dll
 
416
        qscintilla2.dll""".split())
434
417
    # the qt binaries might not be on PATH...
435
418
    # They seem to install to a place like C:\Python25\PyQt4\*
436
419
    # Which is not the same as C:\Python25\Lib\site-packages\PyQt4
475
458
 
476
459
def get_svn_py2exe_info(includes, excludes, packages):
477
460
    packages.append('subvertpy')
478
 
    packages.append('sqlite3')
479
 
 
480
 
 
481
 
def get_fastimport_py2exe_info(includes, excludes, packages):
482
 
    # This is the python-fastimport package, not to be confused with the
483
 
    # bzr-fastimport plugin.
484
 
    packages.append('fastimport')
485
461
 
486
462
 
487
463
if 'bdist_wininst' in sys.argv:
508
484
            # help pages
509
485
            'data_files': find_docs(),
510
486
            # for building pyrex extensions
511
 
            'cmdclass': command_classes,
 
487
            'cmdclass': {'build_ext': build_ext_if_possible},
512
488
           }
513
489
 
514
490
    ARGS.update(META_INFO)
515
491
    ARGS.update(BZRLIB)
516
492
    ARGS.update(PKG_DATA)
517
 
 
 
493
    
518
494
    setup(**ARGS)
519
495
 
520
496
elif 'py2exe' in sys.argv:
545
521
            install_data.run(self)
546
522
 
547
523
            py2exe = self.distribution.get_command_obj('py2exe', False)
548
 
            # GZ 2010-04-19: Setup has py2exe.optimize as 2, but give plugins
549
 
            #                time before living with docstring stripping
550
 
            optimize = 1
 
524
            optimize = py2exe.optimize
551
525
            compile_names = [f for f in self.outfiles if f.endswith('.py')]
552
 
            # Round mtime to nearest even second so that installing on a FAT
553
 
            # filesystem bytecode internal and script timestamps will match
554
 
            for f in compile_names:
555
 
                mtime = os.stat(f).st_mtime
556
 
                remainder = mtime % 2
557
 
                if remainder:
558
 
                    mtime -= remainder
559
 
                    os.utime(f, (mtime, mtime))
560
526
            byte_compile(compile_names,
561
527
                         optimize=optimize,
562
528
                         force=self.force, prefix=self.install_dir,
563
529
                         dry_run=self.dry_run)
564
 
            self.outfiles.extend([f + 'o' for f in compile_names])
 
530
            if optimize:
 
531
                suffix = 'o'
 
532
            else:
 
533
                suffix = 'c'
 
534
            self.outfiles.extend([f + suffix for f in compile_names])
565
535
    # end of class install_data_with_bytecompile
566
536
 
567
537
    target = py2exe.build_exe.Target(script = "bzr",
571
541
                                     version = version_str,
572
542
                                     description = META_INFO['description'],
573
543
                                     author = META_INFO['author'],
574
 
                                     copyright = "(c) Canonical Ltd, 2005-2010",
 
544
                                     copyright = "(c) Canonical Ltd, 2005-2007",
575
545
                                     company_name = "Canonical Ltd.",
576
546
                                     comments = META_INFO['description'],
577
547
                                    )
578
 
    gui_target = copy.copy(target)
579
 
    gui_target.dest_base = "bzrw"
580
548
 
581
549
    packages = BZRLIB['packages']
582
550
    packages.remove('bzrlib')
592
560
    if sys.version.startswith('2.4'):
593
561
        # adding elementtree package
594
562
        additional_packages.add('elementtree')
595
 
    elif sys.version.startswith('2.6') or sys.version.startswith('2.5'):
 
563
    elif sys.version.startswith('2.5'):
596
564
        additional_packages.add('xml.etree')
597
565
    else:
598
566
        import warnings
644
612
            excludes.extend(["bzrlib.plugins." + d for d in dirs])
645
613
        x = []
646
614
        for i in files:
647
 
            # Throw away files we don't want packaged. Note that plugins may
648
 
            # have data files with all sorts of extensions so we need to
649
 
            # be conservative here about what we ditch.
650
 
            ext = os.path.splitext(i)[1]
651
 
            if ext.endswith('~') or ext in [".pyc", ".swp"]:
 
615
            if os.path.splitext(i)[1] not in [".py", ".pyd", ".dll", ".mo"]:
652
616
                continue
653
617
            if i == '__init__.py' and root == 'bzrlib/plugins':
654
618
                continue
667
631
    console_targets = [target,
668
632
                       'tools/win32/bzr_postinstall.py',
669
633
                       ]
670
 
    gui_targets = [gui_target]
 
634
    gui_targets = []
 
635
    com_targets = []
671
636
    data_files = topics_files + plugins_files
672
637
 
673
638
    if 'qbzr' in plugins:
676
641
    if 'svn' in plugins:
677
642
        get_svn_py2exe_info(includes, excludes, packages)
678
643
 
679
 
    if 'fastimport' in plugins:
680
 
        get_fastimport_py2exe_info(includes, excludes, packages)
681
 
 
682
644
    if "TBZR" in os.environ:
683
645
        # TORTOISE_OVERLAYS_MSI_WIN32 must be set to the location of the
684
646
        # TortoiseOverlays MSI installer file. It is in the TSVN svn repo and
703
665
        # print this warning to stderr as output is redirected, so it is seen
704
666
        # at build time.  Also to stdout so it appears in the log
705
667
        for f in (sys.stderr, sys.stdout):
706
 
            f.write("Skipping TBZR binaries - "
707
 
                "please set TBZR to a directory to enable\n")
 
668
            print >> f, \
 
669
                "Skipping TBZR binaries - please set TBZR to a directory to enable"
708
670
 
709
671
    # MSWSOCK.dll is a system-specific library, which py2exe accidentally pulls
710
672
    # in on Vista.
711
 
    dll_excludes.extend(["MSWSOCK.dll",
712
 
                         "MSVCP60.dll",
713
 
                         "MSVCP90.dll",
714
 
                         "powrprof.dll",
715
 
                         "SHFOLDER.dll"])
 
673
    dll_excludes.extend(["MSWSOCK.dll", "MSVCP60.dll", "powrprof.dll"])
716
674
    options_list = {"py2exe": {"packages": packages + list(additional_packages),
717
675
                               "includes": includes,
718
676
                               "excludes": excludes,
719
677
                               "dll_excludes": dll_excludes,
720
678
                               "dist_dir": "win32_bzr.exe",
721
 
                               "optimize": 2,
722
 
                               "custom_boot_script":
723
 
                                        "tools/win32/py2exe_boot_common.py",
 
679
                               "optimize": 1,
724
680
                              },
725
681
                   }
726
682
 
727
 
    # We want the libaray.zip to have optimize = 2, but the exe to have
728
 
    # optimize = 1, so that .py files that get compilied at run time
729
 
    # (e.g. user installed plugins) dont have their doc strings removed.
730
 
    class py2exe_no_oo_exe(py2exe.build_exe.py2exe):
731
 
        def build_executable(self, *args, **kwargs):
732
 
            self.optimize = 1
733
 
            py2exe.build_exe.py2exe.build_executable(self, *args, **kwargs)
734
 
            self.optimize = 2
735
 
 
736
 
    if __name__ == '__main__':
737
 
        command_classes['install_data'] = install_data_with_bytecompile
738
 
        command_classes['py2exe'] = py2exe_no_oo_exe
739
 
        setup(options=options_list,
740
 
              console=console_targets,
741
 
              windows=gui_targets,
742
 
              zipfile='lib/library.zip',
743
 
              data_files=data_files,
744
 
              cmdclass=command_classes,
745
 
              )
 
683
    setup(options=options_list,
 
684
          console=console_targets,
 
685
          windows=gui_targets,
 
686
          com_server=com_targets,
 
687
          zipfile='lib/library.zip',
 
688
          data_files=data_files,
 
689
          cmdclass={'install_data': install_data_with_bytecompile},
 
690
          )
746
691
 
747
692
else:
748
693
    # ad-hoc for easy_install
763
708
    ARGS.update(BZRLIB)
764
709
    ARGS.update(PKG_DATA)
765
710
 
766
 
    if __name__ == '__main__':
767
 
        setup(**ARGS)
 
711
    setup(**ARGS)