1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351135213531354135513561357135813591360136113621363136413651366136713681369137013711372137313741375137613771378137913801381138213831384138513861387138813891390139113921393139413951396139713981399140014011402140314041405140614071408140914101411141214131414141514161417141814191420142114221423142414251426142714281429143014311432143314341435143614371438143914401441144214431444144514461447144814491450145114521453145414551456145714581459146014611462146314641465146614671468146914701471147214731474147514761477147814791480148114821483148414851486148714881489149014911492149314941495149614971498149915001501150215031504150515061507150815091510151115121513151415151516151715181519152015211522152315241525152615271528152915301531153215331534153515361537153815391540154115421543154415451546154715481549155015511552155315541555155615571558155915601561156215631564156515661567156815691570157115721573157415751576157715781579158015811582158315841585158615871588158915901591159215931594159515961597159815991600160116021603160416051606160716081609161016111612161316141615161616171618161916201621162216231624162516261627162816291630163116321633163416351636163716381639164016411642164316441645164616471648164916501651165216531654165516561657165816591660166116621663166416651666166716681669167016711672167316741675167616771678167916801681168216831684168516861687168816891690169116921693169416951696169716981699170017011702170317041705170617071708170917101711171217131714171517161717171817191720172117221723172417251726172717281729173017311732173317341735173617371738173917401741174217431744174517461747174817491750175117521753175417551756175717581759 |
- #!/usr/bin/env python
- ############################################################################
- #
- # MODULE: g.extension
- # AUTHOR(S): Markus Neteler (original shell script)
- # Martin Landa <landa.martin gmail com> (Pythonized & upgraded for GRASS 7)
- # Vaclav Petras <wenzeslaus gmail com> (support for general sources)
- # PURPOSE: Tool to download and install extensions into local installation
- #
- # COPYRIGHT: (C) 2009-2016 by Markus Neteler, and the GRASS Development Team
- #
- # This program is free software under the GNU General
- # Public License (>=v2). Read the file COPYING that
- # comes with GRASS for details.
- #
- # TODO: add sudo support where needed (i.e. check first permission to write into
- # $GISBASE directory)
- #############################################################################
- #%module
- #% label: Maintains GRASS Addons extensions in local GRASS installation.
- #% description: Downloads and installs extensions from GRASS Addons repository or other source into the local GRASS installation or removes installed extensions.
- #% keyword: general
- #% keyword: installation
- #% keyword: extensions
- #% keyword: addons
- #% keyword: download
- #%end
- #%option
- #% key: extension
- #% type: string
- #% key_desc: name
- #% label: Name of extension to install or remove
- #% description: Name of toolbox (set of extensions) when -t flag is given
- #% required: yes
- #%end
- #%option
- #% key: operation
- #% type: string
- #% description: Operation to be performed
- #% required: yes
- #% options: add,remove
- #% answer: add
- #%end
- #%option
- #% key: url
- #% type: string
- #% key_desc: url
- #% label: URL or directory to get the extension from (supported only on Linux and Mac)
- #% description: The official repository is used by default. User can specify a ZIP file, directory or a repository on common hosting services. If not identified, Subversion repository is assumed. See manual for all options.
- #%end
- #%option
- #% key: prefix
- #% type: string
- #% key_desc: path
- #% description: Prefix where to install extension (ignored when flag -s is given)
- #% answer: $GRASS_ADDON_BASE
- #% required: no
- #%end
- #%option
- #% key: proxy
- #% type: string
- #% key_desc: proxy
- #% description: Set the proxy with: "http=<value>,ftp=<value>"
- #% required: no
- #% multiple: yes
- #%end
- #%flag
- #% key: l
- #% description: List available extensions in the official GRASS GIS Addons repository
- #% guisection: Print
- #% suppress_required: yes
- #%end
- #%flag
- #% key: c
- #% description: List available extensions in the official GRASS GIS Addons repository including module description
- #% guisection: Print
- #% suppress_required: yes
- #%end
- #%flag
- #% key: g
- #% description: List available extensions in the official GRASS GIS Addons repository (shell script style)
- #% guisection: Print
- #% suppress_required: yes
- #%end
- #%flag
- #% key: a
- #% description: List locally installed extensions
- #% guisection: Print
- #% suppress_required: yes
- #%end
- #%flag
- #% key: s
- #% description: Install system-wide (may need system administrator rights)
- #% guisection: Install
- #%end
- #%flag
- #% key: d
- #% description: Download source code and exit
- #% guisection: Install
- #%end
- #%flag
- #% key: i
- #% description: Do not install new extension, just compile it
- #% guisection: Install
- #%end
- #%flag
- #% key: f
- #% description: Force removal when uninstalling extension (operation=remove)
- #% guisection: Remove
- #%end
- #%flag
- #% key: t
- #% description: Operate on toolboxes instead of single modules (experimental)
- #% suppress_required: yes
- #%end
- #%rules
- #% required: extension, -l, -c, -g, -a
- #% exclusive: extension, -l, -c, -g
- #% exclusive: extension, -l, -c, -a
- #%end
- # TODO: solve addon-extension(-module) confusion
- from __future__ import print_function
- import os
- import sys
- import re
- import atexit
- import shutil
- import zipfile
- import tempfile
- from distutils.dir_util import copy_tree
- try:
- from urllib2 import HTTPError, URLError, ProxyHandler, build_opener
- from urllib import urlopen, urlretrieve
- except ImportError:
- # there is also HTTPException, perhaps change to list
- from urllib.error import HTTPError, URLError
- from urllib.request import urlopen, urlretrieve, ProxyHandler, build_opener
- try:
- import xml.etree.ElementTree as etree
- except ImportError:
- import elementtree.ElementTree as etree # Python <= 2.4
- # Get the XML parsing exceptions to catch. The behavior changed with Python 2.7
- # and ElementTree 1.3.
- from xml.parsers import expat # TODO: works for any Python?
- if hasattr(etree, 'ParseError'):
- ETREE_EXCEPTIONS = (etree.ParseError, expat.ExpatError)
- else:
- ETREE_EXCEPTIONS = (expat.ExpatError)
- import grass.script as gscript
- from grass.script.utils import try_rmdir
- from grass.script import core as grass
- # i18N
- import gettext
- gettext.install('grassmods', os.path.join(os.getenv("GISBASE"), 'locale'))
- # temp dir
- REMOVE_TMPDIR = True
- PROXIES = {}
- def etree_fromfile(filename):
- """Create XML element tree from a given file name"""
- with open(filename, 'r') as file_:
- return etree.fromstring(file_.read())
- def etree_fromurl(url, proxies=None):
- """Create XML element tree from a given URL"""
- if proxies:
- proxy_handler = ProxyHandler(PROXIES)
- opener = build_opener(proxy_handler)
- with opener.open(url) as file_:
- return etree.fromstring(file_.read())
- else:
- file_ = urlopen(url)
- return etree.fromstring(file_.read())
- def check_progs():
- """Check if the necessary programs are available"""
- # TODO: we need svn for the Subversion repo downloads
- # also git would be tested once supported
- for prog in ('make', 'gcc'):
- if not grass.find_program(prog, '--help'):
- grass.fatal(_("'%s' required. Please install '%s' first.")
- % (prog, prog))
- # expand prefix to class name
- def expand_module_class_name(class_letters):
- """Convert module class (family) letter or letters to class (family) name
- The letter or letters are used in module names, e.g. r.slope.aspect.
- The names are used in directories in Addons but also in the source code.
- >>> expand_module_class_name('r')
- 'raster'
- >>> expand_module_class_name('v')
- 'vector'
- """
- name = {
- 'd': 'display',
- 'db': 'database',
- 'g': 'general',
- 'i': 'imagery',
- 'm': 'misc',
- 'ps': 'postscript',
- 'p': 'paint',
- 'r': 'raster',
- 'r3': 'raster3d',
- 's': 'sites',
- 't': 'temporal',
- 'v': 'vector',
- 'wx': 'gui/wxpython'
- }
- return name.get(class_letters, class_letters)
- def get_module_class_name(module_name):
- """Return class (family) name for a module
- The names are used in directories in Addons but also in the source code.
- >>> get_module_class_name('r.slope.aspect')
- 'raster'
- >>> get_module_class_name('v.to.rast')
- 'vector'
- """
- classchar = module_name.split('.', 1)[0]
- return expand_module_class_name(classchar)
- def get_installed_extensions(force=False):
- """Get list of installed extensions or toolboxes (if -t is set)"""
- if flags['t']:
- return get_installed_toolboxes(force)
- return get_installed_modules(force)
- def list_installed_extensions(toolboxes=False):
- """List installed extensions"""
- elist = get_installed_extensions()
- if elist:
- if toolboxes:
- grass.message(_("List of installed extensions (toolboxes):"))
- else:
- grass.message(_("List of installed extensions (modules):"))
- sys.stdout.write('\n'.join(elist))
- sys.stdout.write('\n')
- else:
- if toolboxes:
- grass.info(_("No extension (toolbox) installed"))
- else:
- grass.info(_("No extension (module) installed"))
- def get_installed_toolboxes(force=False):
- """Get list of installed toolboxes
- Writes toolboxes file if it does not exist.
- Creates a new toolboxes file if it is not possible
- to read the current one.
- """
- xml_file = os.path.join(options['prefix'], 'toolboxes.xml')
- if not os.path.exists(xml_file):
- write_xml_toolboxes(xml_file)
- # read XML file
- try:
- tree = etree_fromfile(xml_file)
- except ETREE_EXCEPTIONS + (OSError, IOError):
- os.remove(xml_file)
- write_xml_toolboxes(xml_file)
- return []
- ret = list()
- for tnode in tree.findall('toolbox'):
- ret.append(tnode.get('code'))
- return ret
- def get_installed_modules(force=False):
- """Get list of installed modules.
- Writes modules file if it does not exist and *force* is set to ``True``.
- Creates a new modules file if it is not possible
- to read the current one.
- """
- xml_file = os.path.join(options['prefix'], 'modules.xml')
- if not os.path.exists(xml_file):
- if force:
- write_xml_modules(xml_file)
- else:
- grass.debug("No addons metadata file available", 1)
- return []
- # read XML file
- try:
- tree = etree_fromfile(xml_file)
- except ETREE_EXCEPTIONS + (OSError, IOError):
- os.remove(xml_file)
- write_xml_modules(xml_file)
- return []
- ret = list()
- for tnode in tree.findall('task'):
- if flags['g']:
- desc, keyw = get_optional_params(tnode)
- ret.append('name={0}'.format(tnode.get('name').strip()))
- ret.append('description={0}'.format(desc))
- ret.append('keywords={0}'.format(keyw))
- ret.append('executables={0}'.format(','.join(
- get_module_executables(tnode))
- ))
- else:
- ret.append(tnode.get('name').strip())
- return ret
- # list extensions (read XML file from grass.osgeo.org/addons)
- def list_available_extensions(url):
- """List available extensions/modules or toolboxes (if -t is given)
- For toolboxes it lists also all modules.
- """
- gscript.debug("list_available_extensions(url={0})".format(url))
- if flags['t']:
- grass.message(_("List of available extensions (toolboxes):"))
- tlist = get_available_toolboxes(url)
- for toolbox_code, toolbox_data in tlist.items():
- if flags['g']:
- print('toolbox_name=' + toolbox_data['name'])
- print('toolbox_code=' + toolbox_code)
- else:
- print('%s (%s)' % (toolbox_data['name'], toolbox_code))
- if flags['c'] or flags['g']:
- list_available_modules(url, toolbox_data['modules'])
- else:
- if toolbox_data['modules']:
- print(os.linesep.join(['* ' + x for x in toolbox_data['modules']]))
- else:
- grass.message(_("List of available extensions (modules):"))
- list_available_modules(url)
- def get_available_toolboxes(url):
- """Return toolboxes available in the repository"""
- tdict = dict()
- url = url + "toolboxes.xml"
- try:
- tree = etree_fromurl(url, proxies=PROXIES)
- for tnode in tree.findall('toolbox'):
- mlist = list()
- clist = list()
- tdict[tnode.get('code')] = {'name': tnode.get('name'),
- 'correlate': clist,
- 'modules': mlist}
- for cnode in tnode.findall('correlate'):
- clist.append(cnode.get('name'))
- for mnode in tnode.findall('task'):
- mlist.append(mnode.get('name'))
- except (HTTPError, IOError, OSError):
- grass.fatal(_("Unable to fetch addons metadata file"))
- return tdict
- def get_toolbox_modules(url, name):
- """Get modules inside a toolbox in toolbox file at given URL
- :param url: URL of the directory (file name will be attached)
- :param name: toolbox name
- """
- tlist = list()
- url = url + "toolboxes.xml"
- try:
- tree = etree_fromurl(url, proxies=PROXIES)
- for tnode in tree.findall('toolbox'):
- if name == tnode.get('code'):
- for mnode in tnode.findall('task'):
- tlist.append(mnode.get('name'))
- break
- except (HTTPError, IOError, OSError):
- grass.fatal(_("Unable to fetch addons metadata file"))
- return tlist
- def get_module_files(mnode):
- """Return list of module files
- :param mnode: XML node for a module
- """
- flist = []
- for file_node in mnode.find('binary').findall('file'):
- filepath = file_node.text
- flist.append(filepath)
- return flist
- def get_module_executables(mnode):
- """Return list of module executables
- :param mnode: XML node for a module
- """
- flist = []
- for filepath in get_module_files(mnode):
- if filepath.startswith(options['prefix'] + os.path.sep + 'bin') or \
- (sys.platform != 'win32' and
- filepath.startswith(options['prefix'] + os.path.sep + 'scripts')):
- filename = os.path.basename(filepath)
- if sys.platform == 'win32':
- filename = os.path.splitext(filename)[0]
- flist.append(filename)
- return flist
- def get_optional_params(mnode):
- """Return description and keywords as a tuple
- :param mnode: XML node for a module
- """
- try:
- desc = mnode.find('description').text
- except AttributeError:
- desc = ''
- if desc is None:
- desc = ''
- try:
- keyw = mnode.find('keywords').text
- except AttributeError:
- keyw = ''
- if keyw is None:
- keyw = ''
- return desc, keyw
- def list_available_modules(url, mlist=None):
- """List modules available in the repository
- Tries to use XML metadata file first. Fallbacks to HTML page with a list.
- :param url: URL of the directory (file name will be attached)
- :param mlist: list only modules in this list
- """
- file_url = url + "modules.xml"
- grass.debug("url=%s" % file_url, 1)
- try:
- tree = etree_fromurl(file_url, proxies=PROXIES)
- except ETREE_EXCEPTIONS:
- grass.warning(_("Unable to parse '%s'. Trying to scan"
- " SVN repository (may take some time)...") % file_url)
- list_available_extensions_svn(url)
- return
- except (HTTPError, URLError, IOError, OSError):
- list_available_extensions_svn(url)
- return
- for mnode in tree.findall('task'):
- name = mnode.get('name').strip()
- if mlist and name not in mlist:
- continue
- if flags['c'] or flags['g']:
- desc, keyw = get_optional_params(mnode)
- if flags['g']:
- print('name=' + name)
- print('description=' + desc)
- print('keywords=' + keyw)
- elif flags['c']:
- if mlist:
- print('*', end='')
- print(name + ' - ' + desc)
- else:
- print(name)
- # TODO: this is now broken/dead code, SVN is basically not used
- # fallback for Trac should parse Trac HTML page
- # this might be useful for potential SVN repos or anything
- # which would list the extensions/addons as list
- # TODO: fail when nothing is accessible
- def list_available_extensions_svn(url):
- """List available extensions from HTML given by URL
- Filename is generated based on the module class/family.
- This works well for the structure which is in grass-addons repository.
- ``<li><a href=...`` is parsed to find module names.
- This works well for HTML page generated by Subversion.
- :param url: a directory URL (filename will be attached)
- """
- gscript.debug("list_available_extensions_svn(url=%s)" % url, 2)
- grass.message(_('Fetching list of extensions from'
- ' GRASS-Addons SVN repository (be patient)...'))
- pattern = re.compile(r'(<li><a href=".+">)(.+)(</a></li>)', re.IGNORECASE)
- if flags['c']:
- grass.warning(
- _("Flag 'c' ignored, addons metadata file not available"))
- if flags['g']:
- grass.warning(
- _("Flag 'g' ignored, addons metadata file not available"))
- prefixes = ['d', 'db', 'g', 'i', 'm', 'ps',
- 'p', 'r', 'r3', 's', 't', 'v']
- for prefix in prefixes:
- modclass = expand_module_class_name(prefix)
- grass.verbose(_("Checking for '%s' modules...") % modclass)
- # construct a full URL of a file
- file_url = '%s/%s' % (url, modclass)
- grass.debug("url = %s" % file_url, debug=2)
- try:
- proxy_handler = ProxyHandler(PROXIES)
- opener = build_opener(proxy_handler)
- file_ = opener.open(url)
- except (HTTPError, IOError, OSError):
- grass.debug(_("Unable to fetch '%s'") % file_url, debug=1)
- continue
- for line in file_.readlines():
- # list extensions
- sline = pattern.search(line)
- if not sline:
- continue
- name = sline.group(2).rstrip('/')
- if name.split('.', 1)[0] == prefix:
- print(name)
- # get_wxgui_extensions(url)
- # TODO: this is a dead code, not clear why not used, but seems not needed
- def get_wxgui_extensions(url):
- """Return list of extensions/addons in wxGUI directory at given URL
- :param url: a directory URL (filename will be attached)
- """
- mlist = list()
- grass.debug('Fetching list of wxGUI extensions from '
- 'GRASS-Addons SVN repository (be patient)...')
- pattern = re.compile(r'(<li><a href=".+">)(.+)(</a></li>)', re.IGNORECASE)
- grass.verbose(_("Checking for '%s' modules...") % 'gui/wxpython')
- # construct a full URL of a file
- url = '%s/%s' % (url, 'gui/wxpython')
- grass.debug("url = %s" % url, debug=2)
- proxy_handler = ProxyHandler(PROXIES)
- opener = build_opener(proxy_handler)
- file_ = opener.open(url)
- if not file_:
- grass.warning(_("Unable to fetch '%s'") % url)
- return
- for line in file.readlines():
- # list extensions
- sline = pattern.search(line)
- if not sline:
- continue
- name = sline.group(2).rstrip('/')
- if name not in ('..', 'Makefile'):
- mlist.append(name)
- return mlist
- def cleanup():
- """Cleanup after the downloads and copilation"""
- if REMOVE_TMPDIR:
- try_rmdir(TMPDIR)
- else:
- grass.message("\n%s\n" % _("Path to the source code:"))
- sys.stderr.write('%s\n' % os.path.join(TMPDIR, options['extension']))
- def write_xml_modules(name, tree=None):
- """Write element tree as a modules matadata file
- If the *tree* is not given, an empty file is created.
- :param name: file name
- :param tree: XML element tree
- """
- file_ = open(name, 'w')
- file_.write('<?xml version="1.0" encoding="UTF-8"?>\n')
- file_.write('<!DOCTYPE task SYSTEM "grass-addons.dtd">\n')
- file_.write('<addons version="%s">\n' % version[0])
- libgis_revison = grass.version()['libgis_revision']
- if tree is not None:
- for tnode in tree.findall('task'):
- indent = 4
- file_.write('%s<task name="%s">\n' %
- (' ' * indent, tnode.get('name')))
- indent += 4
- file_.write('%s<description>%s</description>\n' %
- (' ' * indent, tnode.find('description').text))
- file_.write('%s<keywords>%s</keywords>\n' %
- (' ' * indent, tnode.find('keywords').text))
- bnode = tnode.find('binary')
- if bnode is not None:
- file_.write('%s<binary>\n' % (' ' * indent))
- indent += 4
- for fnode in bnode.findall('file'):
- file_.write('%s<file>%s</file>\n' %
- (' ' * indent, os.path.join(options['prefix'],
- fnode.text)))
- indent -= 4
- file_.write('%s</binary>\n' % (' ' * indent))
- file_.write('%s<libgis revision="%s" />\n' %
- (' ' * indent, libgis_revison))
- indent -= 4
- file_.write('%s</task>\n' % (' ' * indent))
- file_.write('</addons>\n')
- file_.close()
- def write_xml_toolboxes(name, tree=None):
- """Write element tree as a toolboxes matadata file
- If the *tree* is not given, an empty file is created.
- :param name: file name
- :param tree: XML element tree
- """
- file_ = open(name, 'w')
- file_.write('<?xml version="1.0" encoding="UTF-8"?>\n')
- file_.write('<!DOCTYPE toolbox SYSTEM "grass-addons.dtd">\n')
- file_.write('<addons version="%s">\n' % version[0])
- if tree is not None:
- for tnode in tree.findall('toolbox'):
- indent = 4
- file_.write('%s<toolbox name="%s" code="%s">\n' %
- (' ' * indent, tnode.get('name'), tnode.get('code')))
- indent += 4
- for cnode in tnode.findall('correlate'):
- file_.write('%s<correlate code="%s" />\n' %
- (' ' * indent, tnode.get('code')))
- for mnode in tnode.findall('task'):
- file_.write('%s<task name="%s" />\n' %
- (' ' * indent, mnode.get('name')))
- indent -= 4
- file_.write('%s</toolbox>\n' % (' ' * indent))
- file_.write('</addons>\n')
- file_.close()
- def install_extension(source, url, xmlurl):
- """Install extension (e.g. one module) or a toolbox (list of modules)"""
- gisbase = os.getenv('GISBASE')
- if not gisbase:
- grass.fatal(_('$GISBASE not defined'))
- if options['extension'] in get_installed_extensions(force=True):
- grass.warning(_("Extension <%s> already installed. Re-installing...") %
- options['extension'])
- if flags['t']:
- grass.message(_("Installing toolbox <%s>...") % options['extension'])
- mlist = get_toolbox_modules(xmlurl, options['extension'])
- else:
- mlist = [options['extension']]
- if not mlist:
- grass.warning(_("Nothing to install"))
- return
- ret = 0
- for module in mlist:
- if sys.platform == "win32":
- ret += install_extension_win(module)
- else:
- ret += install_extension_std_platforms(module,
- source=source, url=url)
- if len(mlist) > 1:
- print('-' * 60)
- if flags['d']:
- return
- if ret != 0:
- grass.warning(_('Installation failed, sorry.'
- ' Please check above error messages.'))
- else:
- # for now it is reasonable to assume that only official source
- # will provide the metadata file
- if source == 'official':
- grass.message(_("Updating addons metadata file..."))
- blist = install_extension_xml(xmlurl, mlist)
- # the blist was used here, but it seems that it is the same as mlist
- for module in mlist:
- update_manual_page(module)
- grass.message(_("Installation of <%s> successfully finished") %
- options['extension'])
- if not os.getenv('GRASS_ADDON_BASE'):
- grass.warning(_('This add-on module will not function until'
- ' you set the GRASS_ADDON_BASE environment'
- ' variable (see "g.manual variables")'))
- def get_toolboxes_metadata(url):
- """Return metadata for all toolboxes from given URL
- :param url: URL of a modules matadata file
- :param mlist: list of modules to get metadata for
- :returns: tuple where first item is dictionary with module names as keys
- and dictionary with dest, keyw, files keys as value, the second item
- is list of 'binary' files (installation files)
- """
- data = dict()
- try:
- tree = etree_fromurl(url, proxies=PROXIES)
- for tnode in tree.findall('toolbox'):
- clist = list()
- for cnode in tnode.findall('correlate'):
- clist.append(cnode.get('code'))
- mlist = list()
- for mnode in tnode.findall('task'):
- mlist.append(mnode.get('name'))
- code = tnode.get('code')
- data[code] = {
- 'name': tnode.get('name'),
- 'correlate': clist,
- 'modules': mlist,
- }
- except (HTTPError, IOError, OSError):
- grass.error(_("Unable to read addons metadata file "
- "from the remote server"))
- return data
- def install_toolbox_xml(url, name):
- """Update local toolboxes metadata file"""
- # read metadata from remote server (toolboxes)
- url = url + "toolboxes.xml"
- data = get_toolboxes_metadata(url)
- if not data:
- grass.warning(_("No addons metadata available"))
- return
- if name not in data:
- grass.warning(_("No addons metadata available for <%s>") % name)
- return
- xml_file = os.path.join(options['prefix'], 'toolboxes.xml')
- # create an empty file if not exists
- if not os.path.exists(xml_file):
- write_xml_modules(xml_file)
- # read XML file
- with open(xml_file, 'r') as xml:
- tree = etree.fromstring(xml.read())
- # update tree
- tnode = None
- for node in tree.findall('toolbox'):
- if node.get('code') == name:
- tnode = node
- break
- tdata = data[name]
- if tnode is not None:
- # update existing node
- for cnode in tnode.findall('correlate'):
- tnode.remove(cnode)
- for mnode in tnode.findall('task'):
- tnode.remove(mnode)
- else:
- # create new node for task
- tnode = etree.Element(
- 'toolbox', attrib={'name': tdata['name'], 'code': name})
- tree.append(tnode)
- for cname in tdata['correlate']:
- cnode = etree.Element('correlate', attrib={'code': cname})
- tnode.append(cnode)
- for tname in tdata['modules']:
- mnode = etree.Element('task', attrib={'name': tname})
- tnode.append(mnode)
- write_xml_toolboxes(xml_file, tree)
- def get_addons_metadata(url, mlist):
- """Return metadata for list of modules from given URL
- :param url: URL of a modules matadata file
- :param mlist: list of modules to get metadata for
- :returns: tuple where first item is dictionary with module names as keys
- and dictionary with dest, keyw, files keys as value, the second item
- is list of 'binary' files (installation files)
- """
- data = {}
- bin_list = []
- try:
- tree = etree_fromurl(url, proxies=PROXIES)
- except (HTTPError, URLError, IOError, OSError) as error:
- grass.error(_("Unable to read addons metadata file"
- " from the remote server: {0}").format(error))
- return data, bin_list
- except ETREE_EXCEPTIONS as error:
- grass.warning(_("Unable to parse '%s': {0}").format(error) % url)
- return data, bin_list
- for mnode in tree.findall('task'):
- name = mnode.get('name')
- if name not in mlist:
- continue
- file_list = list()
- bnode = mnode.find('binary')
- windows = sys.platform == 'win32'
- if bnode is not None:
- for fnode in bnode.findall('file'):
- path = fnode.text.split('/')
- if path[0] == 'bin':
- bin_list.append(path[-1])
- if windows:
- path[-1] += '.exe'
- elif path[0] == 'scripts':
- bin_list.append(path[-1])
- if windows:
- path[-1] += '.py'
- file_list.append(os.path.sep.join(path))
- desc, keyw = get_optional_params(mnode)
- data[name] = {
- 'desc': desc,
- 'keyw': keyw,
- 'files': file_list,
- }
- return data, bin_list
- def install_extension_xml(url, mlist):
- """Update XML files with metadata about installed modules and toolbox
- Uses the remote/repository XML files for modules to obtain the metadata.
- :returns: list of executables (usable for ``update_manual_page()``)
- """
- if len(mlist) > 1:
- # read metadata from remote server (toolboxes)
- install_toolbox_xml(url, options['extension'])
- # read metadata from remote server (modules)
- url = url + "modules.xml"
- data, bin_list = get_addons_metadata(url, mlist)
- if not data:
- grass.warning(_("No addons metadata available."
- " Addons metadata file not updated."))
- return []
- xml_file = os.path.join(options['prefix'], 'modules.xml')
- # create an empty file if not exists
- if not os.path.exists(xml_file):
- write_xml_modules(xml_file)
- # read XML file
- tree = etree_fromfile(xml_file)
- # update tree
- for name in mlist:
- tnode = None
- for node in tree.findall('task'):
- if node.get('name') == name:
- tnode = node
- break
- if name not in data:
- grass.warning(_("No addons metadata found for <%s>") % name)
- continue
- ndata = data[name]
- if tnode is not None:
- # update existing node
- dnode = tnode.find('description')
- if dnode is not None:
- dnode.text = ndata['desc']
- knode = tnode.find('keywords')
- if knode is not None:
- knode.text = ndata['keyw']
- bnode = tnode.find('binary')
- if bnode is not None:
- tnode.remove(bnode)
- bnode = etree.Element('binary')
- for file_name in ndata['files']:
- fnode = etree.Element('file')
- fnode.text = file_name
- bnode.append(fnode)
- tnode.append(bnode)
- else:
- # create new node for task
- tnode = etree.Element('task', attrib={'name': name})
- dnode = etree.Element('description')
- dnode.text = ndata['desc']
- tnode.append(dnode)
- knode = etree.Element('keywords')
- knode.text = ndata['keyw']
- tnode.append(knode)
- bnode = etree.Element('binary')
- for file_name in ndata['files']:
- fnode = etree.Element('file')
- fnode.text = file_name
- bnode.append(fnode)
- tnode.append(bnode)
- tree.append(tnode)
- write_xml_modules(xml_file, tree)
- return bin_list
- def install_extension_win(name):
- """Install extension on MS Windows"""
- grass.message(_("Downloading precompiled GRASS Addons <%s>...") %
- options['extension'])
- # build base URL
- if build_platform == 'x86_64':
- platform = build_platform
- else:
- platform = 'x86'
- base_url = "http://wingrass.fsv.cvut.cz/" \
- "grass%(major)s%(minor)s/%(platform)s/addons/" \
- "grass-%(major)s.%(minor)s.%(patch)s" % \
- {'platform': platform,
- 'major': version[0], 'minor': version[1],
- 'patch': version[2]}
- # resolve ZIP URL
- source, url = resolve_source_code(url='{0}/{1}.zip'.format(base_url, name))
- # to hide non-error messages from subprocesses
- if grass.verbosity() <= 2:
- outdev = open(os.devnull, 'w')
- else:
- outdev = sys.stdout
- # download Addons ZIP file
- os.chdir(TMPDIR) # this is just to not leave something behind
- srcdir = os.path.join(TMPDIR, name)
- download_source_code(source=source, url=url, name=name,
- outdev=outdev, directory=srcdir, tmpdir=TMPDIR)
- # copy Addons copy tree to destination directory
- move_extracted_files(extract_dir=srcdir, target_dir=options['prefix'],
- files=os.listdir(srcdir))
- return 0
- def download_source_code_svn(url, name, outdev, directory=None):
- """Download source code from a Subversion reporsitory
- .. note:
- Stdout is passed to to *outdev* while stderr is will be just printed.
- :param url: URL of the repository
- (module class/family and name are attached)
- :param name: module name
- :param outdev: output divide for the standard output of the svn command
- :param directory: directory where the source code will be downloaded
- (default is the current directory with name attached)
- :returns: full path to the directory with the source code
- (useful when you not specify directory, if *directory* is specified
- the return value is equal to it)
- """
- if not directory:
- directory = os.path.join(os.getcwd, name)
- classchar = name.split('.', 1)[0]
- moduleclass = expand_module_class_name(classchar)
- url = url + '/' + moduleclass + '/' + name
- if grass.call(['svn', 'checkout',
- url, directory], stdout=outdev) != 0:
- grass.fatal(_("GRASS Addons <%s> not found") % name)
- return directory
- def move_extracted_files(extract_dir, target_dir, files):
- """Fix state of extracted file by moving them to different diretcory
- When extracting, it is not clear what will be the root directory
- or if there will be one at all. So this function moves the files to
- a different directory in the way that if there was one directory extracted,
- the contained files are moved.
- """
- gscript.debug("move_extracted_files({0})".format(locals()))
- if len(files) == 1:
- shutil.copytree(os.path.join(extract_dir, files[0]), target_dir)
- else:
- if not os.path.exists(target_dir):
- os.mkdir(target_dir)
- for file_name in files:
- actual_file = os.path.join(extract_dir, file_name)
- if os.path.isdir(actual_file):
- # shutil.copytree() replaced by copy_tree() because
- # shutil's copytree() fails when subdirectory exists
- copy_tree(actual_file,
- os.path.join(target_dir, file_name))
- else:
- shutil.copy(actual_file, os.path.join(target_dir, file_name))
- # Original copyright and license of the original version of the CRLF function
- # Copyright (c) 2001, 2002, 2003, 2004, 2005, 2006, 2007, 2008, 2009, 2010
- # Python Software Foundation; All Rights Reserved
- # Python Software Foundation License Version 2
- # http://svn.python.org/projects/python/trunk/Tools/scripts/crlf.py
- def fix_newlines(directory):
- """Replace CRLF with LF in all files in the directory
- Binary files are ignored. Recurses into subdirectories.
- """
- for root, unused, files in os.walk(directory):
- for name in files:
- filename = os.path.join(root, name)
- data = open(filename, 'rb').read()
- if '\0' in data:
- continue # ignore binary files
- # we don't expect there would be CRLF file by purpose
- # if we want to allow CRLF files we would have to whitelite .py etc
- newdata = data.replace(b'\r\n', b'\n')
- if newdata != data:
- newfile = open(filename, 'wb')
- newfile.write(newdata)
- newfile.close()
- def extract_zip(name, directory, tmpdir):
- """Extract a ZIP file into a directory"""
- gscript.debug("extract_zip(name={name}, directory={directory},"
- " tmpdir={tmpdir})".format(name=name, directory=directory,
- tmpdir=tmpdir), 3)
- try:
- zip_file = zipfile.ZipFile(name, mode='r')
- file_list = zip_file.namelist()
- # we suppose we can write to parent of the given dir
- # (supposing a tmp dir)
- extract_dir = os.path.join(tmpdir, 'extract_dir')
- os.mkdir(extract_dir)
- for subfile in file_list:
- # this should be safe in Python 2.7.4
- zip_file.extract(subfile, extract_dir)
- files = os.listdir(extract_dir)
- move_extracted_files(extract_dir=extract_dir,
- target_dir=directory, files=files)
- except zipfile.BadZipfile as error:
- gscript.fatal(_("ZIP file is unreadable: {0}").format(error))
- # TODO: solve the other related formats
- def extract_tar(name, directory, tmpdir):
- """Extract a TAR or a similar file into a directory"""
- gscript.debug("extract_tar(name={name}, directory={directory},"
- " tmpdir={tmpdir})".format(name=name, directory=directory,
- tmpdir=tmpdir), 3)
- try:
- import tarfile # we don't need it anywhere else
- tar = tarfile.open(name)
- extract_dir = os.path.join(tmpdir, 'extract_dir')
- os.mkdir(extract_dir)
- tar.extractall(path=extract_dir)
- files = os.listdir(extract_dir)
- move_extracted_files(extract_dir=extract_dir,
- target_dir=directory, files=files)
- except tarfile.TarError as error:
- gscript.fatal(_("Archive file is unreadable: {0}").format(error))
- extract_tar.supported_formats = ['tar.gz', 'gz', 'bz2', 'tar', 'gzip', 'targz']
- def download_source_code(source, url, name, outdev,
- directory=None, tmpdir=None):
- """Get source code to a local directory for compilation"""
- gscript.verbose("Downloading source code for <{name}> from <{url}>"
- " which is identified as '{source}' type of source..."
- .format(source=source, url=url, name=name))
- if source == 'svn':
- download_source_code_svn(url, name, outdev, directory)
- elif source in ['remote_zip', 'official']:
- # we expect that the module.zip file is not by chance in the archive
- zip_name = os.path.join(tmpdir, 'extension.zip')
- f, h = urlretrieve(url, zip_name)
- if h.get('content-type', '') != 'application/zip':
- grass.fatal(_("Extension <%s> not found") % name)
- extract_zip(name=zip_name, directory=directory, tmpdir=tmpdir)
- fix_newlines(directory)
- elif source.startswith('remote_') and \
- source.split('_')[1] in extract_tar.supported_formats:
- # we expect that the module.tar.gz file is not by chance in the archive
- archive_name = os.path.join(tmpdir,
- 'extension.' + source.split('_')[1])
- urlretrieve(url, archive_name)
- extract_tar(name=archive_name, directory=directory, tmpdir=tmpdir)
- fix_newlines(directory)
- elif source == 'zip':
- extract_zip(name=url, directory=directory, tmpdir=tmpdir)
- fix_newlines(directory)
- elif source in extract_tar.supported_formats:
- extract_tar(name=url, directory=directory, tmpdir=tmpdir)
- fix_newlines(directory)
- elif source == 'dir':
- shutil.copytree(url, directory)
- fix_newlines(directory)
- else:
- # probably programmer error
- grass.fatal(_("Unknown extension (addon) source type '{0}'."
- " Please report this to the grass-user mailing list.")
- .format(source))
- assert os.path.isdir(directory)
- def install_extension_std_platforms(name, source, url):
- """Install extension on standard platforms"""
- gisbase = os.getenv('GISBASE')
- source_url = "https://trac.osgeo.org/grass/browser/grass-addons/grass7/"
- if source == 'official':
- gscript.message(_("Fetching <%s> from "
- "GRASS GIS Addons repository (be patient)...") % name)
- else:
- gscript.message(_("Fetching <{name}> from "
- "<{url}> (be patient)...").format(name=name, url=url))
- # to hide non-error messages from subprocesses
- if grass.verbosity() <= 2:
- outdev = open(os.devnull, 'w')
- else:
- outdev = sys.stdout
- os.chdir(TMPDIR) # this is just to not leave something behind
- srcdir = os.path.join(TMPDIR, name)
- download_source_code(source=source, url=url, name=name,
- outdev=outdev, directory=srcdir, tmpdir=TMPDIR)
- os.chdir(srcdir)
- dirs = {
- 'bin': os.path.join(TMPDIR, name, 'bin'),
- 'docs': os.path.join(TMPDIR, name, 'docs'),
- 'html': os.path.join(TMPDIR, name, 'docs', 'html'),
- 'rest': os.path.join(TMPDIR, name, 'docs', 'rest'),
- 'man': os.path.join(TMPDIR, name, 'docs', 'man'),
- 'script': os.path.join(TMPDIR, name, 'scripts'),
- # TODO: handle locales also for addons
- # 'string' : os.path.join(TMPDIR, name, 'locale'),
- 'string': os.path.join(TMPDIR, name),
- 'etc': os.path.join(TMPDIR, name, 'etc'),
- }
- make_cmd = [
- 'make',
- 'MODULE_TOPDIR=%s' % gisbase.replace(' ', r'\ '),
- 'RUN_GISRC=%s' % os.environ['GISRC'],
- 'BIN=%s' % dirs['bin'],
- 'HTMLDIR=%s' % dirs['html'],
- 'RESTDIR=%s' % dirs['rest'],
- 'MANBASEDIR=%s' % dirs['man'],
- 'SCRIPTDIR=%s' % dirs['script'],
- 'STRINGDIR=%s' % dirs['string'],
- 'ETC=%s' % os.path.join(dirs['etc']),
- 'SOURCE_URL=%s' % source_url
- ]
- install_cmd = [
- 'make',
- 'MODULE_TOPDIR=%s' % gisbase,
- 'ARCH_DISTDIR=%s' % os.path.join(TMPDIR, name),
- 'INST_DIR=%s' % options['prefix'],
- 'install'
- ]
- if flags['d']:
- grass.message("\n%s\n" % _("To compile run:"))
- sys.stderr.write(' '.join(make_cmd) + '\n')
- grass.message("\n%s\n" % _("To install run:"))
- sys.stderr.write(' '.join(install_cmd) + '\n')
- return 0
- os.chdir(os.path.join(TMPDIR, name))
- grass.message(_("Compiling..."))
- if not os.path.exists(os.path.join(gisbase, 'include',
- 'Make', 'Module.make')):
- grass.fatal(_("Please install GRASS development package"))
- if 0 != grass.call(make_cmd,
- stdout=outdev):
- grass.fatal(_('Compilation failed, sorry.'
- ' Please check above error messages.'))
- if flags['i']:
- return 0
- grass.message(_("Installing..."))
- return grass.call(install_cmd,
- stdout=outdev)
- def remove_extension(force=False):
- """Remove existing extension (module or toolbox if -t is given)"""
- if flags['t']:
- mlist = get_toolbox_modules(options['prefix'], options['extension'])
- else:
- mlist = [options['extension']]
- if force:
- grass.verbose(_("List of removed files:"))
- else:
- grass.info(_("Files to be removed:"))
- remove_modules(mlist, force)
- if force:
- grass.message(_("Updating addons metadata file..."))
- remove_extension_xml(mlist)
- grass.message(_("Extension <%s> successfully uninstalled.") %
- options['extension'])
- else:
- grass.warning(_("Extension <%s> not removed. "
- "Re-run '%s' with '-f' flag to force removal")
- % (options['extension'], 'g.extension'))
- # remove existing extension(s) (reading XML file)
- def remove_modules(mlist, force=False):
- """Remove extensions/modules specified in a list
- Collects the file names from the file with metadata and fallbacks
- to standard layout of files on prefix path on error.
- """
- # try to read XML metadata file first
- xml_file = os.path.join(options['prefix'], 'modules.xml')
- installed = get_installed_modules()
- if os.path.exists(xml_file):
- tree = etree_fromfile(xml_file)
- else:
- tree = None
- for name in mlist:
- if name not in installed:
- # try even if module does not seem to be available,
- # as the user may be trying to get rid of left over cruft
- grass.warning(_("Extension <%s> not found") % name)
- if tree is not None:
- flist = []
- for task in tree.findall('task'):
- if name == task.get('name') and \
- task.find('binary') is not None:
- flist = get_module_files(task)
- break
- if flist:
- removed = False
- err = list()
- for fpath in flist:
- grass.verbose(fpath)
- if force:
- try:
- os.remove(fpath)
- removed = True
- except OSError:
- msg = "Unable to remove file '%s'"
- err.append((_(msg) % fpath))
- if force and not removed:
- grass.fatal(_("Extension <%s> not found") % name)
- if err:
- for error_line in err:
- grass.error(error_line)
- else:
- remove_extension_std(name, force)
- else:
- remove_extension_std(name, force)
- # remove module libraries directories under GRASS_ADDONS/etc/{name}/*
- libpath = os.path.join(options['prefix'], 'etc', name)
- if os.path.isdir(libpath):
- grass.verbose(libpath)
- if force:
- shutil.rmtree(libpath)
- def remove_extension_std(name, force=False):
- """Remove extension/module expecting the standard layout"""
- for fpath in [os.path.join(options['prefix'], 'bin', name),
- os.path.join(options['prefix'], 'scripts', name),
- os.path.join(
- options['prefix'], 'docs', 'html', name + '.html'),
- os.path.join(
- options['prefix'], 'docs', 'rest', name + '.txt'),
- os.path.join(options['prefix'], 'docs', 'man', 'man1',
- name + '.1')]:
- if os.path.isfile(fpath):
- grass.verbose(fpath)
- if force:
- os.remove(fpath)
- # remove module libraries under GRASS_ADDONS/etc/{name}/*
- libpath = os.path.join(options['prefix'], 'etc', name)
- if os.path.isdir(libpath):
- grass.verbose(libpath)
- if force:
- shutil.rmtree(libpath)
- def remove_from_toolbox_xml(name):
- """Update local meta-file when removing existing toolbox"""
- xml_file = os.path.join(options['prefix'], 'toolboxes.xml')
- if not os.path.exists(xml_file):
- return
- # read XML file
- tree = etree_fromfile(xml_file)
- for node in tree.findall('toolbox'):
- if node.get('code') != name:
- continue
- tree.remove(node)
- write_xml_toolboxes(xml_file, tree)
- def remove_extension_xml(modules):
- """Update local meta-file when removing existing extension"""
- if len(modules) > 1:
- # update also toolboxes metadata
- remove_from_toolbox_xml(options['extension'])
- xml_file = os.path.join(options['prefix'], 'modules.xml')
- if not os.path.exists(xml_file):
- return
- # read XML file
- tree = etree_fromfile(xml_file)
- for name in modules:
- for node in tree.findall('task'):
- if node.get('name') != name:
- continue
- tree.remove(node)
- write_xml_modules(xml_file, tree)
- # check links in CSS
- def check_style_files(fil):
- """Ensures that a specified HTML documentation support file exists
- If the file, e.g. a CSS file does not exist, the file is copied from
- the distribution.
- """
- dist_file = os.path.join(os.getenv('GISBASE'), 'docs', 'html', fil)
- addons_file = os.path.join(options['prefix'], 'docs', 'html', fil)
- if os.path.isfile(addons_file):
- return
- try:
- shutil.copyfile(dist_file, addons_file)
- except OSError as error:
- grass.fatal(_("Unable to create '%s': %s") % (addons_file, error))
- def create_dir(path):
- """Creates the specified directory (with all dirs in between)
- NOOP for existing directory.
- """
- if os.path.isdir(path):
- return
- try:
- os.makedirs(path)
- except OSError as error:
- grass.fatal(_("Unable to create '%s': %s") % (path, error))
- grass.debug("'%s' created" % path)
- def check_dirs():
- """Ensure that the necessary directories in prefix path exist"""
- create_dir(os.path.join(options['prefix'], 'bin'))
- create_dir(os.path.join(options['prefix'], 'docs', 'html'))
- create_dir(os.path.join(options['prefix'], 'docs', 'rest'))
- check_style_files('grass_logo.png')
- check_style_files('grassdocs.css')
- create_dir(os.path.join(options['prefix'], 'etc'))
- create_dir(os.path.join(options['prefix'], 'docs', 'man', 'man1'))
- create_dir(os.path.join(options['prefix'], 'scripts'))
- # fix file URI in manual page
- def update_manual_page(module):
- """Fix manual page for addons which are at different directory then rest"""
- if module.split('.', 1)[0] == 'wx':
- return # skip for GUI modules
- grass.verbose(_("Manual page for <%s> updated") % module)
- # read original html file
- htmlfile = os.path.join(
- options['prefix'], 'docs', 'html', module + '.html')
- try:
- oldfile = open(htmlfile)
- shtml = oldfile.read()
- except IOError as error:
- gscript.fatal(_("Unable to read manual page: %s") % error)
- else:
- oldfile.close()
- pos = []
- # fix logo URL
- pattern = r'''<a href="([^"]+)"><img src="grass_logo.png"'''
- for match in re.finditer(pattern, shtml):
- pos.append(match.start(1))
- # find URIs
- pattern = r'''<a href="([^"]+)">([^>]+)</a>'''
- addons = get_installed_extensions(force=True)
- for match in re.finditer(pattern, shtml):
- if match.group(1)[:4] == 'http':
- continue
- if match.group(1).replace('.html', '') in addons:
- continue
- pos.append(match.start(1))
- if not pos:
- return # no match
- # replace file URIs
- prefix = 'file://' + '/'.join([os.getenv('GISBASE'), 'docs', 'html'])
- ohtml = shtml[:pos[0]]
- for i in range(1, len(pos)):
- ohtml += prefix + '/' + shtml[pos[i - 1]:pos[i]]
- ohtml += prefix + '/' + shtml[pos[-1]:]
- # write updated html file
- try:
- newfile = open(htmlfile, 'w')
- newfile.write(ohtml)
- except IOError as error:
- gscript.fatal(_("Unable for write manual page: %s") % error)
- else:
- newfile.close()
- def resolve_install_prefix(path, to_system):
- """Determine and check the path for installation"""
- if to_system:
- path = os.environ['GISBASE']
- if path == '$GRASS_ADDON_BASE':
- if not os.getenv('GRASS_ADDON_BASE'):
- grass.warning(_("GRASS_ADDON_BASE is not defined, "
- "installing to ~/.grass%s/addons") % version[0])
- path = os.path.join(
- os.environ['HOME'], '.grass%s' % version[0], 'addons')
- else:
- path = os.environ['GRASS_ADDON_BASE']
- if os.path.exists(path) and \
- not os.access(path, os.W_OK):
- grass.fatal(_("You don't have permission to install extension to <{0}>."
- " Try to run {1} with administrator rights"
- " (su or sudo).")
- .format(path, 'g.extension'))
- # ensure dir sep at the end for cases where path is used as URL and pasted
- # together with file names
- if not path.endswith(os.path.sep):
- path = path + os.path.sep
- return os.path.abspath(path) # make likes absolute paths
- def resolve_xmlurl_prefix(url, source=None):
- """Determine and check the URL where the XML metadata files are stored
- It ensures that there is a single slash at the end of URL, so we can attach
- file name easily:
- >>> resolve_xmlurl_prefix('http://grass.osgeo.org/addons')
- 'http://grass.osgeo.org/addons/'
- >>> resolve_xmlurl_prefix('http://grass.osgeo.org/addons/')
- 'http://grass.osgeo.org/addons/'
- """
- gscript.debug("resolve_xmlurl_prefix(url={0}, source={1})".format(url, source))
- if source == 'official':
- # use pregenerated modules XML file
- url = 'http://grass.osgeo.org/addons/grass%s/' % version[0]
- # else try to get modules XMl from SVN repository (provided URL)
- # the exact action depends on subsequent code (somewhere)
- if not url.endswith('/'):
- url = url + '/'
- return url
- KNOWN_HOST_SERVICES_INFO = {
- 'OSGeo Trac': {
- 'domain': 'trac.osgeo.org',
- 'ignored_suffixes': ['format=zip'],
- 'possible_starts': ['', 'https://', 'http://'],
- 'url_start': 'https://',
- 'url_end': '?format=zip',
- },
- 'GitHub': {
- 'domain': 'github.com',
- 'ignored_suffixes': ['.zip', '.tar.gz'],
- 'possible_starts': ['', 'https://', 'http://'],
- 'url_start': 'https://',
- 'url_end': '/archive/master.zip',
- },
- 'GitLab': {
- 'domain': 'gitlab.com',
- 'ignored_suffixes': ['.zip', '.tar.gz', '.tar.bz2', '.tar'],
- 'possible_starts': ['', 'https://', 'http://'],
- 'url_start': 'https://',
- 'url_end': '/repository/archive.zip',
- },
- 'Bitbucket': {
- 'domain': 'bitbucket.org',
- 'ignored_suffixes': ['.zip', '.tar.gz', '.gz', '.bz2'],
- 'possible_starts': ['', 'https://', 'http://'],
- 'url_start': 'https://',
- 'url_end': '/get/default.zip',
- },
- }
- # TODO: support ZIP URLs which don't end with zip
- # https://gitlab.com/user/reponame/repository/archive.zip?ref=b%C3%A9po
- def resolve_known_host_service(url):
- """Determine source type and full URL for known hosting service
- If the service is not determined from the provided URL, tuple with
- is two ``None`` values is returned.
- """
- match = None
- actual_start = None
- for key, value in KNOWN_HOST_SERVICES_INFO.items():
- for start in value['possible_starts']:
- if url.startswith(start + value['domain']):
- match = value
- actual_start = start
- gscript.verbose(_("Identified {0} as known hosting service")
- .format(key))
- for suffix in value['ignored_suffixes']:
- if url.endswith(suffix):
- gscript.verbose(
- _("Not using {service} as known hosting service"
- " because the URL ends with '{suffix}'")
- .format(service=key, suffix=suffix))
- return None, None
- if match:
- if not actual_start:
- actual_start = match['url_start']
- else:
- actual_start = ''
- url = '{prefix}{base}{suffix}'.format(prefix=actual_start,
- base=url.rstrip('/'),
- suffix=match['url_end'])
- gscript.verbose(_("Will use the following URL for download: {0}")
- .format(url))
- return 'remote_zip', url
- else:
- return None, None
- # TODO: add also option to enforce the source type
- def resolve_source_code(url=None, name=None):
- """Return type and URL or path of the source code
- Local paths are not presented as URLs to be usable in standard functions.
- Path is identified as local path if the directory of file exists which
- has the unfortunate consequence that the not existing files are evaluated
- as remote URLs. When path is not evaluated, Subversion is assumed for
- backwards compatibility. When GitHub repository is specified, ZIP file
- link is returned. The ZIP is for master branch, not the default one because
- GitHub does not provide the default branch in the URL (July 2015).
- :returns: tuple with type of source and full URL or path
- Official repository:
- >>> resolve_source_code(name='g.example') # doctest: +SKIP
- ('official', 'https://trac.osgeo.org/.../general/g.example')
- Subversion:
- >>> resolve_source_code('http://svn.osgeo.org/grass/grass-addons/grass7')
- ('svn', 'http://svn.osgeo.org/grass/grass-addons/grass7')
- ZIP files online:
- >>> resolve_source_code('https://trac.osgeo.org/.../r.modis?format=zip')
- ('remote_zip', 'https://trac.osgeo.org/.../r.modis?format=zip')
- Local directories and ZIP files:
- >>> resolve_source_code(os.path.expanduser("~")) # doctest: +ELLIPSIS
- ('dir', '...')
- >>> resolve_source_code('/local/directory/downloaded.zip') # doctest: +SKIP
- ('zip', '/local/directory/downloaded.zip')
- OSGeo Trac:
- >>> resolve_source_code('trac.osgeo.org/.../r.agent.aco')
- ('remote_zip', 'https://trac.osgeo.org/.../r.agent.aco?format=zip')
- >>> resolve_source_code('https://trac.osgeo.org/.../r.agent.aco')
- ('remote_zip', 'https://trac.osgeo.org/.../r.agent.aco?format=zip')
- GitHub:
- >>> resolve_source_code('github.com/user/g.example')
- ('remote_zip', 'https://github.com/user/g.example/archive/master.zip')
- >>> resolve_source_code('github.com/user/g.example/')
- ('remote_zip', 'https://github.com/user/g.example/archive/master.zip')
- >>> resolve_source_code('https://github.com/user/g.example')
- ('remote_zip', 'https://github.com/user/g.example/archive/master.zip')
- >>> resolve_source_code('https://github.com/user/g.example/')
- ('remote_zip', 'https://github.com/user/g.example/archive/master.zip')
- GitLab:
- >>> resolve_source_code('gitlab.com/JoeUser/GrassModule')
- ('remote_zip', 'https://gitlab.com/JoeUser/GrassModule/repository/archive.zip')
- >>> resolve_source_code('https://gitlab.com/JoeUser/GrassModule')
- ('remote_zip', 'https://gitlab.com/JoeUser/GrassModule/repository/archive.zip')
- Bitbucket:
- >>> resolve_source_code('bitbucket.org/joe-user/grass-module')
- ('remote_zip', 'https://bitbucket.org/joe-user/grass-module/get/default.zip')
- >>> resolve_source_code('https://bitbucket.org/joe-user/grass-module')
- ('remote_zip', 'https://bitbucket.org/joe-user/grass-module/get/default.zip')
- """
- if not url and name:
- module_class = get_module_class_name(name)
- trac_url = 'https://trac.osgeo.org/grass/browser/grass-addons/' \
- 'grass{version}/{module_class}/{module_name}?format=zip' \
- .format(version=version[0],
- module_class=module_class, module_name=name)
- return 'official', trac_url
- if os.path.isdir(url):
- return 'dir', os.path.abspath(url)
- elif os.path.exists(url):
- if url.endswith('.zip'):
- return 'zip', os.path.abspath(url)
- for suffix in extract_tar.supported_formats:
- if url.endswith('.' + suffix):
- return suffix, os.path.abspath(url)
- else:
- source, resolved_url = resolve_known_host_service(url)
- if source:
- return source, resolved_url
- # we allow URL to end with =zip or ?zip and not only .zip
- # unfortunately format=zip&version=89612 would require something else
- # special option to force the source type would solve it
- if url.endswith('zip'):
- return 'remote_zip', url
- for suffix in extract_tar.supported_formats:
- if url.endswith(suffix):
- return 'remote_' + suffix, url
- # fallback to the classic behavior
- return 'svn', url
- def main():
- # check dependencies
- if not flags['a'] and sys.platform != "win32":
- check_progs()
- original_url = options['url']
- # manage proxies
- global PROXIES
- if options['proxy']:
- PROXIES = {}
- for ptype, purl in (p.split('=') for p in options['proxy'].split(',')):
- PROXIES[ptype] = purl
- # define path
- options['prefix'] = resolve_install_prefix(path=options['prefix'],
- to_system=flags['s'])
- # list available extensions
- if flags['l'] or flags['c'] or (flags['g'] and not flags['a']):
- # using dummy module, we don't need any module URL now,
- # but will work only as long as the function does not check
- # if the URL is actually valid or something
- source, url = resolve_source_code(name='dummy',
- url=original_url)
- xmlurl = resolve_xmlurl_prefix(original_url, source=source)
- list_available_extensions(xmlurl)
- return 0
- elif flags['a']:
- list_installed_extensions(toolboxes=flags['t'])
- return 0
- if flags['d']:
- if options['operation'] != 'add':
- grass.warning(_("Flag 'd' is relevant only to"
- " 'operation=add'. Ignoring this flag."))
- else:
- global REMOVE_TMPDIR
- REMOVE_TMPDIR = False
- if options['operation'] == 'add':
- check_dirs()
- source, url = resolve_source_code(name=options['extension'],
- url=original_url)
- xmlurl = resolve_xmlurl_prefix(original_url, source=source)
- install_extension(source=source, url=url, xmlurl=xmlurl)
- else: # remove
- remove_extension(force=flags['f'])
- return 0
- if __name__ == "__main__":
- if len(sys.argv) == 2 and sys.argv[1] == '--doctest':
- import doctest
- _ = str # doctest gettext workaround
- sys.exit(doctest.testmod().failed)
- options, flags = grass.parser()
- global TMPDIR
- TMPDIR = tempfile.mkdtemp()
- atexit.register(cleanup)
- grass_version = grass.version()
- version = grass_version['version'].split('.')
- build_platform = grass_version['build_platform'].split('-', 1)[0]
- sys.exit(main())
|