Mercurial > libervia-backend
view sat_frontends/jp/arg_tools.py @ 2758:f4070693814a
plugin XEP-0060: fixed MAM return value on getItems():
callback to generate metadata was using wrong items in return argument.
author | Goffi <goffi@goffi.org> |
---|---|
date | Sun, 06 Jan 2019 17:33:53 +0100 |
parents | 56f94936df1e |
children | 003b8b4b56a7 |
line wrap: on
line source
#!/usr/bin/env python2 # -*- coding: utf-8 -*- # jp: a SàT command line tool # Copyright (C) 2009-2018 Jérôme Poisson (goffi@goffi.org) # This program is free software: you can redistribute it and/or modify # it under the terms of the GNU Affero General Public License as published by # the Free Software Foundation, either version 3 of the License, or # (at your option) any later version. # This program is distributed in the hope that it will be useful, # but WITHOUT ANY WARRANTY; without even the implied warranty of # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the # GNU Affero General Public License for more details. # You should have received a copy of the GNU Affero General Public License # along with this program. If not, see <http://www.gnu.org/licenses/>. from sat.core.i18n import _ from sat.core import exceptions def escape(arg, smart=True): """format arg with quotes @param smart(bool): if True, only escape if needed """ if smart and not " " in arg and not '"' in arg: return arg return u'"' + arg.replace(u'"', u'\\"') + u'"' def get_cmd_choices(cmd=None, parser=None): try: choices = parser._subparsers._group_actions[0].choices return choices[cmd] if cmd is not None else choices except (KeyError, AttributeError): raise exceptions.NotFound def get_use_args(host, args, use, verbose=False, parser=None): """format args for argparse parser with values prefilled @param host(JP): jp instance @param args(list(str)): arguments to use @param use(dict[str, str]): arguments to fill if found in parser @param verbose(bool): if True a message will be displayed when argument is used or not @param parser(argparse.ArgumentParser): parser to use @return (tuple[list[str],list[str]]): 2 args lists: - parser args, i.e. given args corresponding to parsers - use args, i.e. generated args from use """ # FIXME: positional args are not handled correclty # if there is more that one, the position is not corrected if parser is None: parser = host.parser # we check not optional args to see if there # is a corresonding parser # else USE args would not work correctly (only for current parser) parser_args = [] for arg in args: if arg.startswith("-"): break try: parser = get_cmd_choices(arg, parser) except exceptions.NotFound: break parser_args.append(arg) # post_args are remaning given args, # without the ones corresponding to parsers post_args = args[len(parser_args) :] opt_args = [] pos_args = [] actions = {a.dest: a for a in parser._actions} for arg, value in use.iteritems(): try: if arg == u"item" and not u"item" in actions: # small hack when --item is appended to a --items list arg = u"items" action = actions[arg] except KeyError: if verbose: host.disp( _( u"ignoring {name}={value}, not corresponding to any argument (in USE)" ).format(name=arg, value=escape(value)) ) else: if verbose: host.disp( _(u"arg {name}={value} (in USE)").format( name=arg, value=escape(value) ) ) if not action.option_strings: pos_args.append(value) else: opt_args.append(action.option_strings[0]) opt_args.append(value) return parser_args, opt_args + pos_args + post_args