~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to setup.py

  • Committer: Mark Hammond
  • Date: 2008-08-10 11:16:39 UTC
  • mto: (3606.5.3 1.6)
  • mto: This revision was merged to the branch mainline in revision 3626.
  • Revision ID: mhammond@skippinet.com.au-20080810111639-9upz1wig1s8lvz62
install all files with 'restartreplace uninsrestartdelete' and close
tbzrcache at uninstall time.

Show diffs side-by-side

added added

removed removed

Lines of Context:
40
40
    'url':          'http://www.bazaar-vcs.org/',
41
41
    'description':  'Friendly distributed version control system',
42
42
    'license':      'GNU GPL v2',
43
 
    'download_url': 'https://launchpad.net/bzr/+download',
 
43
    'download_url': 'http://bazaar-vcs.org/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'
71
68
                                       ]},
72
69
           }
73
70
 
96
93
BZRLIB['packages'] = get_bzrlib_packages()
97
94
 
98
95
 
99
 
from distutils import log
100
96
from distutils.core import setup
101
97
from distutils.command.install_scripts import install_scripts
102
98
from distutils.command.install_data import install_data
148
144
    """Customized build distutils action.
149
145
    Generate bzr.1.
150
146
    """
151
 
 
152
147
    def run(self):
153
148
        build.run(self)
154
149
 
155
 
        from tools import generate_docs
 
150
        import generate_docs
156
151
        generate_docs.main(argv=["bzr", "man"])
157
152
 
158
153
 
167
162
from distutils.extension import Extension
168
163
ext_modules = []
169
164
try:
170
 
    try:
171
 
        from Pyrex.Distutils import build_ext
172
 
        from Pyrex.Compiler.Version import version as pyrex_version
173
 
    except ImportError:
174
 
        print "No Pyrex, trying Cython..."
175
 
        from Cython.Distutils import build_ext
176
 
        from Cython.Compiler.Version import version as pyrex_version
 
165
    from Pyrex.Distutils import build_ext
177
166
except ImportError:
178
167
    have_pyrex = False
179
168
    # try to build the extension from the prior generated source.
190
179
 
191
180
class build_ext_if_possible(build_ext):
192
181
 
193
 
    user_options = build_ext.user_options + [
194
 
        ('allow-python-fallback', None,
195
 
         "When an extension cannot be built, allow falling"
196
 
         " back to the pure-python implementation.")
197
 
        ]
198
 
 
199
 
    def initialize_options(self):
200
 
        build_ext.initialize_options(self)
201
 
        self.allow_python_fallback = False
202
 
 
203
182
    def run(self):
204
183
        try:
205
184
            build_ext.run(self)
206
185
        except DistutilsPlatformError, e:
207
 
            if not self.allow_python_fallback:
208
 
                log.warn('\n  Cannot build extensions.\n'
209
 
                         '  Use "build_ext --allow-python-fallback" to use'
210
 
                         ' slower python implementations instead.\n')
211
 
                raise
212
186
            log.warn(str(e))
213
 
            log.warn('\n  Extensions cannot be built.\n'
214
 
                     '  Using the slower Python implementations instead.\n')
 
187
            log.warn('Extensions cannot be built, '
 
188
                     'will use the Python versions instead')
215
189
 
216
190
    def build_extension(self, ext):
217
191
        try:
218
192
            build_ext.build_extension(self, ext)
219
193
        except CCompilerError:
220
 
            if not self.allow_python_fallback:
221
 
                log.warn('\n  Cannot build extension "%s".\n'
222
 
                         '  Use "build_ext --allow-python-fallback" to use'
223
 
                         ' slower python implementations instead.\n'
224
 
                         % (ext.name,))
225
 
                raise
226
 
            log.warn('\n  Building of "%s" extension failed.\n'
227
 
                     '  Using the slower Python implementation instead.'
228
 
                     % (ext.name,))
 
194
            log.warn('Building of "%s" extension failed, '
 
195
                     'will use the Python version instead' % (ext.name,))
229
196
 
