~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to setup.py

  • Committer: John Arbash Meinel
  • Author(s): Mark Hammond
  • Date: 2008-09-09 17:02:21 UTC
  • mto: This revision was merged to the branch mainline in revision 3697.
  • Revision ID: john@arbash-meinel.com-20080909170221-svim3jw2mrz0amp3
An updated transparent icon for bzr.

Show diffs side-by-side

added added

removed removed

Lines of Context:
93
93
BZRLIB['packages'] = get_bzrlib_packages()
94
94
 
95
95
 
96
 
from distutils import log
97
96
from distutils.core import setup
98
97
from distutils.command.install_scripts import install_scripts
99
98
from distutils.command.install_data import install_data
145
144
    """Customized build distutils action.
146
145
    Generate bzr.1.
147
146
    """
148
 
 
149
147
    def run(self):
150
148
        build.run(self)
151
149
 
152
 
        from tools import generate_docs
 
150
        import generate_docs
153
151
        generate_docs.main(argv=["bzr", "man"])
154
152
 
155
153
 
177
175
    from distutils.command.build_ext import build_ext
178
176
else:
179
177
    have_pyrex = True
180
 
    from Pyrex.Compiler.Version import version as pyrex_version
181
178
 
182
179
 
183
180
class build_ext_if_possible(build_ext):
184
181
 
185
 
    user_options = build_ext.user_options + [
186
 
        ('allow-python-fallback', None,
187
 
         "When an extension cannot be built, allow falling"
188
 
         " back to the pure-python implementation.")
189
 
        ]
190
 
 
191
 
    def initialize_options(self):
192
 
        build_ext.initialize_options(self)
193
 
        self.allow_python_fallback = False
194
 
 
195
182
    def run(self):
196
183
        try:
197
184
            build_ext.run(self)
198
185
        except DistutilsPlatformError, e:
199
 
            if not self.allow_python_fallback:
200
 
                log.warn('\n  Cannot build extensions.\n'
201
 
                         '  Use "build_ext --allow-python-fallback" to use'
202
 
                         ' slower python implementations instead.\n')
203
 
                raise
204
186
            log.warn(str(e))
205
 
            log.warn('\n  Extensions cannot be built.\n'
206
 
                     '  Using the slower Python implementations instead.\n')
 
187
            log.warn('Extensions cannot be built, '
 
188
                     'will use the Python versions instead')
207
189
 
208
190
    def build_extension(self, ext):
209
191
        try:
210
192
            build_ext.build_extension(self, ext)
211
193
        except CCompilerError:
212
 
            if not self.allow_python_fallback:
213
 
                log.warn('\n  Cannot build extension "%s".\n'
214
 
                         '  Use "build_ext --allow-python-fallback" to use'
215
 
                         ' slower python implementations instead.\n'
216
 
                         % (ext.name,))
217
 
                raise
218
 
            log.warn('\n  Building of "%s" extension failed.\n'
219
 
                     '  Using the slower Python implementation instead.'
220
 
                     % (ext.name,))
 
194
            log.warn('Building of "%s" extension failed, '
 
195
                     'will use the Python version instead' % (ext.name,))
221
196
 
222
197
 
223
198
# Override the build_ext if we have Pyrex available
225
200
unavailable_files = []
226
201
 
227
202
 
228
 
def add_pyrex_extension(module_name, libraries=None, extra_source=[]):
 
