← 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: 2007-07-22 18:09:04 UTC
  • mfrom: (2485.8.63 bzr.connection.sharing)
  • Revision ID: pqm@pqm.ubuntu.com-20070722180904-wy7y7oyi32wbghgf
Transport connection sharing

Show diffs side-by-side

added added

removed removed

Lines of Context:
setup.py
148
148
########################
149
149
 
150
150
command_classes = {'install_scripts': my_install_scripts,
151
 
                  'build': bzr_build}
 
151
                   'build': bzr_build}
 
152
from distutils.extension import Extension
152
153
ext_modules = []
153
154
try:
154
155
    from Pyrex.Distutils import build_ext
155
156
except ImportError:
 
157
    have_pyrex = False
156
158
    # 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
    print
 
160
    print ("The python package 'Pyrex' is not available."
 
161
           " If the .c files are available,")
 
162
    print ("they will be built,"
 
163
           " but modifying the .pyx files will not rebuild them.")
 
164
    print
159
165
    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
166
else:
164
 
    from distutils.extension import Extension
165
 
    #ext_modules.append(
166
 
    #    Extension("bzrlib.modulename", ["bzrlib/foo.pyx"], libraries = []))
 
167
    have_pyrex = True
 
168
# Override the build_ext if we have Pyrex available
167
169
command_classes['build_ext'] = build_ext
 
170
unavailable_files = []
 
171
 
 
172
 
 
173
def add_pyrex_extension(module_name, **kwargs):
 
174
    """Add a pyrex module to build.
 
175
 
 
176
    This will use Pyrex to auto-generate the .c file if it is available.
 
177
    Otherwise it will fall back on the .c file. If the .c file is not
 
178
    available, it will warn, and not add anything.
 
179
 
 
180
    You can pass any extra options to Extension through kwargs. One example is
 
181
    'libraries = []'.
 
182
 
 
183
    :param module_name: The python path to the module. This will be used to
 
184
        determine the .pyx and .c files to use.
 
185
    """
 
186
    path = module_name.replace('.', '/')
 
187
    pyrex_name = path + '.pyx'
 
188
    c_name = path + '.c'
 
189
    if have_pyrex:
 
190
        ext_modules.append(Extension(module_name, [pyrex_name]))
 
191
    else:
 
192
        if not os.path.isfile(c_name):
 
193
            unavailable_files.append(c_name)
 
194
        else:
 
195
            ext_modules.append(Extension(module_name, [c_name]))
 
196
 
 
197
 
 
198
add_pyrex_extension('bzrlib._dirstate_helpers_c')
 
199
add_pyrex_extension('bzrlib._knit_load_data_c')
 
200
 
 
201
 
 
202
if unavailable_files:
 
203
    print 'C extension(s) not found:'
 
204
    print '   %s' % ('\n  '.join(unavailable_files),)
 
205
    print 'The python versions will be used instead.'
 
206
    print
 
207
 
168
208
 
169
209
if 'bdist_wininst' in sys.argv:
170
210
    import glob
171
211
    # doc files
172
212
    docs = glob.glob('doc/*.htm') + ['doc/default.css']
 
213
    dev_docs = glob.glob('doc/developers/*.htm')
173
214
    # python's distutils-based win32 installer
174
215
    ARGS = {'scripts': ['bzr', 'tools/win32/bzr-win32-bdist-postinstall.py'],
 
216
            'ext_modules': ext_modules,
175
217
            # help pages
176
 
            'data_files': [('Doc/Bazaar', docs)],
 
218
            'data_files': [('Doc/Bazaar', docs),
 
219
                           ('Doc/Bazaar/developers', dev_docs),
 
220
                          ],
 
221
            # for building pyrex extensions
 
222
            'cmdclass': {'build_ext': build_ext},
177
223
           }
178
224
 
179
225
    ARGS.update(META_INFO)
220
266
        import warnings
221
267
        warnings.warn('Unknown Python version.\n'
222
268
                      'Please check setup.py script for compatibility.')
 
269
    # email package from std python library use lazy import,
 
270
    # so we need to explicitly add all package
 
271
    additional_packages.append('email')
223
272
 
224
273
    options_list = {"py2exe": {"packages": BZRLIB['packages'] +
225
274
                                           additional_packages,
226
 
                               "excludes": ["Tkinter", "medusa"],
 
275
                               "excludes": ["Tkinter", "medusa", "tools"],
227
276
                               "dist_dir": "win32_bzr.exe",
228
277
                              },
229
278
                   }