175
180
from distutils.command.build_ext import build_ext
177
182
have_pyrex = True
183
from Pyrex.Compiler.Version import version as pyrex_version
180
186
class build_ext_if_possible(build_ext):
188
user_options = build_ext.user_options + [
189
('allow-python-fallback', None,
190
"When an extension cannot be built, allow falling"
191
" back to the pure-python implementation.")
194
def initialize_options(self):
195
build_ext.initialize_options(self)
196
self.allow_python_fallback = False
184
200
build_ext.run(self)
185
201
except DistutilsPlatformError, e:
202
if not self.allow_python_fallback:
203
log.warn('\n Cannot build extensions.\n'
204
' Use "build_ext --allow-python-fallback" to use'
205
' slower python implementations instead.\n')
187
log.warn('Extensions cannot be built, '
188
'will use the Python versions instead')
208
log.warn('\n Extensions cannot be built.\n'
209
' Using the slower Python implementations instead.\n')
190
211
def build_extension(self, ext):
192
213
build_ext.build_extension(self, ext)
193
214
except CCompilerError:
194
log.warn('Building of "%s" extension failed, '
195
'will use the Python version instead' % (ext.name,))
215
if not self.allow_python_fallback:
216
log.warn('\n Cannot build extension "%s".\n'
217
' Use "build_ext --allow-python-fallback" to use'
218
' slower python implementations instead.\n'
221
log.warn('\n Building of "%s" extension failed.\n'
222
' Using the slower Python implementation instead.'
198
226
# Override the build_ext if we have Pyrex available
216
244
path = module_name.replace('.', '/')
217
245
pyrex_name = path + '.pyx'
218
246
c_name = path + '.c'
248
if sys.platform == 'win32':
249
# pyrex uses the macro WIN32 to detect the platform, even though it
250
# should be using something like _WIN32 or MS_WINDOWS, oh well, we can
251
# give it the right value.
252
define_macros.append(('WIN32', None))
220
ext_modules.append(Extension(module_name, [pyrex_name], **kwargs))
254
source = [pyrex_name]
222
256
if not os.path.isfile(c_name):
223
257
unavailable_files.append(c_name)
225
ext_modules.append(Extension(module_name, [c_name], **kwargs))
228
add_pyrex_extension('bzrlib._btree_serializer_c')
229
add_pyrex_extension('bzrlib._dirstate_helpers_c')
230
add_pyrex_extension('bzrlib._knit_load_data_c')
231
add_pyrex_extension('bzrlib._readdir_pyx')
261
source.extend(extra_source)
262
ext_modules.append(Extension(module_name, source,
263
define_macros=define_macros, libraries=libraries))
266
add_pyrex_extension('bzrlib._annotator_pyx')
267
add_pyrex_extension('bzrlib._bencode_pyx')
268
add_pyrex_extension('bzrlib._btree_serializer_pyx')
269
add_pyrex_extension('bzrlib._chunks_to_lines_pyx')
270
add_pyrex_extension('bzrlib._groupcompress_pyx',
271
extra_source=['bzrlib/diff-delta.c'])
272
add_pyrex_extension('bzrlib._knit_load_data_pyx')
273
add_pyrex_extension('bzrlib._known_graph_pyx')
274
add_pyrex_extension('bzrlib._rio_pyx')
232
275
if sys.platform == 'win32':
233
# pyrex uses the macro WIN32 to detect the platform, even though it should
234
# be using something like _WIN32 or MS_WINDOWS, oh well, we can give it the
236
add_pyrex_extension('bzrlib._walkdirs_win32',
237
define_macros=[('WIN32', None)])
238
ext_modules.append(Extension('bzrlib._patiencediff_c', ['bzrlib/_patiencediff_c.c']))
276
add_pyrex_extension('bzrlib._dirstate_helpers_pyx',
277
libraries=['Ws2_32'])
278
add_pyrex_extension('bzrlib._walkdirs_win32')
281
if have_pyrex and pyrex_version == '0.9.4.1':
282
# Pyrex 0.9.4.1 fails to compile this extension correctly
283
# The code it generates re-uses a "local" pointer and
284
# calls "PY_DECREF" after having set it to NULL. (It mixes PY_XDECREF
285
# which is NULL safe with PY_DECREF which is not.)
286
print 'Cannot build extension "bzrlib._dirstate_helpers_pyx" using'
287
print 'your version of pyrex "%s". Please upgrade your pyrex' % (
289
print 'install. For now, the non-compiled (python) version will'
290
print 'be used instead.'
292
add_pyrex_extension('bzrlib._dirstate_helpers_pyx')
293
add_pyrex_extension('bzrlib._readdir_pyx')
295
add_pyrex_extension('bzrlib._chk_map_pyx', libraries=[z_lib])
296
ext_modules.append(Extension('bzrlib._patiencediff_c',
297
['bzrlib/_patiencediff_c.c']))
241
300
if unavailable_files:
248
307
def get_tbzr_py2exe_info(includes, excludes, packages, console_targets,
308
gui_targets, data_files):
250
309
packages.append('tbzrcommands')
252
311
# ModuleFinder can't handle runtime changes to __path__, but
253
312
# win32com uses them. Hook this in so win32com.shell is found.
254
313
import modulefinder
315
import cPickle as pickle
256
316
for p in win32com.__path__[1:]:
257
317
modulefinder.AddPackagePath("win32com", p)
258
318
for extra in ["win32com.shell"]:
271
331
sys.path.append(os.path.join(tbzr_root, "shellext", "python"))
273
333
packages.append("tbzrlib")
335
# collect up our icons.
337
ico_root = os.path.join(tbzr_root, 'tbzrlib', 'resources')
338
icos = [] # list of (path_root, relative_ico_path)
339
# First always bzr's icon and its in the root of the bzr tree.
340
icos.append(('', 'bzr.ico'))
341
for root, dirs, files in os.walk(ico_root):
342
icos.extend([(ico_root, os.path.join(root, f)[len(ico_root)+1:])
343
for f in files if f.endswith('.ico')])
344
# allocate an icon ID for each file and the full path to the ico
345
icon_resources = [(rid, os.path.join(ico_dir, ico_name))
346
for rid, (ico_dir, ico_name) in enumerate(icos)]
347
# create a string resource with the mapping. Might as well save the
348
# runtime some effort and write a pickle.
349
# Runtime expects unicode objects with forward-slash seps.
350
fse = sys.getfilesystemencoding()
351
map_items = [(f.replace('\\', '/').decode(fse), rid)
352
for rid, (_, f) in enumerate(icos)]
353
ico_map = dict(map_items)
354
# Create a new resource type of 'ICON_MAP', and use ID=1
355
other_resources = [ ("ICON_MAP", 1, pickle.dumps(ico_map))]
274
357
excludes.extend("""pywin pywin.dialogs pywin.dialogs.list
275
358
win32ui crawler.Crawler""".split())
360
# NOTE: We still create a DLL version of the Python implemented shell
361
# extension for testing purposes - but it is *not* registered by
362
# default - our C++ one is instead. To discourage people thinking
363
# this DLL is still necessary, its called 'tbzr_old.dll'
278
365
modules=["tbzr"],
279
366
create_exe = False, # we only want a .dll
367
dest_base = 'tbzr_old',
281
369
com_targets.append(tbzr)
283
371
# tbzrcache executables - a "console" version for debugging and a
284
372
# GUI version that is generally used.
285
373
tbzrcache = dict(
286
script = os.path.join(tbzr_root, "Scripts", "tbzrcache.py"),
287
icon_resources = [(0,'bzr.ico')],
374
script = os.path.join(tbzr_root, "scripts", "tbzrcache.py"),
375
icon_resources = icon_resources,
376
other_resources = other_resources,
289
378
console_targets.append(tbzrcache)
303
392
tbzrcommandw["dest_base"]="tbzrcommandw"
304
393
gui_targets.append(tbzrcommandw)
308
script = os.path.join(tbzr_root, "Scripts", "tbzrtest.py"),
310
console_targets.append(tbzrtest)
312
# A utility to see python output from the shell extension - this will
313
# die when we get a c++ extension
314
# any .py file from pywin32's win32 lib will do (other than
315
# win32traceutil itself that is)
317
win32_lib_dir = os.path.dirname(winerror.__file__)
318
tracer = dict(script = os.path.join(win32_lib_dir, "win32traceutil.py"),
319
dest_base="tbzr_tracer")
395
# A utility to see python output from both C++ and Python based shell
397
tracer = dict(script=os.path.join(tbzr_root, "scripts", "tbzrtrace.py"))
320
398
console_targets.append(tracer)
323
def get_qbzr_py2exe_info(includes, excludes, packages):
400
# The C++ implemented shell extensions.
401
dist_dir = os.path.join(tbzr_root, "shellext", "cpp", "tbzrshellext",
403
data_files.append(('', [os.path.join(dist_dir, 'tbzrshellext_x86.dll')]))
404
data_files.append(('', [os.path.join(dist_dir, 'tbzrshellext_x64.dll')]))
407
def get_qbzr_py2exe_info(includes, excludes, packages, data_files):
324
408
# PyQt4 itself still escapes the plugin detection code for some reason...
325
409
packages.append('PyQt4')
326
410
excludes.append('PyQt4.elementtree.ElementTree')
327
411
includes.append('sip') # extension module required for Qt.
328
412
packages.append('pygments') # colorizer for qbzr
413
packages.append('docutils') # html formatting
329
414
# but we can avoid many Qt4 Dlls.
330
415
dll_excludes.extend(
331
416
"""QtAssistantClient4.dll QtCLucene4.dll QtDesigner4.dll
333
418
QtSql4.dll QtTest4.dll QtWebKit4.dll QtXml4.dll
334
419
qscintilla2.dll""".split())
335
420
# the qt binaries might not be on PATH...
336
qt_dir = os.path.join(sys.prefix, "PyQt4", "bin")
337
path = os.environ.get("PATH","")
338
if qt_dir.lower() not in [p.lower() for p in path.split(os.pathsep)]:
339
os.environ["PATH"] = path + os.pathsep + qt_dir
421
# They seem to install to a place like C:\Python25\PyQt4\*
422
# Which is not the same as C:\Python25\Lib\site-packages\PyQt4
423
pyqt_dir = os.path.join(sys.prefix, "PyQt4")
424
pyqt_bin_dir = os.path.join(pyqt_dir, "bin")
425
if os.path.isdir(pyqt_bin_dir):
426
path = os.environ.get("PATH", "")
427
if pyqt_bin_dir.lower() not in [p.lower() for p in path.split(os.pathsep)]:
428
os.environ["PATH"] = path + os.pathsep + pyqt_bin_dir
429
# also add all imageformat plugins to distribution
430
# We will look in 2 places, dirname(PyQt4.__file__) and pyqt_dir
431
base_dirs_to_check = []
432
if os.path.isdir(pyqt_dir):
433
base_dirs_to_check.append(pyqt_dir)
439
pyqt4_base_dir = os.path.dirname(PyQt4.__file__)
440
if pyqt4_base_dir != pyqt_dir:
441
base_dirs_to_check.append(pyqt4_base_dir)
442
if not base_dirs_to_check:
443
log.warn("Can't find PyQt4 installation -> not including imageformat"
447
for base_dir in base_dirs_to_check:
448
plug_dir = os.path.join(base_dir, 'plugins', 'imageformats')
449
if os.path.isdir(plug_dir):
450
for fname in os.listdir(plug_dir):
451
# Include plugin dlls, but not debugging dlls
452
fullpath = os.path.join(plug_dir, fname)
453
if fname.endswith('.dll') and not fname.endswith('d4.dll'):
454
files.append(fullpath)
456
data_files.append(('imageformats', files))
458
log.warn('PyQt4 was found, but we could not find any imageformat'
459
' plugins. Are you sure your configuration is correct?')
462
def get_svn_py2exe_info(includes, excludes, packages):
463
packages.append('subvertpy')
342
466
if 'bdist_wininst' in sys.argv:
482
606
for root, dirs, files in os.walk('bzrlib/plugins'):
483
607
if root == 'bzrlib/plugins':
484
608
plugins = set(dirs)
609
# We ship plugins as normal files on the file-system - however,
610
# the build process can cause *some* of these plugin files to end
611
# up in library.zip. Thus, we saw (eg) "plugins/svn/test" in
612
# library.zip, and then saw import errors related to that as the
613
# rest of the svn plugin wasn't. So we tell py2exe to leave the
614
# plugins out of the .zip file
615
excludes.extend(["bzrlib.plugins." + d for d in dirs])
487
if os.path.splitext(i)[1] not in [".py", ".pyd", ".dll"]:
618
if os.path.splitext(i)[1] not in [".py", ".pyd", ".dll", ".mo"]:
489
620
if i == '__init__.py' and root == 'bzrlib/plugins':
639
data_files = topics_files + plugins_files
509
641
if 'qbzr' in plugins:
510
get_qbzr_py2exe_info(includes, excludes, packages)
642
get_qbzr_py2exe_info(includes, excludes, packages, data_files)
645
get_svn_py2exe_info(includes, excludes, packages)
512
647
if "TBZR" in os.environ:
513
648
# TORTOISE_OVERLAYS_MSI_WIN32 must be set to the location of the
514
649
# TortoiseOverlays MSI installer file. It is in the TSVN svn repo and
515
650
# can be downloaded from (username=guest, blank password):
516
# http://tortoisesvn.tigris.org/svn/tortoisesvn/TortoiseOverlays/version-1.0.4/bin/TortoiseOverlays-1.0.4.11886-win32.msi
517
if not os.path.isfile(os.environ.get('TORTOISE_OVERLAYS_MSI_WIN32',
519
raise RuntimeError("Please set TORTOISE_OVERLAYS_MSI_WIN32 to the"
520
" location of the Win32 TortoiseOverlays .msi"
651
# http://tortoisesvn.tigris.org/svn/tortoisesvn/TortoiseOverlays
652
# look for: version-1.0.4/bin/TortoiseOverlays-1.0.4.11886-win32.msi
653
# Ditto for TORTOISE_OVERLAYS_MSI_X64, pointing at *-x64.msi.
654
for needed in ('TORTOISE_OVERLAYS_MSI_WIN32',
655
'TORTOISE_OVERLAYS_MSI_X64'):
656
url = ('http://guest:@tortoisesvn.tigris.org/svn/tortoisesvn'
658
if not os.path.isfile(os.environ.get(needed, '<nofile>')):
660
"\nPlease set %s to the location of the relevant"
661
"\nTortoiseOverlays .msi installer file."
662
" The installers can be found at"
664
"\ncheck in the version-X.Y.Z/bin/ subdir" % (needed, url))
522
665
get_tbzr_py2exe_info(includes, excludes, packages, console_targets,
666
gui_targets, data_files)
525
668
# print this warning to stderr as output is redirected, so it is seen
526
669
# at build time. Also to stdout so it appears in the log
added
removed
setup.py