203
def add_pyrex_extension(module_name, **kwargs):
229
204
    """Add a pyrex module to build.
230
205
 
231
206
    This will use Pyrex to auto-generate the .c file if it is available.
241
216
    path = module_name.replace('.', '/')
242
217
    pyrex_name = path + '.pyx'
243
218
    c_name = path + '.c'
244
 
    define_macros = []
245
 
    if sys.platform == 'win32':
246
 
        # pyrex uses the macro WIN32 to detect the platform, even though it
247
 
        # should be using something like _WIN32 or MS_WINDOWS, oh well, we can
248
 
        # give it the right value.
249
 
        define_macros.append(('WIN32', None))
250
219
    if have_pyrex:
251
 
        source = [pyrex_name]
 
220
        ext_modules.append(Extension(module_name, [pyrex_name], **kwargs))
252
221
    else:
253
222
        if not os.path.isfile(c_name):
254
223
            unavailable_files.append(c_name)
255
 
            return
256
224
        else:
257
 
            source = [c_name]
258
 
    source.extend(extra_source)
259
 
    ext_modules.append(Extension(module_name, source,
260
 
        define_macros=define_macros, libraries=libraries))
261
 
 
262
 
 
263
 
add_pyrex_extension('bzrlib._annotator_pyx')
264
 
add_pyrex_extension('bzrlib._bencode_pyx')
265
 
add_pyrex_extension('bzrlib._btree_serializer_pyx')
266
 
add_pyrex_extension('bzrlib._chunks_to_lines_pyx')
267
 
add_pyrex_extension('bzrlib._groupcompress_pyx',
268
 
                    extra_source=['bzrlib/diff-delta.c'])
269
 
add_pyrex_extension('bzrlib._knit_load_data_pyx')
270
 
add_pyrex_extension('bzrlib._known_graph_pyx')
271
 
add_pyrex_extension('bzrlib._rio_pyx')
 
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')
272
232
if sys.platform == 'win32':
273
 
    add_pyrex_extension('bzrlib._dirstate_helpers_pyx',
274
 
                        libraries=['Ws2_32'])
275
 
    add_pyrex_extension('bzrlib._walkdirs_win32')
276
 
    z_lib = 'zdll'
277
 
else:
278
 
    if have_pyrex and pyrex_version == '0.9.4.1':
279
 
        # Pyrex 0.9.4.1 fails to compile this extension correctly
280
 
        # The code it generates re-uses a "local" pointer and
281
 
        # calls "PY_DECREF" after having set it to NULL. (It mixes PY_XDECREF
282
 
        # which is NULL safe with PY_DECREF which is not.)
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.'
288
 
    else:
289
 
        add_pyrex_extension('bzrlib._dirstate_helpers_pyx')
290
 
    add_pyrex_extension('bzrlib._readdir_pyx')
291
 
    z_lib = 'z'
292
 
add_pyrex_extension('bzrlib._chk_map_pyx', libraries=[z_lib])
293
 
ext_modules.append(Extension('bzrlib._patiencediff_c',
294
 
                             ['bzrlib/_patiencediff_c.c']))
 
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']))
295
239
 
296
240
 
297
241
if unavailable_files:
302
246
 
303
247
 
304
248
def get_tbzr_py2exe_info(includes, excludes, packages, console_targets,
305
 
                         gui_targets, data_files):
 
249
                         gui_targets):
306
250
    packages.append('tbzrcommands')
307
251
 
308
252
    # ModuleFinder can't handle runtime changes to __path__, but
354
298
    excludes.extend("""pywin pywin.dialogs pywin.dialogs.list
