~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to setup.py

  • Committer: Robert Collins
  • Date: 2007-07-25 00:52:21 UTC
  • mfrom: (2650 +trunk)
  • mto: This revision was merged to the branch mainline in revision 2651.
  • Revision ID: robertc@robertcollins.net-20070725005221-0ysm6il5mqnme3wz
Merge bzr.dev.

Show diffs side-by-side

added added

removed removed

Lines of Context:
7
7
"""
8
8
 
9
9
import os
10
 
import os.path
11
10
import sys
12
11
 
13
 
if sys.version_info < (2, 4):
14
 
    sys.stderr.write("[ERROR] Not a supported Python version. Need 2.4+\n")
15
 
    sys.exit(1)
16
 
 
17
 
# NOTE: The directory containing setup.py, whether run by 'python setup.py' or
18
 
# './setup.py' or the equivalent with another path, should always be at the
19
 
# start of the path, so this should find the right one...
20
12
import bzrlib
21
13
 
22
 
def get_long_description():
23
 
    dirname = os.path.dirname(__file__)
24
 
    readme = os.path.join(dirname, 'README')
25
 
    f = open(readme, 'rb')
26
 
    try:
27
 
        return f.read()
28
 
    finally:
29
 
        f.close()
30
 
 
31
 
 
32
14
##
33
15
# META INFORMATION FOR SETUP
34
 
# see http://docs.python.org/dist/meta-data.html
35
 
META_INFO = {
36
 
    'name':         'bzr',
37
 
    'version':      bzrlib.__version__,
38
 
    'author':       'Canonical Ltd',
39
 
    'author_email': 'bazaar@lists.canonical.com',
40
 
    'url':          'http://www.bazaar-vcs.org/',
41
 
    'description':  'Friendly distributed version control system',
42
 
    'license':      'GNU GPL v2',
43
 
    'download_url': 'http://bazaar-vcs.org/Download',
44
 
    'long_description': get_long_description(),
45
 
    'classifiers': [
46
 
        'Development Status :: 6 - Mature',
47
 
        'Environment :: Console',
48
 
        'Intended Audience :: Developers',
49
 
        'Intended Audience :: System Administrators',
50
 
        'License :: OSI Approved :: GNU General Public License (GPL)',
51
 
        'Operating System :: Microsoft :: Windows',
52
 
        'Operating System :: OS Independent',
53
 
        'Operating System :: POSIX',
54
 
        'Programming Language :: Python',
55
 
        'Programming Language :: C',
56
 
        'Topic :: Software Development :: Version Control',
57
 
        ],
58
 
    }
 
16
 
 
17
META_INFO = {'name':         'bzr',
 
18
             'version':      bzrlib.__version__,
 
19
             'author':       'Canonical Ltd',
 
20
             'author_email': 'bazaar@lists.canonical.com',
 
21
             'url':          'http://www.bazaar-vcs.org/',
 
22
             'description':  'Friendly distributed version control system',
 
23
             'license':      'GNU GPL v2',
 
24
            }
59
25
 
60
26
# The list of packages is automatically generated later. Add other things
61
27
# that are part of BZRLIB here.
64
30
PKG_DATA = {# install files from selftest suite
65
31
            'package_data': {'bzrlib': ['doc/api/*.txt',
66
32
                                        'tests/test_patches_data/*',
67
 
                                        'help_topics/en/*.txt',
68
33
                                       ]},
69
34
           }
70
35
 
 
36
######################################################################
 
37
# Reinvocation stolen from bzr, we need python2.4 by virtue of bzr_man
 
38
# including bzrlib.help
 
39
 
 
40
try:
 
41
    version_info = sys.version_info
 
42
except AttributeError:
 
43
    version_info = 1, 5 # 1.5 or older
 
44
 
 
45
REINVOKE = "__BZR_REINVOKE"
 
46
NEED_VERS = (2, 4)
 
47
KNOWN_PYTHONS = ('python2.4',)
 
48
 
 
49
if version_info < NEED_VERS:
 
50
    if not os.environ.has_key(REINVOKE):
 
51
        # mutating os.environ doesn't work in old Pythons
 
52
        os.putenv(REINVOKE, "1")
 
53
        for python in KNOWN_PYTHONS:
 
54
            try:
 
55
                os.execvp(python, [python] + sys.argv)
 
56
            except OSError:
 
57
                pass
 
58
    print >>sys.stderr, "bzr: error: cannot find a suitable python interpreter"
 
59
    print >>sys.stderr, "  (need %d.%d or later)" % NEED_VERS
 
60
    sys.exit(1)
 
61
if getattr(os, "unsetenv", None) is not None:
 
62
    os.unsetenv(REINVOKE)
 
63
 
71
64
 
72
65
def get_bzrlib_packages():
73
66
    """Recurse through the bzrlib directory, and extract the package names"""
95
88
 
96
89
from distutils.core import setup
97
90
from distutils.command.install_scripts import install_scripts
98
 
from distutils.command.install_data import install_data
99
91
from distutils.command.build import build
100
92
 
101
93
###############################
111
103
 
112
104
        if sys.platform == "win32":
113
105
            try:
114
 
                scripts_dir = os.path.join(sys.prefix, 'Scripts')
 
106
                scripts_dir = self.install_dir
115
107
                script_path = self._quoted_path(os.path.join(scripts_dir,
116
108
                                                             "bzr"))
117
109
                python_exe = self._quoted_path(sys.executable)
118
110
                args = self._win_batch_args()
119
111
                batch_str = "@%s %s %s" % (python_exe, script_path, args)
120
 
                batch_path = os.path.join(self.install_dir, "bzr.bat")
 
112
                batch_path = script_path + ".bat"
121
113
                f = file(batch_path, "w")
122
114
                f.write(batch_str)
123
115
                f.close()
144
136
    """Customized build distutils action.
