/brz/remove-bazaar

To get this branch, use:
bzr branch http://gegoxaren.bato24.eu/bzr/brz/remove-bazaar

« back to all changes in this revision

Viewing changes to setup.py

1st cut merge of bzr.dev r3907

Show diffs side-by-side

added added

removed removed

Lines of Context:
setup.py
95
95
 
96
96
from distutils.core import setup
97
97
from distutils.command.install_scripts import install_scripts
 
98
from distutils.command.install_data import install_data
98
99
from distutils.command.build import build
99
100
 
100
101
###############################
143
144
    """Customized build distutils action.
144
145
    Generate bzr.1.
145
146
    """
 
147
 
146
148
    def run(self):
147
149
        build.run(self)
148
150
 
174
176
    from distutils.command.build_ext import build_ext
175
177
else:
176
178
    have_pyrex = True
 
179
    from Pyrex.Compiler.Version import version as pyrex_version
177
180
 
178
181
 
179
182
class build_ext_if_possible(build_ext):
180
183
 
 
184
    user_options = build_ext.user_options + [
 
185
        ('allow-python-fallback', None,
 
186
         "When an extension cannot be built, allow falling"
 
187
         " back to the pure-python implementation.")
 
188
        ]
 
189
 
 
190
    def initialize_options(self):
 
191
        build_ext.initialize_options(self)
 
192
        self.allow_python_fallback = False
 
193
 
181
194
    def run(self):
182
195
        try:
183
196
            build_ext.run(self)
184
197
        except DistutilsPlatformError, e:
 
198
            if not self.allow_python_fallback:
 
199
                log.warn('\n  Cannot build extensions.\n'
 
200
                         '  Use --allow-python-fallback to use slower'
 
201
                         ' python implementations instead.\n')
 
202
                raise
185
203
            log.warn(str(e))
186
 
            log.warn('Extensions cannot be built, '
187
 
                     'will use the Python versions instead')
 
204
            log.warn('\n  Extensions cannot be built.\n'
 
205
                     '  Using the slower Python implementations instead.\n')
188
206
 
189
207
    def build_extension(self, ext):
190
208
        try:
191
209
            build_ext.build_extension(self, ext)
192
210
        except CCompilerError:
193
 
            log.warn('Building of "%s" extension failed, '
194
 
                     'will use the Python version instead' % (ext.name,))
 
211
            if not self.allow_python_fallback:
 
212
                log.warn('\n  Failed to build "%s".\n'
 
213
                         '  Use --allow-python-fallback to use slower'
 
214
                         ' python implementations instead.\n'
 
215
                         % (ext.name,))
 
216
                raise
 
217
            log.warn('\n  Building of "%s" extension failed.\n'
 
218
                     '  Using the slower Python implementation instead.'
 
219
                     % (ext.name,))
195
220
 
196
221
 
197
222
# Override the build_ext if we have Pyrex available
199
224
unavailable_files = []
200
225
 
201
226
 
202
 
def add_pyrex_extension(module_name, **kwargs):
 