355
299
                       win32ui crawler.Crawler""".split())
356
300
 
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
301
    tbzr = dict(
362
302
        modules=["tbzr"],
363
303
        create_exe = False, # we only want a .dll
364
 
        dest_base = 'tbzr_old',
365
304
    )
366
305
    com_targets.append(tbzr)
367
306
 
368
307
    # tbzrcache executables - a "console" version for debugging and a
369
308
    # GUI version that is generally used.
370
309
    tbzrcache = dict(
371
 
        script = os.path.join(tbzr_root, "scripts", "tbzrcache.py"),
 
310
        script = os.path.join(tbzr_root, "Scripts", "tbzrcache.py"),
372
311
        icon_resources = icon_resources,
373
312
        other_resources = other_resources,
374
313
    )
381
320
 
382
321
    # ditto for the tbzrcommand tool
383
322
    tbzrcommand = dict(
384
 
        script = os.path.join(tbzr_root, "scripts", "tbzrcommand.py"),
 
323
        script = os.path.join(tbzr_root, "Scripts", "tbzrcommand.py"),
385
324
        icon_resources = [(0,'bzr.ico')],
386
325
    )
387
326
    console_targets.append(tbzrcommand)
389
328
    tbzrcommandw["dest_base"]="tbzrcommandw"
390
329
    gui_targets.append(tbzrcommandw)
391
330
    
392
 
    # A utility to see python output from both C++ and Python based shell
393
 
    # extensions
394
 
    tracer = dict(script=os.path.join(tbzr_root, "scripts", "tbzrtrace.py"))
 
331
    # tbzr tests
 
332
    tbzrtest = dict(
 
333
        script = os.path.join(tbzr_root, "Scripts", "tbzrtest.py"),
 
334
    )
 
335
    console_targets.append(tbzrtest)
 
336
 
 
337
    # A utility to see python output from the shell extension - this will
 
338
    # die when we get a c++ extension
 
339
    # any .py file from pywin32's win32 lib will do (other than
 
340
    # win32traceutil itself that is)
 
341
    import winerror
 
342
    win32_lib_dir = os.path.dirname(winerror.__file__)
 
343
    tracer = dict(script = os.path.join(win32_lib_dir, "win32traceutil.py"),
 
344
                  dest_base="tbzr_tracer")
395
345
    console_targets.append(tracer)
396
346
 
397
 
    # The C++ implemented shell extensions.
398
 
    dist_dir = os.path.join(tbzr_root, "shellext", "cpp", "tbzrshellext",
399
 
                            "build", "dist")
400
 
    data_files.append(('', [os.path.join(dist_dir, 'tbzrshellext_x86.dll')]))
401
 
    data_files.append(('', [os.path.join(dist_dir, 'tbzrshellext_x64.dll')]))
402
 
 
403
 
 
404
 
def get_qbzr_py2exe_info(includes, excludes, packages, data_files):
 
347
 
 
348
def get_qbzr_py2exe_info(includes, excludes, packages):
405
349
    # PyQt4 itself still escapes the plugin detection code for some reason...
406
350
    packages.append('PyQt4')
407
351
    excludes.append('PyQt4.elementtree.ElementTree')
408
352
    includes.append('sip') # extension module required for Qt.
409
353
    packages.append('pygments') # colorizer for qbzr
410
 
    packages.append('docutils') # html formatting
411
354
    # but we can avoid many Qt4 Dlls.
412
355
    dll_excludes.extend(
413
356
        """QtAssistantClient4.dll QtCLucene4.dll QtDesigner4.dll
415
358
        QtSql4.dll QtTest4.dll QtWebKit4.dll QtXml4.dll
416
359
        qscintilla2.dll""".split())
417
360
    # 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')
 
361
    qt_dir = os.path.join(sys.prefix, "PyQt4", "bin")
 
362
    path = os.environ.get("PATH","")
 
363
    if qt_dir.lower() not in [p.lower() for p in path.split(os.pathsep)]:
 
364
        os.environ["PATH"] = path + os.pathsep + qt_dir
461
365
 
462
366
 
463
367
if 'bdist_wininst' in sys.argv:
574
478
                  ImaginaryModule cElementTree elementtree.ElementTree
575
479
                  Crypto.PublicKey._fastmath
576
480
                  medusa medusa.filesys medusa.ftp_server
577
 
                  tools
 
481
                  tools tools.doc_generate
578
482
                  resource validate""".split()
579
483
    dll_excludes = []
580
484
 
603
507
    for root, dirs, files in os.walk('bzrlib/plugins'):
604
508
        if root == 'bzrlib/plugins':
605
509
            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
510
        x = []
614
511
        for i in files:
615
 
            if os.path.splitext(i)[1] not in [".py", ".pyd", ".dll", ".mo"]:
 
512
            if os.path.splitext(i)[1] not in [".py", ".pyd", ".dll"]:
616
513
                continue
617
514
            if i == '__init__.py' and root == 'bzrlib/plugins':
618
515
                continue
633
530
                       ]