230
197
 
231
198
# Override the build_ext if we have Pyrex available
233
200
unavailable_files = []
234
201
 
235
202
 
236
 
def add_pyrex_extension(module_name, libraries=None, extra_source=[]):
 
203
def add_pyrex_extension(module_name, **kwargs):
237
204
    """Add a pyrex module to build.
238
205
 
239
206
    This will use Pyrex to auto-generate the .c file if it is available.
249
216
    path = module_name.replace('.', '/')
250
217
    pyrex_name = path + '.pyx'
251
218
    c_name = path + '.c'
252
 
    define_macros = []
253
 
    if sys.platform == 'win32':
254
 
        # pyrex uses the macro WIN32 to detect the platform, even though it
255
 
        # should be using something like _WIN32 or MS_WINDOWS, oh well, we can
256
 
        # give it the right value.
257
 
        define_macros.append(('WIN32', None))
258
219
    if have_pyrex:
259
 
        source = [pyrex_name]
 
220
        ext_modules.append(Extension(module_name, [pyrex_name], **kwargs))
260
221
    else:
261
222
        if not os.path.isfile(c_name):
262
223
            unavailable_files.append(c_name)
263
 
            return
264
224
        else:
265
 
            source = [c_name]
266
 
    source.extend(extra_source)
267
 
    ext_modules.append(Extension(module_name, source,
268
 
        define_macros=define_macros, libraries=libraries))
269
 
 
270
 
 
271
 
add_pyrex_extension('bzrlib._annotator_pyx')
272
 
add_pyrex_extension('bzrlib._bencode_pyx')
273
 
add_pyrex_extension('bzrlib._chunks_to_lines_pyx')
274
 
add_pyrex_extension('bzrlib._groupcompress_pyx',
275
 
                    extra_source=['bzrlib/diff-delta.c'])
276
 
add_pyrex_extension('bzrlib._knit_load_data_pyx')
277
 
add_pyrex_extension('bzrlib._known_graph_pyx')
278
 
add_pyrex_extension('bzrlib._rio_pyx')
 
225
            ext_modules.append(Extension(module_name, [c_name], **kwargs))
 
226
 
 
227
 
 
228
add_pyrex_extension('bzrlib._dirstate_helpers_c')
 
229
add_pyrex_extension('bzrlib._knit_load_data_c')
279
230
if sys.platform == 'win32':
280
 
    add_pyrex_extension('bzrlib._dirstate_helpers_pyx',
281
 
                        libraries=['Ws2_32'])
282
 
    add_pyrex_extension('bzrlib._walkdirs_win32')
283
 
    z_lib = 'zdll'
284
 
else:
285
 
    if have_pyrex and pyrex_version.startswith('0.9.4'):
286
 
        # Pyrex 0.9.4.1 fails to compile this extension correctly
287
 
        # The code it generates re-uses a "local" pointer and
288
 
        # calls "PY_DECREF" after having set it to NULL. (It mixes PY_XDECREF
289
 
        # which is NULL safe with PY_DECREF which is not.)
290
 
        # <https://bugs.edge.launchpad.net/bzr/+bug/449372>
291
 
        # <https://bugs.edge.launchpad.net/bzr/+bug/276868>
292
 
        print 'Cannot build extension "bzrlib._dirstate_helpers_pyx" using'
293
 
        print 'your version of pyrex "%s". Please upgrade your pyrex' % (
294
 
            pyrex_version,)
295
 
        print 'install. For now, the non-compiled (python) version will'
296
 
        print 'be used instead.'
297
 
    else:
298
 
        add_pyrex_extension('bzrlib._dirstate_helpers_pyx')
299
 
    add_pyrex_extension('bzrlib._readdir_pyx')
300
 
    z_lib = 'z'
301
 
add_pyrex_extension('bzrlib._chk_map_pyx', libraries=[z_lib])
302
 
ext_modules.append(Extension('bzrlib._patiencediff_c',
303
 
                             ['bzrlib/_patiencediff_c.c']))
304
 
add_pyrex_extension('bzrlib._simple_set_pyx')
305
 
ext_modules.append(Extension('bzrlib._static_tuple_c',
306
 
                             ['bzrlib/_static_tuple_c.c']))
307
 
add_pyrex_extension('bzrlib._btree_serializer_pyx')
 
231
    # pyrex uses the macro WIN32 to detect the platform, even though it should
 
232
    # be using something like _WIN32 or MS_WINDOWS, oh well, we can give it the
 
233
    # right value.
 
234
    add_pyrex_extension('bzrlib._walkdirs_win32',
 
235
                        define_macros=[('WIN32', None)])
 
236
ext_modules.append(Extension('bzrlib._patiencediff_c', ['bzrlib/_patiencediff_c.c']))
308
237
 
309
238
 
310
239
if unavailable_files:
313
242
    print 'The python versions will be used instead.'
314
243
    print
315
244
 
316
 
 
317
245
def get_tbzr_py2exe_info(includes, excludes, packages, console_targets,
318
 
                         gui_targets, data_files):
 
246
                         gui_targets):
319
247
    packages.append('tbzrcommands')
320
248
 
321
249
    # ModuleFinder can't handle runtime changes to __path__, but
322
250
    # win32com uses them.  Hook this in so win32com.shell is found.
323
251
    import modulefinder
324
252
    import win32com
325
 
    import cPickle as pickle
326
253
    for p in win32com.__path__[1:]:
327
254
        modulefinder.AddPackagePath("win32com", p)
328
255
    for extra in ["win32com.shell"]:
337
264
    # Ensure tbzrlib itself is on sys.path
338
265
    sys.path.append(tbzr_root)
339
266
 
 
267
    # Ensure our COM "entry-point" is on sys.path
 
268
    sys.path.append(os.path.join(tbzr_root, "shellext", "python"))
 
269
 
340
270
    packages.append("tbzrlib")
341
 
 
342
 
    # collect up our icons.
343
 
    cwd = os.getcwd()
344
 
    ico_root = os.path.join(tbzr_root, 'tbzrlib', 'resources')
345
 
    icos = [] # list of (path_root, relative_ico_path)
346
 
    # First always bzr's icon and its in the root of the bzr tree.
347
 
    icos.append(('', 'bzr.ico'))
348
 
    for root, dirs, files in os.walk(ico_root):
349
 
        icos.extend([(ico_root, os.path.join(root, f)[len(ico_root)+1:])
350
 
                     for f in files if f.endswith('.ico')])
351
 
    # allocate an icon ID for each file and the full path to the ico
352
 
    icon_resources = [(rid, os.path.join(ico_dir, ico_name))
353
 
                      for rid, (ico_dir, ico_name) in enumerate(icos)]
354
 
    # create a string resource with the mapping.  Might as well save the
355
 
    # runtime some effort and write a pickle.
356
 
    # Runtime expects unicode objects with forward-slash seps.
357
 
    fse = sys.getfilesystemencoding()
358
 
    map_items = [(f.replace('\\', '/').decode(fse), rid)
359
 
                 for rid, (_, f) in enumerate(icos)]
360
 
    ico_map = dict(map_items)
361
 
    # Create a new resource type of 'ICON_MAP', and use ID=1
362
 
    other_resources = [ ("ICON_MAP", 1, pickle.dumps(ico_map))]
363
 
 
364
271
    excludes.extend("""pywin pywin.dialogs pywin.dialogs.list
