← Back to branch summary

~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to setup.py

  • Committer: Canonical.com Patch Queue Manager
  • Date: 2008-03-16 16:58:03 UTC
  • mfrom: (3224.3.1 news-typo)
  • Revision ID: pqm@pqm.ubuntu.com-20080316165803-tisoc9mpob9z544o
(Matt Nordhoff) Trivial NEWS typo fix

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
 
import glob
14
12
 
15
 
if sys.version_info < (2, 6):
16
 
    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")
17
15
    sys.exit(1)
18
16
 
19
17
# NOTE: The directory containing setup.py, whether run by 'python setup.py' or
39
37
    'version':      bzrlib.__version__,
40
38
    'author':       'Canonical Ltd',
41
39
    'author_email': 'bazaar@lists.canonical.com',
42
 
    'url':          'http://bazaar.canonical.com/',
 
40
    'url':          'http://www.bazaar-vcs.org/',
43
41
    'description':  'Friendly distributed version control system',
44
42
    'license':      'GNU GPL v2',
45
 
    'download_url': 'https://launchpad.net/bzr/+download',
 
43
    'download_url': 'http://bazaar-vcs.org/Download',
46
44
    'long_description': get_long_description(),
47
45
    'classifiers': [
48
46
        'Development Status :: 6 - Mature',
67
65
            'package_data': {'bzrlib': ['doc/api/*.txt',
68
66
                                        'tests/test_patches_data/*',
69
67
                                        'help_topics/en/*.txt',
70
 
                                        'tests/ssl_certs/ca.crt',
71
 
                                        'tests/ssl_certs/server_without_pass.key',
72
 
                                        'tests/ssl_certs/server_with_pass.key',
73
 
                                        'tests/ssl_certs/server.crt',
74
68
                                       ]},
75
69
           }
76
 
I18N_FILES = []
77
 
for filepath in glob.glob("bzrlib/locale/*/LC_MESSAGES/*.mo"):
78
 
    langfile = filepath[len("bzrlib/locale/"):]
79
 
    targetpath = os.path.dirname(os.path.join("share/locale", langfile))
80
 
    I18N_FILES.append((targetpath, [filepath]))
 
70
 
81
71
 
82
72
def get_bzrlib_packages():
83
73
    """Recurse through the bzrlib directory, and extract the package names"""
103
93
BZRLIB['packages'] = get_bzrlib_packages()
104
94
 
105
95
 
106
 
from distutils import log
107
96
from distutils.core import setup
108
97
from distutils.command.install_scripts import install_scripts
109
 
from distutils.command.install_data import install_data
110
98
from distutils.command.build import build
111
99
 
112
100
###############################
132
120
                f = file(batch_path, "w")
133
121
                f.write(batch_str)
134
122
                f.close()
135
 
                print("Created: %s" % batch_path)
136
 
            except Exception:
137
 
                e = sys.exc_info()[1]
138
 
                print("ERROR: Unable to create %s: %s" % (batch_path, e))
 
123
                print "Created:", batch_path
 
124
            except Exception, e:
 
125
                print "ERROR: Unable to create %s: %s" % (batch_path, e)
139
126
 
140
127
    def _quoted_path(self, path):
141
128
        if ' ' in path:
156
143
    """Customized build distutils action.
157
144
    Generate bzr.1.
158
145
    """
159
 
 
160
 
    sub_commands = build.sub_commands + [
161
 
            ('build_mo', lambda _: True),
162
 
            ]
163
 
 
164
146
    def run(self):
165
147
        build.run(self)
166
148
 
167
 
        from tools import generate_docs
 
149
        import generate_docs
168
150
        generate_docs.main(argv=["bzr", "man"])
169
151
 
170
152
 
172
154
## Setup
173
155
########################
174
156
 
175
 
from bzrlib.bzr_distutils import build_mo
176
 
 
177
157
command_classes = {'install_scripts': my_install_scripts,
178
 
                   'build': bzr_build,
179
 
                   'build_mo': build_mo,
180
 
                   }
 
158
                   'build': bzr_build}
