-#----------------------------------------------------------------------
-# Name: __init__.py
-# Purpose: Import logic and common functions for wxaddons module
-#
-# Author: Kevin Ollivier
-#
-# Created: 15-Nov-2006
-# RCS-ID: $Id$
-# Copyright: (c) 2006 Kevin Ollivier
-# Licence: wxWindows license
-#----------------------------------------------------------------------
-
-import sys, os, string
-import xmlrpclib
-import __builtin__
-import wx
-
-# NB: For some reason that I haven't been able to track down, on Mac (at least)
-# calling xmlrpc methods no longer works after the wx.App is started. Therefore,
-# we grab the package URL even before prompting the user if they want to install
-# the package in order for us to have the info we need before the wx.App is started.
-
-domain = 'http://wxaddons.wxcommunity.com'
-builtin_import = __builtin__.__import__
-
-debug = False
-use_gui = True
-checkImports = True
-config = wx.Config("wxaddons")
-if config.Read("PerformChecks", "true") != "true":
- checkImports = False
-
-if use_gui and not wx.App.IsDisplayAvailable():
- use_gui = False
-
-s = xmlrpclib.Server('%s/xmlrpc-server.php' % domain, verbose=(debug == True))
-
-def check_imports(check):
- if check:
- config.Write("PerformChecks", "true")
- else:
- config.Write("PerformChecks", "false")
-
-def version_greater_than_or_equal(version1, version2):
- """
- Checks if version1 >= version2, returning true if so,
- false if otherwise.
- """
- greater_than = True
-
- for index in range(0, len(version1)-1):
- if version1[index] > version2[index]:
- greater_than = True
- break
- elif version[index] < current_version[index]:
- greater_than = False
- break
-
- return greater_than
-
-def prompt_install(name, version):
- should_install = False
- message = "The wxaddon %s is not installed, but was found on the wxaddons site. Would you like to download and install it?" % (name + " " + version)
- if use_gui:
- app = wx.PySimpleApp()
- app.MainLoop()
- result = wx.MessageBox(message, "Install Dependency?", style=wx.YES_NO)
- if result == wx.YES:
- should_install = True
- else:
- result = raw_input(message + " [y/n]")
- if result[0].lower() == "y":
- should_install = True
-
- return should_install
-
-def require_addon_version(name, version=[], canBeNewer=True):
- # Check the install receipt to see if we've got an appropriate version
- config = wx.Config("wxaddons-receipts")
- needs_update = True
- if config.HasGroup(name):
- config.SetPath(name)
- current_version = config.Read("version", "0.0").split(".")
-
- needs_update = version_greater_than_or_equal(version, current_version)
-
- if needs_update:
- comp_xml = s.getComponent(name)
- if not comp_xml:
- raise
-
- comp = xmlrpclib.loads(comp_xml)[0][0]
- comp_version = comp["version"].split(".")
-
- update_comp = False
- if canBeNewer:
- update_comp = version_greater_than_or_equal(comp_version, version)
- else:
- update_comp = (version == comp_version)
-
- if update_comp:
- url = get_url(name, version)
- should_install = prompt_install(name, comp_version)
-
- if should_install:
- dl_and_install_addon(name, comp_version, url)
-
-def get_url(name, version):
- url = ""
- release_xml = s.getReleases(name)
- if not release_xml:
- return ""
-
- releases = xmlrpclib.loads(release_xml)[0][0]
- for release in releases:
- if release['version'] == version:
- url = release['url']
-
- return url
-
-def dl_and_install_addon(name, version, url):
- installed = True
- tempdir = None
- cwd = os.getcwd()
-
- if use_gui:
- progress = wx.ProgressDialog("Installing Dependency",
- "Preparing to install the %s addon module." % name,
- 4,
- style=wx.PD_APP_MODAL|wx.PD_SMOOTH)
-
- message = "Downloading tarball..."
- print message
- if use_gui: progress.Update(1, message)
- import urllib
- temp_archive, headers = urllib.urlretrieve(url)
-
- message = "Extracting files..."
- print message
- if use_gui: progress.Update(2, message)
- import tempfile
- tempdir = tempfile.mkdtemp()
-
- os.chdir(tempdir)
- import tarfile
- tar = tarfile.open(temp_archive, "r:gz")
- for tarinfo in tar:
- tar.extract(tarinfo)
- tar.close()
-
- os.chdir(name)
-
- message = "Installing %s" % name
- if use_gui: progress.Update(3, message)
- # TODO: Add support for modified PYTHONPATH?
- # Also, do we need admin install support here?
- retval = os.system(sys.executable + " " + string.join(("setup.py", "install"), " "))
- if use_gui: progress.Update(4)
-
- if retval == 0:
- message = "The %s addon was successfully installed." % name
- print message
- if use_gui:
- wx.MessageBox(message, "Installation Successful")
- else:
- installed = False
-
- # cleanup
- if use_gui: progress.Destroy()
- os.chdir(cwd)
- import shutil
- shutil.rmtree(tempdir)
- os.remove(temp_archive)
-
- return installed
-
-def import_hook(name, globals=None, locals=None, fromlist=None):
- # Fast path: see if the module has already been imported.
- try:
- return builtin_import(name, globals, locals, fromlist)
- except:
- if name.startswith("wxaddons"):
- print "Querying %s for module." % domain
- try:
- modname = name.split(".")[1]
- comp = None
- comp_xml = s.getComponent(modname)
- if not comp_xml:
- raise
-
- comp = xmlrpclib.loads(comp_xml)[0][0]
- url = get_url(comp["name"], comp["version"])
- should_install = prompt_install(comp["name"], comp["version"])
-
- if should_install:
- try:
- success = dl_and_install_addon(comp["name"], comp["version"], url)
- if not success:
- raise
- except:
- raise
-
-
- except:
- raise
- else:
- raise
-
-def runTests():
- import wxaddons.persistence
- import wxaddons.foo_bar
- import googly
-
-if checkImports:
- __builtin__.__import__ = import_hook
-
projects / wxWidgets.git / blobdiff