365
272
                       win32ui crawler.Crawler""".split())
366
273
 
 
274
    tbzr = dict(
 
275
        modules=["tbzr"],
 
276
        create_exe = False, # we only want a .dll
 
277
    )
 
278
    com_targets.append(tbzr)
 
279
 
367
280
    # tbzrcache executables - a "console" version for debugging and a
368
281
    # GUI version that is generally used.
369
282
    tbzrcache = dict(
370
 
        script = os.path.join(tbzr_root, "scripts", "tbzrcache.py"),
371
 
        icon_resources = icon_resources,
372
 
        other_resources = other_resources,
 
283
        script = os.path.join(tbzr_root, "Scripts", "tbzrcache.py"),
 
284
        icon_resources = [(0,'bzr.ico')],
373
285
    )
374
286
    console_targets.append(tbzrcache)
375
287
 
380
292
 
381
293
    # ditto for the tbzrcommand tool
382
294
    tbzrcommand = dict(
383
 
        script = os.path.join(tbzr_root, "scripts", "tbzrcommand.py"),
 
295
        script = os.path.join(tbzr_root, "Scripts", "tbzrcommand.py"),
384
296
        icon_resources = [(0,'bzr.ico')],
385
297
    )
386
298
    console_targets.append(tbzrcommand)
388
300
    tbzrcommandw["dest_base"]="tbzrcommandw"
389
301
    gui_targets.append(tbzrcommandw)
390
302
    
391
 
    # A utility to see python output from both C++ and Python based shell
392
 
    # extensions
393
 
    tracer = dict(script=os.path.join(tbzr_root, "scripts", "tbzrtrace.py"))
 
303
    # tbzr tests
 
304
    tbzrtest = dict(
 
305
        script = os.path.join(tbzr_root, "Scripts", "tbzrtest.py"),
 
306
    )
 
307
    console_targets.append(tbzrtest)
 
308
 
 
309
    # A utility to see python output from the shell extension - this will
 
310
    # die when we get a c++ extension
 
311
    # any .py file from pywin32's win32 lib will do (other than
 
312
    # win32traceutil itself that is)
 
313
    import winerror
 
314
    win32_lib_dir = os.path.dirname(winerror.__file__)
 
315
    tracer = dict(script = os.path.join(win32_lib_dir, "win32traceutil.py"),
 
316
                  dest_base="tbzr_tracer")
394
317
    console_targets.append(tracer)
395
318
 
396
 
    # The C++ implemented shell extensions.
397
 
    dist_dir = os.path.join(tbzr_root, "shellext", "build")
398
 
    data_files.append(('', [os.path.join(dist_dir, 'tbzrshellext_x86.dll')]))
399
 
    data_files.append(('', [os.path.join(dist_dir, 'tbzrshellext_x64.dll')]))
400
 
 
401
 
 
402
 
def get_qbzr_py2exe_info(includes, excludes, packages, data_files):
 
319
def get_qbzr_py2exe_info(includes, excludes, packages):
403
320
    # PyQt4 itself still escapes the plugin detection code for some reason...
404
321
    packages.append('PyQt4')
405
322
    excludes.append('PyQt4.elementtree.ElementTree')
406
 
    excludes.append('PyQt4.uic.port_v3')
407
323
    includes.append('sip') # extension module required for Qt.
408
324
    packages.append('pygments') # colorizer for qbzr
409
 
    packages.append('docutils') # html formatting
410
 
    includes.append('win32event')  # for qsubprocess stuff
411
325
    # but we can avoid many Qt4 Dlls.
412
326
    dll_excludes.extend(
413
327
        """QtAssistantClient4.dll QtCLucene4.dll QtDesigner4.dll
