← 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: 2010-11-06 16:01:48 UTC
  • mfrom: (5527.1.1 trunk)
  • Revision ID: pqm@pqm.ubuntu.com-20101106160148-8maemz21jbrhpzky
(vila) Move NEWS entry into the correct section (Vincent Ladeuil)

Show diffs side-by-side

added added

removed removed

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