libervia-backend: sat_frontends/jp/cmd

annotate sat_frontends/jp/cmd_pubsub.py @ 2766:93a421de0e3d

tools (common/data_objects): metadata parsing in BlogItems: metadata are parsed to deserialise some well known values like rsm index and count, and some properties have been added to have easier access. A "complete" property (or item in metadata) is set to True if we are on the last page, False if we are not, or None if we don't have enough data to know if we are on the last page or not.

author	Goffi <goffi@goffi.org>
date	Fri, 11 Jan 2019 16:35:13 +0100
parents	92af49cde255
children	003b8b4b56a7

rev	line source
2191 a1a8233f89e8 jp(pubsub/get): pubsub/get command, first draft Goffi <goffi@goffi.org> parents: diff changeset	1 #!/usr/bin/env python2
a1a8233f89e8 jp(pubsub/get): pubsub/get command, first draft Goffi <goffi@goffi.org> parents: diff changeset	2 # -- coding: utf-8 --
a1a8233f89e8 jp(pubsub/get): pubsub/get command, first draft Goffi <goffi@goffi.org> parents: diff changeset	3
a1a8233f89e8 jp(pubsub/get): pubsub/get command, first draft Goffi <goffi@goffi.org> parents: diff changeset	4 # jp: a SàT command line tool
2483 0046283a285d dates update Goffi <goffi@goffi.org> parents: 2456 diff changeset	5 # Copyright (C) 2009-2018 Jérôme Poisson (goffi@goffi.org)
2191 a1a8233f89e8 jp(pubsub/get): pubsub/get command, first draft Goffi <goffi@goffi.org> parents: diff changeset	6
a1a8233f89e8 jp(pubsub/get): pubsub/get command, first draft Goffi <goffi@goffi.org> parents: diff changeset	7 # This program is free software: you can redistribute it and/or modify
a1a8233f89e8 jp(pubsub/get): pubsub/get command, first draft Goffi <goffi@goffi.org> parents: diff changeset	8 # it under the terms of the GNU Affero General Public License as published by
a1a8233f89e8 jp(pubsub/get): pubsub/get command, first draft Goffi <goffi@goffi.org> parents: diff changeset	9 # the Free Software Foundation, either version 3 of the License, or
a1a8233f89e8 jp(pubsub/get): pubsub/get command, first draft Goffi <goffi@goffi.org> parents: diff changeset	10 # (at your option) any later version.
a1a8233f89e8 jp(pubsub/get): pubsub/get command, first draft Goffi <goffi@goffi.org> parents: diff changeset	11
a1a8233f89e8 jp(pubsub/get): pubsub/get command, first draft Goffi <goffi@goffi.org> parents: diff changeset	12 # This program is distributed in the hope that it will be useful,
a1a8233f89e8 jp(pubsub/get): pubsub/get command, first draft Goffi <goffi@goffi.org> parents: diff changeset	13 # but WITHOUT ANY WARRANTY; without even the implied warranty of
a1a8233f89e8 jp(pubsub/get): pubsub/get command, first draft Goffi <goffi@goffi.org> parents: diff changeset	14 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
a1a8233f89e8 jp(pubsub/get): pubsub/get command, first draft Goffi <goffi@goffi.org> parents: diff changeset	15 # GNU Affero General Public License for more details.
a1a8233f89e8 jp(pubsub/get): pubsub/get command, first draft Goffi <goffi@goffi.org> parents: diff changeset	16
a1a8233f89e8 jp(pubsub/get): pubsub/get command, first draft Goffi <goffi@goffi.org> parents: diff changeset	17 # You should have received a copy of the GNU Affero General Public License
a1a8233f89e8 jp(pubsub/get): pubsub/get command, first draft Goffi <goffi@goffi.org> parents: diff changeset	18 # along with this program. If not, see <http://www.gnu.org/licenses/>.
a1a8233f89e8 jp(pubsub/get): pubsub/get command, first draft Goffi <goffi@goffi.org> parents: diff changeset	19
a1a8233f89e8 jp(pubsub/get): pubsub/get command, first draft Goffi <goffi@goffi.org> parents: diff changeset	20
a1a8233f89e8 jp(pubsub/get): pubsub/get command, first draft Goffi <goffi@goffi.org> parents: diff changeset	21 import base
a1a8233f89e8 jp(pubsub/get): pubsub/get command, first draft Goffi <goffi@goffi.org> parents: diff changeset	22 from sat.core.i18n import _
2316 7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	23 from sat.core import exceptions
2191 a1a8233f89e8 jp(pubsub/get): pubsub/get command, first draft Goffi <goffi@goffi.org> parents: diff changeset	24 from sat_frontends.jp.constants import Const as C
2275 64e99bf0dfa2 jp (pubsub/edit): new edit subcommand, which work in a same way as for blog Goffi <goffi@goffi.org> parents: 2274 diff changeset	25 from sat_frontends.jp import common
2316 7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	26 from sat_frontends.jp import arg_tools
2221 a6c9bc4d1de0 jp (pubsub/node): added create and delete commands Goffi <goffi@goffi.org> parents: 2214 diff changeset	27 from functools import partial
2224 87fcd4a7c7e4 jp (pubsub): added uri command to build pubsub URI Goffi <goffi@goffi.org> parents: 2221 diff changeset	28 from sat.tools.common import uri
2316 7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	29 from sat.tools.common.ansi import ANSI as A
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	30 from sat_frontends.tools import jid, strings
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	31 import argparse
2308 0b21d87c91cf jp (pubsub/hook): added create/delete/list hook command to handle new Pubsub hook feature Goffi <goffi@goffi.org> parents: 2292 diff changeset	32 import os.path
2316 7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	33 import re
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	34 import subprocess
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	35 import sys
2191 a1a8233f89e8 jp(pubsub/get): pubsub/get command, first draft Goffi <goffi@goffi.org> parents: diff changeset	36
a1a8233f89e8 jp(pubsub/get): pubsub/get command, first draft Goffi <goffi@goffi.org> parents: diff changeset	37 __commands__ = ["Pubsub"]
a1a8233f89e8 jp(pubsub/get): pubsub/get command, first draft Goffi <goffi@goffi.org> parents: diff changeset	38
2275 64e99bf0dfa2 jp (pubsub/edit): new edit subcommand, which work in a same way as for blog Goffi <goffi@goffi.org> parents: 2274 diff changeset	39 PUBSUB_TMP_DIR = u"pubsub"
2351 3c0a3fae1862 jp (pubsub/node): added schema (set/edit/get) commands to manipulate PubSub node schema Goffi <goffi@goffi.org> parents: 2339 diff changeset	40 PUBSUB_SCHEMA_TMP_DIR = PUBSUB_TMP_DIR + "_schema"
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	41 ALLOWED_SUBSCRIPTIONS_OWNER = ("subscribed", "pending", "none")
2275 64e99bf0dfa2 jp (pubsub/edit): new edit subcommand, which work in a same way as for blog Goffi <goffi@goffi.org> parents: 2274 diff changeset	42
2308 0b21d87c91cf jp (pubsub/hook): added create/delete/list hook command to handle new Pubsub hook feature Goffi <goffi@goffi.org> parents: 2292 diff changeset	43 # TODO: need to split this class in several modules, plugin should handle subcommands
0b21d87c91cf jp (pubsub/hook): added create/delete/list hook command to handle new Pubsub hook feature Goffi <goffi@goffi.org> parents: 2292 diff changeset	44
2191 a1a8233f89e8 jp(pubsub/get): pubsub/get command, first draft Goffi <goffi@goffi.org> parents: diff changeset	45
2235 4db836386641 jp: added use_pubsub and use_pubsub_node_req arguments to CommandBase Goffi <goffi@goffi.org> parents: 2224 diff changeset	46 class NodeInfo(base.CommandBase):
2197 e0e06391ce91 jp (pubsub): added pubsub/node/info command to retrieve node configuration Goffi <goffi@goffi.org> parents: 2195 diff changeset	47 def __init__(self, host):
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	48 base.CommandBase.__init__(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	49 self,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	50 host,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	51 "info",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	52 use_output=C.OUTPUT_DICT,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	53 use_pubsub=True,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	54 pubsub_flags={C.NODE},
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	55 help=_(u"retrieve node configuration"),
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	56 )
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	57 self.need_loop = True
2197 e0e06391ce91 jp (pubsub): added pubsub/node/info command to retrieve node configuration Goffi <goffi@goffi.org> parents: 2195 diff changeset	58
e0e06391ce91 jp (pubsub): added pubsub/node/info command to retrieve node configuration Goffi <goffi@goffi.org> parents: 2195 diff changeset	59 def add_parser_options(self):
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	60 self.parser.add_argument(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	61 "-k",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	62 "--key",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	63 type=base.unicode_decoder,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	64 action="append",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	65 dest="keys",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	66 help=_(u"data key to filter"),
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	67 )
2197 e0e06391ce91 jp (pubsub): added pubsub/node/info command to retrieve node configuration Goffi <goffi@goffi.org> parents: 2195 diff changeset	68
e0e06391ce91 jp (pubsub): added pubsub/node/info command to retrieve node configuration Goffi <goffi@goffi.org> parents: 2195 diff changeset	69 def removePrefix(self, key):
e0e06391ce91 jp (pubsub): added pubsub/node/info command to retrieve node configuration Goffi <goffi@goffi.org> parents: 2195 diff changeset	70 return key[7:] if key.startswith(u"pubsub#") else key
e0e06391ce91 jp (pubsub): added pubsub/node/info command to retrieve node configuration Goffi <goffi@goffi.org> parents: 2195 diff changeset	71
e0e06391ce91 jp (pubsub): added pubsub/node/info command to retrieve node configuration Goffi <goffi@goffi.org> parents: 2195 diff changeset	72 def filterKey(self, key):
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	73 return any((key == k or key == u"pubsub#" + k) for k in self.args.keys)
2197 e0e06391ce91 jp (pubsub): added pubsub/node/info command to retrieve node configuration Goffi <goffi@goffi.org> parents: 2195 diff changeset	74
e0e06391ce91 jp (pubsub): added pubsub/node/info command to retrieve node configuration Goffi <goffi@goffi.org> parents: 2195 diff changeset	75 def psNodeConfigurationGetCb(self, config_dict):
e0e06391ce91 jp (pubsub): added pubsub/node/info command to retrieve node configuration Goffi <goffi@goffi.org> parents: 2195 diff changeset	76 key_filter = (lambda k: True) if not self.args.keys else self.filterKey
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	77 config_dict = {
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	78 self.removePrefix(k): v for k, v in config_dict.iteritems() if key_filter(k)
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	79 }
2197 e0e06391ce91 jp (pubsub): added pubsub/node/info command to retrieve node configuration Goffi <goffi@goffi.org> parents: 2195 diff changeset	80 self.output(config_dict)
e0e06391ce91 jp (pubsub): added pubsub/node/info command to retrieve node configuration Goffi <goffi@goffi.org> parents: 2195 diff changeset	81 self.host.quit()
e0e06391ce91 jp (pubsub): added pubsub/node/info command to retrieve node configuration Goffi <goffi@goffi.org> parents: 2195 diff changeset	82
e0e06391ce91 jp (pubsub): added pubsub/node/info command to retrieve node configuration Goffi <goffi@goffi.org> parents: 2195 diff changeset	83 def psNodeConfigurationGetEb(self, failure_):
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	84 self.disp(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	85 u"can't get node configuration: {reason}".format(reason=failure_), error=True
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	86 )
2197 e0e06391ce91 jp (pubsub): added pubsub/node/info command to retrieve node configuration Goffi <goffi@goffi.org> parents: 2195 diff changeset	87 self.host.quit(C.EXIT_BRIDGE_ERRBACK)
e0e06391ce91 jp (pubsub): added pubsub/node/info command to retrieve node configuration Goffi <goffi@goffi.org> parents: 2195 diff changeset	88
e0e06391ce91 jp (pubsub): added pubsub/node/info command to retrieve node configuration Goffi <goffi@goffi.org> parents: 2195 diff changeset	89 def start(self):
e0e06391ce91 jp (pubsub): added pubsub/node/info command to retrieve node configuration Goffi <goffi@goffi.org> parents: 2195 diff changeset	90 self.host.bridge.psNodeConfigurationGet(
e0e06391ce91 jp (pubsub): added pubsub/node/info command to retrieve node configuration Goffi <goffi@goffi.org> parents: 2195 diff changeset	91 self.args.service,
e0e06391ce91 jp (pubsub): added pubsub/node/info command to retrieve node configuration Goffi <goffi@goffi.org> parents: 2195 diff changeset	92 self.args.node,
e0e06391ce91 jp (pubsub): added pubsub/node/info command to retrieve node configuration Goffi <goffi@goffi.org> parents: 2195 diff changeset	93 self.profile,
e0e06391ce91 jp (pubsub): added pubsub/node/info command to retrieve node configuration Goffi <goffi@goffi.org> parents: 2195 diff changeset	94 callback=self.psNodeConfigurationGetCb,
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	95 errback=self.psNodeConfigurationGetEb,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	96 )
2197 e0e06391ce91 jp (pubsub): added pubsub/node/info command to retrieve node configuration Goffi <goffi@goffi.org> parents: 2195 diff changeset	97
2221 a6c9bc4d1de0 jp (pubsub/node): added create and delete commands Goffi <goffi@goffi.org> parents: 2214 diff changeset	98
2235 4db836386641 jp: added use_pubsub and use_pubsub_node_req arguments to CommandBase Goffi <goffi@goffi.org> parents: 2224 diff changeset	99 class NodeCreate(base.CommandBase):
2221 a6c9bc4d1de0 jp (pubsub/node): added create and delete commands Goffi <goffi@goffi.org> parents: 2214 diff changeset	100 def __init__(self, host):
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	101 base.CommandBase.__init__(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	102 self,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	103 host,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	104 "create",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	105 use_output=C.OUTPUT_DICT,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	106 use_pubsub=True,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	107 pubsub_flags={C.NODE},
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	108 use_verbose=True,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	109 help=_(u"create a node"),
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	110 )
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	111 self.need_loop = True
2221 a6c9bc4d1de0 jp (pubsub/node): added create and delete commands Goffi <goffi@goffi.org> parents: 2214 diff changeset	112
a6c9bc4d1de0 jp (pubsub/node): added create and delete commands Goffi <goffi@goffi.org> parents: 2214 diff changeset	113 def add_parser_options(self):
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	114 self.parser.add_argument(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	115 "-f",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	116 "--field",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	117 type=base.unicode_decoder,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	118 action="append",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	119 nargs=2,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	120 dest="fields",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	121 default=[],
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	122 metavar=(u"KEY", u"VALUE"),
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	123 help=_(u"configuration field to set"),
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	124 )
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	125 self.parser.add_argument(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	126 "-F",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	127 "--full-prefix",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	128 action="store_true",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	129 help=_(u'don\'t prepend "pubsub#" prefix to field names'),
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	130 )
2221 a6c9bc4d1de0 jp (pubsub/node): added create and delete commands Goffi <goffi@goffi.org> parents: 2214 diff changeset	131
a6c9bc4d1de0 jp (pubsub/node): added create and delete commands Goffi <goffi@goffi.org> parents: 2214 diff changeset	132 def psNodeCreateCb(self, node_id):
a6c9bc4d1de0 jp (pubsub/node): added create and delete commands Goffi <goffi@goffi.org> parents: 2214 diff changeset	133 if self.host.verbosity:
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	134 announce = _(u"node created successfully: ")
2221 a6c9bc4d1de0 jp (pubsub/node): added create and delete commands Goffi <goffi@goffi.org> parents: 2214 diff changeset	135 else:
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	136 announce = u""
2221 a6c9bc4d1de0 jp (pubsub/node): added create and delete commands Goffi <goffi@goffi.org> parents: 2214 diff changeset	137 self.disp(announce + node_id)
a6c9bc4d1de0 jp (pubsub/node): added create and delete commands Goffi <goffi@goffi.org> parents: 2214 diff changeset	138 self.host.quit()
a6c9bc4d1de0 jp (pubsub/node): added create and delete commands Goffi <goffi@goffi.org> parents: 2214 diff changeset	139
a6c9bc4d1de0 jp (pubsub/node): added create and delete commands Goffi <goffi@goffi.org> parents: 2214 diff changeset	140 def psNodeCreateEb(self, failure_):
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	141 self.disp(u"can't create: {reason}".format(reason=failure_), error=True)
2221 a6c9bc4d1de0 jp (pubsub/node): added create and delete commands Goffi <goffi@goffi.org> parents: 2214 diff changeset	142 self.host.quit(C.EXIT_BRIDGE_ERRBACK)
a6c9bc4d1de0 jp (pubsub/node): added create and delete commands Goffi <goffi@goffi.org> parents: 2214 diff changeset	143
a6c9bc4d1de0 jp (pubsub/node): added create and delete commands Goffi <goffi@goffi.org> parents: 2214 diff changeset	144 def start(self):
a6c9bc4d1de0 jp (pubsub/node): added create and delete commands Goffi <goffi@goffi.org> parents: 2214 diff changeset	145 if not self.args.full_prefix:
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	146 options = {u"pubsub#" + k: v for k, v in self.args.fields}
2221 a6c9bc4d1de0 jp (pubsub/node): added create and delete commands Goffi <goffi@goffi.org> parents: 2214 diff changeset	147 else:
a6c9bc4d1de0 jp (pubsub/node): added create and delete commands Goffi <goffi@goffi.org> parents: 2214 diff changeset	148 options = dict(self.args.fields)
a6c9bc4d1de0 jp (pubsub/node): added create and delete commands Goffi <goffi@goffi.org> parents: 2214 diff changeset	149 self.host.bridge.psNodeCreate(
a6c9bc4d1de0 jp (pubsub/node): added create and delete commands Goffi <goffi@goffi.org> parents: 2214 diff changeset	150 self.args.service,
a6c9bc4d1de0 jp (pubsub/node): added create and delete commands Goffi <goffi@goffi.org> parents: 2214 diff changeset	151 self.args.node,
a6c9bc4d1de0 jp (pubsub/node): added create and delete commands Goffi <goffi@goffi.org> parents: 2214 diff changeset	152 options,
a6c9bc4d1de0 jp (pubsub/node): added create and delete commands Goffi <goffi@goffi.org> parents: 2214 diff changeset	153 self.profile,
a6c9bc4d1de0 jp (pubsub/node): added create and delete commands Goffi <goffi@goffi.org> parents: 2214 diff changeset	154 callback=self.psNodeCreateCb,
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	155 errback=partial(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	156 self.errback,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	157 msg=_(u"can't create node: {}"),
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	158 exit_code=C.EXIT_BRIDGE_ERRBACK,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	159 ),
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	160 )
2221 a6c9bc4d1de0 jp (pubsub/node): added create and delete commands Goffi <goffi@goffi.org> parents: 2214 diff changeset	161
a6c9bc4d1de0 jp (pubsub/node): added create and delete commands Goffi <goffi@goffi.org> parents: 2214 diff changeset	162
2235 4db836386641 jp: added use_pubsub and use_pubsub_node_req arguments to CommandBase Goffi <goffi@goffi.org> parents: 2224 diff changeset	163 class NodeDelete(base.CommandBase):
2221 a6c9bc4d1de0 jp (pubsub/node): added create and delete commands Goffi <goffi@goffi.org> parents: 2214 diff changeset	164 def __init__(self, host):
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	165 base.CommandBase.__init__(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	166 self,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	167 host,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	168 "delete",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	169 use_pubsub=True,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	170 pubsub_flags={C.NODE},
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	171 help=_(u"delete a node"),
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	172 )
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	173 self.need_loop = True
2221 a6c9bc4d1de0 jp (pubsub/node): added create and delete commands Goffi <goffi@goffi.org> parents: 2214 diff changeset	174
a6c9bc4d1de0 jp (pubsub/node): added create and delete commands Goffi <goffi@goffi.org> parents: 2214 diff changeset	175 def add_parser_options(self):
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	176 self.parser.add_argument(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	177 "-f",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	178 "--force",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	179 action="store_true",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	180 help=_(u"delete node without confirmation"),
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	181 )
2221 a6c9bc4d1de0 jp (pubsub/node): added create and delete commands Goffi <goffi@goffi.org> parents: 2214 diff changeset	182
a6c9bc4d1de0 jp (pubsub/node): added create and delete commands Goffi <goffi@goffi.org> parents: 2214 diff changeset	183 def psNodeDeleteCb(self):
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	184 self.disp(_(u"node [{node}] deleted successfully").format(node=self.args.node))
2221 a6c9bc4d1de0 jp (pubsub/node): added create and delete commands Goffi <goffi@goffi.org> parents: 2214 diff changeset	185 self.host.quit()
a6c9bc4d1de0 jp (pubsub/node): added create and delete commands Goffi <goffi@goffi.org> parents: 2214 diff changeset	186
a6c9bc4d1de0 jp (pubsub/node): added create and delete commands Goffi <goffi@goffi.org> parents: 2214 diff changeset	187 def start(self):
a6c9bc4d1de0 jp (pubsub/node): added create and delete commands Goffi <goffi@goffi.org> parents: 2214 diff changeset	188 if not self.args.force:
a6c9bc4d1de0 jp (pubsub/node): added create and delete commands Goffi <goffi@goffi.org> parents: 2214 diff changeset	189 if not self.args.service:
a6c9bc4d1de0 jp (pubsub/node): added create and delete commands Goffi <goffi@goffi.org> parents: 2214 diff changeset	190 message = _(u"Are you sure to delete pep node [{node_id}] ?").format(
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	191 node_id=self.args.node
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	192 )
2221 a6c9bc4d1de0 jp (pubsub/node): added create and delete commands Goffi <goffi@goffi.org> parents: 2214 diff changeset	193 else:
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	194 message = _(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	195 u"Are you sure to delete node [{node_id}] on service [{service}] ?"
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	196 ).format(node_id=self.args.node, service=self.args.service)
2550 1d754bc14381 jp (base): new confirmOrQuit helper method to ask confirmation to user, and quit if he cancel Goffi <goffi@goffi.org> parents: 2539 diff changeset	197 self.host.confirmOrQuit(message, _(u"node deletion cancelled"))
2221 a6c9bc4d1de0 jp (pubsub/node): added create and delete commands Goffi <goffi@goffi.org> parents: 2214 diff changeset	198
a6c9bc4d1de0 jp (pubsub/node): added create and delete commands Goffi <goffi@goffi.org> parents: 2214 diff changeset	199 self.host.bridge.psNodeDelete(
a6c9bc4d1de0 jp (pubsub/node): added create and delete commands Goffi <goffi@goffi.org> parents: 2214 diff changeset	200 self.args.service,
a6c9bc4d1de0 jp (pubsub/node): added create and delete commands Goffi <goffi@goffi.org> parents: 2214 diff changeset	201 self.args.node,
a6c9bc4d1de0 jp (pubsub/node): added create and delete commands Goffi <goffi@goffi.org> parents: 2214 diff changeset	202 self.profile,
a6c9bc4d1de0 jp (pubsub/node): added create and delete commands Goffi <goffi@goffi.org> parents: 2214 diff changeset	203 callback=self.psNodeDeleteCb,
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	204 errback=partial(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	205 self.errback,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	206 msg=_(u"can't delete node: {}"),
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	207 exit_code=C.EXIT_BRIDGE_ERRBACK,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	208 ),
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	209 )
2221 a6c9bc4d1de0 jp (pubsub/node): added create and delete commands Goffi <goffi@goffi.org> parents: 2214 diff changeset	210
a6c9bc4d1de0 jp (pubsub/node): added create and delete commands Goffi <goffi@goffi.org> parents: 2214 diff changeset	211
2235 4db836386641 jp: added use_pubsub and use_pubsub_node_req arguments to CommandBase Goffi <goffi@goffi.org> parents: 2224 diff changeset	212 class NodeSet(base.CommandBase):
2199 ea0d0a4e2ad8 jp (pubsub/node): added set command Goffi <goffi@goffi.org> parents: 2197 diff changeset	213 def __init__(self, host):
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	214 base.CommandBase.__init__(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	215 self,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	216 host,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	217 "set",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	218 use_output=C.OUTPUT_DICT,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	219 use_pubsub=True,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	220 pubsub_flags={C.NODE},
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	221 use_verbose=True,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	222 help=_(u"set node configuration"),
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	223 )
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	224 self.need_loop = True
2199 ea0d0a4e2ad8 jp (pubsub/node): added set command Goffi <goffi@goffi.org> parents: 2197 diff changeset	225
ea0d0a4e2ad8 jp (pubsub/node): added set command Goffi <goffi@goffi.org> parents: 2197 diff changeset	226 def add_parser_options(self):
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	227 self.parser.add_argument(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	228 "-f",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	229 "--field",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	230 type=base.unicode_decoder,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	231 action="append",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	232 nargs=2,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	233 dest="fields",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	234 required=True,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	235 metavar=(u"KEY", u"VALUE"),
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	236 help=_(u"configuration field to set (required)"),
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	237 )
2199 ea0d0a4e2ad8 jp (pubsub/node): added set command Goffi <goffi@goffi.org> parents: 2197 diff changeset	238
ea0d0a4e2ad8 jp (pubsub/node): added set command Goffi <goffi@goffi.org> parents: 2197 diff changeset	239 def psNodeConfigurationSetCb(self):
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	240 self.disp(_(u"node configuration successful"), 1)
2199 ea0d0a4e2ad8 jp (pubsub/node): added set command Goffi <goffi@goffi.org> parents: 2197 diff changeset	241 self.host.quit()
ea0d0a4e2ad8 jp (pubsub/node): added set command Goffi <goffi@goffi.org> parents: 2197 diff changeset	242
ea0d0a4e2ad8 jp (pubsub/node): added set command Goffi <goffi@goffi.org> parents: 2197 diff changeset	243 def psNodeConfigurationSetEb(self, failure_):
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	244 self.disp(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	245 u"can't set node configuration: {reason}".format(reason=failure_), error=True
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	246 )
2199 ea0d0a4e2ad8 jp (pubsub/node): added set command Goffi <goffi@goffi.org> parents: 2197 diff changeset	247 self.host.quit(C.EXIT_BRIDGE_ERRBACK)
ea0d0a4e2ad8 jp (pubsub/node): added set command Goffi <goffi@goffi.org> parents: 2197 diff changeset	248
ea0d0a4e2ad8 jp (pubsub/node): added set command Goffi <goffi@goffi.org> parents: 2197 diff changeset	249 def getKeyName(self, k):
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	250 if not k.startswith(u"pubsub#"):
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	251 return u"pubsub#" + k
2199 ea0d0a4e2ad8 jp (pubsub/node): added set command Goffi <goffi@goffi.org> parents: 2197 diff changeset	252 else:
ea0d0a4e2ad8 jp (pubsub/node): added set command Goffi <goffi@goffi.org> parents: 2197 diff changeset	253 return k
ea0d0a4e2ad8 jp (pubsub/node): added set command Goffi <goffi@goffi.org> parents: 2197 diff changeset	254
ea0d0a4e2ad8 jp (pubsub/node): added set command Goffi <goffi@goffi.org> parents: 2197 diff changeset	255 def start(self):
ea0d0a4e2ad8 jp (pubsub/node): added set command Goffi <goffi@goffi.org> parents: 2197 diff changeset	256 self.host.bridge.psNodeConfigurationSet(
ea0d0a4e2ad8 jp (pubsub/node): added set command Goffi <goffi@goffi.org> parents: 2197 diff changeset	257 self.args.service,
ea0d0a4e2ad8 jp (pubsub/node): added set command Goffi <goffi@goffi.org> parents: 2197 diff changeset	258 self.args.node,
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	259 {self.getKeyName(k): v for k, v in self.args.fields},
2199 ea0d0a4e2ad8 jp (pubsub/node): added set command Goffi <goffi@goffi.org> parents: 2197 diff changeset	260 self.profile,
ea0d0a4e2ad8 jp (pubsub/node): added set command Goffi <goffi@goffi.org> parents: 2197 diff changeset	261 callback=self.psNodeConfigurationSetCb,
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	262 errback=self.psNodeConfigurationSetEb,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	263 )
2199 ea0d0a4e2ad8 jp (pubsub/node): added set command Goffi <goffi@goffi.org> parents: 2197 diff changeset	264
2197 e0e06391ce91 jp (pubsub): added pubsub/node/info command to retrieve node configuration Goffi <goffi@goffi.org> parents: 2195 diff changeset	265
2235 4db836386641 jp: added use_pubsub and use_pubsub_node_req arguments to CommandBase Goffi <goffi@goffi.org> parents: 2224 diff changeset	266 class NodeAffiliationsGet(base.CommandBase):
2204 afc703419186 jp (pubsub): added affiliations and node/affiliations/get: Goffi <goffi@goffi.org> parents: 2199 diff changeset	267 def __init__(self, host):
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	268 base.CommandBase.__init__(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	269 self,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	270 host,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	271 "get",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	272 use_output=C.OUTPUT_DICT,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	273 use_pubsub=True,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	274 pubsub_flags={C.NODE},
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	275 help=_(u"retrieve node affiliations (for node owner)"),
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	276 )
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	277 self.need_loop = True
2204 afc703419186 jp (pubsub): added affiliations and node/affiliations/get: Goffi <goffi@goffi.org> parents: 2199 diff changeset	278
afc703419186 jp (pubsub): added affiliations and node/affiliations/get: Goffi <goffi@goffi.org> parents: 2199 diff changeset	279 def add_parser_options(self):
2235 4db836386641 jp: added use_pubsub and use_pubsub_node_req arguments to CommandBase Goffi <goffi@goffi.org> parents: 2224 diff changeset	280 pass
2204 afc703419186 jp (pubsub): added affiliations and node/affiliations/get: Goffi <goffi@goffi.org> parents: 2199 diff changeset	281
afc703419186 jp (pubsub): added affiliations and node/affiliations/get: Goffi <goffi@goffi.org> parents: 2199 diff changeset	282 def psNodeAffiliationsGetCb(self, affiliations):
afc703419186 jp (pubsub): added affiliations and node/affiliations/get: Goffi <goffi@goffi.org> parents: 2199 diff changeset	283 self.output(affiliations)
afc703419186 jp (pubsub): added affiliations and node/affiliations/get: Goffi <goffi@goffi.org> parents: 2199 diff changeset	284 self.host.quit()
afc703419186 jp (pubsub): added affiliations and node/affiliations/get: Goffi <goffi@goffi.org> parents: 2199 diff changeset	285
afc703419186 jp (pubsub): added affiliations and node/affiliations/get: Goffi <goffi@goffi.org> parents: 2199 diff changeset	286 def psNodeAffiliationsGetEb(self, failure_):
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	287 self.disp(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	288 u"can't get node affiliations: {reason}".format(reason=failure_), error=True
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	289 )
2204 afc703419186 jp (pubsub): added affiliations and node/affiliations/get: Goffi <goffi@goffi.org> parents: 2199 diff changeset	290 self.host.quit(C.EXIT_BRIDGE_ERRBACK)
afc703419186 jp (pubsub): added affiliations and node/affiliations/get: Goffi <goffi@goffi.org> parents: 2199 diff changeset	291
afc703419186 jp (pubsub): added affiliations and node/affiliations/get: Goffi <goffi@goffi.org> parents: 2199 diff changeset	292 def start(self):
afc703419186 jp (pubsub): added affiliations and node/affiliations/get: Goffi <goffi@goffi.org> parents: 2199 diff changeset	293 self.host.bridge.psNodeAffiliationsGet(
afc703419186 jp (pubsub): added affiliations and node/affiliations/get: Goffi <goffi@goffi.org> parents: 2199 diff changeset	294 self.args.service,
afc703419186 jp (pubsub): added affiliations and node/affiliations/get: Goffi <goffi@goffi.org> parents: 2199 diff changeset	295 self.args.node,
afc703419186 jp (pubsub): added affiliations and node/affiliations/get: Goffi <goffi@goffi.org> parents: 2199 diff changeset	296 self.profile,
afc703419186 jp (pubsub): added affiliations and node/affiliations/get: Goffi <goffi@goffi.org> parents: 2199 diff changeset	297 callback=self.psNodeAffiliationsGetCb,
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	298 errback=self.psNodeAffiliationsGetEb,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	299 )
2204 afc703419186 jp (pubsub): added affiliations and node/affiliations/get: Goffi <goffi@goffi.org> parents: 2199 diff changeset	300
afc703419186 jp (pubsub): added affiliations and node/affiliations/get: Goffi <goffi@goffi.org> parents: 2199 diff changeset	301
2235 4db836386641 jp: added use_pubsub and use_pubsub_node_req arguments to CommandBase Goffi <goffi@goffi.org> parents: 2224 diff changeset	302 class NodeAffiliationsSet(base.CommandBase):
2207 d662bdd682b2 jp (pubsub/node) added pubsub/node/affiliations/set command Goffi <goffi@goffi.org> parents: 2204 diff changeset	303 def __init__(self, host):
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	304 base.CommandBase.__init__(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	305 self,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	306 host,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	307 "set",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	308 use_pubsub=True,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	309 pubsub_flags={C.NODE},
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	310 use_verbose=True,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	311 help=_(u"set affiliations (for node owner)"),
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	312 )
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	313 self.need_loop = True
2207 d662bdd682b2 jp (pubsub/node) added pubsub/node/affiliations/set command Goffi <goffi@goffi.org> parents: 2204 diff changeset	314
d662bdd682b2 jp (pubsub/node) added pubsub/node/affiliations/set command Goffi <goffi@goffi.org> parents: 2204 diff changeset	315 def add_parser_options(self):
d662bdd682b2 jp (pubsub/node) added pubsub/node/affiliations/set command Goffi <goffi@goffi.org> parents: 2204 diff changeset	316 # XXX: we use optional argument syntax for a required one because list of list of 2 elements
d662bdd682b2 jp (pubsub/node) added pubsub/node/affiliations/set command Goffi <goffi@goffi.org> parents: 2204 diff changeset	317 # (uses to construct dicts) don't work with positional arguments
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	318 self.parser.add_argument(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	319 "-a",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	320 "--affiliation",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	321 dest="affiliations",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	322 metavar=("JID", "AFFILIATION"),
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	323 required=True,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	324 type=base.unicode_decoder,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	325 action="append",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	326 nargs=2,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	327 help=_(u"entity/affiliation couple(s)"),
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	328 )
2207 d662bdd682b2 jp (pubsub/node) added pubsub/node/affiliations/set command Goffi <goffi@goffi.org> parents: 2204 diff changeset	329
d662bdd682b2 jp (pubsub/node) added pubsub/node/affiliations/set command Goffi <goffi@goffi.org> parents: 2204 diff changeset	330 def psNodeAffiliationsSetCb(self):
d662bdd682b2 jp (pubsub/node) added pubsub/node/affiliations/set command Goffi <goffi@goffi.org> parents: 2204 diff changeset	331 self.disp(_(u"affiliations have been set"), 1)
d662bdd682b2 jp (pubsub/node) added pubsub/node/affiliations/set command Goffi <goffi@goffi.org> parents: 2204 diff changeset	332 self.host.quit()
d662bdd682b2 jp (pubsub/node) added pubsub/node/affiliations/set command Goffi <goffi@goffi.org> parents: 2204 diff changeset	333
d662bdd682b2 jp (pubsub/node) added pubsub/node/affiliations/set command Goffi <goffi@goffi.org> parents: 2204 diff changeset	334 def psNodeAffiliationsSetEb(self, failure_):
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	335 self.disp(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	336 u"can't set node affiliations: {reason}".format(reason=failure_), error=True
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	337 )
2207 d662bdd682b2 jp (pubsub/node) added pubsub/node/affiliations/set command Goffi <goffi@goffi.org> parents: 2204 diff changeset	338 self.host.quit(C.EXIT_BRIDGE_ERRBACK)
d662bdd682b2 jp (pubsub/node) added pubsub/node/affiliations/set command Goffi <goffi@goffi.org> parents: 2204 diff changeset	339
d662bdd682b2 jp (pubsub/node) added pubsub/node/affiliations/set command Goffi <goffi@goffi.org> parents: 2204 diff changeset	340 def start(self):
d662bdd682b2 jp (pubsub/node) added pubsub/node/affiliations/set command Goffi <goffi@goffi.org> parents: 2204 diff changeset	341 affiliations = dict(self.args.affiliations)
d662bdd682b2 jp (pubsub/node) added pubsub/node/affiliations/set command Goffi <goffi@goffi.org> parents: 2204 diff changeset	342 self.host.bridge.psNodeAffiliationsSet(
d662bdd682b2 jp (pubsub/node) added pubsub/node/affiliations/set command Goffi <goffi@goffi.org> parents: 2204 diff changeset	343 self.args.service,
d662bdd682b2 jp (pubsub/node) added pubsub/node/affiliations/set command Goffi <goffi@goffi.org> parents: 2204 diff changeset	344 self.args.node,
d662bdd682b2 jp (pubsub/node) added pubsub/node/affiliations/set command Goffi <goffi@goffi.org> parents: 2204 diff changeset	345 affiliations,
d662bdd682b2 jp (pubsub/node) added pubsub/node/affiliations/set command Goffi <goffi@goffi.org> parents: 2204 diff changeset	346 self.profile,
d662bdd682b2 jp (pubsub/node) added pubsub/node/affiliations/set command Goffi <goffi@goffi.org> parents: 2204 diff changeset	347 callback=self.psNodeAffiliationsSetCb,
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	348 errback=self.psNodeAffiliationsSetEb,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	349 )
2207 d662bdd682b2 jp (pubsub/node) added pubsub/node/affiliations/set command Goffi <goffi@goffi.org> parents: 2204 diff changeset	350
d662bdd682b2 jp (pubsub/node) added pubsub/node/affiliations/set command Goffi <goffi@goffi.org> parents: 2204 diff changeset	351
2204 afc703419186 jp (pubsub): added affiliations and node/affiliations/get: Goffi <goffi@goffi.org> parents: 2199 diff changeset	352 class NodeAffiliations(base.CommandBase):
2207 d662bdd682b2 jp (pubsub/node) added pubsub/node/affiliations/set command Goffi <goffi@goffi.org> parents: 2204 diff changeset	353 subcommands = (NodeAffiliationsGet, NodeAffiliationsSet)
2204 afc703419186 jp (pubsub): added affiliations and node/affiliations/get: Goffi <goffi@goffi.org> parents: 2199 diff changeset	354
afc703419186 jp (pubsub): added affiliations and node/affiliations/get: Goffi <goffi@goffi.org> parents: 2199 diff changeset	355 def __init__(self, host):
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	356 super(NodeAffiliations, self).__init__(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	357 host,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	358 "affiliations",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	359 use_profile=False,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	360 help=_(u"set or retrieve node affiliations"),
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	361 )
2204 afc703419186 jp (pubsub): added affiliations and node/affiliations/get: Goffi <goffi@goffi.org> parents: 2199 diff changeset	362
afc703419186 jp (pubsub): added affiliations and node/affiliations/get: Goffi <goffi@goffi.org> parents: 2199 diff changeset	363
2339 d94e932be8b3 jp (pubsub/node): added subscriptions subcommand: Goffi <goffi@goffi.org> parents: 2320 diff changeset	364 class NodeSubscriptionsGet(base.CommandBase):
d94e932be8b3 jp (pubsub/node): added subscriptions subcommand: Goffi <goffi@goffi.org> parents: 2320 diff changeset	365 def __init__(self, host):
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	366 base.CommandBase.__init__(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	367 self,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	368 host,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	369 "get",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	370 use_output=C.OUTPUT_DICT,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	371 use_pubsub=True,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	372 pubsub_flags={C.NODE},
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	373 help=_(u"retrieve node subscriptions (for node owner)"),
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	374 )
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	375 self.need_loop = True
2339 d94e932be8b3 jp (pubsub/node): added subscriptions subcommand: Goffi <goffi@goffi.org> parents: 2320 diff changeset	376
d94e932be8b3 jp (pubsub/node): added subscriptions subcommand: Goffi <goffi@goffi.org> parents: 2320 diff changeset	377 def add_parser_options(self):
d94e932be8b3 jp (pubsub/node): added subscriptions subcommand: Goffi <goffi@goffi.org> parents: 2320 diff changeset	378 pass
d94e932be8b3 jp (pubsub/node): added subscriptions subcommand: Goffi <goffi@goffi.org> parents: 2320 diff changeset	379
d94e932be8b3 jp (pubsub/node): added subscriptions subcommand: Goffi <goffi@goffi.org> parents: 2320 diff changeset	380 def psNodeSubscriptionsGetCb(self, subscriptions):
d94e932be8b3 jp (pubsub/node): added subscriptions subcommand: Goffi <goffi@goffi.org> parents: 2320 diff changeset	381 self.output(subscriptions)
d94e932be8b3 jp (pubsub/node): added subscriptions subcommand: Goffi <goffi@goffi.org> parents: 2320 diff changeset	382 self.host.quit()
d94e932be8b3 jp (pubsub/node): added subscriptions subcommand: Goffi <goffi@goffi.org> parents: 2320 diff changeset	383
d94e932be8b3 jp (pubsub/node): added subscriptions subcommand: Goffi <goffi@goffi.org> parents: 2320 diff changeset	384 def psNodeSubscriptionsGetEb(self, failure_):
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	385 self.disp(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	386 u"can't get node subscriptions: {reason}".format(reason=failure_), error=True
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	387 )
2339 d94e932be8b3 jp (pubsub/node): added subscriptions subcommand: Goffi <goffi@goffi.org> parents: 2320 diff changeset	388 self.host.quit(C.EXIT_BRIDGE_ERRBACK)
d94e932be8b3 jp (pubsub/node): added subscriptions subcommand: Goffi <goffi@goffi.org> parents: 2320 diff changeset	389
d94e932be8b3 jp (pubsub/node): added subscriptions subcommand: Goffi <goffi@goffi.org> parents: 2320 diff changeset	390 def start(self):
d94e932be8b3 jp (pubsub/node): added subscriptions subcommand: Goffi <goffi@goffi.org> parents: 2320 diff changeset	391 self.host.bridge.psNodeSubscriptionsGet(
d94e932be8b3 jp (pubsub/node): added subscriptions subcommand: Goffi <goffi@goffi.org> parents: 2320 diff changeset	392 self.args.service,
d94e932be8b3 jp (pubsub/node): added subscriptions subcommand: Goffi <goffi@goffi.org> parents: 2320 diff changeset	393 self.args.node,
d94e932be8b3 jp (pubsub/node): added subscriptions subcommand: Goffi <goffi@goffi.org> parents: 2320 diff changeset	394 self.profile,
d94e932be8b3 jp (pubsub/node): added subscriptions subcommand: Goffi <goffi@goffi.org> parents: 2320 diff changeset	395 callback=self.psNodeSubscriptionsGetCb,
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	396 errback=self.psNodeSubscriptionsGetEb,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	397 )
2339 d94e932be8b3 jp (pubsub/node): added subscriptions subcommand: Goffi <goffi@goffi.org> parents: 2320 diff changeset	398
d94e932be8b3 jp (pubsub/node): added subscriptions subcommand: Goffi <goffi@goffi.org> parents: 2320 diff changeset	399
d94e932be8b3 jp (pubsub/node): added subscriptions subcommand: Goffi <goffi@goffi.org> parents: 2320 diff changeset	400 class StoreSubscriptionAction(argparse.Action):
d94e932be8b3 jp (pubsub/node): added subscriptions subcommand: Goffi <goffi@goffi.org> parents: 2320 diff changeset	401 """Action which handle subscription parameter for owner
d94e932be8b3 jp (pubsub/node): added subscriptions subcommand: Goffi <goffi@goffi.org> parents: 2320 diff changeset	402
d94e932be8b3 jp (pubsub/node): added subscriptions subcommand: Goffi <goffi@goffi.org> parents: 2320 diff changeset	403 list is given by pairs: jid and subscription state
d94e932be8b3 jp (pubsub/node): added subscriptions subcommand: Goffi <goffi@goffi.org> parents: 2320 diff changeset	404 if subscription state is not specified, it default to "subscribed"
d94e932be8b3 jp (pubsub/node): added subscriptions subcommand: Goffi <goffi@goffi.org> parents: 2320 diff changeset	405 """
d94e932be8b3 jp (pubsub/node): added subscriptions subcommand: Goffi <goffi@goffi.org> parents: 2320 diff changeset	406
d94e932be8b3 jp (pubsub/node): added subscriptions subcommand: Goffi <goffi@goffi.org> parents: 2320 diff changeset	407 def __call__(self, parser, namespace, values, option_string):
d94e932be8b3 jp (pubsub/node): added subscriptions subcommand: Goffi <goffi@goffi.org> parents: 2320 diff changeset	408 dest_dict = getattr(namespace, self.dest)
d94e932be8b3 jp (pubsub/node): added subscriptions subcommand: Goffi <goffi@goffi.org> parents: 2320 diff changeset	409 while values:
d94e932be8b3 jp (pubsub/node): added subscriptions subcommand: Goffi <goffi@goffi.org> parents: 2320 diff changeset	410 jid_s = values.pop(0)
d94e932be8b3 jp (pubsub/node): added subscriptions subcommand: Goffi <goffi@goffi.org> parents: 2320 diff changeset	411 try:
d94e932be8b3 jp (pubsub/node): added subscriptions subcommand: Goffi <goffi@goffi.org> parents: 2320 diff changeset	412 subscription = values.pop(0)
d94e932be8b3 jp (pubsub/node): added subscriptions subcommand: Goffi <goffi@goffi.org> parents: 2320 diff changeset	413 except IndexError:
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	414 subscription = "subscribed"
2339 d94e932be8b3 jp (pubsub/node): added subscriptions subcommand: Goffi <goffi@goffi.org> parents: 2320 diff changeset	415 if subscription not in ALLOWED_SUBSCRIPTIONS_OWNER:
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	416 parser.error(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	417 _(u"subscription must be one of {}").format(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	418 u", ".join(ALLOWED_SUBSCRIPTIONS_OWNER)
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	419 )
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	420 )
2339 d94e932be8b3 jp (pubsub/node): added subscriptions subcommand: Goffi <goffi@goffi.org> parents: 2320 diff changeset	421 dest_dict[jid_s] = subscription
d94e932be8b3 jp (pubsub/node): added subscriptions subcommand: Goffi <goffi@goffi.org> parents: 2320 diff changeset	422
d94e932be8b3 jp (pubsub/node): added subscriptions subcommand: Goffi <goffi@goffi.org> parents: 2320 diff changeset	423
d94e932be8b3 jp (pubsub/node): added subscriptions subcommand: Goffi <goffi@goffi.org> parents: 2320 diff changeset	424 class NodeSubscriptionsSet(base.CommandBase):
d94e932be8b3 jp (pubsub/node): added subscriptions subcommand: Goffi <goffi@goffi.org> parents: 2320 diff changeset	425 def __init__(self, host):
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	426 base.CommandBase.__init__(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	427 self,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	428 host,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	429 "set",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	430 use_pubsub=True,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	431 pubsub_flags={C.NODE},
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	432 use_verbose=True,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	433 help=_(u"set/modify subscriptions (for node owner)"),
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	434 )
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	435 self.need_loop = True
2339 d94e932be8b3 jp (pubsub/node): added subscriptions subcommand: Goffi <goffi@goffi.org> parents: 2320 diff changeset	436
d94e932be8b3 jp (pubsub/node): added subscriptions subcommand: Goffi <goffi@goffi.org> parents: 2320 diff changeset	437 def add_parser_options(self):
d94e932be8b3 jp (pubsub/node): added subscriptions subcommand: Goffi <goffi@goffi.org> parents: 2320 diff changeset	438 # XXX: we use optional argument syntax for a required one because list of list of 2 elements
d94e932be8b3 jp (pubsub/node): added subscriptions subcommand: Goffi <goffi@goffi.org> parents: 2320 diff changeset	439 # (uses to construct dicts) don't work with positional arguments
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	440 self.parser.add_argument(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	441 "-S",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	442 "--subscription",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	443 dest="subscriptions",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	444 default={},
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	445 nargs="+",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	446 metavar=("JID [SUSBSCRIPTION]"),
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	447 required=True,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	448 type=base.unicode_decoder,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	449 action=StoreSubscriptionAction,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	450 help=_(u"entity/subscription couple(s)"),
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	451 )
2339 d94e932be8b3 jp (pubsub/node): added subscriptions subcommand: Goffi <goffi@goffi.org> parents: 2320 diff changeset	452
d94e932be8b3 jp (pubsub/node): added subscriptions subcommand: Goffi <goffi@goffi.org> parents: 2320 diff changeset	453 def psNodeSubscriptionsSetCb(self):
d94e932be8b3 jp (pubsub/node): added subscriptions subcommand: Goffi <goffi@goffi.org> parents: 2320 diff changeset	454 self.disp(_(u"subscriptions have been set"), 1)
d94e932be8b3 jp (pubsub/node): added subscriptions subcommand: Goffi <goffi@goffi.org> parents: 2320 diff changeset	455 self.host.quit()
d94e932be8b3 jp (pubsub/node): added subscriptions subcommand: Goffi <goffi@goffi.org> parents: 2320 diff changeset	456
d94e932be8b3 jp (pubsub/node): added subscriptions subcommand: Goffi <goffi@goffi.org> parents: 2320 diff changeset	457 def psNodeSubscriptionsSetEb(self, failure_):
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	458 self.disp(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	459 u"can't set node subscriptions: {reason}".format(reason=failure_), error=True
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	460 )
2339 d94e932be8b3 jp (pubsub/node): added subscriptions subcommand: Goffi <goffi@goffi.org> parents: 2320 diff changeset	461 self.host.quit(C.EXIT_BRIDGE_ERRBACK)
d94e932be8b3 jp (pubsub/node): added subscriptions subcommand: Goffi <goffi@goffi.org> parents: 2320 diff changeset	462
d94e932be8b3 jp (pubsub/node): added subscriptions subcommand: Goffi <goffi@goffi.org> parents: 2320 diff changeset	463 def start(self):
d94e932be8b3 jp (pubsub/node): added subscriptions subcommand: Goffi <goffi@goffi.org> parents: 2320 diff changeset	464 self.host.bridge.psNodeSubscriptionsSet(
d94e932be8b3 jp (pubsub/node): added subscriptions subcommand: Goffi <goffi@goffi.org> parents: 2320 diff changeset	465 self.args.service,
d94e932be8b3 jp (pubsub/node): added subscriptions subcommand: Goffi <goffi@goffi.org> parents: 2320 diff changeset	466 self.args.node,
d94e932be8b3 jp (pubsub/node): added subscriptions subcommand: Goffi <goffi@goffi.org> parents: 2320 diff changeset	467 self.args.subscriptions,
d94e932be8b3 jp (pubsub/node): added subscriptions subcommand: Goffi <goffi@goffi.org> parents: 2320 diff changeset	468 self.profile,
d94e932be8b3 jp (pubsub/node): added subscriptions subcommand: Goffi <goffi@goffi.org> parents: 2320 diff changeset	469 callback=self.psNodeSubscriptionsSetCb,
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	470 errback=self.psNodeSubscriptionsSetEb,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	471 )
2339 d94e932be8b3 jp (pubsub/node): added subscriptions subcommand: Goffi <goffi@goffi.org> parents: 2320 diff changeset	472
d94e932be8b3 jp (pubsub/node): added subscriptions subcommand: Goffi <goffi@goffi.org> parents: 2320 diff changeset	473
d94e932be8b3 jp (pubsub/node): added subscriptions subcommand: Goffi <goffi@goffi.org> parents: 2320 diff changeset	474 class NodeSubscriptions(base.CommandBase):
d94e932be8b3 jp (pubsub/node): added subscriptions subcommand: Goffi <goffi@goffi.org> parents: 2320 diff changeset	475 subcommands = (NodeSubscriptionsGet, NodeSubscriptionsSet)
d94e932be8b3 jp (pubsub/node): added subscriptions subcommand: Goffi <goffi@goffi.org> parents: 2320 diff changeset	476
d94e932be8b3 jp (pubsub/node): added subscriptions subcommand: Goffi <goffi@goffi.org> parents: 2320 diff changeset	477 def __init__(self, host):
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	478 super(NodeSubscriptions, self).__init__(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	479 host,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	480 "subscriptions",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	481 use_profile=False,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	482 help=_(u"get or modify node subscriptions"),
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	483 )
2339 d94e932be8b3 jp (pubsub/node): added subscriptions subcommand: Goffi <goffi@goffi.org> parents: 2320 diff changeset	484
d94e932be8b3 jp (pubsub/node): added subscriptions subcommand: Goffi <goffi@goffi.org> parents: 2320 diff changeset	485
2351 3c0a3fae1862 jp (pubsub/node): added schema (set/edit/get) commands to manipulate PubSub node schema Goffi <goffi@goffi.org> parents: 2339 diff changeset	486 class NodeSchemaSet(base.CommandBase):
3c0a3fae1862 jp (pubsub/node): added schema (set/edit/get) commands to manipulate PubSub node schema Goffi <goffi@goffi.org> parents: 2339 diff changeset	487 def __init__(self, host):
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	488 base.CommandBase.__init__(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	489 self,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	490 host,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	491 "set",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	492 use_pubsub=True,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	493 pubsub_flags={C.NODE},
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	494 use_verbose=True,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	495 help=_(u"set/replace a schema"),
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	496 )
2351 3c0a3fae1862 jp (pubsub/node): added schema (set/edit/get) commands to manipulate PubSub node schema Goffi <goffi@goffi.org> parents: 2339 diff changeset	497 self.need_loop = True
3c0a3fae1862 jp (pubsub/node): added schema (set/edit/get) commands to manipulate PubSub node schema Goffi <goffi@goffi.org> parents: 2339 diff changeset	498
3c0a3fae1862 jp (pubsub/node): added schema (set/edit/get) commands to manipulate PubSub node schema Goffi <goffi@goffi.org> parents: 2339 diff changeset	499 def add_parser_options(self):
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	500 self.parser.add_argument("schema", help=_(u"schema to set (must be XML)"))
2351 3c0a3fae1862 jp (pubsub/node): added schema (set/edit/get) commands to manipulate PubSub node schema Goffi <goffi@goffi.org> parents: 2339 diff changeset	501
3c0a3fae1862 jp (pubsub/node): added schema (set/edit/get) commands to manipulate PubSub node schema Goffi <goffi@goffi.org> parents: 2339 diff changeset	502 def psSchemaSetCb(self):
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	503 self.disp(_(u"schema has been set"), 1)
2351 3c0a3fae1862 jp (pubsub/node): added schema (set/edit/get) commands to manipulate PubSub node schema Goffi <goffi@goffi.org> parents: 2339 diff changeset	504 self.host.quit()
3c0a3fae1862 jp (pubsub/node): added schema (set/edit/get) commands to manipulate PubSub node schema Goffi <goffi@goffi.org> parents: 2339 diff changeset	505
3c0a3fae1862 jp (pubsub/node): added schema (set/edit/get) commands to manipulate PubSub node schema Goffi <goffi@goffi.org> parents: 2339 diff changeset	506 def start(self):
3c0a3fae1862 jp (pubsub/node): added schema (set/edit/get) commands to manipulate PubSub node schema Goffi <goffi@goffi.org> parents: 2339 diff changeset	507 self.host.bridge.psSchemaSet(
3c0a3fae1862 jp (pubsub/node): added schema (set/edit/get) commands to manipulate PubSub node schema Goffi <goffi@goffi.org> parents: 2339 diff changeset	508 self.args.service,
3c0a3fae1862 jp (pubsub/node): added schema (set/edit/get) commands to manipulate PubSub node schema Goffi <goffi@goffi.org> parents: 2339 diff changeset	509 self.args.node,
3c0a3fae1862 jp (pubsub/node): added schema (set/edit/get) commands to manipulate PubSub node schema Goffi <goffi@goffi.org> parents: 2339 diff changeset	510 self.args.schema,
3c0a3fae1862 jp (pubsub/node): added schema (set/edit/get) commands to manipulate PubSub node schema Goffi <goffi@goffi.org> parents: 2339 diff changeset	511 self.profile,
3c0a3fae1862 jp (pubsub/node): added schema (set/edit/get) commands to manipulate PubSub node schema Goffi <goffi@goffi.org> parents: 2339 diff changeset	512 callback=self.psSchemaSetCb,
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	513 errback=partial(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	514 self.errback,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	515 msg=_(u"can't set schema: {}"),
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	516 exit_code=C.EXIT_BRIDGE_ERRBACK,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	517 ),
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	518 )
2351 3c0a3fae1862 jp (pubsub/node): added schema (set/edit/get) commands to manipulate PubSub node schema Goffi <goffi@goffi.org> parents: 2339 diff changeset	519
3c0a3fae1862 jp (pubsub/node): added schema (set/edit/get) commands to manipulate PubSub node schema Goffi <goffi@goffi.org> parents: 2339 diff changeset	520
3c0a3fae1862 jp (pubsub/node): added schema (set/edit/get) commands to manipulate PubSub node schema Goffi <goffi@goffi.org> parents: 2339 diff changeset	521 class NodeSchemaEdit(base.CommandBase, common.BaseEdit):
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	522 use_items = False
2351 3c0a3fae1862 jp (pubsub/node): added schema (set/edit/get) commands to manipulate PubSub node schema Goffi <goffi@goffi.org> parents: 2339 diff changeset	523
3c0a3fae1862 jp (pubsub/node): added schema (set/edit/get) commands to manipulate PubSub node schema Goffi <goffi@goffi.org> parents: 2339 diff changeset	524 def __init__(self, host):
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	525 base.CommandBase.__init__(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	526 self,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	527 host,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	528 "edit",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	529 use_pubsub=True,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	530 pubsub_flags={C.NODE},
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	531 use_draft=True,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	532 use_verbose=True,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	533 help=_(u"edit a schema"),
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	534 )
2351 3c0a3fae1862 jp (pubsub/node): added schema (set/edit/get) commands to manipulate PubSub node schema Goffi <goffi@goffi.org> parents: 2339 diff changeset	535 common.BaseEdit.__init__(self, self.host, PUBSUB_SCHEMA_TMP_DIR)
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	536 self.need_loop = True
2351 3c0a3fae1862 jp (pubsub/node): added schema (set/edit/get) commands to manipulate PubSub node schema Goffi <goffi@goffi.org> parents: 2339 diff changeset	537
3c0a3fae1862 jp (pubsub/node): added schema (set/edit/get) commands to manipulate PubSub node schema Goffi <goffi@goffi.org> parents: 2339 diff changeset	538 def add_parser_options(self):
2532 772447ec070f jp: pubsub options refactoring: Goffi <goffi@goffi.org> parents: 2483 diff changeset	539 pass
2351 3c0a3fae1862 jp (pubsub/node): added schema (set/edit/get) commands to manipulate PubSub node schema Goffi <goffi@goffi.org> parents: 2339 diff changeset	540
3c0a3fae1862 jp (pubsub/node): added schema (set/edit/get) commands to manipulate PubSub node schema Goffi <goffi@goffi.org> parents: 2339 diff changeset	541 def psSchemaSetCb(self):
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	542 self.disp(_(u"schema has been set"), 1)
2351 3c0a3fae1862 jp (pubsub/node): added schema (set/edit/get) commands to manipulate PubSub node schema Goffi <goffi@goffi.org> parents: 2339 diff changeset	543 self.host.quit()
3c0a3fae1862 jp (pubsub/node): added schema (set/edit/get) commands to manipulate PubSub node schema Goffi <goffi@goffi.org> parents: 2339 diff changeset	544
3c0a3fae1862 jp (pubsub/node): added schema (set/edit/get) commands to manipulate PubSub node schema Goffi <goffi@goffi.org> parents: 2339 diff changeset	545 def publish(self, schema):
3c0a3fae1862 jp (pubsub/node): added schema (set/edit/get) commands to manipulate PubSub node schema Goffi <goffi@goffi.org> parents: 2339 diff changeset	546 self.host.bridge.psSchemaSet(
3c0a3fae1862 jp (pubsub/node): added schema (set/edit/get) commands to manipulate PubSub node schema Goffi <goffi@goffi.org> parents: 2339 diff changeset	547 self.args.service,
3c0a3fae1862 jp (pubsub/node): added schema (set/edit/get) commands to manipulate PubSub node schema Goffi <goffi@goffi.org> parents: 2339 diff changeset	548 self.args.node,
3c0a3fae1862 jp (pubsub/node): added schema (set/edit/get) commands to manipulate PubSub node schema Goffi <goffi@goffi.org> parents: 2339 diff changeset	549 schema,
3c0a3fae1862 jp (pubsub/node): added schema (set/edit/get) commands to manipulate PubSub node schema Goffi <goffi@goffi.org> parents: 2339 diff changeset	550 self.profile,
3c0a3fae1862 jp (pubsub/node): added schema (set/edit/get) commands to manipulate PubSub node schema Goffi <goffi@goffi.org> parents: 2339 diff changeset	551 callback=self.psSchemaSetCb,
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	552 errback=partial(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	553 self.errback,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	554 msg=_(u"can't set schema: {}"),
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	555 exit_code=C.EXIT_BRIDGE_ERRBACK,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	556 ),
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	557 )
2351 3c0a3fae1862 jp (pubsub/node): added schema (set/edit/get) commands to manipulate PubSub node schema Goffi <goffi@goffi.org> parents: 2339 diff changeset	558
3c0a3fae1862 jp (pubsub/node): added schema (set/edit/get) commands to manipulate PubSub node schema Goffi <goffi@goffi.org> parents: 2339 diff changeset	559 def psSchemaGetCb(self, schema):
3c0a3fae1862 jp (pubsub/node): added schema (set/edit/get) commands to manipulate PubSub node schema Goffi <goffi@goffi.org> parents: 2339 diff changeset	560 try:
3c0a3fae1862 jp (pubsub/node): added schema (set/edit/get) commands to manipulate PubSub node schema Goffi <goffi@goffi.org> parents: 2339 diff changeset	561 from lxml import etree
3c0a3fae1862 jp (pubsub/node): added schema (set/edit/get) commands to manipulate PubSub node schema Goffi <goffi@goffi.org> parents: 2339 diff changeset	562 except ImportError:
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	563 self.disp(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	564 u'lxml module must be installed to use edit, please install it with "pip install lxml"',
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	565 error=True,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	566 )
2351 3c0a3fae1862 jp (pubsub/node): added schema (set/edit/get) commands to manipulate PubSub node schema Goffi <goffi@goffi.org> parents: 2339 diff changeset	567 self.host.quit(1)
3c0a3fae1862 jp (pubsub/node): added schema (set/edit/get) commands to manipulate PubSub node schema Goffi <goffi@goffi.org> parents: 2339 diff changeset	568 content_file_obj, content_file_path = self.getTmpFile()
2359 47516e90d26a jp (pubsub/node/schema): check that a schema exists before parsing it (fix issue on new schema edition) Goffi <goffi@goffi.org> parents: 2353 diff changeset	569 schema = schema.strip()
47516e90d26a jp (pubsub/node/schema): check that a schema exists before parsing it (fix issue on new schema edition) Goffi <goffi@goffi.org> parents: 2353 diff changeset	570 if schema:
47516e90d26a jp (pubsub/node/schema): check that a schema exists before parsing it (fix issue on new schema edition) Goffi <goffi@goffi.org> parents: 2353 diff changeset	571 parser = etree.XMLParser(remove_blank_text=True)
47516e90d26a jp (pubsub/node/schema): check that a schema exists before parsing it (fix issue on new schema edition) Goffi <goffi@goffi.org> parents: 2353 diff changeset	572 schema_elt = etree.fromstring(schema, parser)
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	573 content_file_obj.write(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	574 etree.tostring(schema_elt, encoding="utf-8", pretty_print=True)
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	575 )
2359 47516e90d26a jp (pubsub/node/schema): check that a schema exists before parsing it (fix issue on new schema edition) Goffi <goffi@goffi.org> parents: 2353 diff changeset	576 content_file_obj.seek(0)
2351 3c0a3fae1862 jp (pubsub/node): added schema (set/edit/get) commands to manipulate PubSub node schema Goffi <goffi@goffi.org> parents: 2339 diff changeset	577 self.runEditor("pubsub_schema_editor_args", content_file_path, content_file_obj)
3c0a3fae1862 jp (pubsub/node): added schema (set/edit/get) commands to manipulate PubSub node schema Goffi <goffi@goffi.org> parents: 2339 diff changeset	578
3c0a3fae1862 jp (pubsub/node): added schema (set/edit/get) commands to manipulate PubSub node schema Goffi <goffi@goffi.org> parents: 2339 diff changeset	579 def start(self):
3c0a3fae1862 jp (pubsub/node): added schema (set/edit/get) commands to manipulate PubSub node schema Goffi <goffi@goffi.org> parents: 2339 diff changeset	580 self.host.bridge.psSchemaGet(
3c0a3fae1862 jp (pubsub/node): added schema (set/edit/get) commands to manipulate PubSub node schema Goffi <goffi@goffi.org> parents: 2339 diff changeset	581 self.args.service,
3c0a3fae1862 jp (pubsub/node): added schema (set/edit/get) commands to manipulate PubSub node schema Goffi <goffi@goffi.org> parents: 2339 diff changeset	582 self.args.node,
3c0a3fae1862 jp (pubsub/node): added schema (set/edit/get) commands to manipulate PubSub node schema Goffi <goffi@goffi.org> parents: 2339 diff changeset	583 self.profile,
3c0a3fae1862 jp (pubsub/node): added schema (set/edit/get) commands to manipulate PubSub node schema Goffi <goffi@goffi.org> parents: 2339 diff changeset	584 callback=self.psSchemaGetCb,
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	585 errback=partial(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	586 self.errback,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	587 msg=_(u"can't edit schema: {}"),
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	588 exit_code=C.EXIT_BRIDGE_ERRBACK,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	589 ),
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	590 )
2351 3c0a3fae1862 jp (pubsub/node): added schema (set/edit/get) commands to manipulate PubSub node schema Goffi <goffi@goffi.org> parents: 2339 diff changeset	591
3c0a3fae1862 jp (pubsub/node): added schema (set/edit/get) commands to manipulate PubSub node schema Goffi <goffi@goffi.org> parents: 2339 diff changeset	592
3c0a3fae1862 jp (pubsub/node): added schema (set/edit/get) commands to manipulate PubSub node schema Goffi <goffi@goffi.org> parents: 2339 diff changeset	593 class NodeSchemaGet(base.CommandBase):
3c0a3fae1862 jp (pubsub/node): added schema (set/edit/get) commands to manipulate PubSub node schema Goffi <goffi@goffi.org> parents: 2339 diff changeset	594 def __init__(self, host):
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	595 base.CommandBase.__init__(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	596 self,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	597 host,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	598 "get",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	599 use_output=C.OUTPUT_XML,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	600 use_pubsub=True,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	601 pubsub_flags={C.NODE},
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	602 use_verbose=True,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	603 help=_(u"get schema"),
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	604 )
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	605 self.need_loop = True
2351 3c0a3fae1862 jp (pubsub/node): added schema (set/edit/get) commands to manipulate PubSub node schema Goffi <goffi@goffi.org> parents: 2339 diff changeset	606
3c0a3fae1862 jp (pubsub/node): added schema (set/edit/get) commands to manipulate PubSub node schema Goffi <goffi@goffi.org> parents: 2339 diff changeset	607 def add_parser_options(self):
3c0a3fae1862 jp (pubsub/node): added schema (set/edit/get) commands to manipulate PubSub node schema Goffi <goffi@goffi.org> parents: 2339 diff changeset	608 pass
3c0a3fae1862 jp (pubsub/node): added schema (set/edit/get) commands to manipulate PubSub node schema Goffi <goffi@goffi.org> parents: 2339 diff changeset	609
3c0a3fae1862 jp (pubsub/node): added schema (set/edit/get) commands to manipulate PubSub node schema Goffi <goffi@goffi.org> parents: 2339 diff changeset	610 def psSchemaGetCb(self, schema):
3c0a3fae1862 jp (pubsub/node): added schema (set/edit/get) commands to manipulate PubSub node schema Goffi <goffi@goffi.org> parents: 2339 diff changeset	611 if not schema:
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	612 self.disp(_(u"no schema found"), 1)
2351 3c0a3fae1862 jp (pubsub/node): added schema (set/edit/get) commands to manipulate PubSub node schema Goffi <goffi@goffi.org> parents: 2339 diff changeset	613 self.host.quit(1)
3c0a3fae1862 jp (pubsub/node): added schema (set/edit/get) commands to manipulate PubSub node schema Goffi <goffi@goffi.org> parents: 2339 diff changeset	614 self.output(schema)
3c0a3fae1862 jp (pubsub/node): added schema (set/edit/get) commands to manipulate PubSub node schema Goffi <goffi@goffi.org> parents: 2339 diff changeset	615 self.host.quit()
3c0a3fae1862 jp (pubsub/node): added schema (set/edit/get) commands to manipulate PubSub node schema Goffi <goffi@goffi.org> parents: 2339 diff changeset	616
3c0a3fae1862 jp (pubsub/node): added schema (set/edit/get) commands to manipulate PubSub node schema Goffi <goffi@goffi.org> parents: 2339 diff changeset	617 def start(self):
3c0a3fae1862 jp (pubsub/node): added schema (set/edit/get) commands to manipulate PubSub node schema Goffi <goffi@goffi.org> parents: 2339 diff changeset	618 self.host.bridge.psSchemaGet(
3c0a3fae1862 jp (pubsub/node): added schema (set/edit/get) commands to manipulate PubSub node schema Goffi <goffi@goffi.org> parents: 2339 diff changeset	619 self.args.service,
3c0a3fae1862 jp (pubsub/node): added schema (set/edit/get) commands to manipulate PubSub node schema Goffi <goffi@goffi.org> parents: 2339 diff changeset	620 self.args.node,
3c0a3fae1862 jp (pubsub/node): added schema (set/edit/get) commands to manipulate PubSub node schema Goffi <goffi@goffi.org> parents: 2339 diff changeset	621 self.profile,
3c0a3fae1862 jp (pubsub/node): added schema (set/edit/get) commands to manipulate PubSub node schema Goffi <goffi@goffi.org> parents: 2339 diff changeset	622 callback=self.psSchemaGetCb,
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	623 errback=partial(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	624 self.errback,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	625 msg=_(u"can't get schema: {}"),
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	626 exit_code=C.EXIT_BRIDGE_ERRBACK,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	627 ),
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	628 )
2351 3c0a3fae1862 jp (pubsub/node): added schema (set/edit/get) commands to manipulate PubSub node schema Goffi <goffi@goffi.org> parents: 2339 diff changeset	629
3c0a3fae1862 jp (pubsub/node): added schema (set/edit/get) commands to manipulate PubSub node schema Goffi <goffi@goffi.org> parents: 2339 diff changeset	630
3c0a3fae1862 jp (pubsub/node): added schema (set/edit/get) commands to manipulate PubSub node schema Goffi <goffi@goffi.org> parents: 2339 diff changeset	631 class NodeSchema(base.CommandBase):
3c0a3fae1862 jp (pubsub/node): added schema (set/edit/get) commands to manipulate PubSub node schema Goffi <goffi@goffi.org> parents: 2339 diff changeset	632 subcommands = (NodeSchemaSet, NodeSchemaEdit, NodeSchemaGet)
3c0a3fae1862 jp (pubsub/node): added schema (set/edit/get) commands to manipulate PubSub node schema Goffi <goffi@goffi.org> parents: 2339 diff changeset	633
3c0a3fae1862 jp (pubsub/node): added schema (set/edit/get) commands to manipulate PubSub node schema Goffi <goffi@goffi.org> parents: 2339 diff changeset	634 def __init__(self, host):
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	635 super(NodeSchema, self).__init__(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	636 host, "schema", use_profile=False, help=_(u"data schema manipulation")
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	637 )
2351 3c0a3fae1862 jp (pubsub/node): added schema (set/edit/get) commands to manipulate PubSub node schema Goffi <goffi@goffi.org> parents: 2339 diff changeset	638
3c0a3fae1862 jp (pubsub/node): added schema (set/edit/get) commands to manipulate PubSub node schema Goffi <goffi@goffi.org> parents: 2339 diff changeset	639
2197 e0e06391ce91 jp (pubsub): added pubsub/node/info command to retrieve node configuration Goffi <goffi@goffi.org> parents: 2195 diff changeset	640 class Node(base.CommandBase):
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	641 subcommands = (
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	642 NodeInfo,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	643 NodeCreate,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	644 NodeDelete,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	645 NodeSet,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	646 NodeAffiliations,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	647 NodeSubscriptions,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	648 NodeSchema,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	649 )
2197 e0e06391ce91 jp (pubsub): added pubsub/node/info command to retrieve node configuration Goffi <goffi@goffi.org> parents: 2195 diff changeset	650
e0e06391ce91 jp (pubsub): added pubsub/node/info command to retrieve node configuration Goffi <goffi@goffi.org> parents: 2195 diff changeset	651 def __init__(self, host):
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	652 super(Node, self).__init__(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	653 host, "node", use_profile=False, help=_("node handling")
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	654 )
2197 e0e06391ce91 jp (pubsub): added pubsub/node/info command to retrieve node configuration Goffi <goffi@goffi.org> parents: 2195 diff changeset	655
e0e06391ce91 jp (pubsub): added pubsub/node/info command to retrieve node configuration Goffi <goffi@goffi.org> parents: 2195 diff changeset	656
2439 6908fe4c6eca jp (pubsub): added new "set" command tu publish or update an item in a non-interactive way (while "edit" do it interactively). Goffi <goffi@goffi.org> parents: 2416 diff changeset	657 class Set(base.CommandBase):
6908fe4c6eca jp (pubsub): added new "set" command tu publish or update an item in a non-interactive way (while "edit" do it interactively). Goffi <goffi@goffi.org> parents: 2416 diff changeset	658 def __init__(self, host):
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	659 base.CommandBase.__init__(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	660 self,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	661 host,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	662 "set",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	663 use_pubsub=True,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	664 pubsub_flags={C.NODE},
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	665 help=_(u"publish a new item or update an existing one"),
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	666 )
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	667 self.need_loop = True
2439 6908fe4c6eca jp (pubsub): added new "set" command tu publish or update an item in a non-interactive way (while "edit" do it interactively). Goffi <goffi@goffi.org> parents: 2416 diff changeset	668
6908fe4c6eca jp (pubsub): added new "set" command tu publish or update an item in a non-interactive way (while "edit" do it interactively). Goffi <goffi@goffi.org> parents: 2416 diff changeset	669 def add_parser_options(self):
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	670 self.parser.add_argument(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	671 "item",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	672 type=base.unicode_decoder,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	673 nargs="?",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	674 default=u"",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	675 help=_(u"id, URL of the item to update, keyword, or nothing for new item"),
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	676 )
2439 6908fe4c6eca jp (pubsub): added new "set" command tu publish or update an item in a non-interactive way (while "edit" do it interactively). Goffi <goffi@goffi.org> parents: 2416 diff changeset	677
6908fe4c6eca jp (pubsub): added new "set" command tu publish or update an item in a non-interactive way (while "edit" do it interactively). Goffi <goffi@goffi.org> parents: 2416 diff changeset	678 def psItemsSendCb(self, published_id):
6908fe4c6eca jp (pubsub): added new "set" command tu publish or update an item in a non-interactive way (while "edit" do it interactively). Goffi <goffi@goffi.org> parents: 2416 diff changeset	679 if published_id:
6908fe4c6eca jp (pubsub): added new "set" command tu publish or update an item in a non-interactive way (while "edit" do it interactively). Goffi <goffi@goffi.org> parents: 2416 diff changeset	680 self.disp(u"Item published at {pub_id}".format(pub_id=published_id))
6908fe4c6eca jp (pubsub): added new "set" command tu publish or update an item in a non-interactive way (while "edit" do it interactively). Goffi <goffi@goffi.org> parents: 2416 diff changeset	681 else:
6908fe4c6eca jp (pubsub): added new "set" command tu publish or update an item in a non-interactive way (while "edit" do it interactively). Goffi <goffi@goffi.org> parents: 2416 diff changeset	682 self.disp(u"Item published")
6908fe4c6eca jp (pubsub): added new "set" command tu publish or update an item in a non-interactive way (while "edit" do it interactively). Goffi <goffi@goffi.org> parents: 2416 diff changeset	683 self.host.quit(C.EXIT_OK)
6908fe4c6eca jp (pubsub): added new "set" command tu publish or update an item in a non-interactive way (while "edit" do it interactively). Goffi <goffi@goffi.org> parents: 2416 diff changeset	684
6908fe4c6eca jp (pubsub): added new "set" command tu publish or update an item in a non-interactive way (while "edit" do it interactively). Goffi <goffi@goffi.org> parents: 2416 diff changeset	685 def start(self):
6908fe4c6eca jp (pubsub): added new "set" command tu publish or update an item in a non-interactive way (while "edit" do it interactively). Goffi <goffi@goffi.org> parents: 2416 diff changeset	686 try:
6908fe4c6eca jp (pubsub): added new "set" command tu publish or update an item in a non-interactive way (while "edit" do it interactively). Goffi <goffi@goffi.org> parents: 2416 diff changeset	687 from lxml import etree
6908fe4c6eca jp (pubsub): added new "set" command tu publish or update an item in a non-interactive way (while "edit" do it interactively). Goffi <goffi@goffi.org> parents: 2416 diff changeset	688 except ImportError:
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	689 self.disp(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	690 u'lxml module must be installed to use edit, please install it with "pip install lxml"',
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	691 error=True,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	692 )
2439 6908fe4c6eca jp (pubsub): added new "set" command tu publish or update an item in a non-interactive way (while "edit" do it interactively). Goffi <goffi@goffi.org> parents: 2416 diff changeset	693 self.host.quit(1)
6908fe4c6eca jp (pubsub): added new "set" command tu publish or update an item in a non-interactive way (while "edit" do it interactively). Goffi <goffi@goffi.org> parents: 2416 diff changeset	694 try:
6908fe4c6eca jp (pubsub): added new "set" command tu publish or update an item in a non-interactive way (while "edit" do it interactively). Goffi <goffi@goffi.org> parents: 2416 diff changeset	695 element = etree.parse(sys.stdin).getroot()
6908fe4c6eca jp (pubsub): added new "set" command tu publish or update an item in a non-interactive way (while "edit" do it interactively). Goffi <goffi@goffi.org> parents: 2416 diff changeset	696 except Exception as e:
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	697 self.parser.error(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	698 _(u"Can't parse the payload XML in input: {msg}").format(msg=e)
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	699 )
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	700 if element.tag in ("item", "{http://jabber.org/protocol/pubsub}item"):
2439 6908fe4c6eca jp (pubsub): added new "set" command tu publish or update an item in a non-interactive way (while "edit" do it interactively). Goffi <goffi@goffi.org> parents: 2416 diff changeset	701 if len(element) > 1:
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	702 self.parser.error(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	703 _(u"<item> can only have one child element (the payload)")
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	704 )
2439 6908fe4c6eca jp (pubsub): added new "set" command tu publish or update an item in a non-interactive way (while "edit" do it interactively). Goffi <goffi@goffi.org> parents: 2416 diff changeset	705 element = element[0]
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	706 payload = etree.tostring(element, encoding="unicode")
2439 6908fe4c6eca jp (pubsub): added new "set" command tu publish or update an item in a non-interactive way (while "edit" do it interactively). Goffi <goffi@goffi.org> parents: 2416 diff changeset	707
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	708 self.host.bridge.psItemSend(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	709 self.args.service,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	710 self.args.node,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	711 payload,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	712 self.args.item,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	713 {},
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	714 self.profile,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	715 callback=self.psItemsSendCb,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	716 errback=partial(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	717 self.errback,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	718 msg=_(u"can't send item: {}"),
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	719 exit_code=C.EXIT_BRIDGE_ERRBACK,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	720 ),
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	721 )
2439 6908fe4c6eca jp (pubsub): added new "set" command tu publish or update an item in a non-interactive way (while "edit" do it interactively). Goffi <goffi@goffi.org> parents: 2416 diff changeset	722
6908fe4c6eca jp (pubsub): added new "set" command tu publish or update an item in a non-interactive way (while "edit" do it interactively). Goffi <goffi@goffi.org> parents: 2416 diff changeset	723
2235 4db836386641 jp: added use_pubsub and use_pubsub_node_req arguments to CommandBase Goffi <goffi@goffi.org> parents: 2224 diff changeset	724 class Get(base.CommandBase):
2191 a1a8233f89e8 jp(pubsub/get): pubsub/get command, first draft Goffi <goffi@goffi.org> parents: diff changeset	725 def __init__(self, host):
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	726 base.CommandBase.__init__(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	727 self,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	728 host,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	729 "get",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	730 use_output=C.OUTPUT_LIST_XML,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	731 use_pubsub=True,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	732 pubsub_flags={C.NODE, C.MULTI_ITEMS},
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	733 help=_(u"get pubsub item(s)"),
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	734 )
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	735 self.need_loop = True
2191 a1a8233f89e8 jp(pubsub/get): pubsub/get command, first draft Goffi <goffi@goffi.org> parents: diff changeset	736
a1a8233f89e8 jp(pubsub/get): pubsub/get command, first draft Goffi <goffi@goffi.org> parents: diff changeset	737 def add_parser_options(self):
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	738 self.parser.add_argument(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	739 "-S",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	740 "--sub-id",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	741 type=base.unicode_decoder,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	742 default=u"",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	743 help=_(u"subscription id"),
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	744 )
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	745 # TODO: a key(s) argument to select keys to display
2191 a1a8233f89e8 jp(pubsub/get): pubsub/get command, first draft Goffi <goffi@goffi.org> parents: diff changeset	746 # TODO: add MAM filters
a1a8233f89e8 jp(pubsub/get): pubsub/get command, first draft Goffi <goffi@goffi.org> parents: diff changeset	747
2274 27f469d40a83 jp (pubsub): psItemGet has been renamed to psItemsGet as several items are gotten Goffi <goffi@goffi.org> parents: 2239 diff changeset	748 def psItemsGetCb(self, ps_result):
2191 a1a8233f89e8 jp(pubsub/get): pubsub/get command, first draft Goffi <goffi@goffi.org> parents: diff changeset	749 self.output(ps_result[0])
a1a8233f89e8 jp(pubsub/get): pubsub/get command, first draft Goffi <goffi@goffi.org> parents: diff changeset	750 self.host.quit(C.EXIT_OK)
a1a8233f89e8 jp(pubsub/get): pubsub/get command, first draft Goffi <goffi@goffi.org> parents: diff changeset	751
2274 27f469d40a83 jp (pubsub): psItemGet has been renamed to psItemsGet as several items are gotten Goffi <goffi@goffi.org> parents: 2239 diff changeset	752 def psItemsGetEb(self, failure_):
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	753 self.disp(u"can't get pubsub items: {reason}".format(reason=failure_), error=True)
2191 a1a8233f89e8 jp(pubsub/get): pubsub/get command, first draft Goffi <goffi@goffi.org> parents: diff changeset	754 self.host.quit(C.EXIT_BRIDGE_ERRBACK)
a1a8233f89e8 jp(pubsub/get): pubsub/get command, first draft Goffi <goffi@goffi.org> parents: diff changeset	755
a1a8233f89e8 jp(pubsub/get): pubsub/get command, first draft Goffi <goffi@goffi.org> parents: diff changeset	756 def start(self):
2274 27f469d40a83 jp (pubsub): psItemGet has been renamed to psItemsGet as several items are gotten Goffi <goffi@goffi.org> parents: 2239 diff changeset	757 self.host.bridge.psItemsGet(
2191 a1a8233f89e8 jp(pubsub/get): pubsub/get command, first draft Goffi <goffi@goffi.org> parents: diff changeset	758 self.args.service,
a1a8233f89e8 jp(pubsub/get): pubsub/get command, first draft Goffi <goffi@goffi.org> parents: diff changeset	759 self.args.node,
a1a8233f89e8 jp(pubsub/get): pubsub/get command, first draft Goffi <goffi@goffi.org> parents: diff changeset	760 self.args.max,
a1a8233f89e8 jp(pubsub/get): pubsub/get command, first draft Goffi <goffi@goffi.org> parents: diff changeset	761 self.args.items,
a1a8233f89e8 jp(pubsub/get): pubsub/get command, first draft Goffi <goffi@goffi.org> parents: diff changeset	762 self.args.sub_id,
2761 4b693ea24d5f jp (base, pubsub, ticket): handle order-by: Goffi <goffi@goffi.org> parents: 2624 diff changeset	763 self.getPubsubExtra(),
2191 a1a8233f89e8 jp(pubsub/get): pubsub/get command, first draft Goffi <goffi@goffi.org> parents: diff changeset	764 self.profile,
2274 27f469d40a83 jp (pubsub): psItemGet has been renamed to psItemsGet as several items are gotten Goffi <goffi@goffi.org> parents: 2239 diff changeset	765 callback=self.psItemsGetCb,
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	766 errback=self.psItemsGetEb,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	767 )
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	768
2191 a1a8233f89e8 jp(pubsub/get): pubsub/get command, first draft Goffi <goffi@goffi.org> parents: diff changeset	769
2281 4af1805cc6df jp (pubsub/delete): delete command implementation (to delete an item) Goffi <goffi@goffi.org> parents: 2280 diff changeset	770 class Delete(base.CommandBase):
4af1805cc6df jp (pubsub/delete): delete command implementation (to delete an item) Goffi <goffi@goffi.org> parents: 2280 diff changeset	771 def __init__(self, host):
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	772 base.CommandBase.__init__(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	773 self,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	774 host,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	775 "delete",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	776 use_pubsub=True,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	777 pubsub_flags={C.NODE, C.SINGLE_ITEM},
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	778 help=_(u"delete an item"),
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	779 )
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	780 self.need_loop = True
2281 4af1805cc6df jp (pubsub/delete): delete command implementation (to delete an item) Goffi <goffi@goffi.org> parents: 2280 diff changeset	781
4af1805cc6df jp (pubsub/delete): delete command implementation (to delete an item) Goffi <goffi@goffi.org> parents: 2280 diff changeset	782 def add_parser_options(self):
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	783 self.parser.add_argument(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	784 "-f", "--force", action="store_true", help=_(u"delete without confirmation")
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	785 )
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	786 self.parser.add_argument(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	787 "-N", "--notify", action="store_true", help=_(u"notify deletion")
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	788 )
2281 4af1805cc6df jp (pubsub/delete): delete command implementation (to delete an item) Goffi <goffi@goffi.org> parents: 2280 diff changeset	789
4af1805cc6df jp (pubsub/delete): delete command implementation (to delete an item) Goffi <goffi@goffi.org> parents: 2280 diff changeset	790 def psItemsDeleteCb(self):
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	791 self.disp(_(u"item {item_id} has been deleted").format(item_id=self.args.item))
2281 4af1805cc6df jp (pubsub/delete): delete command implementation (to delete an item) Goffi <goffi@goffi.org> parents: 2280 diff changeset	792 self.host.quit(C.EXIT_OK)
4af1805cc6df jp (pubsub/delete): delete command implementation (to delete an item) Goffi <goffi@goffi.org> parents: 2280 diff changeset	793
4af1805cc6df jp (pubsub/delete): delete command implementation (to delete an item) Goffi <goffi@goffi.org> parents: 2280 diff changeset	794 def start(self):
4af1805cc6df jp (pubsub/delete): delete command implementation (to delete an item) Goffi <goffi@goffi.org> parents: 2280 diff changeset	795 if not self.args.item:
4af1805cc6df jp (pubsub/delete): delete command implementation (to delete an item) Goffi <goffi@goffi.org> parents: 2280 diff changeset	796 self.parser.error(_(u"You need to specify an item to delete"))
4af1805cc6df jp (pubsub/delete): delete command implementation (to delete an item) Goffi <goffi@goffi.org> parents: 2280 diff changeset	797 if not self.args.force:
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	798 message = _(u"Are you sure to delete item {item_id} ?").format(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	799 item_id=self.args.item
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	800 )
2550 1d754bc14381 jp (base): new confirmOrQuit helper method to ask confirmation to user, and quit if he cancel Goffi <goffi@goffi.org> parents: 2539 diff changeset	801 self.host.confirmOrQuit(message, _(u"item deletion cancelled"))
2281 4af1805cc6df jp (pubsub/delete): delete command implementation (to delete an item) Goffi <goffi@goffi.org> parents: 2280 diff changeset	802 self.host.bridge.psRetractItem(
4af1805cc6df jp (pubsub/delete): delete command implementation (to delete an item) Goffi <goffi@goffi.org> parents: 2280 diff changeset	803 self.args.service,
4af1805cc6df jp (pubsub/delete): delete command implementation (to delete an item) Goffi <goffi@goffi.org> parents: 2280 diff changeset	804 self.args.node,
4af1805cc6df jp (pubsub/delete): delete command implementation (to delete an item) Goffi <goffi@goffi.org> parents: 2280 diff changeset	805 self.args.item,
4af1805cc6df jp (pubsub/delete): delete command implementation (to delete an item) Goffi <goffi@goffi.org> parents: 2280 diff changeset	806 self.args.notify,
4af1805cc6df jp (pubsub/delete): delete command implementation (to delete an item) Goffi <goffi@goffi.org> parents: 2280 diff changeset	807 self.profile,
4af1805cc6df jp (pubsub/delete): delete command implementation (to delete an item) Goffi <goffi@goffi.org> parents: 2280 diff changeset	808 callback=self.psItemsDeleteCb,
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	809 errback=partial(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	810 self.errback,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	811 msg=_(u"can't delete item: {}"),
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	812 exit_code=C.EXIT_BRIDGE_ERRBACK,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	813 ),
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	814 )
2281 4af1805cc6df jp (pubsub/delete): delete command implementation (to delete an item) Goffi <goffi@goffi.org> parents: 2280 diff changeset	815
2191 a1a8233f89e8 jp(pubsub/get): pubsub/get command, first draft Goffi <goffi@goffi.org> parents: diff changeset	816
2275 64e99bf0dfa2 jp (pubsub/edit): new edit subcommand, which work in a same way as for blog Goffi <goffi@goffi.org> parents: 2274 diff changeset	817 class Edit(base.CommandBase, common.BaseEdit):
64e99bf0dfa2 jp (pubsub/edit): new edit subcommand, which work in a same way as for blog Goffi <goffi@goffi.org> parents: 2274 diff changeset	818 def __init__(self, host):
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	819 base.CommandBase.__init__(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	820 self,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	821 host,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	822 "edit",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	823 use_verbose=True,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	824 use_pubsub=True,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	825 pubsub_flags={C.NODE, C.SINGLE_ITEM},
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	826 use_draft=True,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	827 help=_(u"edit an existing or new pubsub item"),
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	828 )
2275 64e99bf0dfa2 jp (pubsub/edit): new edit subcommand, which work in a same way as for blog Goffi <goffi@goffi.org> parents: 2274 diff changeset	829 common.BaseEdit.__init__(self, self.host, PUBSUB_TMP_DIR)
64e99bf0dfa2 jp (pubsub/edit): new edit subcommand, which work in a same way as for blog Goffi <goffi@goffi.org> parents: 2274 diff changeset	830
64e99bf0dfa2 jp (pubsub/edit): new edit subcommand, which work in a same way as for blog Goffi <goffi@goffi.org> parents: 2274 diff changeset	831 def add_parser_options(self):
2532 772447ec070f jp: pubsub options refactoring: Goffi <goffi@goffi.org> parents: 2483 diff changeset	832 pass
2275 64e99bf0dfa2 jp (pubsub/edit): new edit subcommand, which work in a same way as for blog Goffi <goffi@goffi.org> parents: 2274 diff changeset	833
64e99bf0dfa2 jp (pubsub/edit): new edit subcommand, which work in a same way as for blog Goffi <goffi@goffi.org> parents: 2274 diff changeset	834 def edit(self, content_file_path, content_file_obj):
64e99bf0dfa2 jp (pubsub/edit): new edit subcommand, which work in a same way as for blog Goffi <goffi@goffi.org> parents: 2274 diff changeset	835 # we launch editor
64e99bf0dfa2 jp (pubsub/edit): new edit subcommand, which work in a same way as for blog Goffi <goffi@goffi.org> parents: 2274 diff changeset	836 self.runEditor("pubsub_editor_args", content_file_path, content_file_obj)
64e99bf0dfa2 jp (pubsub/edit): new edit subcommand, which work in a same way as for blog Goffi <goffi@goffi.org> parents: 2274 diff changeset	837
64e99bf0dfa2 jp (pubsub/edit): new edit subcommand, which work in a same way as for blog Goffi <goffi@goffi.org> parents: 2274 diff changeset	838 def publish(self, content):
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	839 published_id = self.host.bridge.psItemSend(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	840 self.pubsub_service,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	841 self.pubsub_node,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	842 content,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	843 self.pubsub_item or "",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	844 {},
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	845 self.profile,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	846 )
2275 64e99bf0dfa2 jp (pubsub/edit): new edit subcommand, which work in a same way as for blog Goffi <goffi@goffi.org> parents: 2274 diff changeset	847 if published_id:
64e99bf0dfa2 jp (pubsub/edit): new edit subcommand, which work in a same way as for blog Goffi <goffi@goffi.org> parents: 2274 diff changeset	848 self.disp(u"Item published at {pub_id}".format(pub_id=published_id))
64e99bf0dfa2 jp (pubsub/edit): new edit subcommand, which work in a same way as for blog Goffi <goffi@goffi.org> parents: 2274 diff changeset	849 else:
64e99bf0dfa2 jp (pubsub/edit): new edit subcommand, which work in a same way as for blog Goffi <goffi@goffi.org> parents: 2274 diff changeset	850 self.disp(u"Item published")
64e99bf0dfa2 jp (pubsub/edit): new edit subcommand, which work in a same way as for blog Goffi <goffi@goffi.org> parents: 2274 diff changeset	851
64e99bf0dfa2 jp (pubsub/edit): new edit subcommand, which work in a same way as for blog Goffi <goffi@goffi.org> parents: 2274 diff changeset	852 def getItemData(self, service, node, item):
64e99bf0dfa2 jp (pubsub/edit): new edit subcommand, which work in a same way as for blog Goffi <goffi@goffi.org> parents: 2274 diff changeset	853 try:
64e99bf0dfa2 jp (pubsub/edit): new edit subcommand, which work in a same way as for blog Goffi <goffi@goffi.org> parents: 2274 diff changeset	854 from lxml import etree
64e99bf0dfa2 jp (pubsub/edit): new edit subcommand, which work in a same way as for blog Goffi <goffi@goffi.org> parents: 2274 diff changeset	855 except ImportError:
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	856 self.disp(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	857 u'lxml module must be installed to use edit, please install it with "pip install lxml"',
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	858 error=True,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	859 )
2275 64e99bf0dfa2 jp (pubsub/edit): new edit subcommand, which work in a same way as for blog Goffi <goffi@goffi.org> parents: 2274 diff changeset	860 self.host.quit(1)
64e99bf0dfa2 jp (pubsub/edit): new edit subcommand, which work in a same way as for blog Goffi <goffi@goffi.org> parents: 2274 diff changeset	861 items = [item] if item is not None else []
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	862 item_raw = self.host.bridge.psItemsGet(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	863 service, node, 1, items, "", {}, self.profile
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	864 )[0][0]
2275 64e99bf0dfa2 jp (pubsub/edit): new edit subcommand, which work in a same way as for blog Goffi <goffi@goffi.org> parents: 2274 diff changeset	865 parser = etree.XMLParser(remove_blank_text=True)
64e99bf0dfa2 jp (pubsub/edit): new edit subcommand, which work in a same way as for blog Goffi <goffi@goffi.org> parents: 2274 diff changeset	866 item_elt = etree.fromstring(item_raw, parser)
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	867 item_id = item_elt.get("id")
2275 64e99bf0dfa2 jp (pubsub/edit): new edit subcommand, which work in a same way as for blog Goffi <goffi@goffi.org> parents: 2274 diff changeset	868 try:
64e99bf0dfa2 jp (pubsub/edit): new edit subcommand, which work in a same way as for blog Goffi <goffi@goffi.org> parents: 2274 diff changeset	869 payload = item_elt[0]
64e99bf0dfa2 jp (pubsub/edit): new edit subcommand, which work in a same way as for blog Goffi <goffi@goffi.org> parents: 2274 diff changeset	870 except IndexError:
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	871 self.disp(_(u"Item has not payload"), 1)
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	872 return u""
2280 4bc9a2c2d6c9 jp (pubsub, common): fixed last item edition (keep item id instead of creating a new one) Goffi <goffi@goffi.org> parents: 2279 diff changeset	873 return etree.tostring(payload, encoding="unicode", pretty_print=True), item_id
2275 64e99bf0dfa2 jp (pubsub/edit): new edit subcommand, which work in a same way as for blog Goffi <goffi@goffi.org> parents: 2274 diff changeset	874
64e99bf0dfa2 jp (pubsub/edit): new edit subcommand, which work in a same way as for blog Goffi <goffi@goffi.org> parents: 2274 diff changeset	875 def start(self):
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	876 self.pubsub_service, self.pubsub_node, self.pubsub_item, content_file_path, content_file_obj = (
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	877 self.getItemPath()
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	878 )
2275 64e99bf0dfa2 jp (pubsub/edit): new edit subcommand, which work in a same way as for blog Goffi <goffi@goffi.org> parents: 2274 diff changeset	879 self.edit(content_file_path, content_file_obj)
64e99bf0dfa2 jp (pubsub/edit): new edit subcommand, which work in a same way as for blog Goffi <goffi@goffi.org> parents: 2274 diff changeset	880
64e99bf0dfa2 jp (pubsub/edit): new edit subcommand, which work in a same way as for blog Goffi <goffi@goffi.org> parents: 2274 diff changeset	881
2353 ebc0dfe9c0ca jp (pubsub): added subscribe, unsubscribe and subscriptions methods: Goffi <goffi@goffi.org> parents: 2351 diff changeset	882 class Subscribe(base.CommandBase):
ebc0dfe9c0ca jp (pubsub): added subscribe, unsubscribe and subscriptions methods: Goffi <goffi@goffi.org> parents: 2351 diff changeset	883 def __init__(self, host):
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	884 base.CommandBase.__init__(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	885 self,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	886 host,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	887 "subscribe",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	888 use_pubsub=True,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	889 pubsub_flags={C.NODE},
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	890 use_verbose=True,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	891 help=_(u"subscribe to a node"),
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	892 )
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	893 self.need_loop = True
2353 ebc0dfe9c0ca jp (pubsub): added subscribe, unsubscribe and subscriptions methods: Goffi <goffi@goffi.org> parents: 2351 diff changeset	894
ebc0dfe9c0ca jp (pubsub): added subscribe, unsubscribe and subscriptions methods: Goffi <goffi@goffi.org> parents: 2351 diff changeset	895 def add_parser_options(self):
ebc0dfe9c0ca jp (pubsub): added subscribe, unsubscribe and subscriptions methods: Goffi <goffi@goffi.org> parents: 2351 diff changeset	896 pass
ebc0dfe9c0ca jp (pubsub): added subscribe, unsubscribe and subscriptions methods: Goffi <goffi@goffi.org> parents: 2351 diff changeset	897
2442 b8ffb7f8056b plugin XEP-0060: added "options" argument to psSubscribe, and return subscription id if present Goffi <goffi@goffi.org> parents: 2439 diff changeset	898 def psSubscribeCb(self, sub_id):
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	899 self.disp(_(u"subscription done"), 1)
2442 b8ffb7f8056b plugin XEP-0060: added "options" argument to psSubscribe, and return subscription id if present Goffi <goffi@goffi.org> parents: 2439 diff changeset	900 if sub_id:
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	901 self.disp(_(u"subscription id: {sub_id}").format(sub_id=sub_id))
2353 ebc0dfe9c0ca jp (pubsub): added subscribe, unsubscribe and subscriptions methods: Goffi <goffi@goffi.org> parents: 2351 diff changeset	902 self.host.quit()
ebc0dfe9c0ca jp (pubsub): added subscribe, unsubscribe and subscriptions methods: Goffi <goffi@goffi.org> parents: 2351 diff changeset	903
ebc0dfe9c0ca jp (pubsub): added subscribe, unsubscribe and subscriptions methods: Goffi <goffi@goffi.org> parents: 2351 diff changeset	904 def start(self):
ebc0dfe9c0ca jp (pubsub): added subscribe, unsubscribe and subscriptions methods: Goffi <goffi@goffi.org> parents: 2351 diff changeset	905 self.host.bridge.psSubscribe(
ebc0dfe9c0ca jp (pubsub): added subscribe, unsubscribe and subscriptions methods: Goffi <goffi@goffi.org> parents: 2351 diff changeset	906 self.args.service,
ebc0dfe9c0ca jp (pubsub): added subscribe, unsubscribe and subscriptions methods: Goffi <goffi@goffi.org> parents: 2351 diff changeset	907 self.args.node,
2442 b8ffb7f8056b plugin XEP-0060: added "options" argument to psSubscribe, and return subscription id if present Goffi <goffi@goffi.org> parents: 2439 diff changeset	908 {},
2353 ebc0dfe9c0ca jp (pubsub): added subscribe, unsubscribe and subscriptions methods: Goffi <goffi@goffi.org> parents: 2351 diff changeset	909 self.profile,
ebc0dfe9c0ca jp (pubsub): added subscribe, unsubscribe and subscriptions methods: Goffi <goffi@goffi.org> parents: 2351 diff changeset	910 callback=self.psSubscribeCb,
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	911 errback=partial(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	912 self.errback,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	913 msg=_(u"can't subscribe to node: {}"),
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	914 exit_code=C.EXIT_BRIDGE_ERRBACK,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	915 ),
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	916 )
2353 ebc0dfe9c0ca jp (pubsub): added subscribe, unsubscribe and subscriptions methods: Goffi <goffi@goffi.org> parents: 2351 diff changeset	917
ebc0dfe9c0ca jp (pubsub): added subscribe, unsubscribe and subscriptions methods: Goffi <goffi@goffi.org> parents: 2351 diff changeset	918
ebc0dfe9c0ca jp (pubsub): added subscribe, unsubscribe and subscriptions methods: Goffi <goffi@goffi.org> parents: 2351 diff changeset	919 class Unsubscribe(base.CommandBase):
ebc0dfe9c0ca jp (pubsub): added subscribe, unsubscribe and subscriptions methods: Goffi <goffi@goffi.org> parents: 2351 diff changeset	920 # TODO: voir pourquoi NodeNotFound sur subscribe juste après unsubscribe
ebc0dfe9c0ca jp (pubsub): added subscribe, unsubscribe and subscriptions methods: Goffi <goffi@goffi.org> parents: 2351 diff changeset	921
ebc0dfe9c0ca jp (pubsub): added subscribe, unsubscribe and subscriptions methods: Goffi <goffi@goffi.org> parents: 2351 diff changeset	922 def __init__(self, host):
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	923 base.CommandBase.__init__(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	924 self,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	925 host,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	926 "unsubscribe",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	927 use_pubsub=True,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	928 pubsub_flags={C.NODE},
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	929 use_verbose=True,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	930 help=_(u"unsubscribe from a node"),
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	931 )
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	932 self.need_loop = True
2353 ebc0dfe9c0ca jp (pubsub): added subscribe, unsubscribe and subscriptions methods: Goffi <goffi@goffi.org> parents: 2351 diff changeset	933
ebc0dfe9c0ca jp (pubsub): added subscribe, unsubscribe and subscriptions methods: Goffi <goffi@goffi.org> parents: 2351 diff changeset	934 def add_parser_options(self):
ebc0dfe9c0ca jp (pubsub): added subscribe, unsubscribe and subscriptions methods: Goffi <goffi@goffi.org> parents: 2351 diff changeset	935 pass
ebc0dfe9c0ca jp (pubsub): added subscribe, unsubscribe and subscriptions methods: Goffi <goffi@goffi.org> parents: 2351 diff changeset	936
ebc0dfe9c0ca jp (pubsub): added subscribe, unsubscribe and subscriptions methods: Goffi <goffi@goffi.org> parents: 2351 diff changeset	937 def psUnsubscribeCb(self):
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	938 self.disp(_(u"subscription removed"), 1)
2353 ebc0dfe9c0ca jp (pubsub): added subscribe, unsubscribe and subscriptions methods: Goffi <goffi@goffi.org> parents: 2351 diff changeset	939 self.host.quit()
ebc0dfe9c0ca jp (pubsub): added subscribe, unsubscribe and subscriptions methods: Goffi <goffi@goffi.org> parents: 2351 diff changeset	940
ebc0dfe9c0ca jp (pubsub): added subscribe, unsubscribe and subscriptions methods: Goffi <goffi@goffi.org> parents: 2351 diff changeset	941 def start(self):
ebc0dfe9c0ca jp (pubsub): added subscribe, unsubscribe and subscriptions methods: Goffi <goffi@goffi.org> parents: 2351 diff changeset	942 self.host.bridge.psUnsubscribe(
ebc0dfe9c0ca jp (pubsub): added subscribe, unsubscribe and subscriptions methods: Goffi <goffi@goffi.org> parents: 2351 diff changeset	943 self.args.service,
ebc0dfe9c0ca jp (pubsub): added subscribe, unsubscribe and subscriptions methods: Goffi <goffi@goffi.org> parents: 2351 diff changeset	944 self.args.node,
ebc0dfe9c0ca jp (pubsub): added subscribe, unsubscribe and subscriptions methods: Goffi <goffi@goffi.org> parents: 2351 diff changeset	945 self.profile,
ebc0dfe9c0ca jp (pubsub): added subscribe, unsubscribe and subscriptions methods: Goffi <goffi@goffi.org> parents: 2351 diff changeset	946 callback=self.psUnsubscribeCb,
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	947 errback=partial(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	948 self.errback,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	949 msg=_(u"can't unsubscribe from node: {}"),
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	950 exit_code=C.EXIT_BRIDGE_ERRBACK,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	951 ),
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	952 )
2353 ebc0dfe9c0ca jp (pubsub): added subscribe, unsubscribe and subscriptions methods: Goffi <goffi@goffi.org> parents: 2351 diff changeset	953
ebc0dfe9c0ca jp (pubsub): added subscribe, unsubscribe and subscriptions methods: Goffi <goffi@goffi.org> parents: 2351 diff changeset	954
ebc0dfe9c0ca jp (pubsub): added subscribe, unsubscribe and subscriptions methods: Goffi <goffi@goffi.org> parents: 2351 diff changeset	955 class Subscriptions(base.CommandBase):
ebc0dfe9c0ca jp (pubsub): added subscribe, unsubscribe and subscriptions methods: Goffi <goffi@goffi.org> parents: 2351 diff changeset	956 def __init__(self, host):
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	957 base.CommandBase.__init__(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	958 self,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	959 host,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	960 "subscriptions",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	961 use_output=C.OUTPUT_LIST_DICT,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	962 use_pubsub=True,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	963 help=_(u"retrieve all subscriptions on a service"),
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	964 )
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	965 self.need_loop = True
2353 ebc0dfe9c0ca jp (pubsub): added subscribe, unsubscribe and subscriptions methods: Goffi <goffi@goffi.org> parents: 2351 diff changeset	966
ebc0dfe9c0ca jp (pubsub): added subscribe, unsubscribe and subscriptions methods: Goffi <goffi@goffi.org> parents: 2351 diff changeset	967 def add_parser_options(self):
ebc0dfe9c0ca jp (pubsub): added subscribe, unsubscribe and subscriptions methods: Goffi <goffi@goffi.org> parents: 2351 diff changeset	968 pass
ebc0dfe9c0ca jp (pubsub): added subscribe, unsubscribe and subscriptions methods: Goffi <goffi@goffi.org> parents: 2351 diff changeset	969
ebc0dfe9c0ca jp (pubsub): added subscribe, unsubscribe and subscriptions methods: Goffi <goffi@goffi.org> parents: 2351 diff changeset	970 def psSubscriptionsGetCb(self, subscriptions):
ebc0dfe9c0ca jp (pubsub): added subscribe, unsubscribe and subscriptions methods: Goffi <goffi@goffi.org> parents: 2351 diff changeset	971 self.output(subscriptions)
ebc0dfe9c0ca jp (pubsub): added subscribe, unsubscribe and subscriptions methods: Goffi <goffi@goffi.org> parents: 2351 diff changeset	972 self.host.quit()
ebc0dfe9c0ca jp (pubsub): added subscribe, unsubscribe and subscriptions methods: Goffi <goffi@goffi.org> parents: 2351 diff changeset	973
ebc0dfe9c0ca jp (pubsub): added subscribe, unsubscribe and subscriptions methods: Goffi <goffi@goffi.org> parents: 2351 diff changeset	974 def start(self):
ebc0dfe9c0ca jp (pubsub): added subscribe, unsubscribe and subscriptions methods: Goffi <goffi@goffi.org> parents: 2351 diff changeset	975 self.host.bridge.psSubscriptionsGet(
ebc0dfe9c0ca jp (pubsub): added subscribe, unsubscribe and subscriptions methods: Goffi <goffi@goffi.org> parents: 2351 diff changeset	976 self.args.service,
ebc0dfe9c0ca jp (pubsub): added subscribe, unsubscribe and subscriptions methods: Goffi <goffi@goffi.org> parents: 2351 diff changeset	977 self.args.node,
ebc0dfe9c0ca jp (pubsub): added subscribe, unsubscribe and subscriptions methods: Goffi <goffi@goffi.org> parents: 2351 diff changeset	978 self.profile,
ebc0dfe9c0ca jp (pubsub): added subscribe, unsubscribe and subscriptions methods: Goffi <goffi@goffi.org> parents: 2351 diff changeset	979 callback=self.psSubscriptionsGetCb,
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	980 errback=partial(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	981 self.errback,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	982 msg=_(u"can't retrieve subscriptions: {}"),
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	983 exit_code=C.EXIT_BRIDGE_ERRBACK,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	984 ),
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	985 )
2353 ebc0dfe9c0ca jp (pubsub): added subscribe, unsubscribe and subscriptions methods: Goffi <goffi@goffi.org> parents: 2351 diff changeset	986
ebc0dfe9c0ca jp (pubsub): added subscribe, unsubscribe and subscriptions methods: Goffi <goffi@goffi.org> parents: 2351 diff changeset	987
2235 4db836386641 jp: added use_pubsub and use_pubsub_node_req arguments to CommandBase Goffi <goffi@goffi.org> parents: 2224 diff changeset	988 class Affiliations(base.CommandBase):
2204 afc703419186 jp (pubsub): added affiliations and node/affiliations/get: Goffi <goffi@goffi.org> parents: 2199 diff changeset	989 def __init__(self, host):
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	990 base.CommandBase.__init__(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	991 self,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	992 host,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	993 "affiliations",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	994 use_output=C.OUTPUT_DICT,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	995 use_pubsub=True,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	996 help=_(u"retrieve all affiliations on a service"),
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	997 )
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	998 self.need_loop = True
2204 afc703419186 jp (pubsub): added affiliations and node/affiliations/get: Goffi <goffi@goffi.org> parents: 2199 diff changeset	999
afc703419186 jp (pubsub): added affiliations and node/affiliations/get: Goffi <goffi@goffi.org> parents: 2199 diff changeset	1000 def add_parser_options(self):
2235 4db836386641 jp: added use_pubsub and use_pubsub_node_req arguments to CommandBase Goffi <goffi@goffi.org> parents: 2224 diff changeset	1001 pass
2204 afc703419186 jp (pubsub): added affiliations and node/affiliations/get: Goffi <goffi@goffi.org> parents: 2199 diff changeset	1002
afc703419186 jp (pubsub): added affiliations and node/affiliations/get: Goffi <goffi@goffi.org> parents: 2199 diff changeset	1003 def psAffiliationsGetCb(self, affiliations):
afc703419186 jp (pubsub): added affiliations and node/affiliations/get: Goffi <goffi@goffi.org> parents: 2199 diff changeset	1004 self.output(affiliations)
afc703419186 jp (pubsub): added affiliations and node/affiliations/get: Goffi <goffi@goffi.org> parents: 2199 diff changeset	1005 self.host.quit()
afc703419186 jp (pubsub): added affiliations and node/affiliations/get: Goffi <goffi@goffi.org> parents: 2199 diff changeset	1006
afc703419186 jp (pubsub): added affiliations and node/affiliations/get: Goffi <goffi@goffi.org> parents: 2199 diff changeset	1007 def psAffiliationsGetEb(self, failure_):
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1008 self.disp(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1009 u"can't get node affiliations: {reason}".format(reason=failure_), error=True
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1010 )
2204 afc703419186 jp (pubsub): added affiliations and node/affiliations/get: Goffi <goffi@goffi.org> parents: 2199 diff changeset	1011 self.host.quit(C.EXIT_BRIDGE_ERRBACK)
afc703419186 jp (pubsub): added affiliations and node/affiliations/get: Goffi <goffi@goffi.org> parents: 2199 diff changeset	1012
afc703419186 jp (pubsub): added affiliations and node/affiliations/get: Goffi <goffi@goffi.org> parents: 2199 diff changeset	1013 def start(self):
afc703419186 jp (pubsub): added affiliations and node/affiliations/get: Goffi <goffi@goffi.org> parents: 2199 diff changeset	1014 self.host.bridge.psAffiliationsGet(
afc703419186 jp (pubsub): added affiliations and node/affiliations/get: Goffi <goffi@goffi.org> parents: 2199 diff changeset	1015 self.args.service,
afc703419186 jp (pubsub): added affiliations and node/affiliations/get: Goffi <goffi@goffi.org> parents: 2199 diff changeset	1016 self.args.node,
afc703419186 jp (pubsub): added affiliations and node/affiliations/get: Goffi <goffi@goffi.org> parents: 2199 diff changeset	1017 self.profile,
afc703419186 jp (pubsub): added affiliations and node/affiliations/get: Goffi <goffi@goffi.org> parents: 2199 diff changeset	1018 callback=self.psAffiliationsGetCb,
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1019 errback=self.psAffiliationsGetEb,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1020 )
2204 afc703419186 jp (pubsub): added affiliations and node/affiliations/get: Goffi <goffi@goffi.org> parents: 2199 diff changeset	1021
afc703419186 jp (pubsub): added affiliations and node/affiliations/get: Goffi <goffi@goffi.org> parents: 2199 diff changeset	1022
2316 7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1023 class Search(base.CommandBase):
2764 92af49cde255 jp (base): MAM and RSM arguments can now be used for pubsub commands: Goffi <goffi@goffi.org> parents: 2761 diff changeset	1024 """this command do a search without using MAM
92af49cde255 jp (base): MAM and RSM arguments can now be used for pubsub commands: Goffi <goffi@goffi.org> parents: 2761 diff changeset	1025
92af49cde255 jp (base): MAM and RSM arguments can now be used for pubsub commands: Goffi <goffi@goffi.org> parents: 2761 diff changeset	1026 This commands checks every items it finds by itself,
92af49cde255 jp (base): MAM and RSM arguments can now be used for pubsub commands: Goffi <goffi@goffi.org> parents: 2761 diff changeset	1027 so it may be heavy in resources both for server and client
92af49cde255 jp (base): MAM and RSM arguments can now be used for pubsub commands: Goffi <goffi@goffi.org> parents: 2761 diff changeset	1028 """
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1029
2316 7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1030 RE_FLAGS = re.MULTILINE \| re.UNICODE
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1031 EXEC_ACTIONS = (u"exec", u"external")
2316 7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1032
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1033 def __init__(self, host):
2764 92af49cde255 jp (base): MAM and RSM arguments can now be used for pubsub commands: Goffi <goffi@goffi.org> parents: 2761 diff changeset	1034 # FIXME: C.NO_MAX is not needed here, and this can be globally removed from consts
92af49cde255 jp (base): MAM and RSM arguments can now be used for pubsub commands: Goffi <goffi@goffi.org> parents: 2761 diff changeset	1035 # the only interest is to change the help string, but this can be explained
92af49cde255 jp (base): MAM and RSM arguments can now be used for pubsub commands: Goffi <goffi@goffi.org> parents: 2761 diff changeset	1036 # extensively in man pages (max is for each node found)
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1037 base.CommandBase.__init__(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1038 self,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1039 host,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1040 "search",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1041 use_output=C.OUTPUT_XML,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1042 use_pubsub=True,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1043 pubsub_flags={C.MULTI_ITEMS, C.NO_MAX},
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1044 use_verbose=True,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1045 help=_(u"search items corresponding to filters"),
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1046 )
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1047 self.need_loop = True
2316 7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1048
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1049 @property
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1050 def etree(self):
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1051 """load lxml.etree only if needed"""
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1052 if self._etree is None:
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1053 from lxml import etree
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1054
2316 7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1055 self._etree = etree
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1056 return self._etree
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1057
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1058 def filter_opt(self, value, type_):
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1059 value = base.unicode_decoder(value)
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1060 return (type_, value)
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1061
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1062 def filter_flag(self, value, type_):
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1063 value = C.bool(value)
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1064 return (type_, value)
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1065
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1066 def add_parser_options(self):
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1067 self.parser.add_argument(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1068 "-D",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1069 "--max-depth",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1070 type=int,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1071 default=0,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1072 help=_(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1073 u"maximum depth of recursion (will search linked nodes if > 0, default: 0)"
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1074 ),
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1075 )
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1076 self.parser.add_argument(
2764 92af49cde255 jp (base): MAM and RSM arguments can now be used for pubsub commands: Goffi <goffi@goffi.org> parents: 2761 diff changeset	1077 "-M",
92af49cde255 jp (base): MAM and RSM arguments can now be used for pubsub commands: Goffi <goffi@goffi.org> parents: 2761 diff changeset	1078 "--node-max",
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1079 type=int,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1080 default=30,
2764 92af49cde255 jp (base): MAM and RSM arguments can now be used for pubsub commands: Goffi <goffi@goffi.org> parents: 2761 diff changeset	1081 help=_(u"maximum number of items to get per node ({} to get all items, "
92af49cde255 jp (base): MAM and RSM arguments can now be used for pubsub commands: Goffi <goffi@goffi.org> parents: 2761 diff changeset	1082 u"default: 30)".format( C.NO_LIMIT)),
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1083 )
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1084 self.parser.add_argument(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1085 "-N",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1086 "--namespace",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1087 action="append",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1088 nargs=2,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1089 default=[],
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1090 metavar="NAME NAMESPACE",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1091 help=_(u"namespace to use for xpath"),
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1092 )
2316 7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1093
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1094 # filters
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1095 filter_text = partial(self.filter_opt, type_=u"text")
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1096 filter_re = partial(self.filter_opt, type_=u"regex")
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1097 filter_xpath = partial(self.filter_opt, type_=u"xpath")
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1098 filter_python = partial(self.filter_opt, type_=u"python")
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1099 filters = self.parser.add_argument_group(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1100 _(u"filters"),
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1101 _(u"only items corresponding to following filters will be kept"),
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1102 )
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1103 filters.add_argument(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1104 "-t",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1105 "--text",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1106 action="append",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1107 dest="filters",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1108 type=filter_text,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1109 metavar="TEXT",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1110 help=_(u"full text filter, item must contain this string (XML included)"),
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1111 )
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1112 filters.add_argument(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1113 "-r",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1114 "--regex",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1115 action="append",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1116 dest="filters",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1117 type=filter_re,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1118 metavar="EXPRESSION",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1119 help=_(u"like --text but using a regular expression"),
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1120 )
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1121 filters.add_argument(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1122 "-x",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1123 "--xpath",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1124 action="append",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1125 dest="filters",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1126 type=filter_xpath,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1127 metavar="XPATH",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1128 help=_(u"filter items which has elements matching this xpath"),
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1129 )
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1130 filters.add_argument(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1131 "-P",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1132 "--python",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1133 action="append",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1134 dest="filters",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1135 type=filter_python,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1136 metavar="PYTHON_CODE",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1137 help=_(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1138 u'Python expression which much return a bool (True to keep item, False to reject it). "item" is raw text item, "item_xml" is lxml\'s etree.Element'
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1139 ),
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1140 )
2316 7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1141
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1142 # filters flags
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1143 flag_case = partial(self.filter_flag, type_=u"ignore-case")
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1144 flag_invert = partial(self.filter_flag, type_=u"invert")
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1145 flag_dotall = partial(self.filter_flag, type_=u"dotall")
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1146 flag_matching = partial(self.filter_flag, type_=u"only-matching")
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1147 flags = self.parser.add_argument_group(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1148 _(u"filters flags"),
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1149 _(u"filters modifiers (change behaviour of following filters)"),
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1150 )
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1151 flags.add_argument(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1152 "-C",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1153 "--ignore-case",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1154 action="append",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1155 dest="filters",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1156 type=flag_case,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1157 const=("ignore-case", True),
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1158 nargs="?",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1159 metavar="BOOLEAN",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1160 help=_(u"(don't) ignore case in following filters (default: case sensitive)"),
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1161 )
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1162 flags.add_argument(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1163 "-I",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1164 "--invert",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1165 action="append",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1166 dest="filters",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1167 type=flag_invert,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1168 const=("invert", True),
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1169 nargs="?",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1170 metavar="BOOLEAN",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1171 help=_(u"(don't) invert effect of following filters (default: don't invert)"),
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1172 )
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1173 flags.add_argument(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1174 "-A",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1175 "--dot-all",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1176 action="append",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1177 dest="filters",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1178 type=flag_dotall,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1179 const=("dotall", True),
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1180 nargs="?",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1181 metavar="BOOLEAN",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1182 help=_(u"(don't) use DOTALL option for regex (default: don't use)"),
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1183 )
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1184 flags.add_argument(
2764 92af49cde255 jp (base): MAM and RSM arguments can now be used for pubsub commands: Goffi <goffi@goffi.org> parents: 2761 diff changeset	1185 "-k",
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1186 "--only-matching",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1187 action="append",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1188 dest="filters",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1189 type=flag_matching,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1190 const=("only-matching", True),
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1191 nargs="?",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1192 metavar="BOOLEAN",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1193 help=_(u"keep only the matching part of the item"),
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1194 )
2316 7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1195
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1196 # action
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1197 self.parser.add_argument(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1198 "action",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1199 default="print",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1200 nargs="?",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1201 choices=("print", "exec", "external"),
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1202 help=_(u"action to do on found items (default: print)"),
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1203 )
2316 7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1204 self.parser.add_argument("command", nargs=argparse.REMAINDER)
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1205
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1206 def psItemsGetEb(self, failure_, service, node):
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1207 self.disp(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1208 u"can't get pubsub items at {service} (node: {node}): {reason}".format(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1209 service=service, node=node, reason=failure_
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1210 ),
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1211 error=True,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1212 )
2316 7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1213 self.to_get -= 1
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1214
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1215 def getItems(self, depth, service, node, items):
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1216 search = partial(self.search, depth=depth)
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1217 errback = partial(self.psItemsGetEb, service=service, node=node)
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1218 self.host.bridge.psItemsGet(
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1219 service,
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1220 node,
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1221 self.args.max,
2456 83f9b9ff1a54 jp (pubsub/search): fixed --item argument Goffi <goffi@goffi.org> parents: 2442 diff changeset	1222 items,
2316 7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1223 "",
2761 4b693ea24d5f jp (base, pubsub, ticket): handle order-by: Goffi <goffi@goffi.org> parents: 2624 diff changeset	1224 self.getPubsubExtra(),
2316 7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1225 self.profile,
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1226 callback=search,
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1227 errback=errback,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1228 )
2316 7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1229 self.to_get += 1
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1230
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1231 def _checkPubsubURL(self, match, found_nodes):
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1232 """check that the matched URL is an xmpp: one
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1233
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1234 @param found_nodes(list[unicode]): found_nodes
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1235 this list will be filled while xmpp: URIs are discovered
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1236 """
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1237 url = match.group(0)
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1238 if url.startswith(u"xmpp"):
2316 7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1239 try:
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1240 url_data = uri.parseXMPPUri(url)
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1241 except ValueError:
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1242 return
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1243 if url_data[u"type"] == u"pubsub":
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1244 found_node = {u"service": url_data[u"path"], u"node": url_data[u"node"]}
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1245 if u"item" in url_data:
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1246 found_node[u"item"] = url_data[u"item"]
2316 7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1247 found_nodes.append(found_node)
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1248
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1249 def getSubNodes(self, item, depth):
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1250 """look for pubsub URIs in item, and getItems on the linked nodes"""
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1251 found_nodes = []
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1252 checkURI = partial(self._checkPubsubURL, found_nodes=found_nodes)
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1253 strings.RE_URL.sub(checkURI, item)
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1254 for data in found_nodes:
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1255 self.getItems(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1256 depth + 1,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1257 data[u"service"],
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1258 data[u"node"],
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1259 [data[u"item"]] if u"item" in data else [],
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1260 )
2316 7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1261
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1262 def parseXml(self, item):
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1263 try:
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1264 return self.etree.fromstring(item)
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1265 except self.etree.XMLSyntaxError:
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1266 self.disp(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1267 _(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1268 u"item doesn't looks like XML, you have probably used --only-matching somewhere before and we have no more XML"
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1269 ),
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1270 error=True,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1271 )
2316 7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1272 self.host.quit(C.EXIT_BAD_ARG)
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1273
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1274 def filter(self, item):
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1275 """apply filters given on command line
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1276
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1277 if only-matching is used, item may be modified
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1278 @return (tuple[bool, unicode]): a tuple with:
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1279 - keep: True if item passed the filters
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1280 - item: it is returned in case of modifications
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1281 """
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1282 ignore_case = False
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1283 invert = False
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1284 dotall = False
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1285 only_matching = False
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1286 item_xml = None
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1287 for type_, value in self.args.filters:
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1288 keep = True
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1289
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1290 ## filters
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1291
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1292 if type_ == u"text":
2316 7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1293 if ignore_case:
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1294 if value.lower() not in item.lower():
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1295 keep = False
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1296 else:
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1297 if value not in item:
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1298 keep = False
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1299 if keep and only_matching:
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1300 # doesn't really make sens to keep a fixed string
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1301 # so we raise an error
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1302 self.host.disp(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1303 _(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1304 u"--only-matching used with fixed --text string, are you sure?"
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1305 ),
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1306 error=True,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1307 )
2316 7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1308 self.host.quit(C.EXIT_BAD_ARG)
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1309 elif type_ == u"regex":
2316 7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1310 flags = self.RE_FLAGS
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1311 if ignore_case:
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1312 flags \|= re.IGNORECASE
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1313 if dotall:
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1314 flags \|= re.DOTALL
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1315 match = re.search(value, item, flags)
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1316 keep = match != None
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1317 if keep and only_matching:
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1318 item = match.group()
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1319 item_xml = None
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1320 elif type_ == u"xpath":
2316 7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1321 if item_xml is None:
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1322 item_xml = self.parseXml(item)
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1323 try:
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1324 elts = item_xml.xpath(value, namespaces=self.args.namespace)
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1325 except self.etree.XPathEvalError as e:
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1326 self.disp(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1327 _(u"can't use xpath: {reason}").format(reason=e), error=True
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1328 )
2316 7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1329 self.host.quit(C.EXIT_BAD_ARG)
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1330 keep = bool(elts)
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1331 if keep and only_matching:
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1332 item_xml = elts[0]
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1333 try:
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1334 item = self.etree.tostring(item_xml, encoding="unicode")
2316 7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1335 except TypeError:
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1336 # we have a string only, not an element
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1337 item = unicode(item_xml)
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1338 item_xml = None
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1339 elif type_ == u"python":
2316 7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1340 if item_xml is None:
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1341 item_xml = self.parseXml(item)
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1342 cmd_ns = {u"item": item, u"item_xml": item_xml}
2316 7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1343 try:
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1344 keep = eval(value, cmd_ns)
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1345 except SyntaxError as e:
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1346 self.disp(unicode(e), error=True)
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1347 self.host.quit(C.EXIT_BAD_ARG)
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1348
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1349 ## flags
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1350
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1351 elif type_ == u"ignore-case":
2316 7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1352 ignore_case = value
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1353 elif type_ == u"invert":
2316 7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1354 invert = value
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1355 # we need to continue, else loop would end here
2316 7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1356 continue
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1357 elif type_ == u"dotall":
2316 7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1358 dotall = value
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1359 elif type_ == u"only-matching":
2316 7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1360 only_matching = value
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1361 else:
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1362 raise exceptions.InternalError(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1363 _(u"unknown filter type {type}").format(type=type_)
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1364 )
2316 7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1365
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1366 if invert:
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1367 keep = not keep
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1368 if not keep:
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1369 return False, item
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1370
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1371 return True, item
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1372
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1373 def doItemAction(self, item, metadata):
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1374 """called when item has been kepts and the action need to be done
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1375
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1376 @param item(unicode): accepted item
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1377 """
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1378 action = self.args.action
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1379 if action == u"print" or self.host.verbosity > 0:
2316 7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1380 try:
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1381 self.output(item)
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1382 except self.etree.XMLSyntaxError:
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1383 # item is not valid XML, but a string
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1384 # can happen when --only-matching is used
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1385 self.disp(item)
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1386 if action in self.EXEC_ACTIONS:
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1387 item_elt = self.parseXml(item)
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1388 if action == u"exec":
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1389 use = {
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1390 "service": metadata[u"service"],
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1391 "node": metadata[u"node"],
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1392 "item": item_elt.get("id"),
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1393 "profile": self.profile,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1394 }
2317 f4e05600577b jp (arg_tools): args is not modified anymore in get_use_args + fixed args returned + parser_args are returned separatly (return is now a tuple) Goffi <goffi@goffi.org> parents: 2316 diff changeset	1395 # we need to send a copy of self.args.command
f4e05600577b jp (arg_tools): args is not modified anymore in get_use_args + fixed args returned + parser_args are returned separatly (return is now a tuple) Goffi <goffi@goffi.org> parents: 2316 diff changeset	1396 # else it would be modified
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1397 parser_args, use_args = arg_tools.get_use_args(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1398 self.host, self.args.command, use, verbose=self.host.verbosity > 1
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1399 )
2317 f4e05600577b jp (arg_tools): args is not modified anymore in get_use_args + fixed args returned + parser_args are returned separatly (return is now a tuple) Goffi <goffi@goffi.org> parents: 2316 diff changeset	1400 cmd_args = sys.argv[0:1] + parser_args + use_args
2316 7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1401 else:
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1402 cmd_args = self.args.command
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1403
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1404 self.disp(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1405 u"COMMAND: {command}".format(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1406 command=u" ".join([arg_tools.escape(a) for a in cmd_args])
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1407 ),
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1408 2,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1409 )
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1410 if action == u"exec":
2316 7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1411 ret = subprocess.call(cmd_args)
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1412 else:
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1413 p = subprocess.Popen(cmd_args, stdin=subprocess.PIPE)
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1414 p.communicate(item.encode("utf-8"))
2316 7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1415 ret = p.wait()
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1416 if ret != 0:
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1417 self.disp(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1418 A.color(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1419 C.A_FAILURE,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1420 _(u"executed command failed with exit code {code}").format(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1421 code=ret
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1422 ),
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1423 )
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1424 )
2316 7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1425
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1426 def search(self, items_data, depth):
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1427 """callback of getItems
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1428
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1429 this method filters items, get sub nodes if needed,
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1430 do the requested action, and exit the command when everything is done
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1431 @param items_data(tuple): result of getItems
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1432 @param depth(int): current depth level
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1433 0 for first node, 1 for first children, and so on
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1434 """
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1435 items, metadata = items_data
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1436 for item in items:
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1437 if depth < self.args.max_depth:
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1438 self.getSubNodes(item, depth)
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1439 keep, item = self.filter(item)
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1440 if not keep:
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1441 continue
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1442 self.doItemAction(item, metadata)
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1443
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1444 # we check if we got all getItems results
2316 7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1445 self.to_get -= 1
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1446 if self.to_get == 0:
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1447 # yes, we can quit
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1448 self.host.quit()
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1449 assert self.to_get > 0
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1450
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1451 def start(self):
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1452 if self.args.command:
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1453 if self.args.action not in self.EXEC_ACTIONS:
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1454 self.parser.error(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1455 _(u"Command can only be used with {actions} actions").format(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1456 actions=u", ".join(self.EXEC_ACTIONS)
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1457 )
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1458 )
2316 7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1459 else:
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1460 if self.args.action in self.EXEC_ACTIONS:
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1461 self.parser.error(_(u"you need to specify a command to execute"))
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1462 if not self.args.node:
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1463 # TODO: handle get service affiliations when node is not set
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1464 self.parser.error(_(u"empty node is not handled yet"))
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1465 # to_get is increased on each get and decreased on each answer
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1466 # when it reach 0 again, the command is finished
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1467 self.to_get = 0
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1468 self._etree = None
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1469 if self.args.filters is None:
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1470 self.args.filters = []
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1471 self.args.namespace = dict(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1472 self.args.namespace + [("pubsub", "http://jabber.org/protocol/pubsub")]
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1473 )
2316 7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1474 self.getItems(0, self.args.service, self.args.node, self.args.items)
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1475
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1476
2235 4db836386641 jp: added use_pubsub and use_pubsub_node_req arguments to CommandBase Goffi <goffi@goffi.org> parents: 2224 diff changeset	1477 class Uri(base.CommandBase):
2224 87fcd4a7c7e4 jp (pubsub): added uri command to build pubsub URI Goffi <goffi@goffi.org> parents: 2221 diff changeset	1478 def __init__(self, host):
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1479 base.CommandBase.__init__(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1480 self,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1481 host,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1482 "uri",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1483 use_profile=False,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1484 use_pubsub=True,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1485 pubsub_flags={C.NODE, C.SINGLE_ITEM},
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1486 help=_(u"build URI"),
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1487 )
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1488 self.need_loop = True
2224 87fcd4a7c7e4 jp (pubsub): added uri command to build pubsub URI Goffi <goffi@goffi.org> parents: 2221 diff changeset	1489
87fcd4a7c7e4 jp (pubsub): added uri command to build pubsub URI Goffi <goffi@goffi.org> parents: 2221 diff changeset	1490 def add_parser_options(self):
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1491 self.parser.add_argument(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1492 "-p",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1493 "--profile",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1494 type=base.unicode_decoder,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1495 default=C.PROF_KEY_DEFAULT,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1496 help=_(u"profile (used when no server is specified)"),
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1497 )
2224 87fcd4a7c7e4 jp (pubsub): added uri command to build pubsub URI Goffi <goffi@goffi.org> parents: 2221 diff changeset	1498
87fcd4a7c7e4 jp (pubsub): added uri command to build pubsub URI Goffi <goffi@goffi.org> parents: 2221 diff changeset	1499 def display_uri(self, jid_):
2239 17502e74c046 jp (pubsub/uri): fixed URI generation Goffi <goffi@goffi.org> parents: 2235 diff changeset	1500 uri_args = {}
2224 87fcd4a7c7e4 jp (pubsub): added uri command to build pubsub URI Goffi <goffi@goffi.org> parents: 2221 diff changeset	1501 if not self.args.service:
87fcd4a7c7e4 jp (pubsub): added uri command to build pubsub URI Goffi <goffi@goffi.org> parents: 2221 diff changeset	1502 self.args.service = jid.JID(jid_).bare
87fcd4a7c7e4 jp (pubsub): added uri command to build pubsub URI Goffi <goffi@goffi.org> parents: 2221 diff changeset	1503
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1504 for key in ("node", "service", "item"):
2224 87fcd4a7c7e4 jp (pubsub): added uri command to build pubsub URI Goffi <goffi@goffi.org> parents: 2221 diff changeset	1505 value = getattr(self.args, key)
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1506 if key == "service":
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1507 key = "path"
2224 87fcd4a7c7e4 jp (pubsub): added uri command to build pubsub URI Goffi <goffi@goffi.org> parents: 2221 diff changeset	1508 if value:
87fcd4a7c7e4 jp (pubsub): added uri command to build pubsub URI Goffi <goffi@goffi.org> parents: 2221 diff changeset	1509 uri_args[key] = value
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1510 self.disp(uri.buildXMPPUri(u"pubsub", **uri_args))
2224 87fcd4a7c7e4 jp (pubsub): added uri command to build pubsub URI Goffi <goffi@goffi.org> parents: 2221 diff changeset	1511 self.host.quit()
87fcd4a7c7e4 jp (pubsub): added uri command to build pubsub URI Goffi <goffi@goffi.org> parents: 2221 diff changeset	1512
87fcd4a7c7e4 jp (pubsub): added uri command to build pubsub URI Goffi <goffi@goffi.org> parents: 2221 diff changeset	1513 def start(self):
87fcd4a7c7e4 jp (pubsub): added uri command to build pubsub URI Goffi <goffi@goffi.org> parents: 2221 diff changeset	1514 if not self.args.service:
87fcd4a7c7e4 jp (pubsub): added uri command to build pubsub URI Goffi <goffi@goffi.org> parents: 2221 diff changeset	1515 self.host.bridge.asyncGetParamA(
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1516 u"JabberID",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1517 u"Connection",
2224 87fcd4a7c7e4 jp (pubsub): added uri command to build pubsub URI Goffi <goffi@goffi.org> parents: 2221 diff changeset	1518 profile_key=self.args.profile,
87fcd4a7c7e4 jp (pubsub): added uri command to build pubsub URI Goffi <goffi@goffi.org> parents: 2221 diff changeset	1519 callback=self.display_uri,
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1520 errback=partial(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1521 self.errback,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1522 msg=_(u"can't retrieve jid: {}"),
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1523 exit_code=C.EXIT_BRIDGE_ERRBACK,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1524 ),
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1525 )
2224 87fcd4a7c7e4 jp (pubsub): added uri command to build pubsub URI Goffi <goffi@goffi.org> parents: 2221 diff changeset	1526 else:
87fcd4a7c7e4 jp (pubsub): added uri command to build pubsub URI Goffi <goffi@goffi.org> parents: 2221 diff changeset	1527 self.display_uri(None)
87fcd4a7c7e4 jp (pubsub): added uri command to build pubsub URI Goffi <goffi@goffi.org> parents: 2221 diff changeset	1528
87fcd4a7c7e4 jp (pubsub): added uri command to build pubsub URI Goffi <goffi@goffi.org> parents: 2221 diff changeset	1529
2308 0b21d87c91cf jp (pubsub/hook): added create/delete/list hook command to handle new Pubsub hook feature Goffi <goffi@goffi.org> parents: 2292 diff changeset	1530 class HookCreate(base.CommandBase):
0b21d87c91cf jp (pubsub/hook): added create/delete/list hook command to handle new Pubsub hook feature Goffi <goffi@goffi.org> parents: 2292 diff changeset	1531 def __init__(self, host):
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1532 base.CommandBase.__init__(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1533 self,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1534 host,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1535 "create",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1536 use_pubsub=True,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1537 pubsub_flags={C.NODE},
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1538 help=_(u"create a Pubsub hook"),
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1539 )
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1540 self.need_loop = True
2308 0b21d87c91cf jp (pubsub/hook): added create/delete/list hook command to handle new Pubsub hook feature Goffi <goffi@goffi.org> parents: 2292 diff changeset	1541
0b21d87c91cf jp (pubsub/hook): added create/delete/list hook command to handle new Pubsub hook feature Goffi <goffi@goffi.org> parents: 2292 diff changeset	1542 def add_parser_options(self):
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1543 self.parser.add_argument(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1544 "-t",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1545 "--type",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1546 default=u"python",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1547 choices=("python", "python_file", "python_code"),
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1548 help=_(u"hook type"),
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1549 )
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1550 self.parser.add_argument(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1551 "-P",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1552 "--persistent",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1553 action="store_true",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1554 help=_(u"make hook persistent across restarts"),
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1555 )
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1556 self.parser.add_argument(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1557 "hook_arg",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1558 type=base.unicode_decoder,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1559 help=_(u"argument of the hook (depend of the type)"),
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1560 )
2308 0b21d87c91cf jp (pubsub/hook): added create/delete/list hook command to handle new Pubsub hook feature Goffi <goffi@goffi.org> parents: 2292 diff changeset	1561
0b21d87c91cf jp (pubsub/hook): added create/delete/list hook command to handle new Pubsub hook feature Goffi <goffi@goffi.org> parents: 2292 diff changeset	1562 @staticmethod
0b21d87c91cf jp (pubsub/hook): added create/delete/list hook command to handle new Pubsub hook feature Goffi <goffi@goffi.org> parents: 2292 diff changeset	1563 def checkArgs(self):
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1564 if self.args.type == u"python_file":
2308 0b21d87c91cf jp (pubsub/hook): added create/delete/list hook command to handle new Pubsub hook feature Goffi <goffi@goffi.org> parents: 2292 diff changeset	1565 self.args.hook_arg = os.path.abspath(self.args.hook_arg)
0b21d87c91cf jp (pubsub/hook): added create/delete/list hook command to handle new Pubsub hook feature Goffi <goffi@goffi.org> parents: 2292 diff changeset	1566 if not os.path.isfile(self.args.hook_arg):
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1567 self.parser.error(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1568 _(u"{path} is not a file").format(path=self.args.hook_arg)
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1569 )
2308 0b21d87c91cf jp (pubsub/hook): added create/delete/list hook command to handle new Pubsub hook feature Goffi <goffi@goffi.org> parents: 2292 diff changeset	1570
0b21d87c91cf jp (pubsub/hook): added create/delete/list hook command to handle new Pubsub hook feature Goffi <goffi@goffi.org> parents: 2292 diff changeset	1571 def start(self):
0b21d87c91cf jp (pubsub/hook): added create/delete/list hook command to handle new Pubsub hook feature Goffi <goffi@goffi.org> parents: 2292 diff changeset	1572 self.checkArgs(self)
0b21d87c91cf jp (pubsub/hook): added create/delete/list hook command to handle new Pubsub hook feature Goffi <goffi@goffi.org> parents: 2292 diff changeset	1573 self.host.bridge.psHookAdd(
0b21d87c91cf jp (pubsub/hook): added create/delete/list hook command to handle new Pubsub hook feature Goffi <goffi@goffi.org> parents: 2292 diff changeset	1574 self.args.service,
0b21d87c91cf jp (pubsub/hook): added create/delete/list hook command to handle new Pubsub hook feature Goffi <goffi@goffi.org> parents: 2292 diff changeset	1575 self.args.node,
0b21d87c91cf jp (pubsub/hook): added create/delete/list hook command to handle new Pubsub hook feature Goffi <goffi@goffi.org> parents: 2292 diff changeset	1576 self.args.type,
0b21d87c91cf jp (pubsub/hook): added create/delete/list hook command to handle new Pubsub hook feature Goffi <goffi@goffi.org> parents: 2292 diff changeset	1577 self.args.hook_arg,
0b21d87c91cf jp (pubsub/hook): added create/delete/list hook command to handle new Pubsub hook feature Goffi <goffi@goffi.org> parents: 2292 diff changeset	1578 self.args.persistent,
0b21d87c91cf jp (pubsub/hook): added create/delete/list hook command to handle new Pubsub hook feature Goffi <goffi@goffi.org> parents: 2292 diff changeset	1579 self.profile,
0b21d87c91cf jp (pubsub/hook): added create/delete/list hook command to handle new Pubsub hook feature Goffi <goffi@goffi.org> parents: 2292 diff changeset	1580 callback=self.host.quit,
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1581 errback=partial(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1582 self.errback,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1583 msg=_(u"can't create hook: {}"),
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1584 exit_code=C.EXIT_BRIDGE_ERRBACK,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1585 ),
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1586 )
2308 0b21d87c91cf jp (pubsub/hook): added create/delete/list hook command to handle new Pubsub hook feature Goffi <goffi@goffi.org> parents: 2292 diff changeset	1587
0b21d87c91cf jp (pubsub/hook): added create/delete/list hook command to handle new Pubsub hook feature Goffi <goffi@goffi.org> parents: 2292 diff changeset	1588
0b21d87c91cf jp (pubsub/hook): added create/delete/list hook command to handle new Pubsub hook feature Goffi <goffi@goffi.org> parents: 2292 diff changeset	1589 class HookDelete(base.CommandBase):
0b21d87c91cf jp (pubsub/hook): added create/delete/list hook command to handle new Pubsub hook feature Goffi <goffi@goffi.org> parents: 2292 diff changeset	1590 def __init__(self, host):
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1591 base.CommandBase.__init__(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1592 self,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1593 host,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1594 "delete",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1595 use_pubsub=True,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1596 pubsub_flags={C.NODE},
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1597 help=_(u"delete a Pubsub hook"),
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1598 )
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1599 self.need_loop = True
2308 0b21d87c91cf jp (pubsub/hook): added create/delete/list hook command to handle new Pubsub hook feature Goffi <goffi@goffi.org> parents: 2292 diff changeset	1600
0b21d87c91cf jp (pubsub/hook): added create/delete/list hook command to handle new Pubsub hook feature Goffi <goffi@goffi.org> parents: 2292 diff changeset	1601 def add_parser_options(self):
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1602 self.parser.add_argument(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1603 "-t",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1604 "--type",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1605 default=u"",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1606 choices=("", "python", "python_file", "python_code"),
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1607 help=_(u"hook type to remove, empty to remove all (default: remove all)"),
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1608 )
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1609 self.parser.add_argument(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1610 "-a",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1611 "--arg",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1612 dest="hook_arg",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1613 type=base.unicode_decoder,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1614 default=u"",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1615 help=_(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1616 u"argument of the hook to remove, empty to remove all (default: remove all)"
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1617 ),
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1618 )
2308 0b21d87c91cf jp (pubsub/hook): added create/delete/list hook command to handle new Pubsub hook feature Goffi <goffi@goffi.org> parents: 2292 diff changeset	1619
0b21d87c91cf jp (pubsub/hook): added create/delete/list hook command to handle new Pubsub hook feature Goffi <goffi@goffi.org> parents: 2292 diff changeset	1620 def psHookRemoveCb(self, nb_deleted):
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1621 self.disp(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1622 _(u"{nb_deleted} hook(s) have been deleted").format(nb_deleted=nb_deleted)
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1623 )
2308 0b21d87c91cf jp (pubsub/hook): added create/delete/list hook command to handle new Pubsub hook feature Goffi <goffi@goffi.org> parents: 2292 diff changeset	1624 self.host.quit()
0b21d87c91cf jp (pubsub/hook): added create/delete/list hook command to handle new Pubsub hook feature Goffi <goffi@goffi.org> parents: 2292 diff changeset	1625
0b21d87c91cf jp (pubsub/hook): added create/delete/list hook command to handle new Pubsub hook feature Goffi <goffi@goffi.org> parents: 2292 diff changeset	1626 def start(self):
0b21d87c91cf jp (pubsub/hook): added create/delete/list hook command to handle new Pubsub hook feature Goffi <goffi@goffi.org> parents: 2292 diff changeset	1627 HookCreate.checkArgs(self)
0b21d87c91cf jp (pubsub/hook): added create/delete/list hook command to handle new Pubsub hook feature Goffi <goffi@goffi.org> parents: 2292 diff changeset	1628 self.host.bridge.psHookRemove(
0b21d87c91cf jp (pubsub/hook): added create/delete/list hook command to handle new Pubsub hook feature Goffi <goffi@goffi.org> parents: 2292 diff changeset	1629 self.args.service,
0b21d87c91cf jp (pubsub/hook): added create/delete/list hook command to handle new Pubsub hook feature Goffi <goffi@goffi.org> parents: 2292 diff changeset	1630 self.args.node,
0b21d87c91cf jp (pubsub/hook): added create/delete/list hook command to handle new Pubsub hook feature Goffi <goffi@goffi.org> parents: 2292 diff changeset	1631 self.args.type,
0b21d87c91cf jp (pubsub/hook): added create/delete/list hook command to handle new Pubsub hook feature Goffi <goffi@goffi.org> parents: 2292 diff changeset	1632 self.args.hook_arg,
0b21d87c91cf jp (pubsub/hook): added create/delete/list hook command to handle new Pubsub hook feature Goffi <goffi@goffi.org> parents: 2292 diff changeset	1633 self.profile,
0b21d87c91cf jp (pubsub/hook): added create/delete/list hook command to handle new Pubsub hook feature Goffi <goffi@goffi.org> parents: 2292 diff changeset	1634 callback=self.psHookRemoveCb,
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1635 errback=partial(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1636 self.errback,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1637 msg=_(u"can't delete hook: {}"),
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1638 exit_code=C.EXIT_BRIDGE_ERRBACK,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1639 ),
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1640 )
2308 0b21d87c91cf jp (pubsub/hook): added create/delete/list hook command to handle new Pubsub hook feature Goffi <goffi@goffi.org> parents: 2292 diff changeset	1641
0b21d87c91cf jp (pubsub/hook): added create/delete/list hook command to handle new Pubsub hook feature Goffi <goffi@goffi.org> parents: 2292 diff changeset	1642
0b21d87c91cf jp (pubsub/hook): added create/delete/list hook command to handle new Pubsub hook feature Goffi <goffi@goffi.org> parents: 2292 diff changeset	1643 class HookList(base.CommandBase):
0b21d87c91cf jp (pubsub/hook): added create/delete/list hook command to handle new Pubsub hook feature Goffi <goffi@goffi.org> parents: 2292 diff changeset	1644 def __init__(self, host):
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1645 base.CommandBase.__init__(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1646 self,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1647 host,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1648 "list",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1649 use_output=C.OUTPUT_LIST_DICT,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1650 help=_(u"list hooks of a profile"),
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1651 )
2308 0b21d87c91cf jp (pubsub/hook): added create/delete/list hook command to handle new Pubsub hook feature Goffi <goffi@goffi.org> parents: 2292 diff changeset	1652 self.need_loop = True
0b21d87c91cf jp (pubsub/hook): added create/delete/list hook command to handle new Pubsub hook feature Goffi <goffi@goffi.org> parents: 2292 diff changeset	1653
0b21d87c91cf jp (pubsub/hook): added create/delete/list hook command to handle new Pubsub hook feature Goffi <goffi@goffi.org> parents: 2292 diff changeset	1654 def add_parser_options(self):
0b21d87c91cf jp (pubsub/hook): added create/delete/list hook command to handle new Pubsub hook feature Goffi <goffi@goffi.org> parents: 2292 diff changeset	1655 pass
0b21d87c91cf jp (pubsub/hook): added create/delete/list hook command to handle new Pubsub hook feature Goffi <goffi@goffi.org> parents: 2292 diff changeset	1656
0b21d87c91cf jp (pubsub/hook): added create/delete/list hook command to handle new Pubsub hook feature Goffi <goffi@goffi.org> parents: 2292 diff changeset	1657 def psHookListCb(self, data):
0b21d87c91cf jp (pubsub/hook): added create/delete/list hook command to handle new Pubsub hook feature Goffi <goffi@goffi.org> parents: 2292 diff changeset	1658 if not data:
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1659 self.disp(_(u"No hook found."))
2308 0b21d87c91cf jp (pubsub/hook): added create/delete/list hook command to handle new Pubsub hook feature Goffi <goffi@goffi.org> parents: 2292 diff changeset	1660 self.output(data)
0b21d87c91cf jp (pubsub/hook): added create/delete/list hook command to handle new Pubsub hook feature Goffi <goffi@goffi.org> parents: 2292 diff changeset	1661 self.host.quit()
0b21d87c91cf jp (pubsub/hook): added create/delete/list hook command to handle new Pubsub hook feature Goffi <goffi@goffi.org> parents: 2292 diff changeset	1662
0b21d87c91cf jp (pubsub/hook): added create/delete/list hook command to handle new Pubsub hook feature Goffi <goffi@goffi.org> parents: 2292 diff changeset	1663 def start(self):
0b21d87c91cf jp (pubsub/hook): added create/delete/list hook command to handle new Pubsub hook feature Goffi <goffi@goffi.org> parents: 2292 diff changeset	1664 self.host.bridge.psHookList(
0b21d87c91cf jp (pubsub/hook): added create/delete/list hook command to handle new Pubsub hook feature Goffi <goffi@goffi.org> parents: 2292 diff changeset	1665 self.profile,
0b21d87c91cf jp (pubsub/hook): added create/delete/list hook command to handle new Pubsub hook feature Goffi <goffi@goffi.org> parents: 2292 diff changeset	1666 callback=self.psHookListCb,
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1667 errback=partial(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1668 self.errback,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1669 msg=_(u"can't list hooks: {}"),
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1670 exit_code=C.EXIT_BRIDGE_ERRBACK,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1671 ),
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1672 )
2308 0b21d87c91cf jp (pubsub/hook): added create/delete/list hook command to handle new Pubsub hook feature Goffi <goffi@goffi.org> parents: 2292 diff changeset	1673
0b21d87c91cf jp (pubsub/hook): added create/delete/list hook command to handle new Pubsub hook feature Goffi <goffi@goffi.org> parents: 2292 diff changeset	1674
0b21d87c91cf jp (pubsub/hook): added create/delete/list hook command to handle new Pubsub hook feature Goffi <goffi@goffi.org> parents: 2292 diff changeset	1675 class Hook(base.CommandBase):
0b21d87c91cf jp (pubsub/hook): added create/delete/list hook command to handle new Pubsub hook feature Goffi <goffi@goffi.org> parents: 2292 diff changeset	1676 subcommands = (HookCreate, HookDelete, HookList)
0b21d87c91cf jp (pubsub/hook): added create/delete/list hook command to handle new Pubsub hook feature Goffi <goffi@goffi.org> parents: 2292 diff changeset	1677
0b21d87c91cf jp (pubsub/hook): added create/delete/list hook command to handle new Pubsub hook feature Goffi <goffi@goffi.org> parents: 2292 diff changeset	1678 def __init__(self, host):
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1679 super(Hook, self).__init__(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1680 host,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1681 "hook",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1682 use_profile=False,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1683 use_verbose=True,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1684 help=_("trigger action on Pubsub notifications"),
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1685 )
2308 0b21d87c91cf jp (pubsub/hook): added create/delete/list hook command to handle new Pubsub hook feature Goffi <goffi@goffi.org> parents: 2292 diff changeset	1686
0b21d87c91cf jp (pubsub/hook): added create/delete/list hook command to handle new Pubsub hook feature Goffi <goffi@goffi.org> parents: 2292 diff changeset	1687
2191 a1a8233f89e8 jp(pubsub/get): pubsub/get command, first draft Goffi <goffi@goffi.org> parents: diff changeset	1688 class Pubsub(base.CommandBase):
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1689 subcommands = (
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1690 Set,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1691 Get,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1692 Delete,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1693 Edit,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1694 Subscribe,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1695 Unsubscribe,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1696 Subscriptions,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1697 Node,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1698 Affiliations,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1699 Search,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1700 Hook,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1701 Uri,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1702 )
2191 a1a8233f89e8 jp(pubsub/get): pubsub/get command, first draft Goffi <goffi@goffi.org> parents: diff changeset	1703
a1a8233f89e8 jp(pubsub/get): pubsub/get command, first draft Goffi <goffi@goffi.org> parents: diff changeset	1704 def __init__(self, host):
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1705 super(Pubsub, self).__init__(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1706 host, "pubsub", use_profile=False, help=_("PubSub nodes/items management")
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1707 )

Mercurial > libervia-backend

annotate sat_frontends/jp/cmd_pubsub.py @ 2766:93a421de0e3d