227
def add_pyrex_extension(module_name, libraries=None):
203
228
    """Add a pyrex module to build.
204
229
 
205
230
    This will use Pyrex to auto-generate the .c file if it is available.
215
240
    path = module_name.replace('.', '/')
216
241
    pyrex_name = path + '.pyx'
217
242
    c_name = path + '.c'
 
243
    define_macros = []
 
244
    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.
 
248
        define_macros.append(('WIN32', None))
218
249
    if have_pyrex:
219
 
        ext_modules.append(Extension(module_name, [pyrex_name], **kwargs))
 
250
        ext_modules.append(Extension(module_name, [pyrex_name],
 
251
            define_macros=define_macros, libraries=libraries))
220
252
    else:
221
253
        if not os.path.isfile(c_name):
222
254
            unavailable_files.append(c_name)
223
255
        else:
224
 
            ext_modules.append(Extension(module_name, [c_name], **kwargs))
225
 
 
226
 
 
227
 
add_pyrex_extension('bzrlib._dirstate_helpers_c')
 
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')
 
261
add_pyrex_extension('bzrlib._chunks_to_lines_pyx')
228
262
add_pyrex_extension('bzrlib._knit_load_data_c')
229
263
if sys.platform == 'win32':
230
 
    # pyrex uses the macro WIN32 to detect the platform, even though it should
231
 
    # be using something like _WIN32 or MS_WINDOWS, oh well, we can give it the
232
 
    # right value.
233
 
    add_pyrex_extension('bzrlib._walkdirs_win32',
234
 
                        define_macros=[('WIN32', None)])
 
264
    add_pyrex_extension('bzrlib._dirstate_helpers_c',
 
265
                        libraries=['Ws2_32'])
 
266
    add_pyrex_extension('bzrlib._walkdirs_win32')
 
267
else:
 
268
    if have_pyrex and pyrex_version == '0.9.4.1':
 
269
        # Pyrex 0.9.4.1 fails to compile this extension correctly
 
270
        # The code it generates re-uses a "local" pointer and
 
271
        # calls "PY_DECREF" after having set it to NULL. (It mixes PY_XDECREF
 
272
        # 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.'
 
278
    else:
 
279
        add_pyrex_extension('bzrlib._dirstate_helpers_c')
 
280
    add_pyrex_extension('bzrlib._readdir_pyx')
235
281
ext_modules.append(Extension('bzrlib._patiencediff_c', ['bzrlib/_patiencediff_c.c']))
236
282
 
237
283
 
242
288
    print
243
289
 
244
290
 
 
291
def get_tbzr_py2exe_info(includes, excludes, packages, console_targets,
 
292
                         gui_targets):
 
293
    packages.append('tbzrcommands')
 
294
 
 
295
    # ModuleFinder can't handle runtime changes to __path__, but
 
296
    # win32com uses them.  Hook this in so win32com.shell is found.
 
297
    import modulefinder
 
298
    import win32com
 
299
    import cPickle as pickle
 
300
    for p in win32com.__path__[1:]:
 
301
        modulefinder.AddPackagePath("win32com", p)
 
302
    for extra in ["win32com.shell"]:
 
303
        __import__(extra)
 
304
        m = sys.modules[extra]
 
305
        for p in m.__path__[1:]:
 
306
            modulefinder.AddPackagePath(extra, p)
 
307
 
 
308
    # TBZR points to the TBZR directory
 
309
    tbzr_root = os.environ["TBZR"]
 
310
 
 
311
    # Ensure tbzrlib itself is on sys.path
 
312
    sys.path.append(tbzr_root)
 
313
 
 
314
    # Ensure our COM "entry-point" is on sys.path
 
315
    sys.path.append(os.path.join(tbzr_root, "shellext", "python"))
 
316
 
 
317
    packages.append("tbzrlib")
 
318
 
 
319
    # collect up our icons.
 
320
    cwd = os.getcwd()
 
321
    ico_root = os.path.join(tbzr_root, 'tbzrlib', 'resources')
 
322
    icos = [] # list of (path_root, relative_ico_path)
 
323
    # First always bzr's icon and its in the root of the bzr tree.
 
324
    icos.append(('', 'bzr.ico'))
 
325
    for root, dirs, files in os.walk(ico_root):
 
326
        icos.extend([(ico_root, os.path.join(root, f)[len(ico_root)+1:])
 
327
                     for f in files if f.endswith('.ico')])
 
328
    # allocate an icon ID for each file and the full path to the ico
 
329
    icon_resources = [(rid, os.path.join(ico_dir, ico_name))
 
330
                      for rid, (ico_dir, ico_name) in enumerate(icos)]
 
331
    # create a string resource with the mapping.  Might as well save the
 
332
    # runtime some effort and write a pickle.
 
333
    # Runtime expects unicode objects with forward-slash seps.
 
334
    fse = sys.getfilesystemencoding()
 
335
    map_items = [(f.replace('\\', '/').decode(fse), rid)
 
336
                 for rid, (_, f) in enumerate(icos)]
 
337
    ico_map = dict(map_items)
 
338
    # Create a new resource type of 'ICON_MAP', and use ID=1
 
339
    other_resources = [ ("ICON_MAP", 1, pickle.dumps(ico_map))]
 
340
 
 
341
    excludes.extend("""pywin pywin.dialogs pywin.dialogs.list
 