145
137
    Generate bzr.1.
146
138
    """
147
 
 
148
139
    def run(self):
149
140
        build.run(self)
150
141
 
158
149
 
159
150
command_classes = {'install_scripts': my_install_scripts,
160
151
                   'build': bzr_build}
161
 
from distutils import log
162
 
from distutils.errors import CCompilerError, DistutilsPlatformError
163
152
from distutils.extension import Extension
164
153
ext_modules = []
165
154
try:
176
165
    from distutils.command.build_ext import build_ext
177
166
else:
178
167
    have_pyrex = True
179
 
    from Pyrex.Compiler.Version import version as pyrex_version
180
 
 
181
 
 
182
 
class build_ext_if_possible(build_ext):
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
 
 
194
 
    def run(self):
195
 
        try:
196
 
            build_ext.run(self)
197
 
        except DistutilsPlatformError, e:
198
 
            if not self.allow_python_fallback:
199
 
                log.warn('\n  Cannot build extensions.\n'
200
 
                         '  Use "build_ext --allow-python-fallback" to use'
201
 
                         ' slower python implementations instead.\n')
202
 
                raise
203
 
            log.warn(str(e))
204
 
            log.warn('\n  Extensions cannot be built.\n'
205
 
                     '  Using the slower Python implementations instead.\n')
206
 
 
207
 
    def build_extension(self, ext):
208
 
        try:
209
 
            build_ext.build_extension(self, ext)
210
 
        except CCompilerError:
211
 
            if not self.allow_python_fallback:
212
 
                log.warn('\n  Cannot build extension "%s".\n'
213
 
                         '  Use "build_ext --allow-python-fallback" to use'
214
 
                         ' slower 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,))
220
 
 
221
 
 
222
168
# Override the build_ext if we have Pyrex available
223
 
command_classes['build_ext'] = build_ext_if_possible
 
169
command_classes['build_ext'] = build_ext
224
170
unavailable_files = []
225
171
 
226
172
 
227
 
def add_pyrex_extension(module_name, libraries=None, extra_source=[]):
 
173
def add_pyrex_extension(module_name, **kwargs):
228
174
    """Add a pyrex module to build.
229
175
 
230
176
    This will use Pyrex to auto-generate the .c file if it is available.
240
186
    path = module_name.replace('.', '/')
241
187
    pyrex_name = path + '.pyx'
242
188
    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))
249
189
    if have_pyrex:
250
 
        source = [pyrex_name]
 
190
        ext_modules.append(Extension(module_name, [pyrex_name]))
251
191
    else:
252
192
        if not os.path.isfile(c_name):
253
193
            unavailable_files.append(c_name)
254
 
            return
255
194
        else:
256
 
            source = [c_name]
257
 
    source.extend(extra_source)
258
 
    ext_modules.append(Extension(module_name, source,
259
 
        define_macros=define_macros, libraries=libraries))
260
 
 
261
 
 
262
 
add_pyrex_extension('bzrlib._btree_serializer_c')
263
 
add_pyrex_extension('bzrlib._groupcompress_pyx',
264
 
                    extra_source=['bzrlib/diff-delta.c'])
265
 
add_pyrex_extension('bzrlib._chunks_to_lines_pyx')
 
195
            ext_modules.append(Extension(module_name, [c_name]))
 
196
 
 
197
 
 
198
add_pyrex_extension('bzrlib._dirstate_helpers_c')
266
199
add_pyrex_extension('bzrlib._knit_load_data_c')
267
 
add_pyrex_extension('bzrlib._chk_map_pyx', libraries=['z'])
268
 
if sys.platform == 'win32':
269
 
    add_pyrex_extension('bzrlib._dirstate_helpers_c',
270
 
                        libraries=['Ws2_32'])
271
 
    add_pyrex_extension('bzrlib._walkdirs_win32')
272
 
else:
273
 
    if have_pyrex and pyrex_version == '0.9.4.1':
274
 
        # Pyrex 0.9.4.1 fails to compile this extension correctly
275
 
        # The code it generates re-uses a "local" pointer and
276
 
        # calls "PY_DECREF" after having set it to NULL. (It mixes PY_XDECREF
277
 
        # which is NULL safe with PY_DECREF which is not.)
278
 
        print 'Cannot build extension "bzrlib._dirstate_helpers_c" using'
279
 
        print 'your version of pyrex "%s". Please upgrade your pyrex' % (
280
 
            pyrex_version,)
281
 
        print 'install. For now, the non-compiled (python) version will'
282
 
        print 'be used instead.'
283
 
    else:
284
 
        add_pyrex_extension('bzrlib._dirstate_helpers_c')
285
 
    add_pyrex_extension('bzrlib._readdir_pyx')
286
 
ext_modules.append(Extension('bzrlib._patiencediff_c', ['bzrlib/_patiencediff_c.c']))
287
200
 
288
201
 
289
202
if unavailable_files:
293
206
    print
294
207
 
295
208
 
296
 
def get_tbzr_py2exe_info(includes, excludes, packages, console_targets,
297
 
                         gui_targets, data_files):
298
 
    packages.append('tbzrcommands')
299
 
 
300
 
    # ModuleFinder can't handle runtime changes to __path__, but
301
 
    # win32com uses them.  Hook this in so win32com.shell is found.
302
 
    import modulefinder
303
 
    import win32com
304
 
    import cPickle as pickle
305
 
    for p in win32com.__path__[1:]:
306
 
        modulefinder.AddPackagePath("win32com", p)
307
 
    for extra in ["win32com.shell"]:
308
 
        __import__(extra)
309
 
        m = sys.modules[extra]
310
 
        for p in m.__path__[1:]:
311
 
            modulefinder.AddPackagePath(extra, p)
312
 
 
313
 
    # TBZR points to the TBZR directory
314
 
    tbzr_root = os.environ["TBZR"]
315
 
 
316
 
    # Ensure tbzrlib itself is on sys.path
317
 
    sys.path.append(tbzr_root)
318
 
 
319
 
    # Ensure our COM "entry-point" is on sys.path
320
 
    sys.path.append(os.path.join(tbzr_root, "shellext", "python"))
321
 
 
322
 
    packages.append("tbzrlib")
323
 
 
324
 
    # collect up our icons.
325
 
    cwd = os.getcwd()
326
 
    ico_root = os.path.join(tbzr_root, 'tbzrlib', 'resources')
327
 
    icos = [] # list of (path_root, relative_ico_path)
328
 
    # First always bzr's icon and its in the root of the bzr tree.
329
 
    icos.append(('', 'bzr.ico'))
330
 
    for root, dirs, files in os.walk(ico_root):
331
 
        icos.extend([(ico_root, os.path.join(root, f)[len(ico_root)+1:])
332
 
                     for f in files if f.endswith('.ico')])