415
329
        QtSql4.dll QtTest4.dll QtWebKit4.dll QtXml4.dll
416
330
        qscintilla2.dll""".split())
417
331
    # the qt binaries might not be on PATH...
418
 
    # They seem to install to a place like C:\Python25\PyQt4\*
419
 
    # Which is not the same as C:\Python25\Lib\site-packages\PyQt4
420
 
    pyqt_dir = os.path.join(sys.prefix, "PyQt4")
421
 
    pyqt_bin_dir = os.path.join(pyqt_dir, "bin")
422
 
    if os.path.isdir(pyqt_bin_dir):
423
 
        path = os.environ.get("PATH", "")
424
 
        if pyqt_bin_dir.lower() not in [p.lower() for p in path.split(os.pathsep)]:
425
 
            os.environ["PATH"] = path + os.pathsep + pyqt_bin_dir
426
 
    # also add all imageformat plugins to distribution
427
 
    # We will look in 2 places, dirname(PyQt4.__file__) and pyqt_dir
428
 
    base_dirs_to_check = []
429
 
    if os.path.isdir(pyqt_dir):
430
 
        base_dirs_to_check.append(pyqt_dir)
431
 
    try:
432
 
        import PyQt4
433
 
    except ImportError:
434
 
        pass
435
 
    else:
436
 
        pyqt4_base_dir = os.path.dirname(PyQt4.__file__)
437
 
        if pyqt4_base_dir != pyqt_dir:
438
 
            base_dirs_to_check.append(pyqt4_base_dir)
439
 
    if not base_dirs_to_check:
440
 
        log.warn("Can't find PyQt4 installation -> not including imageformat"
441
 
                 " plugins")
442
 
    else:
443
 
        files = []
444
 
        for base_dir in base_dirs_to_check:
445
 
            plug_dir = os.path.join(base_dir, 'plugins', 'imageformats')
446
 
            if os.path.isdir(plug_dir):
447
 
                for fname in os.listdir(plug_dir):
448
 
                    # Include plugin dlls, but not debugging dlls
449
 
                    fullpath = os.path.join(plug_dir, fname)
450
 
                    if fname.endswith('.dll') and not fname.endswith('d4.dll'):
451
 
                        files.append(fullpath)
452
 
        if files:
453
 
            data_files.append(('imageformats', files))
454
 
        else:
455
 
            log.warn('PyQt4 was found, but we could not find any imageformat'
456
 
                     ' plugins. Are you sure your configuration is correct?')
457
 
 
458
 
 
459
 
def get_svn_py2exe_info(includes, excludes, packages):
460
 
    packages.append('subvertpy')
 
332
    qt_dir = os.path.join(sys.prefix, "PyQt4", "bin")
 
333
    path = os.environ.get("PATH","")
 
334
    if qt_dir.lower() not in [p.lower() for p in path.split(os.pathsep)]:
 
335
        os.environ["PATH"] = path + os.pathsep + qt_dir
461
336
 
462
337
 
463
338
if 'bdist_wininst' in sys.argv:
541
416
                                     version = version_str,
542
417
                                     description = META_INFO['description'],
543
418
                                     author = META_INFO['author'],
544
 
                                     copyright = "(c) Canonical Ltd, 2005-2009",
 
419
                                     copyright = "(c) Canonical Ltd, 2005-2007",
545
420
                                     company_name = "Canonical Ltd.",
546
421
                                     comments = META_INFO['description'],
547
422
                                    )
574
449
                  ImaginaryModule cElementTree elementtree.ElementTree
575
450
                  Crypto.PublicKey._fastmath
576
451
                  medusa medusa.filesys medusa.ftp_server
577
 
                  tools
 
452
                  tools tools.doc_generate
578
453
                  resource validate""".split()
