Adding language support to Birdy

birdy/cli/base.py

@@ -30,17 +30,24 @@ def __init__(self, name=None, url=None, caps_xml=None, desc_xml=None, **attrs):
         self.verify = get_ssl_verify()
         self.caps_xml = caps_xml
         self.desc_xml = desc_xml
-        self.wps = WebProcessingService(self.url, verify=self.verify, skip_caps=True)
+        self._wps = None
         self.commands = OrderedDict()
 
+    @property
+    def wps(self):
+        if self._wps is None:
+            language = self.context_settings['obj'].get('language')
+            self._wps = WebProcessingService(self.url, verify=self.verify, skip_caps=True, language=language)
+        return self._wps
+
     def _update_commands(self):
         if not self.commands:
             try:
                 self.wps.getcapabilities(xml=self.caps_xml)
             except SSLError:
                 raise ConnectionError('SSL verfication of server certificate failed. Set WPS_SSL_VERIFY=false.')
-            except Exception:
-                raise ConnectionError("Web Processing Service not available.")
+            except Exception as e:
+                raise ConnectionError("Could not connect to Web Processing Service ({!r})".format(e))
             for process in self.wps.processes:
                 self.commands[process.identifier] = dict(
                     name=process.identifier,
@@ -50,7 +57,6 @@ def _update_commands(self):
                     options=[])
 
     def list_commands(self, ctx):
-        ctx.obj = True
         self._update_commands()
         return list(self.commands.keys())
 
@@ -65,23 +71,21 @@ def get_command(self, ctx, name):
 
     def _get_command_info(self, name, ctx):
         cmd = self.commands.get(name)
-        if ctx.obj is None or False:
-            pp = self.wps.describeprocess(name, xml=self.desc_xml)
-            for inp in pp.dataInputs:
-                help = inp.title or ''
-                default = BirdyCLI.get_param_default(inp)
-                if default:
-                    help = "{}. Default: {}".format(help, default)
-                cmd['options'].append(dict(
-                    name=inp.identifier,
-                    # default=BirdyCLI.get_param_default(inp),
-                    help=help,
-                    type=BirdyCLI.get_param_type(inp),
-                    multiple=inp.maxOccurs > 1))
-            outputs = []
-            for output in pp.processOutputs:
-                outputs.append((output.identifier, BirdyCLI.get_param_type(output) is COMPLEX))
-            ctx.obj = dict(outputs=outputs)
+        pp = self.wps.describeprocess(name, xml=self.desc_xml)
+        for inp in pp.dataInputs:
+            help = inp.title or ''
+            default = BirdyCLI.get_param_default(inp)
+            if default:
+                help = "{}. Default: {}".format(help, default)
+            cmd['options'].append(dict(
+                name=inp.identifier.replace(' ', '-'),
+                # default=BirdyCLI.get_param_default(inp),
+                help=help,
+                type=BirdyCLI.get_param_type(inp),
+                multiple=inp.maxOccurs > 1))
+        outputs = []
+        for output in pp.processOutputs:
+            outputs.append((output.identifier, BirdyCLI.get_param_type(output) is COMPLEX))
         return cmd
 
     @staticmethod

birdy/cli/run.py

@@ -1,8 +1,29 @@
+import os
+
 import click
 from birdy.cli.base import BirdyCLI
 from birdy.cli.misc import get_ssl_verify
 
