~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to setup.py

  • Committer: Martin
  • Date: 2010-05-03 20:57:39 UTC
  • mto: This revision was merged to the branch mainline in revision 5204.
  • Revision ID: gzlist@googlemail.com-20100503205739-n326zdvevv0rmruh
Retain original stack and error message when translating to ValueError in bencode

Show diffs side-by-side

added added

removed removed

Lines of Context:
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
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
68
                                        'tests/ssl_certs/server_without_pass.key',
72
69
                                        'tests/ssl_certs/server_with_pass.key',
73
 
                                        'tests/ssl_certs/server.crt',
 
70
                                        'tests/ssl_certs/server.crt'
74
71
                                       ]},
75
72
           }
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]))
 
73
 
81
74
 
82
75
def get_bzrlib_packages():
83
76
    """Recurse through the bzrlib directory, and extract the package names"""
105
98
 
106
99
from distutils import log
107
100
from distutils.core import setup
108
 
from distutils.version import LooseVersion
109
101
from distutils.command.install_scripts import install_scripts
110
102
from distutils.command.install_data import install_data
111
103
from distutils.command.build import build
133
125
                f = file(batch_path, "w")
134
126
                f.write(batch_str)
135
127
                f.close()
136
 
                print("Created: %s" % batch_path)
137
 
            except Exception:
138
 
                e = sys.exc_info()[1]
139
 
                print("ERROR: Unable to create %s: %s" % (batch_path, e))
 
128
                print "Created:", batch_path
 
129
            except Exception, e:
 
130
                print "ERROR: Unable to create %s: %s" % (batch_path, e)
140
131
 
141
132
    def _quoted_path(self, path):
142
133
        if ' ' in path:
158
149
    Generate bzr.1.