342
                       win32ui crawler.Crawler""".split())
 
343
 
 
344
    tbzr = dict(
 
345
        modules=["tbzr"],
 
346
        create_exe = False, # we only want a .dll
 
347
    )
 
348
    com_targets.append(tbzr)
 
349
 
 
350
    # tbzrcache executables - a "console" version for debugging and a
 
351
    # GUI version that is generally used.
 
352
    tbzrcache = dict(
 
353
        script = os.path.join(tbzr_root, "Scripts", "tbzrcache.py"),
 
354
        icon_resources = icon_resources,
 
355
        other_resources = other_resources,
 
356
    )
 
357
    console_targets.append(tbzrcache)
 
358
 
 
359
    # Make a windows version which is the same except for the base name.
 
360
    tbzrcachew = tbzrcache.copy()
 
361
    tbzrcachew["dest_base"]="tbzrcachew"
 
362
    gui_targets.append(tbzrcachew)
 
363
 
 
364
    # ditto for the tbzrcommand tool
 
365
    tbzrcommand = dict(
 
366
        script = os.path.join(tbzr_root, "Scripts", "tbzrcommand.py"),
 
367
        icon_resources = [(0,'bzr.ico')],
 
368
    )
 
369
    console_targets.append(tbzrcommand)
 
370
    tbzrcommandw = tbzrcommand.copy()
 
371
    tbzrcommandw["dest_base"]="tbzrcommandw"
 
372
    gui_targets.append(tbzrcommandw)
 
373
    
 
374
    # tbzr tests
 
375
    tbzrtest = dict(
 
376
        script = os.path.join(tbzr_root, "Scripts", "tbzrtest.py"),
 
377
    )
 
378
    console_targets.append(tbzrtest)
 
379
 
 
380
    # A utility to see python output from the shell extension - this will
 
381
    # die when we get a c++ extension
 
382
    # any .py file from pywin32's win32 lib will do (other than
 
383
    # win32traceutil itself that is)
 
384
    import winerror
 
385
    win32_lib_dir = os.path.dirname(winerror.__file__)
 
386
    tracer = dict(script = os.path.join(win32_lib_dir, "win32traceutil.py"),
 
387
                  dest_base="tbzr_tracer")
 
388
    console_targets.append(tracer)
 
389
 
 
390
 
 
391
def get_qbzr_py2exe_info(includes, excludes, packages):
 
392
    # PyQt4 itself still escapes the plugin detection code for some reason...
 
393
    packages.append('PyQt4')
 
394
    excludes.append('PyQt4.elementtree.ElementTree')
 
395
    includes.append('sip') # extension module required for Qt.
 
396
    packages.append('pygments') # colorizer for qbzr
 
397
    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())
 
404
    # 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
 
409
 
 
410
 
245
411
if 'bdist_wininst' in sys.argv:
246
412
    def find_docs():
247
413
        docs = []
292
458
        version_number.append(str(i))
293
459
    version_str = '.'.join(version_number)
294
460
 
 
461
    # An override to install_data used only by py2exe builds, which arranges
 
462
    # to byte-compile any .py files in data_files (eg, our plugins)
 
463
    # Necessary as we can't rely on the user having the relevant permissions
 
464
    # to the "Program Files" directory to generate them on the fly.
 
465
    class install_data_with_bytecompile(install_data):
 
466
        def run(self):
 
467
            from distutils.util import byte_compile
 
468
 
 
469
            install_data.run(self)
 
470
 
 
471
            py2exe = self.distribution.get_command_obj('py2exe', False)
 
472
            optimize = py2exe.optimize
 
473
            compile_names = [f for f in self.outfiles if f.endswith('.py')]
 
474
            byte_compile(compile_names,
 
475
                         optimize=optimize,
 
476
                         force=self.force, prefix=self.install_dir,
 
477
                         dry_run=self.dry_run)
 
478
            if optimize:
 
479
                suffix = 'o'
 
480
            else:
 
481
                suffix = 'c'
 
482
            self.outfiles.extend([f + suffix for f in compile_names])
 
483
    # end of class install_data_with_bytecompile
 
484
 
295
485
    target = py2exe.build_exe.Target(script = "bzr",
296
486
                                     dest_base = "bzr",
297
487
                                     icon_resources = [(0,'bzr.ico')],
324
514
        import warnings
325
515
        warnings.warn('Unknown Python version.\n'
326
516
                      'Please check setup.py script for compatibility.')
 
517
 
 
518
    # Although we currently can't enforce it, we consider it an error for
 
519
    # py2exe to report any files are "missing".  Such modules we know aren't
 
520
    # used should be listed here.
 
521
    excludes = """Tkinter psyco ElementPath r_hmac
 
522
                  ImaginaryModule cElementTree elementtree.ElementTree
 
523
                  Crypto.PublicKey._fastmath
 
524
                  medusa medusa.filesys medusa.ftp_server
 
525
                  tools tools.doc_generate
 