181
159
from distutils import log
182
160
from distutils.errors import CCompilerError, DistutilsPlatformError
183
161
from distutils.extension import Extension
184
162
ext_modules = []
185
163
try:
186
 
    try:
187
 
        from Cython.Distutils import build_ext
188
 
        from Cython.Compiler.Version import version as pyrex_version
189
 
    except ImportError:
190
 
        print("No Cython, trying Pyrex...")
191
 
        from Pyrex.Distutils import build_ext
192
 
        from Pyrex.Compiler.Version import version as pyrex_version
 
164
    from Pyrex.Distutils import build_ext
193
165
except ImportError:
194
166
    have_pyrex = False
195
167
    # try to build the extension from the prior generated source.
196
 
    print("")
197
 
    print("The python package 'Pyrex' is not available."
198
 
          " If the .c files are available,")
199
 
    print("they will be built,"
200
 
          " but modifying the .pyx files will not rebuild them.")
201
 
    print("")
 
168
    print
 
169
    print ("The python package 'Pyrex' is not available."
 
170
           " If the .c files are available,")
 
171
    print ("they will be built,"
 
172
           " but modifying the .pyx files will not rebuild them.")
 
173
    print
202
174
    from distutils.command.build_ext import build_ext
203
175
else:
204
176
    have_pyrex = True
205
 
    pyrex_version_info = tuple(map(int, pyrex_version.rstrip("+").split('.')))
206
177
 
207
178
 
208
179
class build_ext_if_possible(build_ext):
209
180
 
210
 
    user_options = build_ext.user_options + [
211
 
        ('allow-python-fallback', None,
212
 
         "When an extension cannot be built, allow falling"
213
 
         " back to the pure-python implementation.")
214
 
        ]
215
 
 
216
 
    def initialize_options(self):
217
 
        build_ext.initialize_options(self)
218
 
        self.allow_python_fallback = False
219
 
 
220
181
    def run(self):
221
182
        try:
222
183
            build_ext.run(self)
223
 
        except DistutilsPlatformError:
224
 
            e = sys.exc_info()[1]
225
 
            if not self.allow_python_fallback:
226
 
                log.warn('\n  Cannot build extensions.\n'
227
 
                         '  Use "build_ext --allow-python-fallback" to use'
228
 
                         ' slower python implementations instead.\n')
229
 
                raise
 
184
        except DistutilsPlatformError, e:
230
185
            log.warn(str(e))
231
 
            log.warn('\n  Extensions cannot be built.\n'
232
 
                     '  Using the slower Python implementations instead.\n')
 
186
            log.warn('Extensions cannot be built, '
 
187
                     'will use the Python versions instead')
233
188
 
234
189
    def build_extension(self, ext):
235
190
        try:
236
191
            build_ext.build_extension(self, ext)
237
192
        except CCompilerError:
238
 
            if not self.allow_python_fallback:
239
 
                log.warn('\n  Cannot build extension "%s".\n'
240
 
                         '  Use "build_ext --allow-python-fallback" to use'
241
 
                         ' slower python implementations instead.\n'
242
 
                         % (ext.name,))
243
 
                raise
244
 
            log.warn('\n  Building of "%s" extension failed.\n'
245
 
                     '  Using the slower Python implementation instead.'
246
 
                     % (ext.name,))
 
193
            log.warn('Building of "%s" extension failed, '
 
194
                     'will use the Python version instead' % (ext.name,))
247
195
 
248
196
 
249
197
# Override the build_ext if we have Pyrex available
251
199
unavailable_files = []
252
200
 
253
201
 
254
 
def add_pyrex_extension(module_name, libraries=None, extra_source=[]):
 