159
150
    """
160
151
 
161
 
    sub_commands = build.sub_commands + [
162
 
            ('build_mo', lambda _: True),
163
 
            ]
164
 
 
165
152
    def run(self):
166
153
        build.run(self)
167
154
 
173
160
## Setup
174
161
########################
175
162
 
176
 
from bzrlib.bzr_distutils import build_mo
177
 
 
178
163
command_classes = {'install_scripts': my_install_scripts,
179
 
                   'build': bzr_build,
180
 
                   'build_mo': build_mo,
181
 
                   }
 
164
                   'build': bzr_build}
182
165
from distutils import log
183
166
from distutils.errors import CCompilerError, DistutilsPlatformError
184
167
from distutils.extension import Extension
185
168
ext_modules = []
186
169
try:
187
170
    try:
 
171
        from Pyrex.Distutils import build_ext
 
172
        from Pyrex.Compiler.Version import version as pyrex_version
 
173
    except ImportError:
 
174
        print "No Pyrex, trying Cython..."
188
175
        from Cython.Distutils import build_ext
189
176
        from Cython.Compiler.Version import version as pyrex_version
190
 
    except ImportError:
191
 
        print("No Cython, trying Pyrex...")
192
 
        from Pyrex.Distutils import build_ext
193
 
        from Pyrex.Compiler.Version import version as pyrex_version
194
177
except ImportError:
195
178
    have_pyrex = False
196
179
    # try to build the extension from the prior generated source.
197
 
    print("")
198
 
    print("The python package 'Pyrex' is not available."
199
 
          " If the .c files are available,")
200
 
    print("they will be built,"
201
 
          " but modifying the .pyx files will not rebuild them.")
202
 
    print("")
 
180
    print
 
181
    print ("The python package 'Pyrex' is not available."
 
182
           " If the .c files are available,")
 
183
    print ("they will be built,"
 
184
           " but modifying the .pyx files will not rebuild them.")
 
185
    print
203
186
    from distutils.command.build_ext import build_ext
204
187
else:
205
188
    have_pyrex = True
206
 
    pyrex_version_info = LooseVersion(pyrex_version)
 
189
    pyrex_version_info = tuple(map(int, pyrex_version.split('.')))
207
190
 
208
191
 
209
192
class build_ext_if_possible(build_ext):
221
204
    def run(self):
222
205
        try:
223
206
            build_ext.run(self)
224
 
        except DistutilsPlatformError:
225
 
            e = sys.exc_info()[1]
 
207
        except DistutilsPlatformError, e:
226
208
            if not self.allow_python_fallback:
227
209
                log.warn('\n  Cannot build extensions.\n'
228
210
                         '  Use "build_ext --allow-python-fallback" to use'
299
281
    add_pyrex_extension('bzrlib._dirstate_helpers_pyx',
300
282
                        libraries=['Ws2_32'])
301
283
    add_pyrex_extension('bzrlib._walkdirs_win32')
 
284
    z_lib = 'zdll'
302
285
else:
303
 
    if have_pyrex and pyrex_version_info == LooseVersion("0.9.4.1"):
 
286
    if have_pyrex and pyrex_version_info[:3] == (0,9,4):
304
287
        # Pyrex 0.9.4.1 fails to compile this extension correctly
305
288
        # The code it generates re-uses a "local" pointer and
306
289
        # calls "PY_DECREF" after having set it to NULL. (It mixes PY_XDECREF
307
290
        # which is NULL safe with PY_DECREF which is not.)
308
 
        # <https://bugs.launchpad.net/bzr/+bug/449372>
309
 
        # <https://bugs.launchpad.net/bzr/+bug/276868>
310
 
        print('Cannot build extension "bzrlib._dirstate_helpers_pyx" using')
311
 
        print('your version of pyrex "%s". Please upgrade your pyrex'
312
 
              % (pyrex_version,))
313
 
        print('install. For now, the non-compiled (python) version will')
314
 
        print('be used instead.')
 
291
        # <https://bugs.edge.launchpad.net/bzr/+bug/449372>
 
292
        # <https://bugs.edge.launchpad.net/bzr/+bug/276868>
 
293
        print 'Cannot build extension "bzrlib._dirstate_helpers_pyx" using'
 
294
        print 'your version of pyrex "%s". Please upgrade your pyrex' % (
 
295
            pyrex_version,)
 
296
        print 'install. For now, the non-compiled (python) version will'
 
297
        print 'be used instead.'
315
298
    else:
316
299
        add_pyrex_extension('bzrlib._dirstate_helpers_pyx')
317
300
    add_pyrex_extension('bzrlib._readdir_pyx')
318
 
add_pyrex_extension('bzrlib._chk_map_pyx')
 
301
    z_lib = 'z'
 
302
add_pyrex_extension('bzrlib._chk_map_pyx', libraries=[z_lib])
319
303
ext_modules.append(Extension('bzrlib._patiencediff_c',
320
304
                             ['bzrlib/_patiencediff_c.c']))
321
 
if have_pyrex and pyrex_version_info < LooseVersion("0.9.6.3"):
322
 
    print("")
323
 
    print('Your Pyrex/Cython version %s is too old to build the simple_set' % (
324
 
        pyrex_version))
325
 
    print('and static_tuple extensions.')
326
 
    print('Please upgrade to at least Pyrex 0.9.6.3')
327
 
    print("")
 
305
if have_pyrex and pyrex_version_info < (0, 9, 6, 3):
 
306
    print
 
307
    print 'Your Pyrex/Cython version %s is too old to build the simple_set' % (
 
308
        pyrex_version)
 
309
    print 'and static_tuple extensions.'
 
310
    print 'Please upgrade to at least Pyrex 0.9.6.3'
 
311
    print
328
312
    # TODO: Should this be a fatal error?
329
313
else:
330
314
    # We only need 0.9.6.3 to build _simple_set_pyx, but static_tuple depends
336
320
 
337
321
 
338
322
if unavailable_files:
339
 
    print('C extension(s) not found:')
340
 
    print('   %s' % ('\n  '.join(unavailable_files),))
341
 
    print('The python versions will be used instead.')
342
 
    print("")
 
323
    print 'C extension(s) not found:'
 
324
    print '   %s' % ('\n  '.join(unavailable_files),)
 
325
    print 'The python versions will be used instead.'
 
326
    print
343
327
 
344
328
 
345
329
def get_tbzr_py2exe_info(includes, excludes, packages, console_targets,
409
393
    # ditto for the tbzrcommand tool
410
394
    tbzrcommand = dict(
411
395
        script = os.path.join(tbzr_root, "scripts", "tbzrcommand.py"),
412
 
        icon_resources = icon_resources,
413
 
        other_resources = other_resources,
 
396
        icon_resources = [(0,'bzr.ico')],
414
397
    )
415
398
    console_targets.append(tbzrcommand)
416
399
    tbzrcommandw = tbzrcommand.copy()
430
413
 
431
414
def get_qbzr_py2exe_info(includes, excludes, packages, data_files):
432
415
    # PyQt4 itself still escapes the plugin detection code for some reason...
433
 
    includes.append('PyQt4.QtCore')
434
 
    includes.append('PyQt4.QtGui')
435
 
    includes.append('PyQt4.QtTest')
 
416
    packages.append('PyQt4')
 
417
    excludes.append('PyQt4.elementtree.ElementTree')
 
418
    excludes.append('PyQt4.uic.port_v3')
436
419
    includes.append('sip') # extension module required for Qt.
437
420
    packages.append('pygments') # colorizer for qbzr
438
421
    packages.append('docutils') # html formatting
439
422
    includes.append('win32event')  # for qsubprocess stuff
 
423
    # but we can avoid many Qt4 Dlls.
 
424
    dll_excludes.extend(
 
425
        """QtAssistantClient4.dll QtCLucene4.dll QtDesigner4.dll
 
