123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374137513761377137813791380138113821383138413851386138713881389139013911392139313941395139613971398139914001401140214031404140514061407140814091410141114121413141414151416141714181419142014211422142314241425142614271428142914301431143214331434143514361437143814391440144114421443144414451446144714481449145014511452145314541455145614571458145914601461146214631464146514661467146814691470147114721473147414751476147714781479148014811482148314841485148614871488148914901491149214931494149514961497149814991500150115021503150415051506150715081509151015111512151315141515151615171518151915201521152215231524152515261527152815291530153115321533153415351536153715381539154015411542154315441545154615471548154915501551155215531554155515561557155815591560156115621563156415651566156715681569157015711572157315741575157615771578157915801581158215831584158515861587158815891590159115921593159415951596159715981599160016011602160316041605160616071608160916101611161216131614161516161617161816191620162116221623162416251626162716281629163016311632163316341635163616371638163916401641164216431644164516461647164816491650165116521653165416551656165716581659166016611662166316641665166616671668166916701671167216731674167516761677167816791680168116821683168416851686168716881689169016911692169316941695169616971698169917001701170217031704170517061707170817091710171117121713171417151716171717181719172017211722172317241725172617271728172917301731173217331734173517361737173817391740174117421743174417451746174717481749175017511752175317541755175617571758175917601761176217631764176517661767176817691770177117721773177417751776177717781779178017811782178317841785178617871788178917901791179217931794179517961797179817991800180118021803180418051806180718081809181018111812181318141815181618171818181918201821182218231824182518261827182818291830183118321833183418351836183718381839184018411842184318441845184618471848184918501851185218531854185518561857185818591860186118621863186418651866186718681869187018711872187318741875187618771878187918801881188218831884188518861887188818891890189118921893189418951896189718981899190019011902190319041905190619071908190919101911191219131914191519161917191819191920192119221923192419251926192719281929193019311932193319341935193619371938193919401941194219431944194519461947194819491950195119521953195419551956195719581959196019611962196319641965196619671968196919701971197219731974197519761977197819791980198119821983198419851986198719881989199019911992199319941995199619971998199920002001200220032004200520062007200820092010201120122013201420152016201720182019202020212022202320242025202620272028202920302031203220332034203520362037203820392040204120422043204420452046204720482049205020512052205320542055205620572058205920602061206220632064206520662067206820692070207120722073207420752076207720782079208020812082208320842085208620872088208920902091209220932094209520962097209820992100210121022103210421052106210721082109211021112112211321142115211621172118211921202121212221232124212521262127212821292130213121322133213421352136213721382139214021412142214321442145214621472148214921502151215221532154215521562157215821592160216121622163216421652166216721682169217021712172217321742175217621772178217921802181218221832184218521862187218821892190219121922193219421952196219721982199220022012202220322042205220622072208220922102211221222132214221522162217221822192220222122222223222422252226222722282229223022312232223322342235223622372238223922402241224222432244224522462247224822492250225122522253225422552256225722582259226022612262226322642265226622672268226922702271227222732274227522762277227822792280228122822283228422852286228722882289229022912292229322942295229622972298229923002301230223032304230523062307230823092310231123122313231423152316231723182319232023212322232323242325232623272328232923302331233223332334233523362337233823392340234123422343234423452346234723482349235023512352235323542355235623572358235923602361236223632364236523662367236823692370237123722373237423752376237723782379238023812382238323842385238623872388238923902391239223932394239523962397239823992400240124022403240424052406240724082409241024112412241324142415241624172418241924202421242224232424242524262427242824292430243124322433243424352436243724382439244024412442244324442445244624472448244924502451245224532454245524562457245824592460246124622463246424652466246724682469247024712472247324742475247624772478247924802481248224832484248524862487248824892490249124922493249424952496249724982499250025012502250325042505250625072508250925102511251225132514251525162517251825192520252125222523252425252526252725282529253025312532253325342535253625372538253925402541254225432544254525462547254825492550255125522553255425552556255725582559256025612562256325642565256625672568256925702571257225732574257525762577257825792580258125822583258425852586258725882589259025912592259325942595259625972598259926002601260226032604260526062607260826092610261126122613261426152616261726182619262026212622262326242625262626272628262926302631263226332634263526362637263826392640264126422643264426452646264726482649265026512652265326542655265626572658265926602661266226632664 |
- #!/usr/bin/env python3
- ############################################################################
- #
- # 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-2021 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: - update temporary workaround of using grass7 subdir of addon-repo, see
- # https://github.com/OSGeo/grass-addons/issues/528
- # - add sudo support where needed (i.e. check first permission to write into
- # $GISBASE directory)
- # - fix toolbox support in install_private_extension_xml()
- #############################################################################
- # %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
- # %option
- # % key: branch
- # % type: string
- # % key_desc: branch
- # % description: Specific branch to fetch addon from (only used when fetching from git)
- # % required: no
- # % multiple: no
- # %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
- # %flag
- # % key: o
- # % description: url refers to a fork of the official extension repository
- # %end
- # %flag
- # % key: j
- # % description: Generates JSON file containing the download URLs of the official Addons
- # % guisection: Install
- # % suppress_required: yes
- # %end
- # %rules
- # % required: extension, -l, -c, -g, -a, -j
- # % exclusive: extension, -l, -c, -g
- # % exclusive: extension, -l, -c, -a
- # % requires: -o, url
- # % requires: branch, url
- # %end
- # TODO: solve addon-extension(-module) confusion
- from __future__ import print_function
- import fileinput
- import http
- import os
- import codecs
- import sys
- import re
- import atexit
- import shutil
- import zipfile
- import tempfile
- import json
- import xml.etree.ElementTree as etree
- from distutils.dir_util import copy_tree
- from six.moves.urllib import request as urlrequest
- from six.moves.urllib.error import HTTPError, URLError
- from six.moves.urllib.parse import urlparse
- # 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
- from grass.script import task as gtask
- # temp dir
- REMOVE_TMPDIR = True
- PROXIES = {}
- HEADERS = {
- "User-Agent": "Mozilla/5.0",
- }
- HTTP_STATUS_CODES = list(http.HTTPStatus)
- GIT_URL = "https://github.com/OSGeo/grass-addons"
- # MAKE command
- # GRASS Makefile are type of GNU Make and not BSD Make
- # On FreeBSD (and other BSD and maybe unix) we have to
- # use GNU Make program often "gmake" to distinct with the (bsd) "make"
- if sys.platform.startswith("freebsd"):
- MAKE = "gmake"
- else:
- MAKE = "make"
- def replace_shebang_win(python_file):
- """
- Replaces "python" with "python3" in python files
- using UTF8 encoding on MS Windows
- """
- cur_dir = os.path.dirname(python_file)
- tmp_name = os.path.join(cur_dir, gscript.tempname(12))
- with codecs.open(python_file, "r", encoding="utf8") as in_file, codecs.open(
- tmp_name, "w", encoding="utf8"
- ) as out_file:
- for line in in_file:
- new_line = line.replace(
- "#!/usr/bin/env python\n", "#!/usr/bin/env python3\n"
- )
- out_file.write(new_line)
- os.remove(python_file) # remove original
- os.rename(tmp_name, python_file) # rename temp to original name
- def urlretrieve(url, filename, *args, **kwargs):
- """Same function as 'urlretrieve', but with the ability to
- define headers.
- """
- request = urlrequest.Request(url, headers=HEADERS)
- response = urlrequest.urlopen(request, *args, **kwargs)
- with open(filename, "wb") as f:
- f.write(response.read())
- def urlopen(url, *args, **kwargs):
- """Wrapper around urlopen. Same function as 'urlopen', but with the
- ability to define headers.
- """
- request = urlrequest.Request(url, headers=HEADERS)
- return urlrequest.urlopen(request, *args, **kwargs)
- def get_version_branch(major_version):
- """Check if version branch for the current GRASS version exists,
- if not, take branch for the previous version
- For the official repo we assume that at least one version branch is present"""
- version_branch = f"grass{major_version}"
- try:
- urlrequest.urlopen(f"{GIT_URL}/tree/{version_branch}/src")
- except URLError:
- version_branch = "grass{}".format(int(major_version) - 1)
- return version_branch
- def get_github_branches(
- github_api_url="https://api.github.com/repos/OSGeo/grass-addons/branches",
- version_only=True,
- ):
- """Get ordered list of branch names in repo using github API
- For the official repo we assume that at least one version branch is present
- Due to strict rate limits in the github API (60 calls per hour) this function
- is currently not used."""
- req = urlrequest.urlopen(github_api_url)
- content = json.loads(req.read())
- branches = [repo_branch["name"] for repo_branch in content]
- if version_only:
- branches = [
- version_branch
- for version_branch in branches
- if version_branch.startswith("grass")
- ]
- branches.sort()
- return branches
- def get_default_branch(full_url):
- """Get default branch for repository in known hosting services
- (currently only implemented for github, gitlab and bitbucket API)
- In all other cases "main" is used as default"""
- # Parse URL
- url_parts = urlparse(full_url)
- # Get organization and repository component
- try:
- organization, repository = url_parts.path.split("/")[1:3]
- except URLError:
- gscript.fatal(
- _(
- "Cannot retrieve organization and repository from URL: <{}>.".format(
- full_url
- )
- )
- )
- # Construct API call and retrieve default branch
- api_calls = {
- "github.com": f"https://api.github.com/repos/{organization}/{repository}",
- "gitlab.com": f"https://gitlab.com/api/v4/projects/{organization}%2F{repository}",
- "bitbucket.org": f"https://api.bitbucket.org/2.0/repositories/{organization}/{repository}/branching-model?",
- }
- # Try to get default branch via API. The API call is known to fail a) if the full_url
- # does not belong to an implemented hosting service or b) if the rate limit of the
- # API is exceeded
- try:
- req = urlrequest.urlopen(api_calls.get(url_parts.netloc))
- content = json.loads(req.read())
- # For github and gitlab
- default_branch = content.get("default_branch")
- # For bitbucket
- if not default_branch:
- default_branch = content.get("development").get("name")
- except URLError:
- default_branch = "main"
- return default_branch
- def download_addons_paths_file(url, response_format, *args, **kwargs):
- """Generates JSON file containing the download URLs of the official
- Addons
- :param str url: url address
- :param str response_format: content type
- :return response: urllib.request.urlopen response object or None
- """
- try:
- response = urlopen(url, *args, **kwargs)
- if not response.code == 200:
- index = HTTP_STATUS_CODES.index(response.code)
- desc = HTTP_STATUS_CODES[index].description
- gscript.fatal(
- _(
- "Download file from <{url}>, "
- "return status code {code}, "
- "{desc}".format(
- url=url,
- code=response.code,
- desc=desc,
- ),
- ),
- )
- if response_format not in response.getheader("Content-Type"):
- gscript.fatal(
- _(
- "Wrong downloaded file format. "
- "Check url <{url}>. Allowed file format is "
- "{response_format}.".format(
- url=url,
- response_format=response_format,
- ),
- ),
- )
- return response
- except HTTPError as err:
- if err.code == 403 and err.msg == "rate limit exceeded":
- gscript.warning(
- _(
- "The download of the json file with add-ons paths "
- "from the github server wasn't successful, "
- "{}. The previous downloaded json file "
- " will be used if exists.".format(err.msg)
- ),
- )
- else:
- return download_addons_paths_file(
- url=url.replace("main", "master"),
- response_format=response_format,
- )
- except URLError:
- gscript.fatal(
- _(
- "Download file from <{url}>, "
- "failed. Check internet connection.".format(
- url=url,
- ),
- ),
- )
- 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):
- """Create XML element tree from a given URL"""
- try:
- file_ = urlopen(url)
- except URLError:
- gscript.fatal(
- _(
- "Download file from <{url}>,"
- " failed. File is not on the server or"
- " check your internet connection.".format(
- url=url,
- ),
- ),
- )
- return etree.fromstring(file_.read())
- def check_progs():
- """Check if the necessary programs are available"""
- # git to be tested once supported instead of `svn`
- for prog in (MAKE, "gcc", "svn"):
- 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": "db",
- "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)
- # TODO: extension != module
- 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)
- tkeys = sorted(tlist.keys())
- for toolbox_code in tkeys:
- toolbox_data = tlist[toolbox_code]
- 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):"))
- # TODO: extensions with several modules + lib
- 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)
- 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_extensions(url, name):
- """Get extensions inside a toolbox in toolbox file at given URL
- :param url: URL of the directory (file name will be attached)
- :param name: toolbox name
- """
- # dictionary of extensions
- edict = dict()
- url = url + "toolboxes.xml"
- try:
- tree = etree_fromurl(url)
- for tnode in tree.findall("toolbox"):
- if name == tnode.get("code"):
- for enode in tnode.findall("task"):
- # extension name
- ename = enode.get("name")
- edict[ename] = dict()
- # list of modules installed by this extension
- edict[ename]["mlist"] = list()
- # list of files installed by this extension
- edict[ename]["flist"] = list()
- break
- except (HTTPError, IOError, OSError):
- grass.fatal(_("Unable to fetch addons metadata file"))
- return edict
- def get_module_files(mnode):
- """Return list of module files
- :param mnode: XML node for a module
- """
- flist = []
- if mnode.find("binary") is None:
- return 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 of a module 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)
- 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:
- file_ = urlopen(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)
- file_ = urlopen(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_extensions(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
- # extension name
- 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))
- """
- # extension files
- 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))
- # extension modules
- mnode = tnode.find("modules")
- if mnode is not None:
- file_.write("%s<modules>\n" % (" " * indent))
- indent += 4
- for fnode in mnode.findall("module"):
- file_.write("%s<module>%s</module>\n" % (" " * indent, fnode.text))
- indent -= 4
- file_.write("%s</modules>\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, branch):
- """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"]
- )
- # create a dictionary of extensions
- # for each extension
- # - a list of modules installed by this extension
- # - a list of files installed by this extension
- edict = None
- if flags["t"]:
- grass.message(_("Installing toolbox <%s>...") % options["extension"])
- edict = get_toolbox_extensions(xmlurl, options["extension"])
- else:
- edict = dict()
- edict[options["extension"]] = dict()
- # list of modules installed by this extension
- edict[options["extension"]]["mlist"] = list()
- # list of files installed by this extension
- edict[options["extension"]]["flist"] = list()
- if not edict:
- grass.warning(_("Nothing to install"))
- return
- ret = 0
- tmp_dir = None
- new_modules = list()
- for extension in edict:
- ret1 = 0
- new_modules_ext = None
- if sys.platform == "win32":
- ret1, new_modules_ext, new_files_ext = install_extension_win(extension)
- else:
- (
- ret1,
- new_modules_ext,
- new_files_ext,
- tmp_dir,
- ) = install_extension_std_platforms(
- extension, source=source, url=url, branch=branch
- )
- if not flags["d"] and not flags["i"]:
- edict[extension]["mlist"].extend(new_modules_ext)
- edict[extension]["flist"].extend(new_files_ext)
- new_modules.extend(new_modules_ext)
- ret += ret1
- if len(edict) > 1:
- print("-" * 60)
- if flags["d"] or flags["i"]:
- return
- if ret != 0:
- grass.warning(
- _("Installation failed, sorry." " Please check above error messages.")
- )
- else:
- # update extensions metadata file
- grass.message(_("Updating extensions metadata file..."))
- install_extension_xml(edict)
- # update modules metadata file
- grass.message(_("Updating extension modules metadata file..."))
- install_module_xml(new_modules)
- for module in new_modules:
- 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)
- 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)
- """
- # TODO: extensions with multiple modules
- data = {}
- bin_list = []
- try:
- tree = etree_fromurl(url)
- 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(edict):
- """Update XML files with metadata about installed modules and toolbox
- of an private addon
- """
- # TODO toolbox
- # if len(mlist) > 1:
- # # read metadata from remote server (toolboxes)
- # install_toolbox_xml(url, options['extension'])
- xml_file = os.path.join(options["prefix"], "extensions.xml")
- # create an empty file if not exists
- if not os.path.exists(xml_file):
- write_xml_extensions(xml_file)
- # read XML file
- tree = etree_fromfile(xml_file)
- # update tree
- for name in edict:
- # so far extensions do not have description or keywords
- # only modules have
- """
- try:
- desc = gtask.parse_interface(name).description
- # mname = gtask.parse_interface(name).name
- keywords = gtask.parse_interface(name).keywords
- except Exception as e:
- grass.warning(_("No addons metadata available."
- " Addons metadata file not updated."))
- return []
- """
- tnode = None
- for node in tree.findall("task"):
- if node.get("name") == name:
- tnode = node
- break
- if tnode is None:
- # create new node for task
- tnode = etree.Element("task", attrib={"name": name})
- """
- dnode = etree.Element('description')
- dnode.text = desc
- tnode.append(dnode)
- knode = etree.Element('keywords')
- knode.text = (',').join(keywords)
- tnode.append(knode)
- """
- # create binary
- bnode = etree.Element("binary")
- # list of all installed files for this extension
- for file_name in edict[name]["flist"]:
- fnode = etree.Element("file")
- fnode.text = file_name
- bnode.append(fnode)
- tnode.append(bnode)
- # create modules
- msnode = etree.Element("modules")
- # list of all installed modules for this extension
- for module_name in edict[name]["mlist"]:
- mnode = etree.Element("module")
- mnode.text = module_name
- msnode.append(mnode)
- tnode.append(msnode)
- tree.append(tnode)
- else:
- grass.verbose(
- "Extension already listed in metadata file; metadata not updated!"
- )
- write_xml_extensions(xml_file, tree)
- return None
- def get_multi_addon_addons_which_install_only_html_man_page():
- """Get multi-addon addons which install only manual html page
- :return list addons: list of multi-addon addons which install
- only manual html page
- """
- addons = []
- all_addon_dirs = []
- addon_dirs_with_source_module = [] # *.py, *.c file
- addon_pattern = re.compile(r".*{}".format(options["extension"]))
- addon_src_file_pattern = re.compile(r".*.py$|.*.c$")
- addons_paths_file = os.path.join(
- options["prefix"],
- get_addons_paths.json_file,
- )
- if not os.path.exists(addons_paths_file):
- get_addons_paths(gg_addons_base_dir=options["prefix"])
- with open(addons_paths_file) as f:
- addons_paths = json.loads(f.read())
- for addon in addons_paths["tree"]:
- if re.match(addon_pattern, addon["path"]) and addon["type"] == "blob":
- if re.match(addon_src_file_pattern, addon["path"]):
- # Add addon dirs which contains source module *.py, *.c file
- addon_dirs_with_source_module.append(
- os.path.dirname(addon["path"]),
- )
- elif re.match(addon_pattern, addon["path"]) and addon["type"] == "tree":
- # Add all addon dirs
- all_addon_dirs.append(addon["path"])
- for addon in set(all_addon_dirs) ^ set(addon_dirs_with_source_module):
- addons.append(os.path.basename(addon))
- return addons
- def filter_multi_addon_addons(mlist):
- """Filter out list of multi-addon addons which contains
- and installs only *.html manual page, without source/binary
- excutable module and doesn't need to check metadata.
- e.g. the i.sentinel multi-addon consists of several full i.sentinel.*
- addons along with a i.sentinel.html overview file.
- :param list mlist: list of multi-addons (groups of addons
- with respective addon overview HTML pages)
- :return list mlist: list of individual multi-addons without respective
- addon overview HTML pages
- """
- # Filters out add-ons that only contain the *.html man page,
- # e.g. multi-addon i.sentinel (root directory) contains only
- # the *.html manual page for installation, it does not need
- # to check if metadata is available if there is no executable module.
- for addon in get_multi_addon_addons_which_install_only_html_man_page():
- if addon in mlist:
- mlist.pop(mlist.index(addon))
- return mlist
- def install_module_xml(mlist):
- """Update XML files with metadata about installed modules and toolbox
- of an private addon
- """
- 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)
- # Filter multi-addon addons
- if len(mlist) > 1:
- mlist = filter_multi_addon_addons(
- mlist.copy()
- ) # mlist.copy() keep the original list of add-ons
- # update tree
- for name in mlist:
- try:
- desc = gtask.parse_interface(name).description
- # mname = gtask.parse_interface(name).name
- keywords = gtask.parse_interface(name).keywords
- except Exception as error:
- grass.warning(
- _("No metadata available for module '{name}': {error}").format(
- name=name, error=error
- )
- )
- continue
- tnode = None
- for node in tree.findall("task"):
- if node.get("name") == name:
- tnode = node
- break
- if tnode is None:
- # create new node for task
- tnode = etree.Element("task", attrib={"name": name})
- dnode = etree.Element("description")
- dnode.text = desc
- tnode.append(dnode)
- knode = etree.Element("keywords")
- knode.text = (",").join(keywords)
- tnode.append(knode)
- # binary files installed with an extension are now
- # listed in extensions.xml
- """
- # create binary
- bnode = etree.Element('binary')
- list_of_binary_files = []
- for file_name in os.listdir(url):
- file_type = os.path.splitext(file_name)[-1]
- file_n = os.path.splitext(file_name)[0]
- html_path = os.path.join(options['prefix'], 'docs', 'html')
- c_path = os.path.join(options['prefix'], 'bin')
- py_path = os.path.join(options['prefix'], 'scripts')
- # html or image file
- if file_type in ['.html', '.jpg', '.png'] \
- and file_n in os.listdir(html_path):
- list_of_binary_files.append(os.path.join(html_path, file_name))
- # c file
- elif file_type in ['.c'] and file_name in os.listdir(c_path):
- list_of_binary_files.append(os.path.join(c_path, file_n))
- # python file
- elif file_type in ['.py'] and file_name in os.listdir(py_path):
- list_of_binary_files.append(os.path.join(py_path, file_n))
- # man file
- man_path = os.path.join(options['prefix'], 'docs', 'man', 'man1')
- if name + '.1' in os.listdir(man_path):
- list_of_binary_files.append(os.path.join(man_path, name + '.1'))
- # add binaries to xml file
- for binary_file_name in list_of_binary_files:
- fnode = etree.Element('file')
- fnode.text = binary_file_name
- bnode.append(fnode)
- tnode.append(bnode)
- """
- tree.append(tnode)
- else:
- grass.verbose(
- "Extension module already listed in metadata file; metadata not updated!"
- )
- write_xml_modules(xml_file, tree)
- return mlist
- def install_extension_win(name):
- """Install extension on MS Windows"""
- grass.message(
- _("Downloading precompiled GRASS Addons <{}>...").format(options["extension"])
- )
- # build base URL
- base_url = (
- "http://wingrass.fsv.cvut.cz/"
- "grass{major}{minor}/addons/"
- "grass-{major}.{minor}.{patch}".format(
- 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,
- )
- # collect module names and file names
- module_list = list()
- for r, d, f in os.walk(srcdir):
- for file in f:
- # Filter GRASS module name patterns
- if re.search(r"^[d,db,g,i,m,p,ps,r,r3,s,t,v,wx]\..*[\.py,\.exe]$", file):
- modulename = os.path.splitext(file)[0]
- module_list.append(modulename)
- # remove duplicates in case there are .exe wrappers for python scripts
- module_list = set(module_list)
- # change shebang from python to python3
- pyfiles = []
- for r, d, f in os.walk(srcdir):
- for file in f:
- if file.endswith(".py"):
- pyfiles.append(os.path.join(r, file))
- for filename in pyfiles:
- replace_shebang_win(filename)
- # collect old files
- old_file_list = list()
- for r, d, f in os.walk(options["prefix"]):
- for filename in f:
- fullname = os.path.join(r, filename)
- old_file_list.append(fullname)
- # copy Addons copy tree to destination directory
- move_extracted_files(
- extract_dir=srcdir, target_dir=options["prefix"], files=os.listdir(srcdir)
- )
- # collect new files
- file_list = list()
- for r, d, f in os.walk(options["prefix"]):
- for filename in f:
- fullname = os.path.join(r, filename)
- if fullname not in old_file_list:
- file_list.append(fullname)
- return 0, module_list, file_list
- def download_source_code_svn(url, name, outdev, directory=None):
- """Download source code from a Subversion repository
- .. 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 download_source_code_official_github(url, name, outdev, directory=None):
- """Download source code from a official GitHub repository
- .. 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)
- if grass.call(["svn", "export", 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 files by moving them to different directory
- 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.
- """
- # skip binary files
- # see https://stackoverflow.com/a/7392391
- textchars = bytearray({7, 8, 9, 10, 12, 13, 27} | set(range(0x20, 0x100)) - {0x7F})
- def is_binary_string(bytes):
- return bool(bytes.translate(None, textchars))
- for root, unused, files in os.walk(directory):
- for name in files:
- filename = os.path.join(root, name)
- if is_binary_string(open(filename, "rb").read(1024)):
- continue # ignore binary files
- # read content of text file
- with open(filename, "rb") as fd:
- data = fd.read()
- # 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:
- with open(filename, "wb") as newfile:
- newfile.write(newdata)
- 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:
- if "__pycache__" in subfile:
- continue
- 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, branch=None
- ):
- """Get source code to a local directory for compilation"""
- gscript.verbose(_("Type of source identified as '{source}'.").format(source=source))
- if source == "official":
- gscript.message(
- _("Fetching <%s> from " "GRASS GIS Addons repository (be patient)...")
- % name
- )
- download_source_code_official_github(url, name, outdev, directory)
- elif source == "official_fork":
- gscript.message(
- _("Fetching <{name}> from " "<{url}> (be patient)...").format(
- name=name, url=url
- )
- )
- download_source_code_official_github(url, name, outdev, directory)
- elif source == "svn":
- gscript.message(
- _("Fetching <{name}> from " "<{url}> (be patient)...").format(
- name=name, url=url
- )
- )
- download_source_code_svn(url, name, outdev, directory)
- elif source in ["remote_zip"]: # , 'official'
- gscript.message(
- _("Fetching <{name}> from " "<{url}> (be patient)...").format(
- name=name, url=url
- )
- )
- # we expect that the module.zip file is not by chance in the archive
- zip_name = os.path.join(tmpdir, "extension.zip")
- try:
- response = urlopen(url)
- except URLError:
- # Try download add-on from 'master' branch if default "main" fails
- if not branch:
- try:
- url = url.replace("main", "master")
- gscript.message(
- _(
- "Expected default branch not found. "
- "Trying again from <{url}>..."
- ).format(url=url)
- )
- response = urlopen(url)
- except URLError:
- grass.fatal(
- _(
- "Extension <{name}> not found. Please check "
- "'url' and 'branch' options".format(name=name)
- )
- )
- else:
- grass.fatal(_("Extension <%s> not found") % name)
- with open(zip_name, "wb") as out_file:
- shutil.copyfileobj(response, out_file)
- 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, branch):
- """Install extension on standard platforms"""
- gisbase = os.getenv("GISBASE")
- # 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,
- branch=branch,
- )
- os.chdir(srcdir)
- pgm_not_found_message = _(
- "Module name not found." " Check module Makefile syntax (PGM variable)."
- )
- # collect module names
- module_list = list()
- for r, d, f in os.walk(srcdir):
- for filename in f:
- if filename == "Makefile":
- # get the module name: PGM = <module name>
- with open(os.path.join(r, "Makefile")) as fp:
- for line in fp.readlines():
- if re.match(r"PGM.*.=|PGM=", line):
- try:
- modulename = line.split("=")[1].strip()
- if modulename:
- if modulename not in module_list:
- module_list.append(modulename)
- else:
- grass.fatal(pgm_not_found_message)
- except IndexError:
- grass.fatal(pgm_not_found_message)
- # change shebang from python to python3
- pyfiles = []
- # r=root, d=directories, f = files
- for r, d, f in os.walk(srcdir):
- for file in f:
- if file.endswith(".py"):
- pyfiles.append(os.path.join(r, file))
- for filename in pyfiles:
- with fileinput.FileInput(filename, inplace=True) as file:
- for line in file:
- print(
- line.replace("#!/usr/bin/env python\n", "#!/usr/bin/env python3\n"),
- end="",
- )
- 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" % 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, None, None, None
- 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, None, None, None
- # collect old files
- old_file_list = list()
- for r, d, f in os.walk(options["prefix"]):
- for filename in f:
- fullname = os.path.join(r, filename)
- old_file_list.append(fullname)
- grass.message(_("Installing..."))
- ret = grass.call(install_cmd, stdout=outdev)
- # collect new files
- file_list = list()
- for r, d, f in os.walk(options["prefix"]):
- for filename in f:
- fullname = os.path.join(r, filename)
- if fullname not in old_file_list:
- file_list.append(fullname)
- return ret, module_list, file_list, os.path.join(TMPDIR, name)
- def remove_extension(force=False):
- """Remove existing extension
- extension or toolbox with extensions if -t is given)"""
- if flags["t"]:
- edict = get_toolbox_extensions(options["prefix"], options["extension"])
- else:
- edict = dict()
- edict[options["extension"]] = dict()
- # list of modules installed by this extension
- edict[options["extension"]]["mlist"] = list()
- # list of files installed by this extension
- edict[options["extension"]]["flist"] = list()
- # collect modules and files installed by these extensions
- mlist = list()
- xml_file = os.path.join(options["prefix"], "extensions.xml")
- if os.path.exists(xml_file):
- # read XML file
- tree = None
- try:
- tree = etree_fromfile(xml_file)
- except ETREE_EXCEPTIONS + (OSError, IOError):
- os.remove(xml_file)
- write_xml_extensions(xml_file)
- if tree is not None:
- for tnode in tree.findall("task"):
- ename = tnode.get("name").strip()
- if ename in edict:
- # modules installed by this extension
- mnode = tnode.find("modules")
- if mnode:
- for fnode in mnode.findall("module"):
- mname = fnode.text.strip()
- edict[ename]["mlist"].append(mname)
- mlist.append(mname)
- # files installed by this extension
- bnode = tnode.find("binary")
- if bnode:
- for fnode in bnode.findall("file"):
- bname = fnode.text.strip()
- edict[ename]["flist"].append(bname)
- else:
- if force:
- write_xml_extensions(xml_file)
- 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)
- # read XML file
- tree = None
- try:
- tree = etree_fromfile(xml_file)
- except ETREE_EXCEPTIONS + (OSError, IOError):
- os.remove(xml_file)
- write_xml_modules(xml_file)
- return []
- if tree is not None:
- for tnode in tree.findall("task"):
- ename = tnode.get("name").strip()
- if ename in edict:
- # assume extension name == module name
- edict[ename]["mlist"].append(ename)
- mlist.append(ename)
- # files installed by this extension
- bnode = tnode.find("binary")
- if bnode:
- for fnode in bnode.findall("file"):
- bname = fnode.text.strip()
- edict[ename]["flist"].append(bname)
- if force:
- grass.verbose(_("List of removed files:"))
- else:
- grass.info(_("Files to be removed:"))
- eremoved = remove_extension_files(edict, force)
- if force:
- if len(eremoved) > 0:
- grass.message(_("Updating addons metadata file..."))
- remove_extension_xml(mlist, edict)
- for ename in edict:
- if ename in eremoved:
- grass.message(_("Extension <%s> successfully uninstalled.") % ename)
- else:
- if flags["t"]:
- grass.warning(
- _(
- "Toolbox <%s> not removed. "
- "Re-run '%s' with '-f' flag to force removal"
- )
- % (options["extension"], "g.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_extension_files(edict, force=False):
- """Remove extensions specified in a dictionary
- Uses the file names from the file list of the dictionary
- 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"], "extensions.xml")
- einstalled = list()
- eremoved = list()
- if os.path.exists(xml_file):
- tree = etree_fromfile(xml_file)
- if tree is not None:
- for task in tree.findall("task"):
- ename = task.get("name").strip()
- einstalled.append(ename)
- else:
- tree = None
- for name in edict:
- removed = True
- if len(edict[name]["flist"]) > 0:
- err = list()
- for fpath in edict[name]["flist"]:
- grass.verbose(fpath)
- if force:
- try:
- os.remove(fpath)
- except OSError:
- msg = "Unable to remove file '%s'"
- err.append((_(msg) % fpath))
- removed = False
- if len(err) > 0:
- for error_line in err:
- grass.error(error_line)
- else:
- if name not in einstalled:
- # 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)
- remove_extension_std(name, force)
- removed = False
- if removed is True:
- eremoved.append(name)
- return eremoved
- def remove_extension_std(name, force=False):
- """Remove extension/module expecting the standard layout
- Any images for manuals or files installed in etc will not be
- removed
- """
- 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(mlist, edict):
- """Update local meta-file when removing existing extension"""
- if len(edict) > 1:
- # update also toolboxes metadata
- remove_from_toolbox_xml(options["extension"])
- # modules
- xml_file = os.path.join(options["prefix"], "modules.xml")
- if os.path.exists(xml_file):
- # read XML file
- tree = etree_fromfile(xml_file)
- for name in mlist:
- for node in tree.findall("task"):
- if node.get("name") != name:
- continue
- tree.remove(node)
- write_xml_modules(xml_file, tree)
- # extensions
- xml_file = os.path.join(options["prefix"], "extensions.xml")
- if os.path.exists(xml_file):
- # read XML file
- tree = etree_fromfile(xml_file)
- for name in edict:
- for node in tree.findall("task"):
- if node.get("name") != name:
- continue
- tree.remove(node)
- write_xml_extensions(xml_file, tree)
- # check links in CSS
- def check_style_file(name):
- """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.
- If the files are missing, a warning is issued.
- """
- dist_file = os.path.join(os.getenv("GISBASE"), "docs", "html", name)
- addons_file = os.path.join(options["prefix"], "docs", "html", name)
- if os.path.isfile(addons_file):
- return
- try:
- shutil.copyfile(dist_file, addons_file)
- except OSError as error:
- grass.warning(
- _(
- "Unable to create '{filename}': {error}."
- " Is the GRASS GIS documentation package installed?"
- " Installation continues,"
- " but documentation may not look right."
- ).format(filename=addons_file, error=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_file("grass_logo.png")
- check_style_file("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
- than core modules"""
- 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):
- if match.group(1)[:4] == "http":
- continue
- pos.append(match.start(1))
- # find URIs
- pattern = r"""<a href="([^"]+)">([^>]+)</a>"""
- addons = get_installed_extensions(force=True)
- # Multi-addon
- if len(addons) > 1:
- for a in get_multi_addon_addons_which_install_only_html_man_page():
- # Add multi-addon addons which install only manual html page
- addons.append(a)
- 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('https://grass.osgeo.org/addons')
- 'https://grass.osgeo.org/addons/'
- >>> resolve_xmlurl_prefix('https://grass.osgeo.org/addons/')
- 'https://grass.osgeo.org/addons/'
- """
- gscript.debug("resolve_xmlurl_prefix(url={0}, source={1})".format(url, source))
- if source in ("official", "official_fork"):
- # use pregenerated modules XML file
- # Define branch to fetch from (latest or current version)
- version_branch = get_version_branch(version[0])
- url = "https://grass.osgeo.org/addons/{}/".format(version_branch)
- # else try to get extensions 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/{branch}.zip",
- },
- "GitLab": {
- "domain": "gitlab.com",
- "ignored_suffixes": [".zip", ".tar.gz", ".tar.bz2", ".tar"],
- "possible_starts": ["", "https://", "http://"],
- "url_start": "https://",
- "url_end": "/-/archive/{branch}/{name}-{branch}.zip",
- },
- "Bitbucket": {
- "domain": "bitbucket.org",
- "ignored_suffixes": [".zip", ".tar.gz", ".gz", ".bz2"],
- "possible_starts": ["", "https://", "http://"],
- "url_start": "https://",
- "url_end": "/get/{branch}.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, name, branch):
- """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.
- :param url: URL
- :param name: module name
- """
- 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 = ""
- if "branch" in match["url_end"]:
- suffix = match["url_end"].format(
- name=name,
- branch=branch if branch else get_default_branch(url),
- )
- else:
- suffix = match["url_end"].format(name=name)
- url = "{prefix}{base}{suffix}".format(
- prefix=actual_start, base=url.rstrip("/"), suffix=suffix
- )
- 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
- # TODO: workaround, https://github.com/OSGeo/grass-addons/issues/528
- def resolve_source_code(url=None, name=None, branch=None, fork=False):
- """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 {branch} 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('https://svn.osgeo.org/grass/grass-addons/grass7')
- ('svn', 'https://svn.osgeo.org/grass/grass-addons/grass7')
- ZIP files online:
- >>> resolve_source_code('https://trac.osgeo.org/.../r.modis?format=zip') # doctest: +SKIP
- ('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') # doctest: +SKIP
- ('remote_zip', 'https://trac.osgeo.org/.../r.agent.aco?format=zip')
- >>> resolve_source_code('https://trac.osgeo.org/.../r.agent.aco') # doctest: +SKIP
- ('remote_zip', 'https://trac.osgeo.org/.../r.agent.aco?format=zip')
- GitHub:
- >>> resolve_source_code('github.com/user/g.example') # doctest: +SKIP
- ('remote_zip', 'https://github.com/user/g.example/archive/master.zip')
- >>> resolve_source_code('github.com/user/g.example/') # doctest: +SKIP
- ('remote_zip', 'https://github.com/user/g.example/archive/master.zip')
- >>> resolve_source_code('https://github.com/user/g.example') # doctest: +SKIP
- ('remote_zip', 'https://github.com/user/g.example/archive/master.zip')
- >>> resolve_source_code('https://github.com/user/g.example/') # doctest: +SKIP
- ('remote_zip', 'https://github.com/user/g.example/archive/master.zip')
- GitLab:
- >>> resolve_source_code('gitlab.com/JoeUser/GrassModule') # doctest: +SKIP
- ('remote_zip', 'https://gitlab.com/JoeUser/GrassModule/-/archive/master/GrassModule-master.zip')
- >>> resolve_source_code('https://gitlab.com/JoeUser/GrassModule') # doctest: +SKIP
- ('remote_zip', 'https://gitlab.com/JoeUser/GrassModule/-/archive/master/GrassModule-master.zip')
- Bitbucket:
- >>> resolve_source_code('bitbucket.org/joe-user/grass-module') # doctest: +SKIP
- ('remote_zip', 'https://bitbucket.org/joe-user/grass-module/get/default.zip')
- >>> resolve_source_code('https://bitbucket.org/joe-user/grass-module') # doctest: +SKIP
- ('remote_zip', 'https://bitbucket.org/joe-user/grass-module/get/default.zip')
- """
- # Handle URL for the offical repo
- if name and (not url or fork):
- module_class = get_module_class_name(name)
- # note: 'trunk' is required to make URL usable for 'svn export' call
- # and fetches the default branch
- if not branch:
- # Fetch from default branch
- version_branch = get_version_branch(version[0])
- try:
- url = url.rstrip("/") if url else GIT_URL
- urlrequest.urlopen(f"{url}/tree/{version_branch}/src")
- svn_reference = "branches/{}".format(version_branch)
- except URLError:
- svn_reference = "trunk"
- else:
- svn_reference = "branches/{}".format(branch)
- if not url or url == GIT_URL:
- # Set URL for the given GRASS version
- git_url = f"{GIT_URL}/{svn_reference}/src/{module_class}/{name}"
- return "official", git_url
- else:
- # Forks from the official repo should reflect the current structure
- url = url.rstrip("/")
- git_url = f"{url}/{svn_reference}/src/{module_class}/{name}"
- return "official_fork", git_url
- # Check if URL can be found
- # Catch corner case if local URL is given starting with file://
- url = url[6:] if url.startswith("file://") else url
- if not os.path.exists(url):
- url_validated = False
- if url.startswith("http"):
- try:
- open_url = urlopen(url)
- open_url.close()
- url_validated = True
- except URLError:
- pass
- else:
- try:
- open_url = urlopen("http://" + url)
- open_url.close()
- url_validated = True
- except URLError:
- pass
- try:
- open_url = urlopen("https://" + url)
- open_url.close()
- url_validated = True
- except URLError:
- pass
- if not url_validated:
- grass.fatal(_("Cannot open URL: {}".format(url)))
- # Handle local URLs
- 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)
- # Handle remote URLs
- else:
- source, resolved_url = resolve_known_host_service(url, name, branch)
- 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 get_addons_paths(gg_addons_base_dir):
- """Get and save addons paths from GRASS GIS Addons GitHub repo API
- as 'addons_paths.json' file in the gg_addons_base_dir. The file
- serves as a list of all addons, and their paths (required for
- mkhmtl.py tool)
- :param str gg_addons_base_dir: dir path where addons are installed
- """
- # Define branch to fetch from (latest or current version)
- addons_branch = get_version_branch(version[0])
- url = f"https://api.github.com/repos/OSGeo/grass-addons/git/trees/{addons_branch}?recursive=1"
- response = download_addons_paths_file(
- url=url,
- response_format="application/json",
- )
- if response:
- addons_paths = json.loads(gscript.decode(response.read()))
- with open(
- os.path.join(gg_addons_base_dir, get_addons_paths.json_file), "w"
- ) as f:
- json.dump(addons_paths, f)
- get_addons_paths.json_file = "addons_paths.json"
- def main():
- # check dependencies
- if not flags["a"] and sys.platform != "win32":
- check_progs()
- original_url = options["url"]
- branch = options["branch"]
- # manage proxies
- global PROXIES
- if options["proxy"]:
- PROXIES = {}
- for ptype, purl in (p.split("=") for p in options["proxy"].split(",")):
- PROXIES[ptype] = purl
- proxy = urlrequest.ProxyHandler(PROXIES)
- opener = urlrequest.build_opener(proxy)
- urlrequest.install_opener(opener)
- # Required for mkhtml.py script (get addon git commit from GitHub API server)
- os.environ["GRASS_PROXY"] = options["proxy"]
- # define path
- options["prefix"] = resolve_install_prefix(
- path=options["prefix"], to_system=flags["s"]
- )
- if flags["j"]:
- get_addons_paths(gg_addons_base_dir=options["prefix"])
- return 0
- # list available extensions
- if flags["l"] or flags["c"] or (flags["g"] and not flags["a"]):
- # using dummy extension, we don't need any extension 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, branch=branch, fork=flags["o"]
- )
- 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"] or flags["i"]:
- flag = "d" if flags["d"] else "i"
- if options["operation"] != "add":
- grass.warning(
- _(
- "Flag '{}' is relevant only to"
- " 'operation=add'. Ignoring this flag."
- ).format(flag)
- )
- else:
- global REMOVE_TMPDIR
- REMOVE_TMPDIR = False
- if options["operation"] == "add":
- check_dirs()
- if original_url == "" or flags["o"]:
- """
- Query GitHub API only if extension will be downloaded
- from official GRASS GIS addon repository
- """
- get_addons_paths(gg_addons_base_dir=options["prefix"])
- source, url = resolve_source_code(
- name=options["extension"], url=original_url, branch=branch, fork=flags["o"]
- )
- xmlurl = resolve_xmlurl_prefix(original_url, source=source)
- install_extension(source=source, url=url, xmlurl=xmlurl, branch=branch)
- else: # remove
- remove_extension(force=flags["f"])
- return 0
- if __name__ == "__main__":
- if len(sys.argv) == 2 and sys.argv[1] == "--doctest":
- import doctest
- 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(".")
- sys.exit(main())
|