← Back to branch summary

~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to setup.py

  • Committer: Alexander Belchenko
  • Date: 2007-07-03 17:12:17 UTC
  • mto: This revision was merged to the branch mainline in revision 2582.
  • Revision ID: bialix@ukr.net-20070703171217-8vypfjd2uso3air0
Build pyrex/C extensions for bzr.exe

Show diffs side-by-side

added added

removed removed

Lines of Context:
setup.py
1
1
#! /usr/bin/env python
2
2
 
3
 
# This is an installation script for bzr.  Run it with
4
 
# './setup.py install', or
5
 
# './setup.py --help' for more options
 
3
"""Installation script for bzr.
 
4
Run it with
 
5
 './setup.py install', or
 
6
 './setup.py --help' for more options
 
7
"""
 
8
 
 
9
import os
 
10
import sys
 
11
 
 
12
import bzrlib
 
13
 
 
14
##
 
15
# 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
            }
 
25
 
 
26
# The list of packages is automatically generated later. Add other things
 
27
# that are part of BZRLIB here.
 
28
BZRLIB = {}
 
29
 
 
30
PKG_DATA = {# install files from selftest suite
 
31
            'package_data': {'bzrlib': ['doc/api/*.txt',
 
32
                                        'tests/test_patches_data/*',
 
33
                                       ]},
 
34
           }
 
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
 
 
64
 
 
65
def get_bzrlib_packages():
 
66
    """Recurse through the bzrlib directory, and extract the package names"""
 
67
 
 
68
    packages = []
 
69
    base_path = os.path.dirname(os.path.abspath(bzrlib.__file__))
 
70
    for root, dirs, files in os.walk(base_path):
 
71
        if '__init__.py' in files:
 
72
            assert root.startswith(base_path)
 
73
            # Get just the path below bzrlib
 
74
            package_path = root[len(base_path):]
 
75
            # Remove leading and trailing slashes
 
76
            package_path = package_path.strip('\\/')
 
77
            if not package_path:
 
78
                package_name = 'bzrlib'
 
79
            else:
 
80
                package_name = ('bzrlib.' +
 
81
                            package_path.replace('/', '.').replace('\\', '.'))
 
82
            packages.append(package_name)
 
83
    return sorted(packages)
 
84
 
 
85
 
 
86
BZRLIB['packages'] = get_bzrlib_packages()
 
87
 
6
88
 
7
89
from distutils.core import setup
8
 
 
9
 
setup(name='bzr',
10
 
      version='0.0.0',
11
 
      author='Martin Pool',
12
 
      author_email='mbp@sourcefrog.net',
13
 
      url='http://www.bazaar-ng.org/',
14
 
      description='Friendly distributed version control system',
15
 
      license='GNU GPL v2',
16
 
      packages=['bzrlib'],
17
 
      scripts=['bzr'])
 
90
from distutils.command.install_scripts import install_scripts
 
91
from distutils.command.build import build
 
92
 
 
93
###############################
 
94
# Overridden distutils actions
 
95
###############################
 
96
 
 
97
class my_install_scripts(install_scripts):
 
98
    """ Customized install_scripts distutils action.
 
99
    Create bzr.bat for win32.
 
100
    """
 
101
    def run(self):
 
102
        install_scripts.run(self)   # standard action
 
103
 
 
104
        if sys.platform == "win32":
 
105
            try:
 
106
                scripts_dir = self.install_dir
 
107
                script_path = self._quoted_path(os.path.join(scripts_dir,
 
108
                                                             "bzr"))
 
109
                python_exe = self._quoted_path(sys.executable)
 
110
                args = self._win_batch_args()
 
111
                batch_str = "@%s %s %s" % (python_exe, script_path, args)
 
112
                batch_path = script_path + ".bat"
 
113
                f = file(batch_path, "w")
 
114
                f.write(batch_str)
 
115
                f.close()
 
116
                print "Created:", batch_path
 
117
            except Exception, e:
 
118
                print "ERROR: Unable to create %s: %s" % (batch_path, e)
 
119
 
 
120
    def _quoted_path(self, path):
 
121
        if ' ' in path:
 
122
            return '"' + path + '"'
 
123
        else:
 
124
            return path
 
125
 
 
126
    def _win_batch_args(self):
 
127
        from bzrlib.win32utils import winver
 
128
        if winver == 'Windows NT':
 
129
            return '%*'
 
130
        else:
 
131
            return '%1 %2 %3 %4 %5 %6 %7 %8 %9'
 
132
#/class my_install_scripts
 
133
 
 
134
 
 
135
class bzr_build(build):
 
136
    """Customized build distutils action.
 
137
    Generate bzr.1.
 
138
    """
 
139
    def run(self):
 
140
        build.run(self)
 
141
 
 
142
        import generate_docs
 