202
def add_pyrex_extension(module_name, **kwargs):
255
203
    """Add a pyrex module to build.
256
204
 
257
205
    This will use Pyrex to auto-generate the .c file if it is available.
267
215
    path = module_name.replace('.', '/')
268
216
    pyrex_name = path + '.pyx'
269
217
    c_name = path + '.c'
270
 
    define_macros = []
271
 
    if sys.platform == 'win32':
272
 
        # pyrex uses the macro WIN32 to detect the platform, even though it
273
 
        # should be using something like _WIN32 or MS_WINDOWS, oh well, we can
274
 
        # give it the right value.
275
 
        define_macros.append(('WIN32', None))
276
218
    if have_pyrex:
277
 
        source = [pyrex_name]
 
219
        ext_modules.append(Extension(module_name, [pyrex_name]))
278
220
    else:
279
221
        if not os.path.isfile(c_name):
280
222
            unavailable_files.append(c_name)
281
 
            return
282
223
        else:
283
 
            source = [c_name]
284
 
    source.extend(extra_source)
285
 
    ext_modules.append(Extension(module_name, source,
286
 
        define_macros=define_macros, libraries=libraries))
287
 
 
288
 
 
289
 
add_pyrex_extension('bzrlib._annotator_pyx')
290
 
add_pyrex_extension('bzrlib._bencode_pyx')
291
 
add_pyrex_extension('bzrlib._chunks_to_lines_pyx')
292
 
add_pyrex_extension('bzrlib._groupcompress_pyx',
293
 
                    extra_source=['bzrlib/diff-delta.c'])
294
 
add_pyrex_extension('bzrlib._knit_load_data_pyx')
295
 
add_pyrex_extension('bzrlib._known_graph_pyx')
296
 
add_pyrex_extension('bzrlib._rio_pyx')
297
 
if sys.platform == 'win32':
298
 
    add_pyrex_extension('bzrlib._dirstate_helpers_pyx',
299
 
                        libraries=['Ws2_32'])
300
 
    add_pyrex_extension('bzrlib._walkdirs_win32')
301
 
else:
302
 
    if have_pyrex and pyrex_version_info[:3] == (0,9,4):
303
 
        # Pyrex 0.9.4.1 fails to compile this extension correctly
304
 
        # The code it generates re-uses a "local" pointer and
305
 
        # calls "PY_DECREF" after having set it to NULL. (It mixes PY_XDECREF
306
 
        # which is NULL safe with PY_DECREF which is not.)
307
 
        # <https://bugs.launchpad.net/bzr/+bug/449372>
308
 
        # <https://bugs.launchpad.net/bzr/+bug/276868>
309
 
        print('Cannot build extension "bzrlib._dirstate_helpers_pyx" using')
310
 
        print('your version of pyrex "%s". Please upgrade your pyrex'
311
 
              % (pyrex_version,))
312
 
        print('install. For now, the non-compiled (python) version will')
313
 
        print('be used instead.')
314
 
    else:
315
 
        add_pyrex_extension('bzrlib._dirstate_helpers_pyx')
316
 
    add_pyrex_extension('bzrlib._readdir_pyx')
317
 
add_pyrex_extension('bzrlib._chk_map_pyx')
318
 
ext_modules.append(Extension('bzrlib._patiencediff_c',
319
 
                             ['bzrlib/_patiencediff_c.c']))
320
 
if have_pyrex and pyrex_version_info < (0, 9, 6, 3):
321
 
    print("")
322
 
    print('Your Pyrex/Cython version %s is too old to build the simple_set' % (
323
 
        pyrex_version))
324
 
    print('and static_tuple extensions.')
325
 
    print('Please upgrade to at least Pyrex 0.9.6.3')
326
 
    print("")
327
 
    # TODO: Should this be a fatal error?
328
 
else:
329
 
    # We only need 0.9.6.3 to build _simple_set_pyx, but static_tuple depends
330
 
    # on simple_set
331
 
    add_pyrex_extension('bzrlib._simple_set_pyx')
332
 
    ext_modules.append(Extension('bzrlib._static_tuple_c',
333
 
                                 ['bzrlib/_static_tuple_c.c']))
334
 
add_pyrex_extension('bzrlib._btree_serializer_pyx')
 
224
            ext_modules.append(Extension(module_name, [c_name]))
 
225
 
 
226
 
 
227
add_pyrex_extension('bzrlib._dirstate_helpers_c')
 
228
add_pyrex_extension('bzrlib._knit_load_data_c')
 
229
ext_modules.append(Extension('bzrlib._patiencediff_c', ['bzrlib/_patiencediff_c.c']))
335
230
 
336
231
 
337
232
if unavailable_files:
338
 
    print('C extension(s) not found:')
339
 
    print('   %s' % ('\n  '.join(unavailable_files),))
340
 
    print('The python versions will be used instead.')
341
 
    print("")
342
 
 
343
 
 
344
 
def get_tbzr_py2exe_info(includes, excludes, packages, console_targets,
345
 
                         gui_targets, data_files):
346
 
    packages.append('tbzrcommands')
347
 
 
348
 
    # ModuleFinder can't handle runtime changes to __path__, but
349
 
    # win32com uses them.  Hook this in so win32com.shell is found.
350
 
    import modulefinder
351
 
    import win32com
352
 
    import cPickle as pickle
353
 
    for p in win32com.__path__[1:]:
354
 
        modulefinder.AddPackagePath("win32com", p)
355
 
    for extra in ["win32com.shell"]:
356
 
        __import__(extra)
357
 
        m = sys.modules[extra]
358
 
        for p in m.__path__[1:]:
359
 
            modulefinder.AddPackagePath(extra, p)
360
 
 
361
 
    # TBZR points to the TBZR directory
362
 
    tbzr_root = os.environ["TBZR"]
363
 
 
364
 
    # Ensure tbzrlib itself is on sys.path
365
 
    sys.path.append(tbzr_root)
366
 
 
367
 
    packages.append("tbzrlib")
368
 
 
369
 
    # collect up our icons.
370
 
    cwd = os.getcwd()
371
 
    ico_root = os.path.join(tbzr_root, 'tbzrlib', 'resources')
372
 
    icos = [] # list of (path_root, relative_ico_path)
373
 
    # First always bzr's icon and its in the root of the bzr tree.
374
 
    icos.append(('', 'bzr.ico'))
375
 
    for root, dirs, files in os.walk(ico_root):
376
 
        icos.extend([(ico_root, os.path.join(root, f)[len(ico_root)+1:])
377
 
                     for f in files if f.endswith('.ico')])
378
 
    # allocate an icon ID for each file and the full path to the ico
379
 
    icon_resources = [(rid, os.path.join(ico_dir, ico_name))
380
 
                      for rid, (ico_dir, ico_name) in enumerate(icos)]
381
 
    # create a string resource with the mapping.  Might as well save the
382
 
    # runtime some effort and write a pickle.
383
 
    # Runtime expects unicode objects with forward-slash seps.
384
 
    fse = sys.getfilesystemencoding()
385
 
    map_items = [(f.replace('\\', '/').decode(fse), rid)
386
 
                 for rid, (_, f) in enumerate(icos)]
387
 
    ico_map = dict(map_items)
388
 
    # Create a new resource type of 'ICON_MAP', and use ID=1
389
 
    other_resources = [ ("ICON_MAP", 1, pickle.dumps(ico_map))]
390
 
 
391
 
    excludes.extend("""pywin pywin.dialogs pywin.dialogs.list