333
 
    # allocate an icon ID for each file and the full path to the ico
334
 
    icon_resources = [(rid, os.path.join(ico_dir, ico_name))
335
 
                      for rid, (ico_dir, ico_name) in enumerate(icos)]
336
 
    # create a string resource with the mapping.  Might as well save the
337
 
    # runtime some effort and write a pickle.
338
 
    # Runtime expects unicode objects with forward-slash seps.
339
 
    fse = sys.getfilesystemencoding()
340
 
    map_items = [(f.replace('\\', '/').decode(fse), rid)
341
 
                 for rid, (_, f) in enumerate(icos)]
342
 
    ico_map = dict(map_items)
343
 
    # Create a new resource type of 'ICON_MAP', and use ID=1
344
 
    other_resources = [ ("ICON_MAP", 1, pickle.dumps(ico_map))]
345
 
 
346
 
    excludes.extend("""pywin pywin.dialogs pywin.dialogs.list
347
 
                       win32ui crawler.Crawler""".split())
348
 
 
349
 
    # NOTE: We still create a DLL version of the Python implemented shell
350
 
    # extension for testing purposes - but it is *not* registered by
351
 
    # default - our C++ one is instead.  To discourage people thinking
352
 
    # this DLL is still necessary, its called 'tbzr_old.dll'
353
 
    tbzr = dict(
354
 
        modules=["tbzr"],
355
 
        create_exe = False, # we only want a .dll
356
 
        dest_base = 'tbzr_old',
357
 
    )
358
 
    com_targets.append(tbzr)
359
 
 
360
 
    # tbzrcache executables - a "console" version for debugging and a
361
 
    # GUI version that is generally used.
362
 
    tbzrcache = dict(
363
 
        script = os.path.join(tbzr_root, "scripts", "tbzrcache.py"),
364
 
        icon_resources = icon_resources,
365
 
        other_resources = other_resources,
366
 
    )
367
 
    console_targets.append(tbzrcache)
368
 
 
369
 
    # Make a windows version which is the same except for the base name.
370
 
    tbzrcachew = tbzrcache.copy()
371
 
    tbzrcachew["dest_base"]="tbzrcachew"
372
 
    gui_targets.append(tbzrcachew)
373
 
 
374
 
    # ditto for the tbzrcommand tool
375
 
    tbzrcommand = dict(
376
 
        script = os.path.join(tbzr_root, "scripts", "tbzrcommand.py"),
377
 
        icon_resources = [(0,'bzr.ico')],
378
 
    )
379
 
    console_targets.append(tbzrcommand)
380
 
    tbzrcommandw = tbzrcommand.copy()
381
 
    tbzrcommandw["dest_base"]="tbzrcommandw"
382
 
    gui_targets.append(tbzrcommandw)
383
 
    
384
 
    # A utility to see python output from both C++ and Python based shell
385
 
    # extensions
386
 
    tracer = dict(script=os.path.join(tbzr_root, "scripts", "tbzrtrace.py"))
387
 
    console_targets.append(tracer)
388
 
 
389
 
    # The C++ implemented shell extensions.
390
 
    dist_dir = os.path.join(tbzr_root, "shellext", "cpp", "tbzrshellext",
391
 
                            "build", "dist")
392
 
    data_files.append(('', [os.path.join(dist_dir, 'tbzrshellext_x86.dll')]))
393
 
    data_files.append(('', [os.path.join(dist_dir, 'tbzrshellext_x64.dll')]))
394
 
 
395
 
 
396
 
def get_qbzr_py2exe_info(includes, excludes, packages):
397
 
    # PyQt4 itself still escapes the plugin detection code for some reason...
398
 
    packages.append('PyQt4')
399
 
    excludes.append('PyQt4.elementtree.ElementTree')
400
 
    includes.append('sip') # extension module required for Qt.
401
 
    packages.append('pygments') # colorizer for qbzr
402
 
    packages.append('docutils') # html formatting
403
 
    # but we can avoid many Qt4 Dlls.