-CONTEXT_SETTINGS = dict(help_option_names=['-h', '--help'])
+from owslib.wps import WebProcessingService
+
+CONTEXT_OBJ = dict(language=None)
+CONTEXT_SETTINGS = dict(help_option_names=['-h', '--help'], obj=CONTEXT_OBJ)
+DEFAULT_URL = "http://localhost:5000/wps"
+
+
+def _show_languages(ctx, param, value):
+    if not value or ctx.resilient_parsing:
+        return
+    url = os.environ.get('WPS_SERVICE') or DEFAULT_URL
+    wps = WebProcessingService(url, verify=get_ssl_verify())
+    click.echo(','.join(wps.languages.supported))
+    ctx.exit()
+
+
+def _set_language(ctx, param, value):
+    if not value or ctx.resilient_parsing:
+        return
+    CONTEXT_OBJ['language'] = value
 
 
 @click.command(cls=BirdyCLI, context_settings=CONTEXT_SETTINGS,
@@ -12,6 +33,12 @@
 @click.option('--send', '-S', is_flag=True, help="Send client side certificate to WPS. Default: false")
 @click.option("--sync", '-s', is_flag=True, help="Execute process in sync mode. Default: async mode.")
 @click.option("--token", "-t", help="Token to access the WPS service.")
+@click.option(
+    "--language", "-l", expose_value=False, is_eager=True, callback=_set_language,
+    help="Set the accepted language to send to the WPS service.")
+@click.option(
+    "--show-languages", "-L", expose_value=False, is_flag=True, is_eager=True, callback=_show_languages,
+    help="Show a list of accepted languages for the WPS service.")
 @click.pass_context
 def cli(ctx, cert, send, sync, token):
     """

birdy/client/base.py

@@ -45,6 +45,7 @@ def __init__(
         version=WPS_DEFAULT_VERSION,
         caps_xml=None,
         desc_xml=None,
+        language=None,
     ):
         """
         Args:
@@ -63,6 +64,8 @@ def __init__(
             verbose (str): passed to :class:`owslib.wps.WebProcessingService`
             progress (bool): If True, enable interactive user mode.
             version (str): WPS version to use.
+            language (str): passed to :class:`owslib.wps.WebProcessingService`
+                ex: 'fr-CA', 'en_US'.
         """
         self._converters = converters
         self._interactive = progress
@@ -102,6 +105,7 @@ def __init__(
             verify=verify,
             cert=cert,
             skip_caps=True,
+            language=language
         )
 
         try:
@@ -125,6 +129,18 @@ def __init__(
 
         self.__doc__ = utils.build_wps_client_doc(self._wps, self._processes)
 
+    @property
+    def language(self):
+        return self._wps.language
+
+    @language.setter
+    def language(self, value):
+        self._wps.language = value
+
+    @property
+    def languages(self):
+        return self._wps.languages
+
     def _get_process_description(self, processes=None, xml=None):
         """Return the description for each process.
 

birdy/templates/cmd.py.j2

@@ -31,7 +31,8 @@ def cli(ctx, **options):
     if send and cert:
         with open(cert, 'r') as fh:
             headers = {'X-Ssl-Client-Cert': quote(fh.read())}
-    wps = WebProcessingService('{{ url }}', skip_caps=True, verify=verify, cert=cert, headers=headers)
+    language = ctx.obj.get('language')
+    wps = WebProcessingService('{{ url }}', skip_caps=True, verify=verify, cert=cert, headers=headers, language=language)
     inputs = []
     for opt in options.keys():
         if not options[opt]:

requirements.txt

@@ -1,7 +1,7 @@
 click
 jinja2
 lxml
-owslib>=0.18
+owslib>=0.19.2
 wrapt
 funcsigs
 boltons

tests/resources/wps_emu_caps.xml

@@ -248,7 +248,8 @@
             <ows:Language>en-US</ows:Language>
         </wps:Default>
         <wps:Supported>
-            <ows:Language>lang</ows:Language>
+            <ows:Language>en-US</ows:Language>
+            <ows:Language>fr-CA</ows:Language>
         </wps:Supported>
     </wps:Languages>
 </wps:Capabilities>

tests/test_cli.py

@@ -1,31 +1,33 @@
+from click.testing import CliRunner
 import pytest
 
-import click
-from click.testing import CliRunner
+import birdy.cli.run
 
-from birdy.cli.base import BirdyCLI
-from .common import (
-    URL_EMU,
-    EMU_CAPS_XML,
-    # EMU_DESC_XML,
-)
+from .common import EMU_CAPS_XML, URL_EMU
 
 
-@click.command(cls=BirdyCLI,
-               url=URL_EMU,
-               caps_xml=EMU_CAPS_XML,
-               # desc_xml=EMU_DESC_XML
-               )
-def cli():
-    pass
+cli = birdy.cli.run.cli
+cli.url = URL_EMU
+cli.caps_xml = EMU_CAPS_XML
 
 
+@pytest.mark.online
 def test_help():
     runner = CliRunner()
     result = runner.invoke(cli, ['--help'])
     assert result.exit_code == 0
     assert 'hello' in result.output
     assert 'wordcount' in result.output
+    assert 'language' in result.output
+    assert 'show-languages' in result.output
+
+
+@pytest.mark.online
+def test_show_languages():
+    runner = CliRunner()
+    result = runner.invoke(cli, ['--show-languages'])
+    assert result.exit_code == 0
+    assert 'en-US' in result.output
 
 
 @pytest.mark.online

tests/test_client.py

@@ -51,6 +51,16 @@ def test_emu_offline():
     assert 'Hello' in wps.hello.__doc__
 
 
+def test_wps_supported_languages():
+    wps = WPSClient(URL_EMU, caps_xml=EMU_CAPS_XML, desc_xml=EMU_DESC_XML)
+    assert wps.languages.supported == ['en-US', 'fr-CA']
+
+
+def test_wps_with_language_arg():
+    wps = WPSClient(URL_EMU, caps_xml=EMU_CAPS_XML, desc_xml=EMU_DESC_XML, language='fr-CA')
+    assert wps.language == 'fr-CA'
+
+
 @pytest.mark.online
 @pytest.mark.xfail(reason="a wps process has invalid defaultValue Inf")
 def test_52north():