526
                  resource validate""".split()
 
527
    dll_excludes = []
 
528
 
327
529
    # email package from std python library use lazy import,
328
530
    # so we need to explicitly add all package
329
531
    additional_packages.add('email')
 
532
    # And it uses funky mappings to conver to 'Oldname' to 'newname'.  As
 
533
    # a result, packages like 'email.Parser' show as missing.  Tell py2exe
 
534
    # to exclude them.
 
535
    import email
 
536
    for oldname in getattr(email, '_LOWERNAMES', []):
 
537
        excludes.append("email." + oldname)
 
538
    for oldname in getattr(email, '_MIMENAMES', []):
 
539
        excludes.append("email.MIME" + oldname)
330
540
 
331
541
    # text files for help topis
332
542
    text_topics = glob.glob('bzrlib/help_topics/en/*.txt')
334
544
 
335
545
    # built-in plugins
336
546
    plugins_files = []
 
547
    # XXX - should we consider having the concept of an 'official' build,
 
548
    # which hard-codes the list of plugins, gets more upset if modules are
 
549
    # missing, etc?
 
550
    plugins = None # will be a set after plugin sniffing...
337
551
    for root, dirs, files in os.walk('bzrlib/plugins'):
 
552
        if root == 'bzrlib/plugins':
 
553
            plugins = set(dirs)
338
554
        x = []
339
555
        for i in files:
340
 
            if not i.endswith('.py'):
 
556
            if os.path.splitext(i)[1] not in [".py", ".pyd", ".dll", ".mo"]:
341
557
                continue
342
558
            if i == '__init__.py' and root == 'bzrlib/plugins':
343
559
                continue
351
567
    mf.run_package('bzrlib/plugins')
352
568
    packs, mods = mf.get_result()
353
569
    additional_packages.update(packs)
 
570
    includes.extend(mods)
 
571
 
 
572
    console_targets = [target,
 
573
                       'tools/win32/bzr_postinstall.py',
 
574
                       ]
 
575
    gui_targets = []
 
576
    com_targets = []
 
577
 
 
578
    if 'qbzr' in plugins:
 
579
        get_qbzr_py2exe_info(includes, excludes, packages)
 
580
 
 
581
    if "TBZR" in os.environ:
 
582
        # TORTOISE_OVERLAYS_MSI_WIN32 must be set to the location of the
 
583
        # TortoiseOverlays MSI installer file. It is in the TSVN svn repo and
 
584
        # can be downloaded from (username=guest, blank password):
 
585
        # http://tortoisesvn.tigris.org/svn/tortoisesvn/TortoiseOverlays/version-1.0.4/bin/TortoiseOverlays-1.0.4.11886-win32.msi
 
586
        if not os.path.isfile(os.environ.get('TORTOISE_OVERLAYS_MSI_WIN32',
 
587
                                             '<nofile>')):
 
588
            raise RuntimeError("Please set TORTOISE_OVERLAYS_MSI_WIN32 to the"
 
589
                               " location of the Win32 TortoiseOverlays .msi"
 
590
                               " installer file")
 
591
        get_tbzr_py2exe_info(includes, excludes, packages, console_targets,
 
592
                             gui_targets)
 
593
    else:
 
594
        # print this warning to stderr as output is redirected, so it is seen
 
595
        # at build time.  Also to stdout so it appears in the log
 
596
        for f in (sys.stderr, sys.stdout):
 
597
            print >> f, \
 
598
                "Skipping TBZR binaries - please set TBZR to a directory to enable"
354
599
 
355
600
    # MSWSOCK.dll is a system-specific library, which py2exe accidentally pulls
356
601
    # in on Vista.
 
602
    dll_excludes.append("MSWSOCK.dll")
357
603
    options_list = {"py2exe": {"packages": packages + list(additional_packages),
358
 
                               "includes": includes + mods,
359
 
                               "excludes": ["Tkinter", "medusa", "tools"],
360
 
                               "dll_excludes": ["MSWSOCK.dll"],
 
604
                               "includes": includes,
 
605
                               "excludes": excludes,
 
606
                               "dll_excludes": dll_excludes,
361
607
                               "dist_dir": "win32_bzr.exe",
 
608
                               "optimize": 1,
362
609
                              },
363
610
                   }
 
611
 
364
612
    setup(options=options_list,
365
 
          console=[target,
366
 
                   'tools/win32/bzr_postinstall.py',
367
 
                  ],
 
613
          console=console_targets,
 
614
          windows=gui_targets,
 
615
          com_server=com_targets,
368
616
          zipfile='lib/library.zip',
369
617
          data_files=topics_files + plugins_files,
 
618
          cmdclass={'install_data': install_data_with_bytecompile},
370
619
          )
371
620
 
372
621
else: