diff frontends/src/jp/common.py @ 2551:b27165bf160c

jp (merge-request/set): if service and node are not specified, URIFinder is now used + ask confirmation before publishing
author Goffi <goffi@goffi.org>
date Sat, 31 Mar 2018 18:23:06 +0200
parents 772447ec070f
children 0062d3e79d12
line wrap: on
line diff
--- a/frontends/src/jp/common.py	Sat Mar 31 18:21:56 2018 +0200
+++ b/frontends/src/jp/common.py	Sat Mar 31 18:23:06 2018 +0200
@@ -22,9 +22,11 @@
 from sat.core import exceptions
 from sat.tools.common import regex
 from sat.tools.common.ansi import ANSI as A
+from sat.tools.common import uri as xmpp_uri
 from sat.tools import config
 from ConfigParser import NoSectionError, NoOptionError
 from collections import namedtuple
+from functools import partial
 import json
 import os
 import os.path
@@ -663,3 +665,37 @@
         kwargs_ = {'col_sep':u' ', 'head_line_sep':u' ', 'show_borders':False}
         kwargs_.update(kwargs)
         return self.display(**kwargs_)
+
+
+class URIFinder(object):
+    """Helper class to find URIs in well-known locations"""
+
+    def __init__(self, command, path, key, callback):
+        if not command.args.service and not command.args.node:
+            self.host = command.host
+            self.args = command.args
+            self.key = key
+            self.callback = callback
+            self.host.bridge.URIFind(path,
+                                     [key],
+                                     callback=self.URIFindCb,
+                                     errback=partial(command.errback,
+                                                     msg=_(u"can't find " + key + u" URI: {}"),
+                                                     exit_code=C.EXIT_BRIDGE_ERRBACK))
+        else:
+            callback()
+
+    def URIFindCb(self, uri_data):
+        try:
+            uri = uri_data[self.key]
+        except KeyError:
+            self.host.disp(_(u"No {key} URI specified for this project, please specify service and node").format(key=self.key), error=True)
+            self.host.quit(C.EXIT_NOT_FOUND)
+        parsed_uri = xmpp_uri.parseXMPPUri(uri)
+        try:
+            self.args.service = parsed_uri[u'path']
+            self.args.node = parsed_uri[u'node']
+        except KeyError:
+            self.host.disp(_(u"Invalid URI found: {uri}").format(uri=uri), error=True)
+            self.host.quit(C.EXIT_DATA_ERROR)
+        self.callback()