404
 
    dll_excludes.extend(
405
 
        """QtAssistantClient4.dll QtCLucene4.dll QtDesigner4.dll
406
 
        QtHelp4.dll QtNetwork4.dll QtOpenGL4.dll QtScript4.dll
407
 
        QtSql4.dll QtTest4.dll QtWebKit4.dll QtXml4.dll
408
 
        qscintilla2.dll""".split())
409
 
    # the qt binaries might not be on PATH...
410
 
    qt_dir = os.path.join(sys.prefix, "PyQt4", "bin")
411
 
    path = os.environ.get("PATH","")
412
 
    if qt_dir.lower() not in [p.lower() for p in path.split(os.pathsep)]:
413
 
        os.environ["PATH"] = path + os.pathsep + qt_dir
414
 
 
415
 
 
416
 
def get_svn_py2exe_info(includes, excludes, packages):
417
 
    packages.append('subvertpy')
418
 
 
419
 
 
420
209
if 'bdist_wininst' in sys.argv:
421
 
    def find_docs():
422
 
        docs = []
423
 
        for root, dirs, files in os.walk('doc'):
424
 
            r = []
425
 
            for f in files:
426
 
                if (os.path.splitext(f)[1] in ('.html','.css','.png','.pdf')
427
 
                    or f == 'quick-start-summary.svg'):
428
 
                    r.append(os.path.join(root, f))
429
 
            if r:
430
 
                relative = root[4:]
431
 
                if relative:
432
 
                    target = os.path.join('Doc\\Bazaar', relative)
433
 
                else:
434
 
                    target = 'Doc\\Bazaar'
435
 
                docs.append((target, r))
436
 
        return docs
437
 
 
 
210
    import glob
 
211
    # doc files
 
212
    docs = glob.glob('doc/*.htm') + ['doc/default.css']
 
213
    dev_docs = glob.glob('doc/developers/*.htm')
438
214
    # python's distutils-based win32 installer
439
215
    ARGS = {'scripts': ['bzr', 'tools/win32/bzr-win32-bdist-postinstall.py'],
440
216
            'ext_modules': ext_modules,
441
217
            # help pages
442
 
            'data_files': find_docs(),
 
218
            'data_files': [('Doc/Bazaar', docs),
 
219
                           ('Doc/Bazaar/developers', dev_docs),
 
220
                          ],
443
221
            # for building pyrex extensions
444
 
            'cmdclass': {'build_ext': build_ext_if_possible},
 
222
            'cmdclass': {'build_ext': build_ext},
445
223
           }
446
224
 
447
225
    ARGS.update(META_INFO)
451
229
    setup(**ARGS)
452
230
 
453
231
elif 'py2exe' in sys.argv:
454
 
    import glob
455
232
    # py2exe setup
456
233
    import py2exe
457
234
 
467
244
        version_number.append(str(i))
468
245
    version_str = '.'.join(version_number)
469
246
 
470
 
    # An override to install_data used only by py2exe builds, which arranges
471
 
    # to byte-compile any .py files in data_files (eg, our plugins)
472
 
    # Necessary as we can't rely on the user having the relevant permissions
473
 
    # to the "Program Files" directory to generate them on the fly.
474
 
    class install_data_with_bytecompile(install_data):
475
 
        def run(self):
476
 
            from distutils.util import byte_compile
477
 
 
478
 
            install_data.run(self)
479
 
 
480
 
            py2exe = self.distribution.get_command_obj('py2exe', False)
481
 
            optimize = py2exe.optimize
482
 
            compile_names = [f for f in self.outfiles if f.endswith('.py')]
483
 
            byte_compile(compile_names,
484
 
                         optimize=optimize,
485
 
                         force=self.force, prefix=self.install_dir,
486
 
                         dry_run=self.dry_run)
487
 
            if optimize:
488
 
                suffix = 'o'
489
 
            else:
490
 
                suffix = 'c'
491
 
            self.outfiles.extend([f + suffix for f in compile_names])
492
 
    # end of class install_data_with_bytecompile
493
 
 
494
247
    target = py2exe.build_exe.Target(script = "bzr",
495
248
                                     dest_base = "bzr",
496
249
                                     icon_resources = [(0,'bzr.ico')],
503
256
                                     comments = META_INFO['description'],
504
257
                                    )
505
258
 
506
 
    packages = BZRLIB['packages']