634
531
    gui_targets = []
635
532
    com_targets = []
636
 
    data_files = topics_files + plugins_files
637
533
 
638
534
    if 'qbzr' in plugins:
639
 
        get_qbzr_py2exe_info(includes, excludes, packages, data_files)
640
 
 
641
 
    if 'svn' in plugins:
642
 
        get_svn_py2exe_info(includes, excludes, packages)
 
535
        get_qbzr_py2exe_info(includes, excludes, packages)
643
536
 
644
537
    if "TBZR" in os.environ:
645
538
        # TORTOISE_OVERLAYS_MSI_WIN32 must be set to the location of the
646
539
        # TortoiseOverlays MSI installer file. It is in the TSVN svn repo and
647
540
        # can be downloaded from (username=guest, blank password):
648
 
        # http://tortoisesvn.tigris.org/svn/tortoisesvn/TortoiseOverlays
649
 
        # look for: version-1.0.4/bin/TortoiseOverlays-1.0.4.11886-win32.msi
650
 
        # Ditto for TORTOISE_OVERLAYS_MSI_X64, pointing at *-x64.msi.
651
 
        for needed in ('TORTOISE_OVERLAYS_MSI_WIN32',
652
 
                       'TORTOISE_OVERLAYS_MSI_X64'):
653
 
            url = ('http://guest:@tortoisesvn.tigris.org/svn/tortoisesvn'
654
 
                   '/TortoiseOverlays')
655
 
            if not os.path.isfile(os.environ.get(needed, '<nofile>')):
656
 
                raise RuntimeError(
657
 
                    "\nPlease set %s to the location of the relevant"
658
 
                    "\nTortoiseOverlays .msi installer file."
659
 
                    " The installers can be found at"
660
 
                    "\n  %s"
661
 
                    "\ncheck in the version-X.Y.Z/bin/ subdir" % (needed, url))
 
541
        # http://tortoisesvn.tigris.org/svn/tortoisesvn/TortoiseOverlays/version-1.0.4/bin/TortoiseOverlays-1.0.4.11886-win32.msi
 
542
        if not os.path.isfile(os.environ.get('TORTOISE_OVERLAYS_MSI_WIN32',
 
543
                                             '<nofile>')):
 
544
            raise RuntimeError("Please set TORTOISE_OVERLAYS_MSI_WIN32 to the"
 
545
                               " location of the Win32 TortoiseOverlays .msi"
 
546
                               " installer file")
662
547
        get_tbzr_py2exe_info(includes, excludes, packages, console_targets,
663
 
                             gui_targets, data_files)
 
548
                             gui_targets)
664
549
    else:
665
550
        # print this warning to stderr as output is redirected, so it is seen
666
551
        # at build time.  Also to stdout so it appears in the log
670
555
 
671
556
    # MSWSOCK.dll is a system-specific library, which py2exe accidentally pulls
672
557
    # in on Vista.
673
 
    dll_excludes.extend(["MSWSOCK.dll", "MSVCP60.dll", "powrprof.dll"])
 
558
    dll_excludes.append("MSWSOCK.dll")
674
559
    options_list = {"py2exe": {"packages": packages + list(additional_packages),
675
560
                               "includes": includes,
676
561
                               "excludes": excludes,
685
570
          windows=gui_targets,
686
571
          com_server=com_targets,
687
572
          zipfile='lib/library.zip',
688
 
          data_files=data_files,
 
573
          data_files=topics_files + plugins_files,
689
574
          cmdclass={'install_data': install_data_with_bytecompile},
690
575
          )
691
576
 
693
578
    # ad-hoc for easy_install
694
579
    DATA_FILES = []
695
580
    if not 'bdist_egg' in sys.argv:
696
 
        # generate and install bzr.1 only with plain install, not the
697
 
        # easy_install one
 
581
        # generate and install bzr.1 only with plain install, not easy_install one
698
582
        DATA_FILES = [('man/man1', ['bzr.1'])]
699
583
 
700
584
    # std setup