392
 
                       win32ui crawler.Crawler""".split())
393
 
 
394
 
    # tbzrcache executables - a "console" version for debugging and a
395
 
    # GUI version that is generally used.
396
 
    tbzrcache = dict(
397
 
        script = os.path.join(tbzr_root, "scripts", "tbzrcache.py"),
398
 
        icon_resources = icon_resources,
399
 
        other_resources = other_resources,
400
 
    )
401
 
    console_targets.append(tbzrcache)
402
 
 
403
 
    # Make a windows version which is the same except for the base name.
404
 
    tbzrcachew = tbzrcache.copy()
405
 
    tbzrcachew["dest_base"]="tbzrcachew"
406
 
    gui_targets.append(tbzrcachew)
407
 
 
408
 
    # ditto for the tbzrcommand tool
409
 
    tbzrcommand = dict(
410
 
        script = os.path.join(tbzr_root, "scripts", "tbzrcommand.py"),
411
 
        icon_resources = icon_resources,
412
 
        other_resources = other_resources,
413
 
    )
414
 
    console_targets.append(tbzrcommand)
415
 
    tbzrcommandw = tbzrcommand.copy()
416
 
    tbzrcommandw["dest_base"]="tbzrcommandw"
417
 
    gui_targets.append(tbzrcommandw)
418
 
    
419
 
    # A utility to see python output from both C++ and Python based shell
420
 
    # extensions
421
 
    tracer = dict(script=os.path.join(tbzr_root, "scripts", "tbzrtrace.py"))
422
 
    console_targets.append(tracer)
423
 
 
424
 
    # The C++ implemented shell extensions.
425
 
    dist_dir = os.path.join(tbzr_root, "shellext", "build")
426
 
    data_files.append(('', [os.path.join(dist_dir, 'tbzrshellext_x86.dll')]))
427
 
    data_files.append(('', [os.path.join(dist_dir, 'tbzrshellext_x64.dll')]))
428
 
 
429
 
 
430
 
def get_qbzr_py2exe_info(includes, excludes, packages, data_files):
431
 
    # PyQt4 itself still escapes the plugin detection code for some reason...
432
 
    includes.append('PyQt4.QtCore')
433
 
    includes.append('PyQt4.QtGui')
434
 
    includes.append('sip') # extension module required for Qt.
435
 
    packages.append('pygments') # colorizer for qbzr
436
 
    packages.append('docutils') # html formatting
437
 
    includes.append('win32event')  # for qsubprocess stuff
438
 
    # the qt binaries might not be on PATH...
439
 
    # They seem to install to a place like C:\Python25\PyQt4\*
440
 
    # Which is not the same as C:\Python25\Lib\site-packages\PyQt4
441
 
    pyqt_dir = os.path.join(sys.prefix, "PyQt4")
442
 
    pyqt_bin_dir = os.path.join(pyqt_dir, "bin")
443
 
    if os.path.isdir(pyqt_bin_dir):
444
 
        path = os.environ.get("PATH", "")
445
 
        if pyqt_bin_dir.lower() not in [p.lower() for p in path.split(os.pathsep)]:
446
 
            os.environ["PATH"] = path + os.pathsep + pyqt_bin_dir
447
 
    # also add all imageformat plugins to distribution
448
 
    # We will look in 2 places, dirname(PyQt4.__file__) and pyqt_dir
449
 
    base_dirs_to_check = []
450
 
    if os.path.isdir(pyqt_dir):
451
 
        base_dirs_to_check.append(pyqt_dir)
452
 
    try:
453
 
        import PyQt4
454
 
    except ImportError:
455
 
        pass
456
 
    else:
457
 
        pyqt4_base_dir = os.path.dirname(PyQt4.__file__)
458
 
        if pyqt4_base_dir != pyqt_dir:
459
 
            base_dirs_to_check.append(pyqt4_base_dir)
460
 
    if not base_dirs_to_check:
461
 
        log.warn("Can't find PyQt4 installation -> not including imageformat"
462
 
                 " plugins")
463
 
    else:
464
 
        files = []
465
 
        for base_dir in base_dirs_to_check:
466
 
            plug_dir = os.path.join(base_dir, 'plugins', 'imageformats')
467
 
            if os.path.isdir(plug_dir):
468
 
                for fname in os.listdir(plug_dir):
469
 
                    # Include plugin dlls, but not debugging dlls
470
 
                    fullpath = os.path.join(plug_dir, fname)
471
 
                    if fname.endswith('.dll') and not fname.endswith('d4.dll'):
472
 
                        files.append(fullpath)
473
 
        if files:
474
 
            data_files.append(('imageformats', files))
475
 
        else:
476
 
            log.warn('PyQt4 was found, but we could not find any imageformat'
477
 
                     ' plugins. Are you sure your configuration is correct?')
478
 
 
479
 
 
480
 
def get_svn_py2exe_info(includes, excludes, packages):
481
 
    packages.append('subvertpy')
482
 
    packages.append('sqlite3')
483
 
 
484
 
 
485
 
def get_fastimport_py2exe_info(includes, excludes, packages):
486
 
    # This is the python-fastimport package, not to be confused with the
487
 
    # bzr-fastimport plugin.
488
 
    packages.append('fastimport')
 
233
    print 'C extension(s) not found:'
 
234
    print '   %s' % ('\n  '.join(unavailable_files),)
 
235
    print 'The python versions will be used instead.'
 
236
    print
489
237
 
490
238
 
491
239
if 'bdist_wininst' in sys.argv:
512
260
            # help pages
513
261
            'data_files': find_docs(),
514
262
            # for building pyrex extensions
515
 
            'cmdclass': command_classes,
 
263
            'cmdclass': {'build_ext': build_ext_if_possible},
516
264
           }
517
265
 
518
266
    ARGS.update(META_INFO)
519
267
    ARGS.update(BZRLIB)
520
 
    PKG_DATA['package_data']['bzrlib'].append('locale/*/LC_MESSAGES/*.mo')
521
268
    ARGS.update(PKG_DATA)
522
 
 
 
269
    
523
270
    setup(**ARGS)
524
271
 
525
272
elif 'py2exe' in sys.argv:
 
273
    import glob
526
274
    # py2exe setup
527
275
    import py2exe
528
276
 
538
286
        version_number.append(str(i))
539
287
    version_str = '.'.join(version_number)
540
288
 
541
 
    # An override to install_data used only by py2exe builds, which arranges
542
 
    # to byte-compile any .py files in data_files (eg, our plugins)
543
 
    # Necessary as we can't rely on the user having the relevant permissions
544
 
    # to the "Program Files" directory to generate them on the fly.
545
 
    class install_data_with_bytecompile(install_data):
546
 
        def run(self):
547
 
            from distutils.util import byte_compile
548
 
 
549
 
            install_data.run(self)
550
 
 
551
 
            py2exe = self.distribution.get_command_obj('py2exe', False)
552
 
            # GZ 2010-04-19: Setup has py2exe.optimize as 2, but give plugins
553
 
            #                time before living with docstring stripping
554
 
            optimize = 1
555
 
            compile_names = [f for f in self.outfiles if f.endswith('.py')]
556
 
            # Round mtime to nearest even second so that installing on a FAT
557
 
            # filesystem bytecode internal and script timestamps will match
558
 
            for f in compile_names:
559
 
                mtime = os.stat(f).st_mtime
560
 
                remainder = mtime % 2
561
 
                if remainder:
562
 
                    mtime -= remainder
563
 
                    os.utime(f, (mtime, mtime))
564
 
            byte_compile(compile_names,
565
 
                         optimize=optimize,
566
 
                         force=self.force, prefix=self.install_dir,
567
 
                         dry_run=self.dry_run)
568
 
            self.outfiles.extend([f + 'o' for f in compile_names])
569
 
    # end of class install_data_with_bytecompile
570
 
 
571
289
    target = py2exe.build_exe.Target(script = "bzr",
572
290
                                     dest_base = "bzr",
573
291
                                     icon_resources = [(0,'bzr.ico')],
575
293
                                     version = version_str,
576
294
                                     description = META_INFO['description'],
577
295
                                     author = META_INFO['author'],
578
 
                                     copyright = "(c) Canonical Ltd, 2005-2010",
 
296
                                     copyright = "(c) Canonical Ltd, 2005-2007",
579
297
                                     company_name = "Canonical Ltd.",
580
298
                                     comments = META_INFO['description'],
581
299
                                    )
582
 
    gui_target = copy.copy(target)
583
 
    gui_target.dest_base = "bzrw"
584
300
 
585
301
    packages = BZRLIB['packages']
586
302
    packages.remove('bzrlib')
596
312
    if sys.version.startswith('2.4'):
597
313
        # adding elementtree package
598
314
        additional_packages.add('elementtree')
599
 
    elif sys.version.startswith('2.6') or sys.version.startswith('2.5'):
 
315
    elif sys.version.startswith('2.5'):
600
316
        additional_packages.add('xml.etree')
601
317
    else:
602
318
        import warnings
603
319
        warnings.warn('Unknown Python version.\n'
604
320
                      'Please check setup.py script for compatibility.')
605
 
 
606
 
    # Although we currently can't enforce it, we consider it an error for
607
 
    # py2exe to report any files are "missing".  Such modules we know aren't
608
 
    # used should be listed here.
609
 
    excludes = """Tkinter psyco ElementPath r_hmac