507
 
    packages.remove('bzrlib')
508
 
    packages = [i for i in packages if not i.startswith('bzrlib.plugins')]
509
 
    includes = []
510
 
    for i in glob.glob('bzrlib\\*.py'):
511
 
        module = i[:-3].replace('\\', '.')
512
 
        if module.endswith('__init__'):
513
 
            module = module[:-len('__init__')]
514
 
        includes.append(module)
515
 
 
516
 
    additional_packages = set()
 
259
    additional_packages =  []
517
260
    if sys.version.startswith('2.4'):
518
261
        # adding elementtree package
519
 
        additional_packages.add('elementtree')
 
262
        additional_packages.append('elementtree')
520
263
    elif sys.version.startswith('2.5'):
521
 
        additional_packages.add('xml.etree')
 
264
        additional_packages.append('xml.etree')
522
265
    else:
523
266
        import warnings
524
267
        warnings.warn('Unknown Python version.\n'
525
268
                      'Please check setup.py script for compatibility.')
526
 
 
527
 
    # Although we currently can't enforce it, we consider it an error for
528
 
    # py2exe to report any files are "missing".  Such modules we know aren't
529
 
    # used should be listed here.
530
 
    excludes = """Tkinter psyco ElementPath r_hmac
531
 
                  ImaginaryModule cElementTree elementtree.ElementTree
532
 
                  Crypto.PublicKey._fastmath
533
 
                  medusa medusa.filesys medusa.ftp_server
534
 
                  tools tools.doc_generate
535
 
                  resource validate""".split()
536
 
    dll_excludes = []
537
 
 
538
269
    # email package from std python library use lazy import,
539
270
    # so we need to explicitly add all package
540
 
    additional_packages.add('email')
541
 
    # And it uses funky mappings to conver to 'Oldname' to 'newname'.  As
542
 
    # a result, packages like 'email.Parser' show as missing.  Tell py2exe
543
 
    # to exclude them.
544
 
    import email
545
 
    for oldname in getattr(email, '_LOWERNAMES', []):
546
 
        excludes.append("email." + oldname)
547
 
    for oldname in getattr(email, '_MIMENAMES', []):
548
 
        excludes.append("email.MIME" + oldname)
549
 
 
550
 
    # text files for help topis
551
 
    text_topics = glob.glob('bzrlib/help_topics/en/*.txt')
552
 
    topics_files = [('lib/help_topics/en', text_topics)]
553
 
 
554
 
    # built-in plugins
555
 
    plugins_files = []
556
 
    # XXX - should we consider having the concept of an 'official' build,
557
 
    # which hard-codes the list of plugins, gets more upset if modules are
558
 
    # missing, etc?
559
 
    plugins = None # will be a set after plugin sniffing...
560
 
    for root, dirs, files in os.walk('bzrlib/plugins'):
561
 
        if root == 'bzrlib/plugins':
562
 
            plugins = set(dirs)
563
 
            # We ship plugins as normal files on the file-system - however,
564
 
            # the build process can cause *some* of these plugin files to end
565
 
            # up in library.zip. Thus, we saw (eg) "plugins/svn/test" in
566
 
            # library.zip, and then saw import errors related to that as the
567
 
            # rest of the svn plugin wasn't. So we tell py2exe to leave the
568
 
            # plugins out of the .zip file
569
 
            excludes.extend(["bzrlib.plugins." + d for d in dirs])
570
 
        x = []
571
 
        for i in files:
572
 
            if os.path.splitext(i)[1] not in [".py", ".pyd", ".dll", ".mo"]:
573
 
                continue
574
 
            if i == '__init__.py' and root == 'bzrlib/plugins':
575
 
                continue
576
 
            x.append(os.path.join(root, i))
577
 
        if x:
578
 
            target_dir = root[len('bzrlib/'):]  # install to 'plugins/...'
579
 
            plugins_files.append((target_dir, x))
580
 
    # find modules for built-in plugins
581
 
    import tools.package_mf
582
 
    mf = tools.package_mf.CustomModuleFinder()
583
 
    mf.run_package('bzrlib/plugins')
584
 
    packs, mods = mf.get_result()
585
 
    additional_packages.update(packs)
586
 
    includes.extend(mods)