579
454
    dll_excludes = []
580
455
 
603
478
    for root, dirs, files in os.walk('bzrlib/plugins'):
604
479
        if root == 'bzrlib/plugins':
605
480
            plugins = set(dirs)
606
 
            # We ship plugins as normal files on the file-system - however,
607
 
            # the build process can cause *some* of these plugin files to end
608
 
            # up in library.zip. Thus, we saw (eg) "plugins/svn/test" in
609
 
            # library.zip, and then saw import errors related to that as the
610
 
            # rest of the svn plugin wasn't. So we tell py2exe to leave the
611
 
            # plugins out of the .zip file
612
 
            excludes.extend(["bzrlib.plugins." + d for d in dirs])
613
481
        x = []
614
482
        for i in files:
615
 
            if os.path.splitext(i)[1] not in [".py", ".pyd", ".dll", ".mo"]:
 
483
            if os.path.splitext(i)[1] not in [".py", ".pyd", ".dll"]:
616
484
                continue
617
485
            if i == '__init__.py' and root == 'bzrlib/plugins':
618
486
                continue
632
500
                       'tools/win32/bzr_postinstall.py',
633
501
                       ]
634
502
    gui_targets = []
635
 
    data_files = topics_files + plugins_files
 
503
    com_targets = []
636
504
 