610
 
                  ImaginaryModule cElementTree elementtree.ElementTree
611
 
                  Crypto.PublicKey._fastmath
612
 
                  medusa medusa.filesys medusa.ftp_server
613
 
                  tools
614
 
                  resource validate""".split()
615
 
    dll_excludes = []
616
 
 
617
321
    # email package from std python library use lazy import,
618
322
    # so we need to explicitly add all package
619
323
    additional_packages.add('email')
620
 
    # And it uses funky mappings to conver to 'Oldname' to 'newname'.  As
621
 
    # a result, packages like 'email.Parser' show as missing.  Tell py2exe
622
 
    # to exclude them.
623
 
    import email
624
 
    for oldname in getattr(email, '_LOWERNAMES', []):
625
 
        excludes.append("email." + oldname)
626
 
    for oldname in getattr(email, '_MIMENAMES', []):
627
 
        excludes.append("email.MIME" + oldname)
628
324
 
629
325
    # text files for help topis
630
326
    text_topics = glob.glob('bzrlib/help_topics/en/*.txt')
632
328
 
633
329
    # built-in plugins
634
330
    plugins_files = []
635
 
    # XXX - should we consider having the concept of an 'official' build,
636
 
    # which hard-codes the list of plugins, gets more upset if modules are
637
 
    # missing, etc?
638
 
    plugins = None # will be a set after plugin sniffing...
639
331
    for root, dirs, files in os.walk('bzrlib/plugins'):
640
 
        if root == 'bzrlib/plugins':
641
 
            plugins = set(dirs)
642
 
            # We ship plugins as normal files on the file-system - however,
643
 
            # the build process can cause *some* of these plugin files to end
644
 
            # up in library.zip. Thus, we saw (eg) "plugins/svn/test" in
645
 
            # library.zip, and then saw import errors related to that as the
646
 
            # rest of the svn plugin wasn't. So we tell py2exe to leave the
647
 
            # plugins out of the .zip file
648
 
            excludes.extend(["bzrlib.plugins." + d for d in dirs])
649
332
        x = []
650
333
        for i in files:
651
 
            # Throw away files we don't want packaged. Note that plugins may
652
 
            # have data files with all sorts of extensions so we need to
653
 
            # be conservative here about what we ditch.
654
 
            ext = os.path.splitext(i)[1]
655
 
            if ext.endswith('~') or ext in [".pyc", ".swp"]:
 
334
            if not i.endswith('.py'):
656
335
                continue
657
336
            if i == '__init__.py' and root == 'bzrlib/plugins':
658
337
                continue
666
345
    mf.run_package('bzrlib/plugins')
667
346
    packs, mods = mf.get_result()
668
347
    additional_packages.update(packs)
669
 
    includes.extend(mods)
670
 
 
671
 
    console_targets = [target,
672
 
                       'tools/win32/bzr_postinstall.py',
673
 
                       ]
674
 
    gui_targets = [gui_target]
675
 
    data_files = topics_files + plugins_files + I18N_FILES
676
 
 
677
 
    if 'qbzr' in plugins:
678
 
        get_qbzr_py2exe_info(includes, excludes, packages, data_files)
679
 
 
680
 
    if 'svn' in plugins:
681
 
        get_svn_py2exe_info(includes, excludes, packages)
682
 
 
683
 
    if 'fastimport' in plugins:
684
 
        get_fastimport_py2exe_info(includes, excludes, packages)
685
 
 
686
 
    if "TBZR" in os.environ:
687
 
        # TORTOISE_OVERLAYS_MSI_WIN32 must be set to the location of the
688
 
        # TortoiseOverlays MSI installer file. It is in the TSVN svn repo and
689
 
        # can be downloaded from (username=guest, blank password):
690
 
        # http://tortoisesvn.tigris.org/svn/tortoisesvn/TortoiseOverlays
691
 
        # look for: version-1.0.4/bin/TortoiseOverlays-1.0.4.11886-win32.msi
692
 
        # Ditto for TORTOISE_OVERLAYS_MSI_X64, pointing at *-x64.msi.
693
 
        for needed in ('TORTOISE_OVERLAYS_MSI_WIN32',
694
 
                       'TORTOISE_OVERLAYS_MSI_X64'):
695
 
            url = ('http://guest:@tortoisesvn.tigris.org/svn/tortoisesvn'
696
 
                   '/TortoiseOverlays')
697
 
            if not os.path.isfile(os.environ.get(needed, '<nofile>')):
698
 
                raise RuntimeError(
699
 
                    "\nPlease set %s to the location of the relevant"
700
 
                    "\nTortoiseOverlays .msi installer file."
701
 
                    " The installers can be found at"
702
 
                    "\n  %s"
703
 
                    "\ncheck in the version-X.Y.Z/bin/ subdir" % (needed, url))
704
 
        get_tbzr_py2exe_info(includes, excludes, packages, console_targets,
705
 
                             gui_targets, data_files)
706
 
    else:
707
 
        # print this warning to stderr as output is redirected, so it is seen
708
 
        # at build time.  Also to stdout so it appears in the log
709
 
        for f in (sys.stderr, sys.stdout):
710
 
            f.write("Skipping TBZR binaries - "
711
 
                "please set TBZR to a directory to enable\n")
712
 
 
713
 
    # MSWSOCK.dll is a system-specific library, which py2exe accidentally pulls
714
 
    # in on Vista.
715
 
    dll_excludes.extend(["MSWSOCK.dll",
716
 
                         "MSVCP60.dll",
717
 
                         "MSVCP90.dll",
718
 
                         "powrprof.dll",
719
 
                         "SHFOLDER.dll"])
 
348
 
720
349
    options_list = {"py2exe": {"packages": packages + list(additional_packages),
721
 
                               "includes": includes,
722
 
                               "excludes": excludes,
723
 
                               "dll_excludes": dll_excludes,
 
350
                               "includes": includes + mods,
 
351
                               "excludes": ["Tkinter", "medusa", "tools"],
724
352
                               "dist_dir": "win32_bzr.exe",
725
 
                               "optimize": 2,
726
 
                               "custom_boot_script":
727
 
                                        "tools/win32/py2exe_boot_common.py",
728
353
                              },
729
354
                   }
730
 
 
731
 
    # We want the libaray.zip to have optimize = 2, but the exe to have
732
 
    # optimize = 1, so that .py files that get compilied at run time
733
 
    # (e.g. user installed plugins) dont have their doc strings removed.
734
 
    class py2exe_no_oo_exe(py2exe.build_exe.py2exe):
735
 
        def build_executable(self, *args, **kwargs):
736
 
            self.optimize = 1
737
 
            py2exe.build_exe.py2exe.build_executable(self, *args, **kwargs)
738
 
            self.optimize = 2
739
 
 
740
 
    if __name__ == '__main__':
741
 
        command_classes['install_data'] = install_data_with_bytecompile
742
 
        command_classes['py2exe'] = py2exe_no_oo_exe
743
 
        setup(options=options_list,
744
 
              console=console_targets,
745
 
              windows=gui_targets,
746
 
              zipfile='lib/library.zip',
747
 
              data_files=data_files,
748
 
              cmdclass=command_classes,
749
 
              )
 
355
    setup(options=options_list,
 
356
          console=[target,
 
357
                   'tools/win32/bzr_postinstall.py',
 
358
                  ],
 
359
          zipfile='lib/library.zip',
 
360
          data_files=topics_files + plugins_files,
 
361
          )
750
362
 
751
363
else:
752
364
    # ad-hoc for easy_install
753
365
    DATA_FILES = []
754
366
    if not 'bdist_egg' in sys.argv:
755
 
        # generate and install bzr.1 only with plain install, not the
756
 
        # easy_install one
 
367
        # generate and install bzr.1 only with plain install, not easy_install one
757
368
        DATA_FILES = [('man/man1', ['bzr.1'])]
758
369
 
759
 
    DATA_FILES = DATA_FILES + I18N_FILES
760
370
    # std setup
761
371
    ARGS = {'scripts': ['bzr'],
762
372
            'data_files': DATA_FILES,
768
378
    ARGS.update(BZRLIB)
769
379
    ARGS.update(PKG_DATA)
770
380
 
771
 
    if __name__ == '__main__':
772
 
        setup(**ARGS)
 
381
    setup(**ARGS)