587
 
 
588
 
    console_targets = [target,
589
 
                       'tools/win32/bzr_postinstall.py',
590
 
                       ]
591
 
    gui_targets = []
592
 
    com_targets = []
593
 
    data_files = topics_files + plugins_files
594
 
 
595
 
    if 'qbzr' in plugins:
596
 
        get_qbzr_py2exe_info(includes, excludes, packages)
597
 
 
598
 
    if 'svn' in plugins:
599
 
        get_svn_py2exe_info(includes, excludes, packages)
600
 
 
601
 
    if "TBZR" in os.environ:
602
 
        # TORTOISE_OVERLAYS_MSI_WIN32 must be set to the location of the
603
 
        # TortoiseOverlays MSI installer file. It is in the TSVN svn repo and
604
 
        # can be downloaded from (username=guest, blank password):
605
 
        # http://tortoisesvn.tigris.org/svn/tortoisesvn/TortoiseOverlays/version-1.0.4/bin/TortoiseOverlays-1.0.4.11886-win32.msi
606
 
        # Ditto for TORTOISE_OVERLAYS_MSI_X64, pointing at *-x64.msi.
607
 
        for needed in ('TORTOISE_OVERLAYS_MSI_WIN32',
608
 
                       'TORTOISE_OVERLAYS_MSI_X64'):
609
 
            if not os.path.isfile(os.environ.get(needed, '<nofile>')):
610
 
                raise RuntimeError("Please set %s to the"
611
 
                                   " location of the relevant TortoiseOverlays"
612
 
                                   " .msi installer file" % needed)
613
 
        get_tbzr_py2exe_info(includes, excludes, packages, console_targets,
614
 
                             gui_targets, data_files)
615
 
    else:
616
 
        # print this warning to stderr as output is redirected, so it is seen
617
 
        # at build time.  Also to stdout so it appears in the log
618
 
        for f in (sys.stderr, sys.stdout):
619
 
            print >> f, \
620
 
                "Skipping TBZR binaries - please set TBZR to a directory to enable"
621
 
 
622
 
    # MSWSOCK.dll is a system-specific library, which py2exe accidentally pulls
623
 
    # in on Vista.
624
 
    dll_excludes.extend(["MSWSOCK.dll", "MSVCP60.dll", "powrprof.dll"])
625
 
    options_list = {"py2exe": {"packages": packages + list(additional_packages),
626
 
                               "includes": includes,
627
 
                               "excludes": excludes,
628
 
                               "dll_excludes": dll_excludes,
 
271
    additional_packages.append('email')
 
272
 
 
273
    options_list = {"py2exe": {"packages": BZRLIB['packages'] +
 
274
                                           additional_packages,
 
275
                               "excludes": ["Tkinter", "medusa", "tools"],
629
276
                               "dist_dir": "win32_bzr.exe",
630
 
                               "optimize": 1,
631
277
                              },
632
278
                   }
633
 
 
634
279
    setup(options=options_list,
635
 
          console=console_targets,
636
 
          windows=gui_targets,
637
 
          com_server=com_targets,
638
 
          zipfile='lib/library.zip',
639
 
          data_files=data_files,
640
 
          cmdclass={'install_data': install_data_with_bytecompile},
641
 
          )
 
280
          console=[target,
 
281
                   'tools/win32/bzr_postinstall.py',
 
282
                  ],
 
283
          zipfile='lib/library.zip')
642
284
 
643
285
else:
644
 
    # ad-hoc for easy_install
645
 
    DATA_FILES = []
646
 
    if not 'bdist_egg' in sys.argv:
647
 
        # generate and install bzr.1 only with plain install, not easy_install one
648
 
        DATA_FILES = [('man/man1', ['bzr.1'])]
649
 
 
650
286
    # std setup
651
287
    ARGS = {'scripts': ['bzr'],
652
 
            'data_files': DATA_FILES,
 
288
            'data_files': [('man/man1', ['bzr.1'])],
653
289
            'cmdclass': command_classes,
654
290
            'ext_modules': ext_modules,
655
291
           }
656
 
 
 
292
    
657
293
    ARGS.update(META_INFO)
658
294
    ARGS.update(BZRLIB)
659
295
    ARGS.update(PKG_DATA)