Beginings of supporting multiple versions installed side-by-side
git-svn-id: https://svn.wxwidgets.org/svn/wx/wxWidgets/trunk@29517 c3d73ce0-8a6f-49c7-b76d-6d57e0e08775
This commit is contained in:
Robin Dunn
210
wxPython/wxversion/wxversion.py
Normal file
210
wxPython/wxversion/wxversion.py
Normal file
@@ -0,0 +1,210 @@
|
||||
#----------------------------------------------------------------------
|
||||
# Name: wxversion
|
||||
# Purpose: Allows a wxPython program to search for alternate
|
||||
# installations of the wxPython packages and modify sys.path
|
||||
# so they will be found when "import wx" is done.
|
||||
#
|
||||
# Author: Robin Dunn
|
||||
#
|
||||
# Created: 24-Sept-2004
|
||||
# RCS-ID: $Id$
|
||||
# Copyright: (c) 2004 by Total Control Software
|
||||
# Licence: wxWindows license
|
||||
#----------------------------------------------------------------------
|
||||
|
||||
"""
|
||||
If you have more than one version of wxPython installed this module
|
||||
allows your application to choose which version of wxPython will be
|
||||
imported when it does 'import wx'. You use it like this:
|
||||
|
||||
import wxversion
|
||||
wxversion.require('2.4')
|
||||
import wx
|
||||
|
||||
Of course the default wxPython version can also be controlled by
|
||||
setting PYTHONPATH or by editing the wx.pth path configuration file,
|
||||
but using wxversion will allow an application to manage the version
|
||||
selection itself rather than depend on the user to setup the
|
||||
environment correctly.
|
||||
|
||||
It works by searching the sys.path for directories matching wx-* and
|
||||
then comparing them to what was passed to the require function. If a
|
||||
match is found then that path is inserted into sys.path.
|
||||
"""
|
||||
|
||||
import sys, os, glob, fnmatch
|
||||
|
||||
|
||||
|
||||
|
||||
def require(versions):
|
||||
"""
|
||||
Search for a wxPython installation that matches version.
|
||||
|
||||
:param version: Specifies the version to look for, it can either be
|
||||
a sting or a list of strings. Each string is
|
||||
compared to the installed wxPythons and the best
|
||||
match is added to the sys.path, allowing an 'import
|
||||
wx' to find that version.
|
||||
|
||||
The version string is composed of the dotted
|
||||
version number (at least 2 of the 4 components)
|
||||
optionally followed by hyphen ('-') separated
|
||||
options (wx port, unicode/ansi, flavour, etc.) A
|
||||
match is determined by how much of the installed
|
||||
version matches what is given in the version
|
||||
parameter. If the version number components don't
|
||||
match then the score is zero, otherwise the score
|
||||
is increased for every specified optional component
|
||||
that is specified and that matches.
|
||||
"""
|
||||
assert not sys.modules.has_key('wx') and not sys.modules.has_key('wxPython'), \
|
||||
"wxversion.require() must be called before wxPython is imported"
|
||||
|
||||
bestMatch = None
|
||||
bestScore = 0
|
||||
if type(versions) == str:
|
||||
versions = [versions]
|
||||
|
||||
packages = _find_installed()
|
||||
for pkg in packages:
|
||||
for ver in versions:
|
||||
score = pkg.Score(_wxPackageInfo(ver))
|
||||
if score > bestScore:
|
||||
bestMatch = pkg
|
||||
bestScore = score
|
||||
|
||||
assert bestMatch is not None, \
|
||||
"Required version of wxPython not found"
|
||||
|
||||
sys.path.insert(0, bestMatch.pathname)
|
||||
|
||||
|
||||
|
||||
|
||||
_pattern = "wx-[0-9].*"
|
||||
def _find_installed():
|
||||
installed = []
|
||||
for pth in sys.path:
|
||||
|
||||
# empty means to look in the current dir
|
||||
if not pth:
|
||||
pth = '.'
|
||||
|
||||
# skip it if it's not a package dir
|
||||
if not os.path.isdir(pth):
|
||||
continue
|
||||
|
||||
base = os.path.basename(pth)
|
||||
|
||||
# if it's a wx path that's already in the sys.path then skip it
|
||||
if fnmatch.fnmatchcase(base, _pattern):
|
||||
continue
|
||||
|
||||
# now look in the dir for matching subdirs
|
||||
for name in glob.glob(os.path.join(pth, _pattern)):
|
||||
# make sure it's a directory
|
||||
if not os.path.isdir(name):
|
||||
continue
|
||||
# and has a wx subdir
|
||||
if not os.path.exists(os.path.join(name, 'wx')):
|
||||
continue
|
||||
installed.append(_wxPackageInfo(name, True))
|
||||
|
||||
installed.sort()
|
||||
installed.reverse()
|
||||
return installed
|
||||
|
||||
|
||||
class _wxPackageInfo(object):
|
||||
def __init__(self, pathname, stripFirst=False):
|
||||
self.pathname = pathname
|
||||
base = os.path.basename(pathname)
|
||||
segments = base.split('-')
|
||||
if stripFirst:
|
||||
segments = segments[1:]
|
||||
self.version = tuple([int(x) for x in segments[0].split('.')])
|
||||
self.options = segments[1:]
|
||||
|
||||
|
||||
def Score(self, other):
|
||||
score = 0
|
||||
# whatever version components given in other must match exactly
|
||||
if len(self.version) > len(other.version):
|
||||
v = self.version[:len(other.version)]
|
||||
else:
|
||||
v = self.version
|
||||
if v != other.version:
|
||||
return 0
|
||||
score += 1
|
||||
for opt in other.options:
|
||||
if opt in self.options:
|
||||
score += 1
|
||||
return score
|
||||
|
||||
|
||||
# TODO: factor self.options into the sort order?
|
||||
def __lt__(self, other):
|
||||
return self.version < other.version
|
||||
def __gt__(self, other):
|
||||
return self.version > other.version
|
||||
def __eq__(self, other):
|
||||
return self.version == other.version
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
if __name__ == '__main__':
|
||||
def test(version):
|
||||
savepath = sys.path[:]
|
||||
require(version)
|
||||
print "Asked for %s:\t got: %s" % (version, sys.path[0])
|
||||
sys.path = savepath[:]
|
||||
|
||||
|
||||
# make some test dirs
|
||||
names = ['wx-2.4',
|
||||
'wx-2.5.2',
|
||||
'wx-2.5.2.9-gtk2-unicode',
|
||||
'wx-2.5.2.9-gtk-ansi',
|
||||
'wx-2.5.1',
|
||||
'wx-2.5.2.8-gtk2-unicode',
|
||||
'wx-2.5.3']
|
||||
for name in names:
|
||||
d = os.path.join('/tmp', name)
|
||||
os.mkdir(d)
|
||||
os.mkdir(os.path.join(d, 'wx'))
|
||||
|
||||
# setup sys.path to see those dirs
|
||||
sys.path.append('/tmp')
|
||||
|
||||
|
||||
# now run some tests
|
||||
test("2.4")
|
||||
test("2.5")
|
||||
test("2.5-gtk2")
|
||||
test("2.5.2")
|
||||
test("2.5-ansi")
|
||||
test("2.5-unicode")
|
||||
|
||||
# There isn't a unicode match for this one, but it will give the best
|
||||
# available 2.4. Should it give an error instead? I don't think so...
|
||||
test("2.4-unicode")
|
||||
|
||||
try:
|
||||
# expecting an error on this one
|
||||
test("2.6")
|
||||
except AssertionError:
|
||||
print "Asked for 2.6:\t got: Assertion"
|
||||
|
||||
# Try asking for multiple versions
|
||||
test(["2.6", "2.5.3", "2.5.2-gtk2"])
|
||||
|
||||
# cleanup
|
||||
for name in names:
|
||||
d = os.path.join('/tmp', name)
|
||||
os.rmdir(os.path.join(d, 'wx'))
|
||||
os.rmdir(d)
|
||||
|
||||
|
||||
Reference in New Issue
Block a user