http://www.mediawiki.org/wiki/Special:Code/pywikipedia/11524
Revision: 11524 Author: drtrigon Date: 2013-05-12 15:36:25 +0000 (Sun, 12 May 2013) Log Message: ----------- new feature; introduce "externals" like in rewrite additional feature is automatic installation of missing code
Modified Paths: -------------- trunk/pywikipedia/subster.py
Added Paths: ----------- trunk/pywikipedia/externals/ trunk/pywikipedia/externals/__init__.py
Removed Paths: ------------- trunk/pywikipedia/crontab/ trunk/pywikipedia/odf/ trunk/pywikipedia/openpyxl/
Added: trunk/pywikipedia/externals/__init__.py =================================================================== --- trunk/pywikipedia/externals/__init__.py (rev 0) +++ trunk/pywikipedia/externals/__init__.py 2013-05-12 15:36:25 UTC (rev 11524) @@ -0,0 +1,293 @@ +#!/usr/bin/python +# -*- coding: utf-8 -*- +""" +Externals modules automatic setup checker and installer for various OS. +""" + +# +# (C) DrTrigon, 2013 +# (C) Pywikipedia team, 2013 +# +# Distributed under the terms of the MIT license. +# +# Strongly inspired by files beeing part of VisTrails distribution +# utils/installbundle.py +# utils/requirements.py +# Copyright (C) 2006-2010 University of Utah. All rights reserved. +# GNU General Public License version 2.0 by the Free Software Foundation +# +__version__ = '$Id$' +# + + +# supports: 1. package management system (yum, apt-get, ...) +# 2. download from url (or svn, git repo) +# (?. checkout from svn/mercurial repo) +# 3. svn:externals +modules_needed = { +# TODO: vvv +# 'crontab' has to be moved and integrated (in)to externals as well + 'crontab': ({}, + #{ 'url': 'https://github.com/josiahcarlson/parse-crontab/archive/master.zip', + # 'path': 'parse-crontab-master/crontab',}), # OK + { 'url': 'https://github.com/josiahcarlson/parse-crontab/archive/1ec538ff67df6a207993a...', + 'path': 'parse-crontab-1ec538ff67df6a207993a6c5b6988f4f628c5776/crontab',}),# OK +# TODO: vvv +# 'odf' has to be moved and integrated (in)to externals as well + 'odf': ({}, + #{ 'url': 'https://pypi.python.org/packages/source/o/odfpy/odfpy-0.9.6.tar.gz', + # 'path': 'odfpy-0.9.6/odf',}), # OK + { 'url': 'https://pypi.python.org/packages/source/o/odfpy/odfpy-0.9.4.tar.gz', + 'path': 'odfpy-0.9.4/odf',}), # OK +# TODO: vvv +# 'openpyxl' has to be moved and integrated (in)to externals as well + 'openpyxl': ({}, + { 'url': 'https://bitbucket.org/ericgazoni/openpyxl/get/1.5.6.tar.gz', + 'path': 'ericgazoni-openpyxl-e5934500ffac/openpyxl',}),# OK +# TODO: vvv +# 'simplejson' has to be moved and integrated (in)to externals as well +# 'spelling' has to be moved and integrated (in)to externals as well +# 'i18n' has to be moved and integrated (in)to externals as well +# $ svn propget svn:externals pywikipedia/ +# spelling http://svn.wikimedia.org/svnroot/pywikipedia/trunk/spelling/ +# simplejson http://simplejson.googlecode.com/svn/tags/simplejson-2.1.3/simplejson/ +# i18n http://svn.wikimedia.org/svnroot/pywikipedia/branches/rewrite/scripts/i18n + 'colormath': ({'linux-fedora': [], + 'linux-ubuntu': ['python-colormath'],}, + { 'url': 'https://github.com/gtaylor/python-colormath/archive/master.zip', + 'path': 'python-colormath-master/colormath',}), # OK + 'jseg': ({}, + { 'url': 'http://vision.ece.ucsb.edu/segmentation/jseg/software/jseg.zip', + 'path': 'jseg', + #$ diff -Nau --exclude="*.o" --exclude="*.pyc" --exclude="segdist_cpp*" TEST_jseg/ jseg/ > patch-jseg + 'patch': 'patch-jseg',}), # OK + 'jseg/jpeg-6b': ({}, + { 'url': 'http://vision.ece.ucsb.edu/segmentation/jseg/software/jpeg-6b.zip', + 'path': 'jpeg-6b',}), # OK +# TODO: vvv (future; enable for and use in 'catimages.py') +# '_mlpy': ({}, +# { 'url': 'http://downloads.sourceforge.net/project/mlpy/mlpy%203.5.0/mlpy-3.5.0.tar.gz', +# 'path': 'mlpy-3.5.0/mlpy',}), # OK + '_music21': ({}, + { 'url': 'http://music21.googlecode.com/files/music21-1.4.0.tar.gz', + 'path': 'music21-1.4.0', + #$ diff -Naur --exclude="*.pyc" TEST__music21/ _music21/ > patch-music21 + 'patch': 'patch-music21',}), # OK +# TODO: vvv (future; enable for and use in 'catimages.py') +# mercurial: $ hg clone -r ocropus-0.6pre3 https://code.google.com/p/ocropus +# '_ocropus': ({}, +# {}), # OPEN +# TODO: vvv (further clean-up and unlink - check with 'svn list') +# 'opencv': $ svn propedit svn:externals . +# bagofwords_classification.cpp https://svn.toolserver.org/svnroot/drtrigon/externals/opencv/bagofwords_clas... +# bagofwords_classification_python.cpp https://svn.toolserver.org/svnroot/drtrigon/externals/opencv/bagofwords_clas... +# camera_virtual_default https://svn.toolserver.org/svnroot/drtrigon/externals/opencv/camera_virtual_... +# facetest.pl https://svn.toolserver.org/svnroot/drtrigon/externals/opencv/facetest.pl +# __init.py__ https://svn.toolserver.org/svnroot/drtrigon/externals/opencv/__init__.py +# makefile https://svn.toolserver.org/svnroot/drtrigon/externals/opencv/makefile +# peopledetect.py https://svn.toolserver.org/svnroot/drtrigon/externals/opencv/peopledetect.py +# $ svn propedit svn:externals haarcascades/haartraining/ +# HaarTraining https://svn.toolserver.org/svnroot/drtrigon/externals/haartraining/HaarTrain... +# HaarTraining.tar.gz https://svn.toolserver.org/svnroot/drtrigon/externals/haartraining/HaarTrain... +# convert_cascade.c https://svn.toolserver.org/svnroot/drtrigon/externals/haartraining/convert_c... +# create_pos_neg.py https://svn.toolserver.org/svnroot/drtrigon/externals/haartraining/create_po... +# createtestsamples.pl https://svn.toolserver.org/svnroot/drtrigon/externals/haartraining/createtes... +# createtrainsamples.pl https://svn.toolserver.org/svnroot/drtrigon/externals/haartraining/createtra... +'opencv/haarcascades': ({}, + { 'url': 'https://svn.toolserver.org/svnroot/drtrigon/externals/haarcascades-full.tar....', + 'path': 'haarcascades',}), # OK +# 'pdfminer' is not used anymore/at the moment... +# 'pycolorname': $ svn propset svn:externals 'pycolorname https://svn.toolserver.org/svnroot/drtrigon/externals/pycolorname' . + '_pydmtx': ({'linux-fedora': ['python-libdmtx'], + 'linux-ubuntu': ['libdmtx-dev'],}, + { 'url': 'https://github.com/dmtx/dmtx-wrappers/archive/master.zip', + 'path': 'dmtx-wrappers-master/python', + #$ diff -Naur TEST__pydmtx/ _pydmtx/ > patch-pydmtx + 'patch': 'patch-pydmtx',}), # OK + 'py_w3c': ({}, + { 'url': 'https://bitbucket.org/nmb10/py_w3c/downloads/py_w3c-v0.1.0.tar.gz', + 'path': 'py_w3c-0.1.0/py_w3c',}), # OK +# TODO: vvv (include) +# 'TEST_slic': ({}, +# { 'url': 'https://github.com/amueller/slic-python/archive/master.zip', +# 'path': 'slic-python-master',}), # OPEN +# 'TEST_slic': ({}, +# { 'url': 'http://ivrg.epfl.ch/files/content/sites/ivrg/files/supplementary_material/RK...', +# 'path': 'SLICSuperpixelsAndSupervoxelsCode/SLICSuperpixels',}),# OPEN +# (2 download sources to same dir) + patch (at least for '__init__.py') needed + '_zbar': ({'linux-fedora': ['zbar'], + 'linux-ubuntu': ['python-zbar'],}, + { 'url': 'https://pypi.python.org/packages/source/z/zbar/zbar-0.10.tar.bz2', + 'path': 'zbar-0.10', + #$ diff -Nau --exclude="*.pyc" TEST__zbar/ _zbar/ > patch-zbar + 'patch': 'patch-zbar',}), # OK +# TODO: vvv (include) +# 'TEST__bob': ({}, +# { 'url': 'https://www.idiap.ch/software/bob/packages/bob-1.1.2.zip', +# 'path': 'bob-1.1.2',}), # OPEN +# (complex compilation) + patch (at least for '__init__.py') needed +# 'TEST_xbob_flandmark': ({}, +# { 'url': 'https://pypi.python.org/packages/source/x/xbob.flandmark/xbob.flandmark-1.0....', +# 'path': 'xbob.flandmark-1.0.9',}), # OPEN +# (complex compilation, dependent on '_bob') + patch (at least for '__init__.py') needed +} + +#modules_order = ['crontab', 'odf', 'openpyxl', +# 'colormath', 'jseg', 'jseg/jpeg-6b', '_mlpy', '_music21', +# '_ocropus', 'opencv', 'opencv/haarcascades', '_pydmtx', +# 'py_w3c', 'slic', '_zbar', '_bob', 'xbob_flandmark',] +modules_order = [#'crontab', 'odf', 'openpyxl', + 'colormath', 'jseg', 'jseg/jpeg-6b', '_music21', + 'opencv/haarcascades', '_pydmtx', 'py_w3c', '_zbar',] + + +import os, sys + +import wikipedia as pywikibot + + +sys.path.append(os.path.dirname(os.path.abspath(os.path.join(os.curdir, __file__)))) + + +### BEGIN of VisTrails inspired and copied code ### ### ### ### ### ### ### ### + +def guess_system(): + import platform + return ("%s-%s" % (platform.system(), platform.dist()[0])).lower() + +def show_question(which_files): + pywikibot.output("Required package missing") + pywikibot.output("A required package is missing, but externals can" + " automatically install it." + " If you say Yes, externals will need administrator" + " privileges, and you might be asked for the administrator" + " password.") + pywikibot.output("Give externals permission to try to install package?" + " (y/N)") + v = raw_input().upper() + return v == 'Y' or v == 'YES' + + +def linux_ubuntu_install(package_name): + cmd = 'apt-get install -y' + + if type(package_name) == str: + cmd += ' ' + package_name + elif type(package_name) == list: + for package in package_name: + if type(package) != str: + raise TypeError("Expected string or list of strings") + cmd += ' ' + package + + pywikibot.warning("externals wants to install package(s) '%s'" % + package_name) + sucmd = "sudo %s" % cmd + + result = os.system(sucmd) + + return (result == 0) # 0 indicates success + +def linux_fedora_install(package_name): + cmd = 'yum -y install' + + if type(package_name) == str: + cmd += ' ' + package_name + elif type(package_name) == list: + for package in package_name: + if type(package) != str: + raise TypeError("Expected string or list of strings") + cmd += ' ' + package + + pywikibot.warning("externals wants to install package(s) '%s'" % + package_name) + sucmd = "su -c'%s'" % cmd + + result = os.system(sucmd) + + return (result == 0) + +def linux_install(dependency_dictionary): + """Tries to import a python module. If unsuccessful, tries to install +the appropriate bundle and then reimport. py_import tries to be smart +about which system it runs on.""" + + # Ugly fix to avoid circular import + distro = guess_system() + if not dependency_dictionary.has_key(distro): + return False + else: + files = dependency_dictionary[distro] + if files and show_question(files): + callable_ = globals()[distro.replace('-', '_') + '_install'] + return callable_(files) + else: + return False + +### END of VisTrails inspired and copied code ### ### ### ### ### ### ### ### + + +def download_install(package, module, path): + if package: + pywikibot.warning(u'Download package "%s" from %s' + % (module, package['url'])) + import urllib2, mimetypes + response = urllib2.urlopen(package['url']) + pywikibot.warning(u'Size of download: %s byte(s)' + % response.headers['Content-Length']) + #mime = response.headers['Content-Type'].lower().split('/') + mime = mimetypes.guess_type(package['url'], strict=True)[0].lower().split('/') + pywikibot.warning(u'MIME type: %s' % mime) + + pywikibot.warning(u'Extract package "%s" to %s' + % (module, os.path.join(path, module))) + if len(mime) > 1: + if mime[1] == 'zip': + import zipfile, StringIO + arch = zipfile.ZipFile(StringIO.StringIO(response.read())) + elif mime[1] == 'x-tar': + import tarfile, StringIO + arch = tarfile.open(fileobj=StringIO.StringIO(response.read())) + arch.extractall(os.path.join(path, '__setup_tmp/')) + arch.close() + import shutil + shutil.move(os.path.join(path, '__setup_tmp/', package['path']), + os.path.join(path, module)) + shutil.rmtree(os.path.join(path, '__setup_tmp/')) + + result = 0 + if 'patch' in package: + pywikibot.warning(u'Install package "%s" by applying patch to %s.' + % (module, os.path.join(path, module))) + cmd = 'patch -p0 -d %s < %s' % (path, os.path.join(path, package['patch'])) + result = os.system(cmd) + + pywikibot.warning(u'Package "%s" installed to %s.' + % (module, os.path.join(path, module))) + return (result == 0) + + return False + + +def check_setup(m): + path = os.path.dirname(os.path.abspath(os.path.join(os.curdir, __file__))) + mf = os.path.join(path, m) + + #__import__(mf) + if not os.path.exists(mf): + # install the missing module + if linux_install(modules_needed[m][0]): + return + if download_install(modules_needed[m][1], m, path): + return +# if svn_repo_install(modules_needed[m][2]): +# return + pywikibot.error(u'Package "%s" could not be found nor installed!' + % m) + +def check_setup_all(): + #for m in modules_needed: + for m in modules_order: + check_setup(m) + + +#check_setup_all()
Property changes on: trunk/pywikipedia/externals/__init__.py ___________________________________________________________________ Added: svn:keywords + Id Added: svn:eol-style + native
Modified: trunk/pywikipedia/subster.py =================================================================== --- trunk/pywikipedia/subster.py 2013-05-10 20:05:45 UTC (rev 11523) +++ trunk/pywikipedia/subster.py 2013-05-12 15:36:25 UTC (rev 11524) @@ -70,8 +70,6 @@ import BeautifulSoup import StringIO, zipfile, csv import mailbox, mimetypes, datetime, email.utils -import openpyxl.reader.excel -import crontab import logging import ast import shelve, pprint @@ -81,6 +79,12 @@ import wikipedia as pywikibot from pywikibot import i18n from pywikibot.comms import http +import externals # check for and install needed 'externals' +externals.check_setup('crontab') # +externals.check_setup('odf') # +externals.check_setup('openpyxl') # +import openpyxl.reader.excel +import crontab
# TODO: think about what config to move to 'subster-config.css' (per wiki)