143
        generate_docs.main(argv=["bzr", "man"])
 
144
 
 
145
 
 
146
########################
 
147
## Setup
 
148
########################
 
149
 
 
150
command_classes = {'install_scripts': my_install_scripts,
 
151
                   'build': bzr_build}
 
152
ext_modules = []
 
153
try:
 
154
    from Pyrex.Distutils import build_ext
 
155
except ImportError:
 
156
    # try to build the extension from the prior generated source.
 
157
    print ("Pyrex not available, while bzr will build, "
 
158
           "you cannot modify the C extensions.")
 
159
    from distutils.command.build_ext import build_ext
 
160
    from distutils.extension import Extension
 
161
    #ext_modules.append(
 
162
    #    Extension("bzrlib.modulename", ["bzrlib/foo.c"], libraries = []))
 
163
else:
 
164
    from distutils.extension import Extension
 
165
    #ext_modules.append(
 
166
    #    Extension("bzrlib.modulename", ["bzrlib/foo.pyx"], libraries = []))
 
167
command_classes['build_ext'] = build_ext
 
168
 
 
169
if 'bdist_wininst' in sys.argv:
 
170
    import glob
 
171
    # doc files
 
172
    docs = glob.glob('doc/*.htm') + ['doc/default.css']
 
173
    dev_docs = glob.glob('doc/developers/*.htm')
 
174
    # python's distutils-based win32 installer
 
175
    ARGS = {'scripts': ['bzr', 'tools/win32/bzr-win32-bdist-postinstall.py'],
 
176
            'ext_modules': ext_modules,
 
177
            # help pages
 
178
            'data_files': [('Doc/Bazaar', docs),
 
179
                           ('Doc/Bazaar/developers', dev_docs),
 
180
                          ],
 
181
            # for building pyrex extensions
 
182
            'cmdclass': {'build_ext': build_ext},
 
183
           }
 
184
 
 
185
    ARGS.update(META_INFO)
 
186
    ARGS.update(BZRLIB)
 
187
    ARGS.update(PKG_DATA)
 
188
    
 
189
    setup(**ARGS)
 
190
 
 
191
elif 'py2exe' in sys.argv:
 
192
    # py2exe setup
 
193
    import py2exe
 
194
 
 
195
    # pick real bzr version
 
196
    import bzrlib
 
197
 
 
198
    version_number = []
 
199
    for i in bzrlib.version_info[:4]:
 
200
        try:
 
201
            i = int(i)
 
202
        except ValueError:
 
203
            i = 0
 
204
        version_number.append(str(i))
 
205
    version_str = '.'.join(version_number)
 
206
 
 
207
    target = py2exe.build_exe.Target(script = "bzr",
 
208
                                     dest_base = "bzr",
 
209
                                     icon_resources = [(0,'bzr.ico')],
 
210
                                     name = META_INFO['name'],
 
211
                                     version = version_str,
 
212
                                     description = META_INFO['description'],
 
213
                                     author = META_INFO['author'],
 
214
                                     copyright = "(c) Canonical Ltd, 2005-2007",
 
215
                                     company_name = "Canonical Ltd.",
 
216
                                     comments = META_INFO['description'],
 
217
                                    )
 
218
 
 
219
    additional_packages =  []
 
220
    if sys.version.startswith('2.4'):
 
221
        # adding elementtree package
 
222
        additional_packages.append('elementtree')
 
223
    elif sys.version.startswith('2.5'):
 
224
        additional_packages.append('xml.etree')
 
225
    else:
 
226
        import warnings
 
227
        warnings.warn('Unknown Python version.\n'
 
228
                      'Please check setup.py script for compatibility.')
 
229
    # email package from std python library use lazy import,
 
230
    # so we need to explicitly add all package
 
231
    additional_packages.append('email')
 
232
 
 
233
    options_list = {"py2exe": {"packages": BZRLIB['packages'] +
 
234
                                           additional_packages,
 
235
                               "excludes": ["Tkinter", "medusa", "tools"],
 
236
                               "dist_dir": "win32_bzr.exe",
 
237
                              },
 
238
                   }
 
239
    setup(options=options_list,
 
240
          console=[target,
 
241
                   'tools/win32/bzr_postinstall.py',
 
242
                  ],
 
243
          zipfile='lib/library.zip')
 
244
 
 
245
else:
 
246
    # std setup
 
247
    ARGS = {'scripts': ['bzr'],
 
248
            'data_files': [('man/man1', ['bzr.1'])],
 
249
            'cmdclass': command_classes,
 
250
            'ext_modules': ext_modules,
 
251
           }
 
252
    
 
253
    ARGS.update(META_INFO)
 
254
    ARGS.update(BZRLIB)
 
255
    ARGS.update(PKG_DATA)
 
256
 
 
257
    setup(**ARGS)