426
        QtHelp4.dll QtNetwork4.dll QtOpenGL4.dll QtScript4.dll
 
427
        QtSql4.dll QtTest4.dll QtWebKit4.dll QtXml4.dll
 
428
        qscintilla2.dll""".split())
440
429
    # the qt binaries might not be on PATH...
441
430
    # They seem to install to a place like C:\Python25\PyQt4\*
442
431
    # Which is not the same as C:\Python25\Lib\site-packages\PyQt4
484
473
    packages.append('sqlite3')
485
474
 
486
475
 
487
 
def get_git_py2exe_info(includes, excludes, packages):
488
 
    packages.append('dulwich')
489
 
 
490
 
 
491
 
def get_fastimport_py2exe_info(includes, excludes, packages):
492
 
    # This is the python-fastimport package, not to be confused with the
493
 
    # bzr-fastimport plugin.
494
 
    packages.append('fastimport')
495
 
 
496
 
 
497
476
if 'bdist_wininst' in sys.argv:
498
477
    def find_docs():
499
478
        docs = []
518
497
            # help pages
519
498
            'data_files': find_docs(),
520
499
            # for building pyrex extensions
521
 
            'cmdclass': command_classes,
 
500
            'cmdclass': {'build_ext': build_ext_if_possible},
522
501
           }
523
502
 
524
503
    ARGS.update(META_INFO)
525
504
    ARGS.update(BZRLIB)
526
 
    PKG_DATA['package_data']['bzrlib'].append('locale/*/LC_MESSAGES/*.mo')
527
505
    ARGS.update(PKG_DATA)
528
 
 
 
506
    
529
507
    setup(**ARGS)
530
508
 
531
509
elif 'py2exe' in sys.argv:
 
510
    import glob
532
511
    # py2exe setup
533
512
    import py2exe
534
513
 
555
534
            install_data.run(self)
556
535
 
557
536
            py2exe = self.distribution.get_command_obj('py2exe', False)
558
 
            # GZ 2010-04-19: Setup has py2exe.optimize as 2, but give plugins
559
 
            #                time before living with docstring stripping
560
 
            optimize = 1
 
537
            optimize = py2exe.optimize
561
538
            compile_names = [f for f in self.outfiles if f.endswith('.py')]
562
 
            # Round mtime to nearest even second so that installing on a FAT
563
 
            # filesystem bytecode internal and script timestamps will match
564
 
            for f in compile_names:
565
 
                mtime = os.stat(f).st_mtime
566
 
                remainder = mtime % 2
567
 
                if remainder:
568
 
                    mtime -= remainder
569
 
                    os.utime(f, (mtime, mtime))
570
539
            byte_compile(compile_names,
571
540
                         optimize=optimize,
572
541
                         force=self.force, prefix=self.install_dir,
573
542
                         dry_run=self.dry_run)
574
 
            self.outfiles.extend([f + 'o' for f in compile_names])
 
543
            if optimize:
 
544
                suffix = 'o'
 
545
            else:
 
546
                suffix = 'c'
 
547
            self.outfiles.extend([f + suffix for f in compile_names])
575
548
    # end of class install_data_with_bytecompile
576
549
 
577
550
    target = py2exe.build_exe.Target(script = "bzr",
585
558
                                     company_name = "Canonical Ltd.",
586
559
                                     comments = META_INFO['description'],
587
560
                                    )
588
 
    gui_target = copy.copy(target)
589
 
    gui_target.dest_base = "bzrw"
590
561
 
591
562
    packages = BZRLIB['packages']
592
563
    packages.remove('bzrlib')
677
648
    console_targets = [target,
678
649
                       'tools/win32/bzr_postinstall.py',
679
650
                       ]
680
 
    gui_targets = [gui_target]
681
 
    data_files = topics_files + plugins_files + I18N_FILES
 
651
    gui_targets = []
 
652
    data_files = topics_files + plugins_files
682
653
 
683
654
    if 'qbzr' in plugins:
684
655
        get_qbzr_py2exe_info(includes, excludes, packages, data_files)
686
657
    if 'svn' in plugins:
687
658
        get_svn_py2exe_info(includes, excludes, packages)
688
659
 
689
 
    if 'git' in plugins:
690
 
        get_git_py2exe_info(includes, excludes, packages)
691
 
 
692
 
    if 'fastimport' in plugins:
693
 
        get_fastimport_py2exe_info(includes, excludes, packages)
694
 
 
695
660
    if "TBZR" in os.environ:
696
661
        # TORTOISE_OVERLAYS_MSI_WIN32 must be set to the location of the
697
662
        # TortoiseOverlays MSI installer file. It is in the TSVN svn repo and
716
681
        # print this warning to stderr as output is redirected, so it is seen
717
682
        # at build time.  Also to stdout so it appears in the log
718
683
        for f in (sys.stderr, sys.stdout):
719
 
            f.write("Skipping TBZR binaries - "
720
 
                "please set TBZR to a directory to enable\n")
 
684
            print >> f, \
 
685
                "Skipping TBZR binaries - please set TBZR to a directory to enable"
721
686
 
722
687
    # MSWSOCK.dll is a system-specific library, which py2exe accidentally pulls
723
688
    # in on Vista.
724
 
    dll_excludes.extend(["MSWSOCK.dll",
725
 
                         "MSVCP60.dll",
726
 
                         "MSVCP90.dll",
727
 
                         "powrprof.dll",
728
 
                         "SHFOLDER.dll"])
 
689
    dll_excludes.extend(["MSWSOCK.dll", "MSVCP60.dll", "powrprof.dll"])
729
690
    options_list = {"py2exe": {"packages": packages + list(additional_packages),
730
691
                               "includes": includes,
731
692
                               "excludes": excludes,
732
693
                               "dll_excludes": dll_excludes,
733
694
                               "dist_dir": "win32_bzr.exe",
734
 
                               "optimize": 2,
735
 
                               "custom_boot_script":
736
 
                                        "tools/win32/py2exe_boot_common.py",
 
695
                               "optimize": 1,
737
696
                              },
738
697
                   }
739
698
 
740
 
    # We want the libaray.zip to have optimize = 2, but the exe to have
741
 
    # optimize = 1, so that .py files that get compilied at run time
742
 
    # (e.g. user installed plugins) dont have their doc strings removed.
743
 
    class py2exe_no_oo_exe(py2exe.build_exe.py2exe):
744
 
        def build_executable(self, *args, **kwargs):
745
 
            self.optimize = 1
746
 
            py2exe.build_exe.py2exe.build_executable(self, *args, **kwargs)
747
 
            self.optimize = 2
748
 
 
749
 
    if __name__ == '__main__':
750
 
        command_classes['install_data'] = install_data_with_bytecompile
751
 
        command_classes['py2exe'] = py2exe_no_oo_exe
752
 
        setup(options=options_list,
753
 
              console=console_targets,
754
 
              windows=gui_targets,
755
 
              zipfile='lib/library.zip',
756
 
              data_files=data_files,
757
 
              cmdclass=command_classes,
758
 
              )
 
699
    setup(options=options_list,
 
700
          console=console_targets,
 
701
          windows=gui_targets,
 
702
          zipfile='lib/library.zip',
 
703
          data_files=data_files,
 
704
          cmdclass={'install_data': install_data_with_bytecompile},
 
705
          )
759
706
 
760
707
else:
761
708
    # ad-hoc for easy_install
765
712
        # easy_install one
766
713
        DATA_FILES = [('man/man1', ['bzr.1'])]
767
714
 
768
 
    DATA_FILES = DATA_FILES + I18N_FILES
 
715
    if sys.platform != 'win32':
 
716
        # see https://wiki.kubuntu.org/Apport/DeveloperHowTo
 
717
        #
 
718
        # checking the paths and hardcoding the check for root is a bit gross,
 
719
        # but I don't see a cleaner way to find out the locations in a way
 
720
        # that's going to align with the hardcoded paths in apport.
 
721
        if os.geteuid() == 0:
 
722
            DATA_FILES += [
 
723
                ('/usr/share/apport/package-hooks',
 
724
                    ['apport/source_bzr.py']),
 
725
                ('/etc/apport/crashdb.conf.d/',
 
726
                    ['apport/bzr-crashdb.conf']),]
 
727
 
769
728
    # std setup
770
729
    ARGS = {'scripts': ['bzr'],
771
730
            'data_files': DATA_FILES,
777
736
    ARGS.update(BZRLIB)
778
737
    ARGS.update(PKG_DATA)
779
738
 
780
 
    if __name__ == '__main__':
781
 
        setup(**ARGS)
 
739
    setup(**ARGS)