637
505
    if 'qbzr' in plugins:
638
 
        get_qbzr_py2exe_info(includes, excludes, packages, data_files)
639
 
 
640
 
    if 'svn' in plugins:
641
 
        get_svn_py2exe_info(includes, excludes, packages)
 
506
        get_qbzr_py2exe_info(includes, excludes, packages)
642
507
 
643
508
    if "TBZR" in os.environ:
644
 
        # TORTOISE_OVERLAYS_MSI_WIN32 must be set to the location of the
645
 
        # TortoiseOverlays MSI installer file. It is in the TSVN svn repo and
646
 
        # can be downloaded from (username=guest, blank password):
647
 
        # http://tortoisesvn.tigris.org/svn/tortoisesvn/TortoiseOverlays
648
 
        # look for: version-1.0.4/bin/TortoiseOverlays-1.0.4.11886-win32.msi
649
 
        # Ditto for TORTOISE_OVERLAYS_MSI_X64, pointing at *-x64.msi.
650
 
        for needed in ('TORTOISE_OVERLAYS_MSI_WIN32',
651
 
                       'TORTOISE_OVERLAYS_MSI_X64'):
652
 
            url = ('http://guest:@tortoisesvn.tigris.org/svn/tortoisesvn'
653
 
                   '/TortoiseOverlays')
654
 
            if not os.path.isfile(os.environ.get(needed, '<nofile>')):
655
 
                raise RuntimeError(
656
 
                    "\nPlease set %s to the location of the relevant"
657
 
                    "\nTortoiseOverlays .msi installer file."
658
 
                    " The installers can be found at"
659
 
                    "\n  %s"
660
 
                    "\ncheck in the version-X.Y.Z/bin/ subdir" % (needed, url))
 
509
        # Eg: via SVN: http://tortoisesvn.tigris.org/svn/tortoisesvn/TortoiseOverlays/version-1.0.4/bin/TortoiseOverlays-1.0.4.11886-win32.msi
 
510
        if not os.path.isfile(os.environ.get('TOVMSI_WIN32', '<nofile>')):
 
511
            raise RuntimeError, "Please set TOVMSI_WIN32 to the location of " \
 
512
                                "the TortoiseOverlays .msi installerfile"
661
513
        get_tbzr_py2exe_info(includes, excludes, packages, console_targets,
662
 
                             gui_targets, data_files)
 
514
                             gui_targets)
663
515
    else:
664
516
        # print this warning to stderr as output is redirected, so it is seen
665
517
        # at build time.  Also to stdout so it appears in the log
669
521
 
670
522
    # MSWSOCK.dll is a system-specific library, which py2exe accidentally pulls
671
523
    # in on Vista.
672
 
    dll_excludes.extend(["MSWSOCK.dll", "MSVCP60.dll", "powrprof.dll"])
 
524
    dll_excludes.append("MSWSOCK.dll")
673
525
    options_list = {"py2exe": {"packages": packages + list(additional_packages),
674
526
                               "includes": includes,
675
527
                               "excludes": excludes,
682
534
    setup(options=options_list,
683
535
          console=console_targets,
684
536
          windows=gui_targets,
 
537
          com_server=com_targets,
685
538
          zipfile='lib/library.zip',
686
 
          data_files=data_files,
 
539
          data_files=topics_files + plugins_files,
687
540
          cmdclass={'install_data': install_data_with_bytecompile},
688
541
          )
689
542
 
691
544
    # ad-hoc for easy_install
692
545
    DATA_FILES = []
693
546
    if not 'bdist_egg' in sys.argv:
694
 
        # generate and install bzr.1 only with plain install, not the
695
 
        # easy_install one
 
547
        # generate and install bzr.1 only with plain install, not easy_install one
696
548
        DATA_FILES = [('man/man1', ['bzr.1'])]
697
549
 
698
550
    # std setup