libervia-backend: sat_frontends/jp/cmd

annotate sat_frontends/jp/cmd_pubsub.py @ 3498:d78b5eae912a

tests: update following names change

author	Goffi <goffi@goffi.org>
date	Fri, 16 Apr 2021 18:32:34 +0200
parents	be6d91572633
children	62f490eff51c

rev	line source
3137 559a625a236b fixed shebangs Goffi <goffi@goffi.org> parents: 3136 diff changeset	1 #!/usr/bin/env python3
559a625a236b fixed shebangs Goffi <goffi@goffi.org> parents: 3136 diff changeset	2
2191 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
3479 be6d91572633 date update Goffi <goffi@goffi.org> parents: 3467 diff changeset	5 # Copyright (C) 2009-2021 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
3040 fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	21 import argparse
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	22 import os.path
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	23 import re
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	24 import sys
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	25 import subprocess
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	26 import asyncio
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2910 diff changeset	27 from . import base
2191 a1a8233f89e8 jp(pubsub/get): pubsub/get command, first draft Goffi <goffi@goffi.org> parents: diff changeset	28 from sat.core.i18n import _
2316 7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	29 from sat.core import exceptions
2191 a1a8233f89e8 jp(pubsub/get): pubsub/get command, first draft Goffi <goffi@goffi.org> parents: diff changeset	30 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	31 from sat_frontends.jp import common
2316 7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	32 from sat_frontends.jp import arg_tools
2777 ff1b40823b07 jp (pubsub): new "transform" command: Goffi <goffi@goffi.org> parents: 2773 diff changeset	33 from sat_frontends.jp import xml_tools
2221 a6c9bc4d1de0 jp (pubsub/node): added create and delete commands Goffi <goffi@goffi.org> parents: 2214 diff changeset	34 from functools import partial
3100 cea52c9ddfd9 plugin XEP-0060, jp (pubsub/set): publish-options implementation: Goffi <goffi@goffi.org> parents: 3040 diff changeset	35 from sat.tools.common import data_format
2224 87fcd4a7c7e4 jp (pubsub): added uri command to build pubsub URI Goffi <goffi@goffi.org> parents: 2221 diff changeset	36 from sat.tools.common import uri
2316 7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	37 from sat.tools.common.ansi import ANSI as A
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	38 from sat_frontends.tools import jid, strings
3413 f4d417593e77 jp (pubsub/node/info, pubsub/get): better exit code: Goffi <goffi@goffi.org> parents: 3412 diff changeset	39 from sat_frontends.bridge.bridge_frontend import BridgeException
2191 a1a8233f89e8 jp(pubsub/get): pubsub/get command, first draft Goffi <goffi@goffi.org> parents: diff changeset	40
a1a8233f89e8 jp(pubsub/get): pubsub/get command, first draft Goffi <goffi@goffi.org> parents: diff changeset	41 __commands__ = ["Pubsub"]
a1a8233f89e8 jp(pubsub/get): pubsub/get command, first draft Goffi <goffi@goffi.org> parents: diff changeset	42
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2910 diff changeset	43 PUBSUB_TMP_DIR = "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	44 PUBSUB_SCHEMA_TMP_DIR = PUBSUB_TMP_DIR + "_schema"
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	45 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	46
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	47 # 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	48
2191 a1a8233f89e8 jp(pubsub/get): pubsub/get command, first draft Goffi <goffi@goffi.org> parents: diff changeset	49
2235 4db836386641 jp: added use_pubsub and use_pubsub_node_req arguments to CommandBase Goffi <goffi@goffi.org> parents: 2224 diff changeset	50 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	51 def __init__(self, host):
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	52 base.CommandBase.__init__(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	53 self,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	54 host,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	55 "info",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	56 use_output=C.OUTPUT_DICT,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	57 use_pubsub=True,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	58 pubsub_flags={C.NODE},
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2910 diff changeset	59 help=_("retrieve node configuration"),
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	60 )
2197 e0e06391ce91 jp (pubsub): added pubsub/node/info command to retrieve node configuration Goffi <goffi@goffi.org> parents: 2195 diff changeset	61
e0e06391ce91 jp (pubsub): added pubsub/node/info command to retrieve node configuration Goffi <goffi@goffi.org> parents: 2195 diff changeset	62 def add_parser_options(self):
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	63 self.parser.add_argument(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	64 "-k",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	65 "--key",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	66 action="append",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	67 dest="keys",
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2910 diff changeset	68 help=_("data key to filter"),
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	69 )
2197 e0e06391ce91 jp (pubsub): added pubsub/node/info command to retrieve node configuration Goffi <goffi@goffi.org> parents: 2195 diff changeset	70
e0e06391ce91 jp (pubsub): added pubsub/node/info command to retrieve node configuration Goffi <goffi@goffi.org> parents: 2195 diff changeset	71 def removePrefix(self, key):
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2910 diff changeset	72 return key[7:] if key.startswith("pubsub#") else key
2197 e0e06391ce91 jp (pubsub): added pubsub/node/info command to retrieve node configuration Goffi <goffi@goffi.org> parents: 2195 diff changeset	73
e0e06391ce91 jp (pubsub): added pubsub/node/info command to retrieve node configuration Goffi <goffi@goffi.org> parents: 2195 diff changeset	74 def filterKey(self, key):
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2910 diff changeset	75 return any((key == k or key == "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	76
3040 fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	77 async def start(self):
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	78 try:
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	79 config_dict = await self.host.bridge.psNodeConfigurationGet(
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	80 self.args.service,
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	81 self.args.node,
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	82 self.profile,
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	83 )
3413 f4d417593e77 jp (pubsub/node/info, pubsub/get): better exit code: Goffi <goffi@goffi.org> parents: 3412 diff changeset	84 except BridgeException as e:
f4d417593e77 jp (pubsub/node/info, pubsub/get): better exit code: Goffi <goffi@goffi.org> parents: 3412 diff changeset	85 if e.condition == 'item-not-found':
f4d417593e77 jp (pubsub/node/info, pubsub/get): better exit code: Goffi <goffi@goffi.org> parents: 3412 diff changeset	86 self.disp(
f4d417593e77 jp (pubsub/node/info, pubsub/get): better exit code: Goffi <goffi@goffi.org> parents: 3412 diff changeset	87 f"The node {self.args.node} doesn't exist on {self.args.service}",
f4d417593e77 jp (pubsub/node/info, pubsub/get): better exit code: Goffi <goffi@goffi.org> parents: 3412 diff changeset	88 error=True
f4d417593e77 jp (pubsub/node/info, pubsub/get): better exit code: Goffi <goffi@goffi.org> parents: 3412 diff changeset	89 )
f4d417593e77 jp (pubsub/node/info, pubsub/get): better exit code: Goffi <goffi@goffi.org> parents: 3412 diff changeset	90 self.host.quit(C.EXIT_NOT_FOUND)
f4d417593e77 jp (pubsub/node/info, pubsub/get): better exit code: Goffi <goffi@goffi.org> parents: 3412 diff changeset	91 else:
f4d417593e77 jp (pubsub/node/info, pubsub/get): better exit code: Goffi <goffi@goffi.org> parents: 3412 diff changeset	92 self.disp(f"can't get node configuration: {e}", error=True)
f4d417593e77 jp (pubsub/node/info, pubsub/get): better exit code: Goffi <goffi@goffi.org> parents: 3412 diff changeset	93 self.host.quit(C.EXIT_BRIDGE_ERRBACK)
3040 fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	94 except Exception as e:
3413 f4d417593e77 jp (pubsub/node/info, pubsub/get): better exit code: Goffi <goffi@goffi.org> parents: 3412 diff changeset	95 self.disp(f"Internal error: {e}", error=True)
f4d417593e77 jp (pubsub/node/info, pubsub/get): better exit code: Goffi <goffi@goffi.org> parents: 3412 diff changeset	96 self.host.quit(C.EXIT_INTERNAL_ERROR)
3040 fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	97 else:
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	98 key_filter = (lambda k: True) if not self.args.keys else self.filterKey
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	99 config_dict = {
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	100 self.removePrefix(k): v for k, v in config_dict.items() if key_filter(k)
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	101 }
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	102 await self.output(config_dict)
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	103 self.host.quit()
2197 e0e06391ce91 jp (pubsub): added pubsub/node/info command to retrieve node configuration Goffi <goffi@goffi.org> parents: 2195 diff changeset	104
2221 a6c9bc4d1de0 jp (pubsub/node): added create and delete commands Goffi <goffi@goffi.org> parents: 2214 diff changeset	105
2235 4db836386641 jp: added use_pubsub and use_pubsub_node_req arguments to CommandBase Goffi <goffi@goffi.org> parents: 2224 diff changeset	106 class NodeCreate(base.CommandBase):
2221 a6c9bc4d1de0 jp (pubsub/node): added create and delete commands Goffi <goffi@goffi.org> parents: 2214 diff changeset	107 def __init__(self, host):
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	108 base.CommandBase.__init__(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	109 self,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	110 host,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	111 "create",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	112 use_output=C.OUTPUT_DICT,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	113 use_pubsub=True,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	114 pubsub_flags={C.NODE},
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	115 use_verbose=True,
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2910 diff changeset	116 help=_("create a node"),
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	117 )
2221 a6c9bc4d1de0 jp (pubsub/node): added create and delete commands Goffi <goffi@goffi.org> parents: 2214 diff changeset	118
3100 cea52c9ddfd9 plugin XEP-0060, jp (pubsub/set): publish-options implementation: Goffi <goffi@goffi.org> parents: 3040 diff changeset	119 @staticmethod
cea52c9ddfd9 plugin XEP-0060, jp (pubsub/set): publish-options implementation: Goffi <goffi@goffi.org> parents: 3040 diff changeset	120 def add_node_config_options(parser):
cea52c9ddfd9 plugin XEP-0060, jp (pubsub/set): publish-options implementation: Goffi <goffi@goffi.org> parents: 3040 diff changeset	121 parser.add_argument(
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	122 "-f",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	123 "--field",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	124 action="append",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	125 nargs=2,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	126 dest="fields",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	127 default=[],
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2910 diff changeset	128 metavar=("KEY", "VALUE"),
ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2910 diff changeset	129 help=_("configuration field to set"),
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	130 )
3100 cea52c9ddfd9 plugin XEP-0060, jp (pubsub/set): publish-options implementation: Goffi <goffi@goffi.org> parents: 3040 diff changeset	131 parser.add_argument(
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	132 "-F",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	133 "--full-prefix",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	134 action="store_true",
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2910 diff changeset	135 help=_('don\'t prepend "pubsub#" prefix to field names'),
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	136 )
2221 a6c9bc4d1de0 jp (pubsub/node): added create and delete commands Goffi <goffi@goffi.org> parents: 2214 diff changeset	137
3100 cea52c9ddfd9 plugin XEP-0060, jp (pubsub/set): publish-options implementation: Goffi <goffi@goffi.org> parents: 3040 diff changeset	138 def add_parser_options(self):
cea52c9ddfd9 plugin XEP-0060, jp (pubsub/set): publish-options implementation: Goffi <goffi@goffi.org> parents: 3040 diff changeset	139 self.add_node_config_options(self.parser)
cea52c9ddfd9 plugin XEP-0060, jp (pubsub/set): publish-options implementation: Goffi <goffi@goffi.org> parents: 3040 diff changeset	140
cea52c9ddfd9 plugin XEP-0060, jp (pubsub/set): publish-options implementation: Goffi <goffi@goffi.org> parents: 3040 diff changeset	141 @staticmethod
cea52c9ddfd9 plugin XEP-0060, jp (pubsub/set): publish-options implementation: Goffi <goffi@goffi.org> parents: 3040 diff changeset	142 def get_config_options(args):
cea52c9ddfd9 plugin XEP-0060, jp (pubsub/set): publish-options implementation: Goffi <goffi@goffi.org> parents: 3040 diff changeset	143 if not args.full_prefix:
cea52c9ddfd9 plugin XEP-0060, jp (pubsub/set): publish-options implementation: Goffi <goffi@goffi.org> parents: 3040 diff changeset	144 return {"pubsub#" + k: v for k, v in args.fields}
cea52c9ddfd9 plugin XEP-0060, jp (pubsub/set): publish-options implementation: Goffi <goffi@goffi.org> parents: 3040 diff changeset	145 else:
cea52c9ddfd9 plugin XEP-0060, jp (pubsub/set): publish-options implementation: Goffi <goffi@goffi.org> parents: 3040 diff changeset	146 return dict(args.fields)
cea52c9ddfd9 plugin XEP-0060, jp (pubsub/set): publish-options implementation: Goffi <goffi@goffi.org> parents: 3040 diff changeset	147
3040 fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	148 async def start(self):
3100 cea52c9ddfd9 plugin XEP-0060, jp (pubsub/set): publish-options implementation: Goffi <goffi@goffi.org> parents: 3040 diff changeset	149 options = self.get_config_options(self.args)
3040 fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	150 try:
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	151 node_id = await self.host.bridge.psNodeCreate(
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	152 self.args.service,
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	153 self.args.node,
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	154 options,
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	155 self.profile,
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	156 )
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	157 except Exception as e:
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	158 self.disp(msg=_(f"can't create node: {e}"), error=True)
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	159 self.host.quit(C.EXIT_BRIDGE_ERRBACK)
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	160 else:
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	161 if self.host.verbosity:
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	162 announce = _("node created successfully: ")
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	163 else:
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	164 announce = ""
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	165 self.disp(announce + node_id)
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	166 self.host.quit()
2221 a6c9bc4d1de0 jp (pubsub/node): added create and delete commands Goffi <goffi@goffi.org> parents: 2214 diff changeset	167
a6c9bc4d1de0 jp (pubsub/node): added create and delete commands Goffi <goffi@goffi.org> parents: 2214 diff changeset	168
2802 f61a50790fae jp (pubsub/node): added purge command Goffi <goffi@goffi.org> parents: 2777 diff changeset	169 class NodePurge(base.CommandBase):
f61a50790fae jp (pubsub/node): added purge command Goffi <goffi@goffi.org> parents: 2777 diff changeset	170
f61a50790fae jp (pubsub/node): added purge command Goffi <goffi@goffi.org> parents: 2777 diff changeset	171 def __init__(self, host):
f61a50790fae jp (pubsub/node): added purge command Goffi <goffi@goffi.org> parents: 2777 diff changeset	172 super(NodePurge, self).__init__(
f61a50790fae jp (pubsub/node): added purge command Goffi <goffi@goffi.org> parents: 2777 diff changeset	173 host,
f61a50790fae jp (pubsub/node): added purge command Goffi <goffi@goffi.org> parents: 2777 diff changeset	174 "purge",
f61a50790fae jp (pubsub/node): added purge command Goffi <goffi@goffi.org> parents: 2777 diff changeset	175 use_pubsub=True,
f61a50790fae jp (pubsub/node): added purge command Goffi <goffi@goffi.org> parents: 2777 diff changeset	176 pubsub_flags={C.NODE},
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2910 diff changeset	177 help=_("purge a node (i.e. remove all items from it)"),
2802 f61a50790fae jp (pubsub/node): added purge command Goffi <goffi@goffi.org> parents: 2777 diff changeset	178 )
f61a50790fae jp (pubsub/node): added purge command Goffi <goffi@goffi.org> parents: 2777 diff changeset	179
f61a50790fae jp (pubsub/node): added purge command Goffi <goffi@goffi.org> parents: 2777 diff changeset	180 def add_parser_options(self):
f61a50790fae jp (pubsub/node): added purge command Goffi <goffi@goffi.org> parents: 2777 diff changeset	181 self.parser.add_argument(
f61a50790fae jp (pubsub/node): added purge command Goffi <goffi@goffi.org> parents: 2777 diff changeset	182 "-f",
f61a50790fae jp (pubsub/node): added purge command Goffi <goffi@goffi.org> parents: 2777 diff changeset	183 "--force",
f61a50790fae jp (pubsub/node): added purge command Goffi <goffi@goffi.org> parents: 2777 diff changeset	184 action="store_true",
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2910 diff changeset	185 help=_("purge node without confirmation"),
2802 f61a50790fae jp (pubsub/node): added purge command Goffi <goffi@goffi.org> parents: 2777 diff changeset	186 )
f61a50790fae jp (pubsub/node): added purge command Goffi <goffi@goffi.org> parents: 2777 diff changeset	187
3040 fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	188 async def start(self):
2802 f61a50790fae jp (pubsub/node): added purge command Goffi <goffi@goffi.org> parents: 2777 diff changeset	189 if not self.args.force:
f61a50790fae jp (pubsub/node): added purge command Goffi <goffi@goffi.org> parents: 2777 diff changeset	190 if not self.args.service:
3040 fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	191 message = _(
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	192 f"Are you sure to purge PEP node [{self.args.node}]? This will "
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	193 f"delete ALL items from it!")
2802 f61a50790fae jp (pubsub/node): added purge command Goffi <goffi@goffi.org> parents: 2777 diff changeset	194 else:
f61a50790fae jp (pubsub/node): added purge command Goffi <goffi@goffi.org> parents: 2777 diff changeset	195 message = _(
3040 fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	196 f"Are you sure to delete node [{self.args.node}] on service "
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	197 f"[{self.args.service}]? This will delete ALL items from it!")
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	198 await self.host.confirmOrQuit(message, _("node purge cancelled"))
2802 f61a50790fae jp (pubsub/node): added purge command Goffi <goffi@goffi.org> parents: 2777 diff changeset	199
3040 fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	200 try:
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	201 await self.host.bridge.psNodePurge(
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	202 self.args.service,
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	203 self.args.node,
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	204 self.profile,
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	205 )
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	206 except Exception as e:
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	207 self.disp(msg=_(f"can't purge node: {e}"), error=True)
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	208 self.host.quit(C.EXIT_BRIDGE_ERRBACK)
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	209 else:
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	210 self.disp(_(f"node [{self.args.node}] purged successfully"))
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	211 self.host.quit()
2802 f61a50790fae jp (pubsub/node): added purge command Goffi <goffi@goffi.org> parents: 2777 diff changeset	212
f61a50790fae jp (pubsub/node): added purge command Goffi <goffi@goffi.org> parents: 2777 diff changeset	213
2235 4db836386641 jp: added use_pubsub and use_pubsub_node_req arguments to CommandBase Goffi <goffi@goffi.org> parents: 2224 diff changeset	214 class NodeDelete(base.CommandBase):
2221 a6c9bc4d1de0 jp (pubsub/node): added create and delete commands Goffi <goffi@goffi.org> parents: 2214 diff changeset	215 def __init__(self, host):
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	216 base.CommandBase.__init__(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	217 self,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	218 host,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	219 "delete",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	220 use_pubsub=True,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	221 pubsub_flags={C.NODE},
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2910 diff changeset	222 help=_("delete a node"),
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	223 )
2221 a6c9bc4d1de0 jp (pubsub/node): added create and delete commands Goffi <goffi@goffi.org> parents: 2214 diff changeset	224
a6c9bc4d1de0 jp (pubsub/node): added create and delete commands Goffi <goffi@goffi.org> parents: 2214 diff changeset	225 def add_parser_options(self):
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	226 self.parser.add_argument(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	227 "-f",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	228 "--force",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	229 action="store_true",
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2910 diff changeset	230 help=_("delete node without confirmation"),
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	231 )
2221 a6c9bc4d1de0 jp (pubsub/node): added create and delete commands Goffi <goffi@goffi.org> parents: 2214 diff changeset	232
3040 fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	233 async def start(self):
2221 a6c9bc4d1de0 jp (pubsub/node): added create and delete commands Goffi <goffi@goffi.org> parents: 2214 diff changeset	234 if not self.args.force:
a6c9bc4d1de0 jp (pubsub/node): added create and delete commands Goffi <goffi@goffi.org> parents: 2214 diff changeset	235 if not self.args.service:
3040 fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	236 message = _(f"Are you sure to delete PEP node [{self.args.node}] ?")
2221 a6c9bc4d1de0 jp (pubsub/node): added create and delete commands Goffi <goffi@goffi.org> parents: 2214 diff changeset	237 else:
3040 fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	238 message = _(f"Are you sure to delete node [{self.args.node}] on "
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	239 f"service [{self.args.service}]?")
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	240 await self.host.confirmOrQuit(message, _("node deletion cancelled"))
2221 a6c9bc4d1de0 jp (pubsub/node): added create and delete commands Goffi <goffi@goffi.org> parents: 2214 diff changeset	241
3040 fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	242 try:
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	243 await self.host.bridge.psNodeDelete(
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	244 self.args.service,
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	245 self.args.node,
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	246 self.profile,
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	247 )
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	248 except Exception as e:
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	249 self.disp(f"can't delete node: {e}", error=True)
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	250 self.host.quit(C.EXIT_BRIDGE_ERRBACK)
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	251 else:
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	252 self.disp(_(f"node [{self.args.node}] deleted successfully"))
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	253 self.host.quit()
2221 a6c9bc4d1de0 jp (pubsub/node): added create and delete commands Goffi <goffi@goffi.org> parents: 2214 diff changeset	254
a6c9bc4d1de0 jp (pubsub/node): added create and delete commands Goffi <goffi@goffi.org> parents: 2214 diff changeset	255
2235 4db836386641 jp: added use_pubsub and use_pubsub_node_req arguments to CommandBase Goffi <goffi@goffi.org> parents: 2224 diff changeset	256 class NodeSet(base.CommandBase):
2199 ea0d0a4e2ad8 jp (pubsub/node): added set command Goffi <goffi@goffi.org> parents: 2197 diff changeset	257 def __init__(self, host):
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	258 base.CommandBase.__init__(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	259 self,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	260 host,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	261 "set",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	262 use_output=C.OUTPUT_DICT,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	263 use_pubsub=True,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	264 pubsub_flags={C.NODE},
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	265 use_verbose=True,
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2910 diff changeset	266 help=_("set node configuration"),
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	267 )
2199 ea0d0a4e2ad8 jp (pubsub/node): added set command Goffi <goffi@goffi.org> parents: 2197 diff changeset	268
ea0d0a4e2ad8 jp (pubsub/node): added set command Goffi <goffi@goffi.org> parents: 2197 diff changeset	269 def add_parser_options(self):
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	270 self.parser.add_argument(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	271 "-f",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	272 "--field",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	273 action="append",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	274 nargs=2,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	275 dest="fields",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	276 required=True,
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2910 diff changeset	277 metavar=("KEY", "VALUE"),
ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2910 diff changeset	278 help=_("configuration field to set (required)"),
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	279 )
3040 fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	280 self.parser.add_argument(
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	281 "-F",
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	282 "--full-prefix",
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	283 action="store_true",
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	284 help=_('don\'t prepend "pubsub#" prefix to field names'),
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	285 )
2199 ea0d0a4e2ad8 jp (pubsub/node): added set command Goffi <goffi@goffi.org> parents: 2197 diff changeset	286
ea0d0a4e2ad8 jp (pubsub/node): added set command Goffi <goffi@goffi.org> parents: 2197 diff changeset	287 def getKeyName(self, k):
3040 fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	288 if self.args.full_prefix or k.startswith("pubsub#"):
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	289 return k
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	290 else:
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2910 diff changeset	291 return "pubsub#" + k
2199 ea0d0a4e2ad8 jp (pubsub/node): added set command Goffi <goffi@goffi.org> parents: 2197 diff changeset	292
3040 fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	293 async def start(self):
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	294 try:
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	295 await self.host.bridge.psNodeConfigurationSet(
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	296 self.args.service,
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	297 self.args.node,
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	298 {self.getKeyName(k): v for k, v in self.args.fields},
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	299 self.profile,
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	300 )
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	301 except Exception as e:
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	302 self.disp(f"can't set node configuration: {e}", error=True)
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	303 self.host.quit(C.EXIT_BRIDGE_ERRBACK)
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	304 else:
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	305 self.disp(_("node configuration successful"), 1)
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	306 self.host.quit()
2199 ea0d0a4e2ad8 jp (pubsub/node): added set command Goffi <goffi@goffi.org> parents: 2197 diff changeset	307
2197 e0e06391ce91 jp (pubsub): added pubsub/node/info command to retrieve node configuration Goffi <goffi@goffi.org> parents: 2195 diff changeset	308
2804 710de41da2f2 jp (pubsub/node): new "import" command, to publish many nodes from an XML file Goffi <goffi@goffi.org> parents: 2803 diff changeset	309 class NodeImport(base.CommandBase):
710de41da2f2 jp (pubsub/node): new "import" command, to publish many nodes from an XML file Goffi <goffi@goffi.org> parents: 2803 diff changeset	310
710de41da2f2 jp (pubsub/node): new "import" command, to publish many nodes from an XML file Goffi <goffi@goffi.org> parents: 2803 diff changeset	311 def __init__(self, host):
710de41da2f2 jp (pubsub/node): new "import" command, to publish many nodes from an XML file Goffi <goffi@goffi.org> parents: 2803 diff changeset	312 super(NodeImport, self).__init__(
710de41da2f2 jp (pubsub/node): new "import" command, to publish many nodes from an XML file Goffi <goffi@goffi.org> parents: 2803 diff changeset	313 host,
710de41da2f2 jp (pubsub/node): new "import" command, to publish many nodes from an XML file Goffi <goffi@goffi.org> parents: 2803 diff changeset	314 "import",
710de41da2f2 jp (pubsub/node): new "import" command, to publish many nodes from an XML file Goffi <goffi@goffi.org> parents: 2803 diff changeset	315 use_pubsub=True,
710de41da2f2 jp (pubsub/node): new "import" command, to publish many nodes from an XML file Goffi <goffi@goffi.org> parents: 2803 diff changeset	316 pubsub_flags={C.NODE},
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2910 diff changeset	317 help=_("import raw XML to a node"),
2804 710de41da2f2 jp (pubsub/node): new "import" command, to publish many nodes from an XML file Goffi <goffi@goffi.org> parents: 2803 diff changeset	318 )
710de41da2f2 jp (pubsub/node): new "import" command, to publish many nodes from an XML file Goffi <goffi@goffi.org> parents: 2803 diff changeset	319
710de41da2f2 jp (pubsub/node): new "import" command, to publish many nodes from an XML file Goffi <goffi@goffi.org> parents: 2803 diff changeset	320 def add_parser_options(self):
710de41da2f2 jp (pubsub/node): new "import" command, to publish many nodes from an XML file Goffi <goffi@goffi.org> parents: 2803 diff changeset	321 self.parser.add_argument(
710de41da2f2 jp (pubsub/node): new "import" command, to publish many nodes from an XML file Goffi <goffi@goffi.org> parents: 2803 diff changeset	322 "--admin",
710de41da2f2 jp (pubsub/node): new "import" command, to publish many nodes from an XML file Goffi <goffi@goffi.org> parents: 2803 diff changeset	323 action="store_true",
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2910 diff changeset	324 help=_("do a pubsub admin request, needed to change publisher"),
2804 710de41da2f2 jp (pubsub/node): new "import" command, to publish many nodes from an XML file Goffi <goffi@goffi.org> parents: 2803 diff changeset	325 )
710de41da2f2 jp (pubsub/node): new "import" command, to publish many nodes from an XML file Goffi <goffi@goffi.org> parents: 2803 diff changeset	326 self.parser.add_argument(
710de41da2f2 jp (pubsub/node): new "import" command, to publish many nodes from an XML file Goffi <goffi@goffi.org> parents: 2803 diff changeset	327 "import_file",
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2910 diff changeset	328 type=argparse.FileType(),
ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2910 diff changeset	329 help=_("path to the XML file with data to import. The file must contain "
ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2910 diff changeset	330 "whole XML of each item to import."),
2804 710de41da2f2 jp (pubsub/node): new "import" command, to publish many nodes from an XML file Goffi <goffi@goffi.org> parents: 2803 diff changeset	331 )
710de41da2f2 jp (pubsub/node): new "import" command, to publish many nodes from an XML file Goffi <goffi@goffi.org> parents: 2803 diff changeset	332
3040 fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	333 async def start(self):
2804 710de41da2f2 jp (pubsub/node): new "import" command, to publish many nodes from an XML file Goffi <goffi@goffi.org> parents: 2803 diff changeset	334 try:
710de41da2f2 jp (pubsub/node): new "import" command, to publish many nodes from an XML file Goffi <goffi@goffi.org> parents: 2803 diff changeset	335 element, etree = xml_tools.etreeParse(self, self.args.import_file,
710de41da2f2 jp (pubsub/node): new "import" command, to publish many nodes from an XML file Goffi <goffi@goffi.org> parents: 2803 diff changeset	336 reraise=True)
710de41da2f2 jp (pubsub/node): new "import" command, to publish many nodes from an XML file Goffi <goffi@goffi.org> parents: 2803 diff changeset	337 except Exception as e:
710de41da2f2 jp (pubsub/node): new "import" command, to publish many nodes from an XML file Goffi <goffi@goffi.org> parents: 2803 diff changeset	338 from lxml.etree import XMLSyntaxError
710de41da2f2 jp (pubsub/node): new "import" command, to publish many nodes from an XML file Goffi <goffi@goffi.org> parents: 2803 diff changeset	339 if isinstance(e, XMLSyntaxError) and e.code == 5:
710de41da2f2 jp (pubsub/node): new "import" command, to publish many nodes from an XML file Goffi <goffi@goffi.org> parents: 2803 diff changeset	340 # we have extra content, this probaby means that item are not wrapped
710de41da2f2 jp (pubsub/node): new "import" command, to publish many nodes from an XML file Goffi <goffi@goffi.org> parents: 2803 diff changeset	341 # so we wrap them here and try again
710de41da2f2 jp (pubsub/node): new "import" command, to publish many nodes from an XML file Goffi <goffi@goffi.org> parents: 2803 diff changeset	342 self.args.import_file.seek(0)
710de41da2f2 jp (pubsub/node): new "import" command, to publish many nodes from an XML file Goffi <goffi@goffi.org> parents: 2803 diff changeset	343 xml_buf = "<import>" + self.args.import_file.read() + "</import>"
710de41da2f2 jp (pubsub/node): new "import" command, to publish many nodes from an XML file Goffi <goffi@goffi.org> parents: 2803 diff changeset	344 element, etree = xml_tools.etreeParse(self, xml_buf)
710de41da2f2 jp (pubsub/node): new "import" command, to publish many nodes from an XML file Goffi <goffi@goffi.org> parents: 2803 diff changeset	345
710de41da2f2 jp (pubsub/node): new "import" command, to publish many nodes from an XML file Goffi <goffi@goffi.org> parents: 2803 diff changeset	346 # we reverse element as we expect to have most recently published element first
710de41da2f2 jp (pubsub/node): new "import" command, to publish many nodes from an XML file Goffi <goffi@goffi.org> parents: 2803 diff changeset	347 # TODO: make this more explicit and add an option
710de41da2f2 jp (pubsub/node): new "import" command, to publish many nodes from an XML file Goffi <goffi@goffi.org> parents: 2803 diff changeset	348 element[:] = reversed(element)
710de41da2f2 jp (pubsub/node): new "import" command, to publish many nodes from an XML file Goffi <goffi@goffi.org> parents: 2803 diff changeset	349
710de41da2f2 jp (pubsub/node): new "import" command, to publish many nodes from an XML file Goffi <goffi@goffi.org> parents: 2803 diff changeset	350 if not all([i.tag == '{http://jabber.org/protocol/pubsub}item' for i in element]):
710de41da2f2 jp (pubsub/node): new "import" command, to publish many nodes from an XML file Goffi <goffi@goffi.org> parents: 2803 diff changeset	351 self.disp(
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2910 diff changeset	352 _("You are not using list of pubsub items, we can't import this file"),
2804 710de41da2f2 jp (pubsub/node): new "import" command, to publish many nodes from an XML file Goffi <goffi@goffi.org> parents: 2803 diff changeset	353 error=True)
710de41da2f2 jp (pubsub/node): new "import" command, to publish many nodes from an XML file Goffi <goffi@goffi.org> parents: 2803 diff changeset	354 self.host.quit(C.EXIT_DATA_ERROR)
3040 fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	355 return
2804 710de41da2f2 jp (pubsub/node): new "import" command, to publish many nodes from an XML file Goffi <goffi@goffi.org> parents: 2803 diff changeset	356
3040 fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	357 items = [etree.tostring(i, encoding="unicode") for i in element]
2804 710de41da2f2 jp (pubsub/node): new "import" command, to publish many nodes from an XML file Goffi <goffi@goffi.org> parents: 2803 diff changeset	358 if self.args.admin:
3040 fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	359 method = self.host.bridge.psAdminItemsSend
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	360 else:
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	361 self.disp(_("Items are imported without using admin mode, publisher can't "
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	362 "be changed"))
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	363 method = self.host.bridge.psItemsSend
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	364
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	365 try:
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	366 items_ids = await method(
2804 710de41da2f2 jp (pubsub/node): new "import" command, to publish many nodes from an XML file Goffi <goffi@goffi.org> parents: 2803 diff changeset	367 self.args.service,
710de41da2f2 jp (pubsub/node): new "import" command, to publish many nodes from an XML file Goffi <goffi@goffi.org> parents: 2803 diff changeset	368 self.args.node,
710de41da2f2 jp (pubsub/node): new "import" command, to publish many nodes from an XML file Goffi <goffi@goffi.org> parents: 2803 diff changeset	369 items,
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2910 diff changeset	370 "",
2804 710de41da2f2 jp (pubsub/node): new "import" command, to publish many nodes from an XML file Goffi <goffi@goffi.org> parents: 2803 diff changeset	371 self.profile,
710de41da2f2 jp (pubsub/node): new "import" command, to publish many nodes from an XML file Goffi <goffi@goffi.org> parents: 2803 diff changeset	372 )
3040 fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	373 except Exception as e:
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	374 self.disp(f"can't send items: {e}", error=True)
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	375 self.host.quit(C.EXIT_BRIDGE_ERRBACK)
2804 710de41da2f2 jp (pubsub/node): new "import" command, to publish many nodes from an XML file Goffi <goffi@goffi.org> parents: 2803 diff changeset	376 else:
3040 fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	377 if items_ids:
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	378 self.disp(_('items published with id(s) {items_ids}').format(
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	379 items_ids=', '.join(items_ids)))
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	380 else:
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	381 self.disp(_('items published'))
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	382 self.host.quit()
2804 710de41da2f2 jp (pubsub/node): new "import" command, to publish many nodes from an XML file Goffi <goffi@goffi.org> parents: 2803 diff changeset	383
710de41da2f2 jp (pubsub/node): new "import" command, to publish many nodes from an XML file Goffi <goffi@goffi.org> parents: 2803 diff changeset	384
2235 4db836386641 jp: added use_pubsub and use_pubsub_node_req arguments to CommandBase Goffi <goffi@goffi.org> parents: 2224 diff changeset	385 class NodeAffiliationsGet(base.CommandBase):
2204 afc703419186 jp (pubsub): added affiliations and node/affiliations/get: Goffi <goffi@goffi.org> parents: 2199 diff changeset	386 def __init__(self, host):
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	387 base.CommandBase.__init__(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	388 self,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	389 host,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	390 "get",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	391 use_output=C.OUTPUT_DICT,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	392 use_pubsub=True,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	393 pubsub_flags={C.NODE},
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2910 diff changeset	394 help=_("retrieve node affiliations (for node owner)"),
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	395 )
2204 afc703419186 jp (pubsub): added affiliations and node/affiliations/get: Goffi <goffi@goffi.org> parents: 2199 diff changeset	396
afc703419186 jp (pubsub): added affiliations and node/affiliations/get: Goffi <goffi@goffi.org> parents: 2199 diff changeset	397 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	398 pass
2204 afc703419186 jp (pubsub): added affiliations and node/affiliations/get: Goffi <goffi@goffi.org> parents: 2199 diff changeset	399
3040 fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	400 async def start(self):
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	401 try:
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	402 affiliations = await self.host.bridge.psNodeAffiliationsGet(
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	403 self.args.service,
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	404 self.args.node,
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	405 self.profile,
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	406 )
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	407 except Exception as e:
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	408 self.disp(f"can't get node affiliations: {e}", error=True)
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	409 self.host.quit(C.EXIT_BRIDGE_ERRBACK)
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	410 else:
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	411 await self.output(affiliations)
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	412 self.host.quit()
2204 afc703419186 jp (pubsub): added affiliations and node/affiliations/get: Goffi <goffi@goffi.org> parents: 2199 diff changeset	413
afc703419186 jp (pubsub): added affiliations and node/affiliations/get: Goffi <goffi@goffi.org> parents: 2199 diff changeset	414
2235 4db836386641 jp: added use_pubsub and use_pubsub_node_req arguments to CommandBase Goffi <goffi@goffi.org> parents: 2224 diff changeset	415 class NodeAffiliationsSet(base.CommandBase):
2207 d662bdd682b2 jp (pubsub/node) added pubsub/node/affiliations/set command Goffi <goffi@goffi.org> parents: 2204 diff changeset	416 def __init__(self, host):
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	417 base.CommandBase.__init__(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	418 self,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	419 host,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	420 "set",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	421 use_pubsub=True,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	422 pubsub_flags={C.NODE},
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	423 use_verbose=True,
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2910 diff changeset	424 help=_("set affiliations (for node owner)"),
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	425 )
2207 d662bdd682b2 jp (pubsub/node) added pubsub/node/affiliations/set command Goffi <goffi@goffi.org> parents: 2204 diff changeset	426
d662bdd682b2 jp (pubsub/node) added pubsub/node/affiliations/set command Goffi <goffi@goffi.org> parents: 2204 diff changeset	427 def add_parser_options(self):
d662bdd682b2 jp (pubsub/node) added pubsub/node/affiliations/set command Goffi <goffi@goffi.org> parents: 2204 diff changeset	428 # XXX: we use optional argument syntax for a required one because list of list of 2 elements
3325 7ebda4b54170 jp (file/share): added commands to manage affiliations and configuration + documentation Goffi <goffi@goffi.org> parents: 3312 diff changeset	429 # (used to construct dicts) don't work with positional arguments
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	430 self.parser.add_argument(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	431 "-a",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	432 "--affiliation",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	433 dest="affiliations",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	434 metavar=("JID", "AFFILIATION"),
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	435 required=True,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	436 action="append",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	437 nargs=2,
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2910 diff changeset	438 help=_("entity/affiliation couple(s)"),
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	439 )
2207 d662bdd682b2 jp (pubsub/node) added pubsub/node/affiliations/set command Goffi <goffi@goffi.org> parents: 2204 diff changeset	440
3040 fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	441 async def start(self):
2207 d662bdd682b2 jp (pubsub/node) added pubsub/node/affiliations/set command Goffi <goffi@goffi.org> parents: 2204 diff changeset	442 affiliations = dict(self.args.affiliations)
3040 fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	443 try:
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	444 await self.host.bridge.psNodeAffiliationsSet(
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	445 self.args.service,
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	446 self.args.node,
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	447 affiliations,
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	448 self.profile,
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	449 )
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	450 except Exception as e:
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	451 self.disp(f"can't set node affiliations: {e}", error=True)
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	452 self.host.quit(C.EXIT_BRIDGE_ERRBACK)
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	453 else:
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	454 self.disp(_("affiliations have been set"), 1)
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	455 self.host.quit()
2207 d662bdd682b2 jp (pubsub/node) added pubsub/node/affiliations/set command Goffi <goffi@goffi.org> parents: 2204 diff changeset	456
d662bdd682b2 jp (pubsub/node) added pubsub/node/affiliations/set command Goffi <goffi@goffi.org> parents: 2204 diff changeset	457
2204 afc703419186 jp (pubsub): added affiliations and node/affiliations/get: Goffi <goffi@goffi.org> parents: 2199 diff changeset	458 class NodeAffiliations(base.CommandBase):
2207 d662bdd682b2 jp (pubsub/node) added pubsub/node/affiliations/set command Goffi <goffi@goffi.org> parents: 2204 diff changeset	459 subcommands = (NodeAffiliationsGet, NodeAffiliationsSet)
2204 afc703419186 jp (pubsub): added affiliations and node/affiliations/get: Goffi <goffi@goffi.org> parents: 2199 diff changeset	460
afc703419186 jp (pubsub): added affiliations and node/affiliations/get: Goffi <goffi@goffi.org> parents: 2199 diff changeset	461 def __init__(self, host):
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	462 super(NodeAffiliations, self).__init__(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	463 host,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	464 "affiliations",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	465 use_profile=False,
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2910 diff changeset	466 help=_("set or retrieve node affiliations"),
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	467 )
2204 afc703419186 jp (pubsub): added affiliations and node/affiliations/get: Goffi <goffi@goffi.org> parents: 2199 diff changeset	468
afc703419186 jp (pubsub): added affiliations and node/affiliations/get: Goffi <goffi@goffi.org> parents: 2199 diff changeset	469
2339 d94e932be8b3 jp (pubsub/node): added subscriptions subcommand: Goffi <goffi@goffi.org> parents: 2320 diff changeset	470 class NodeSubscriptionsGet(base.CommandBase):
d94e932be8b3 jp (pubsub/node): added subscriptions subcommand: Goffi <goffi@goffi.org> parents: 2320 diff changeset	471 def __init__(self, host):
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	472 base.CommandBase.__init__(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	473 self,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	474 host,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	475 "get",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	476 use_output=C.OUTPUT_DICT,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	477 use_pubsub=True,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	478 pubsub_flags={C.NODE},
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2910 diff changeset	479 help=_("retrieve node subscriptions (for node owner)"),
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	480 )
2339 d94e932be8b3 jp (pubsub/node): added subscriptions subcommand: Goffi <goffi@goffi.org> parents: 2320 diff changeset	481
d94e932be8b3 jp (pubsub/node): added subscriptions subcommand: Goffi <goffi@goffi.org> parents: 2320 diff changeset	482 def add_parser_options(self):
d94e932be8b3 jp (pubsub/node): added subscriptions subcommand: Goffi <goffi@goffi.org> parents: 2320 diff changeset	483 pass
d94e932be8b3 jp (pubsub/node): added subscriptions subcommand: Goffi <goffi@goffi.org> parents: 2320 diff changeset	484
3040 fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	485 async def start(self):
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	486 try:
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	487 subscriptions = await self.host.bridge.psNodeSubscriptionsGet(
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	488 self.args.service,
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	489 self.args.node,
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	490 self.profile,
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	491 )
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	492 except Exception as e:
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	493 self.disp(f"can't get node subscriptions: {e}", error=True)
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	494 self.host.quit(C.EXIT_BRIDGE_ERRBACK)
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	495 else:
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	496 await self.output(subscriptions)
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	497 self.host.quit()
2339 d94e932be8b3 jp (pubsub/node): added subscriptions subcommand: Goffi <goffi@goffi.org> parents: 2320 diff changeset	498
d94e932be8b3 jp (pubsub/node): added subscriptions subcommand: Goffi <goffi@goffi.org> parents: 2320 diff changeset	499
d94e932be8b3 jp (pubsub/node): added subscriptions subcommand: Goffi <goffi@goffi.org> parents: 2320 diff changeset	500 class StoreSubscriptionAction(argparse.Action):
d94e932be8b3 jp (pubsub/node): added subscriptions subcommand: Goffi <goffi@goffi.org> parents: 2320 diff changeset	501 """Action which handle subscription parameter for owner
d94e932be8b3 jp (pubsub/node): added subscriptions subcommand: Goffi <goffi@goffi.org> parents: 2320 diff changeset	502
d94e932be8b3 jp (pubsub/node): added subscriptions subcommand: Goffi <goffi@goffi.org> parents: 2320 diff changeset	503 list is given by pairs: jid and subscription state
d94e932be8b3 jp (pubsub/node): added subscriptions subcommand: Goffi <goffi@goffi.org> parents: 2320 diff changeset	504 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	505 """
d94e932be8b3 jp (pubsub/node): added subscriptions subcommand: Goffi <goffi@goffi.org> parents: 2320 diff changeset	506
d94e932be8b3 jp (pubsub/node): added subscriptions subcommand: Goffi <goffi@goffi.org> parents: 2320 diff changeset	507 def __call__(self, parser, namespace, values, option_string):
d94e932be8b3 jp (pubsub/node): added subscriptions subcommand: Goffi <goffi@goffi.org> parents: 2320 diff changeset	508 dest_dict = getattr(namespace, self.dest)
d94e932be8b3 jp (pubsub/node): added subscriptions subcommand: Goffi <goffi@goffi.org> parents: 2320 diff changeset	509 while values:
d94e932be8b3 jp (pubsub/node): added subscriptions subcommand: Goffi <goffi@goffi.org> parents: 2320 diff changeset	510 jid_s = values.pop(0)
d94e932be8b3 jp (pubsub/node): added subscriptions subcommand: Goffi <goffi@goffi.org> parents: 2320 diff changeset	511 try:
d94e932be8b3 jp (pubsub/node): added subscriptions subcommand: Goffi <goffi@goffi.org> parents: 2320 diff changeset	512 subscription = values.pop(0)
d94e932be8b3 jp (pubsub/node): added subscriptions subcommand: Goffi <goffi@goffi.org> parents: 2320 diff changeset	513 except IndexError:
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	514 subscription = "subscribed"
2339 d94e932be8b3 jp (pubsub/node): added subscriptions subcommand: Goffi <goffi@goffi.org> parents: 2320 diff changeset	515 if subscription not in ALLOWED_SUBSCRIPTIONS_OWNER:
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	516 parser.error(
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2910 diff changeset	517 _("subscription must be one of {}").format(
ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2910 diff changeset	518 ", ".join(ALLOWED_SUBSCRIPTIONS_OWNER)
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	519 )
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	520 )
2339 d94e932be8b3 jp (pubsub/node): added subscriptions subcommand: Goffi <goffi@goffi.org> parents: 2320 diff changeset	521 dest_dict[jid_s] = subscription
d94e932be8b3 jp (pubsub/node): added subscriptions subcommand: Goffi <goffi@goffi.org> parents: 2320 diff changeset	522
d94e932be8b3 jp (pubsub/node): added subscriptions subcommand: Goffi <goffi@goffi.org> parents: 2320 diff changeset	523
d94e932be8b3 jp (pubsub/node): added subscriptions subcommand: Goffi <goffi@goffi.org> parents: 2320 diff changeset	524 class NodeSubscriptionsSet(base.CommandBase):
d94e932be8b3 jp (pubsub/node): added subscriptions subcommand: Goffi <goffi@goffi.org> parents: 2320 diff changeset	525 def __init__(self, host):
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	526 base.CommandBase.__init__(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	527 self,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	528 host,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	529 "set",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	530 use_pubsub=True,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	531 pubsub_flags={C.NODE},
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	532 use_verbose=True,
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2910 diff changeset	533 help=_("set/modify subscriptions (for node owner)"),
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	534 )
2339 d94e932be8b3 jp (pubsub/node): added subscriptions subcommand: Goffi <goffi@goffi.org> parents: 2320 diff changeset	535
d94e932be8b3 jp (pubsub/node): added subscriptions subcommand: Goffi <goffi@goffi.org> parents: 2320 diff changeset	536 def add_parser_options(self):
d94e932be8b3 jp (pubsub/node): added subscriptions subcommand: Goffi <goffi@goffi.org> parents: 2320 diff changeset	537 # 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	538 # (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	539 self.parser.add_argument(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	540 "-S",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	541 "--subscription",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	542 dest="subscriptions",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	543 default={},
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	544 nargs="+",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	545 metavar=("JID [SUSBSCRIPTION]"),
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	546 required=True,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	547 action=StoreSubscriptionAction,
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2910 diff changeset	548 help=_("entity/subscription couple(s)"),
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	549 )
2339 d94e932be8b3 jp (pubsub/node): added subscriptions subcommand: Goffi <goffi@goffi.org> parents: 2320 diff changeset	550
3040 fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	551 async def start(self):
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	552 try:
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	553 self.host.bridge.psNodeSubscriptionsSet(
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	554 self.args.service,
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	555 self.args.node,
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	556 self.args.subscriptions,
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	557 self.profile,
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	558 )
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	559 except Exception as e:
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	560 self.disp(f"can't set node subscriptions: {e}", error=True)
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	561 self.host.quit(C.EXIT_BRIDGE_ERRBACK)
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	562 else:
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	563 self.disp(_("subscriptions have been set"), 1)
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	564 self.host.quit()
2339 d94e932be8b3 jp (pubsub/node): added subscriptions subcommand: Goffi <goffi@goffi.org> parents: 2320 diff changeset	565
d94e932be8b3 jp (pubsub/node): added subscriptions subcommand: Goffi <goffi@goffi.org> parents: 2320 diff changeset	566
d94e932be8b3 jp (pubsub/node): added subscriptions subcommand: Goffi <goffi@goffi.org> parents: 2320 diff changeset	567 class NodeSubscriptions(base.CommandBase):
d94e932be8b3 jp (pubsub/node): added subscriptions subcommand: Goffi <goffi@goffi.org> parents: 2320 diff changeset	568 subcommands = (NodeSubscriptionsGet, NodeSubscriptionsSet)
d94e932be8b3 jp (pubsub/node): added subscriptions subcommand: Goffi <goffi@goffi.org> parents: 2320 diff changeset	569
d94e932be8b3 jp (pubsub/node): added subscriptions subcommand: Goffi <goffi@goffi.org> parents: 2320 diff changeset	570 def __init__(self, host):
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	571 super(NodeSubscriptions, self).__init__(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	572 host,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	573 "subscriptions",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	574 use_profile=False,
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2910 diff changeset	575 help=_("get or modify node subscriptions"),
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	576 )
2339 d94e932be8b3 jp (pubsub/node): added subscriptions subcommand: Goffi <goffi@goffi.org> parents: 2320 diff changeset	577
d94e932be8b3 jp (pubsub/node): added subscriptions subcommand: Goffi <goffi@goffi.org> parents: 2320 diff changeset	578
2351 3c0a3fae1862 jp (pubsub/node): added schema (set/edit/get) commands to manipulate PubSub node schema Goffi <goffi@goffi.org> parents: 2339 diff changeset	579 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	580 def __init__(self, host):
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	581 base.CommandBase.__init__(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	582 self,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	583 host,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	584 "set",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	585 use_pubsub=True,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	586 pubsub_flags={C.NODE},
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	587 use_verbose=True,
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2910 diff changeset	588 help=_("set/replace a schema"),
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	589 )
2351 3c0a3fae1862 jp (pubsub/node): added schema (set/edit/get) commands to manipulate PubSub node schema Goffi <goffi@goffi.org> parents: 2339 diff changeset	590
3c0a3fae1862 jp (pubsub/node): added schema (set/edit/get) commands to manipulate PubSub node schema Goffi <goffi@goffi.org> parents: 2339 diff changeset	591 def add_parser_options(self):
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2910 diff changeset	592 self.parser.add_argument("schema", help=_("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	593
3040 fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	594 async def start(self):
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	595 try:
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	596 await self.host.bridge.psSchemaSet(
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	597 self.args.service,
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	598 self.args.node,
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	599 self.args.schema,
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	600 self.profile,
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	601 )
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	602 except Exception as e:
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	603 self.disp(f"can't set schema: {e}", error=True)
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	604 self.host.quit(C.EXIT_BRIDGE_ERRBACK)
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	605 else:
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	606 self.disp(_("schema has been set"), 1)
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	607 self.host.quit()
2351 3c0a3fae1862 jp (pubsub/node): added schema (set/edit/get) commands to manipulate PubSub node schema Goffi <goffi@goffi.org> parents: 2339 diff changeset	608
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 class NodeSchemaEdit(base.CommandBase, common.BaseEdit):
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	611 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	612
3c0a3fae1862 jp (pubsub/node): added schema (set/edit/get) commands to manipulate PubSub node schema Goffi <goffi@goffi.org> parents: 2339 diff changeset	613 def __init__(self, host):
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	614 base.CommandBase.__init__(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	615 self,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	616 host,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	617 "edit",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	618 use_pubsub=True,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	619 pubsub_flags={C.NODE},
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	620 use_draft=True,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	621 use_verbose=True,
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2910 diff changeset	622 help=_("edit a schema"),
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	623 )
2351 3c0a3fae1862 jp (pubsub/node): added schema (set/edit/get) commands to manipulate PubSub node schema Goffi <goffi@goffi.org> parents: 2339 diff changeset	624 common.BaseEdit.__init__(self, self.host, PUBSUB_SCHEMA_TMP_DIR)
3c0a3fae1862 jp (pubsub/node): added schema (set/edit/get) commands to manipulate PubSub node schema Goffi <goffi@goffi.org> parents: 2339 diff changeset	625
3c0a3fae1862 jp (pubsub/node): added schema (set/edit/get) commands to manipulate PubSub node schema Goffi <goffi@goffi.org> parents: 2339 diff changeset	626 def add_parser_options(self):
2532 772447ec070f jp: pubsub options refactoring: Goffi <goffi@goffi.org> parents: 2483 diff changeset	627 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	628
3040 fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	629 async def publish(self, schema):
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	630 try:
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	631 await self.host.bridge.psSchemaSet(
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	632 self.args.service,
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	633 self.args.node,
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	634 schema,
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	635 self.profile,
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	636 )
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	637 except Exception as e:
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	638 self.disp(f"can't set schema: {e}", error=True)
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	639 self.host.quit(C.EXIT_BRIDGE_ERRBACK)
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	640 else:
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	641 self.disp(_("schema has been set"), 1)
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	642 self.host.quit()
2351 3c0a3fae1862 jp (pubsub/node): added schema (set/edit/get) commands to manipulate PubSub node schema Goffi <goffi@goffi.org> parents: 2339 diff changeset	643
3040 fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	644 async def psSchemaGetCb(self, schema):
2351 3c0a3fae1862 jp (pubsub/node): added schema (set/edit/get) commands to manipulate PubSub node schema Goffi <goffi@goffi.org> parents: 2339 diff changeset	645 try:
3c0a3fae1862 jp (pubsub/node): added schema (set/edit/get) commands to manipulate PubSub node schema Goffi <goffi@goffi.org> parents: 2339 diff changeset	646 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	647 except ImportError:
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2910 diff changeset	648 self.disp('lxml module must be installed to use edit, please install it '
ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2910 diff changeset	649 'with "pip install lxml"',
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	650 error=True,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	651 )
2351 3c0a3fae1862 jp (pubsub/node): added schema (set/edit/get) commands to manipulate PubSub node schema Goffi <goffi@goffi.org> parents: 2339 diff changeset	652 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	653 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	654 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	655 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	656 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	657 schema_elt = etree.fromstring(schema, parser)
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	658 content_file_obj.write(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	659 etree.tostring(schema_elt, encoding="utf-8", pretty_print=True)
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	660 )
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	661 content_file_obj.seek(0)
3040 fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	662 await self.runEditor("pubsub_schema_editor_args", content_file_path, content_file_obj)
2351 3c0a3fae1862 jp (pubsub/node): added schema (set/edit/get) commands to manipulate PubSub node schema Goffi <goffi@goffi.org> parents: 2339 diff changeset	663
3040 fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	664 async def start(self):
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	665 try:
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	666 schema = await self.host.bridge.psSchemaGet(
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	667 self.args.service,
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	668 self.args.node,
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	669 self.profile,
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	670 )
3453 864485605d12 jp (pubsub/node/schema): handle not-found conditions properly Goffi <goffi@goffi.org> parents: 3413 diff changeset	671 except BridgeException as e:
864485605d12 jp (pubsub/node/schema): handle not-found conditions properly Goffi <goffi@goffi.org> parents: 3413 diff changeset	672 if e.condition == 'item-not-found' or e.classname=="NotFound":
864485605d12 jp (pubsub/node/schema): handle not-found conditions properly Goffi <goffi@goffi.org> parents: 3413 diff changeset	673 schema = ""
864485605d12 jp (pubsub/node/schema): handle not-found conditions properly Goffi <goffi@goffi.org> parents: 3413 diff changeset	674 else:
864485605d12 jp (pubsub/node/schema): handle not-found conditions properly Goffi <goffi@goffi.org> parents: 3413 diff changeset	675 self.disp(f"can't edit schema: {e}", error=True)
864485605d12 jp (pubsub/node/schema): handle not-found conditions properly Goffi <goffi@goffi.org> parents: 3413 diff changeset	676 self.host.quit(C.EXIT_BRIDGE_ERRBACK)
864485605d12 jp (pubsub/node/schema): handle not-found conditions properly Goffi <goffi@goffi.org> parents: 3413 diff changeset	677
864485605d12 jp (pubsub/node/schema): handle not-found conditions properly Goffi <goffi@goffi.org> parents: 3413 diff changeset	678 await self.psSchemaGetCb(schema)
2351 3c0a3fae1862 jp (pubsub/node): added schema (set/edit/get) commands to manipulate PubSub node schema Goffi <goffi@goffi.org> parents: 2339 diff changeset	679
3c0a3fae1862 jp (pubsub/node): added schema (set/edit/get) commands to manipulate PubSub node schema Goffi <goffi@goffi.org> parents: 2339 diff changeset	680
3c0a3fae1862 jp (pubsub/node): added schema (set/edit/get) commands to manipulate PubSub node schema Goffi <goffi@goffi.org> parents: 2339 diff changeset	681 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	682 def __init__(self, host):
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	683 base.CommandBase.__init__(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	684 self,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	685 host,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	686 "get",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	687 use_output=C.OUTPUT_XML,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	688 use_pubsub=True,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	689 pubsub_flags={C.NODE},
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	690 use_verbose=True,
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2910 diff changeset	691 help=_("get schema"),
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	692 )
2351 3c0a3fae1862 jp (pubsub/node): added schema (set/edit/get) commands to manipulate PubSub node schema Goffi <goffi@goffi.org> parents: 2339 diff changeset	693
3c0a3fae1862 jp (pubsub/node): added schema (set/edit/get) commands to manipulate PubSub node schema Goffi <goffi@goffi.org> parents: 2339 diff changeset	694 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	695 pass
3c0a3fae1862 jp (pubsub/node): added schema (set/edit/get) commands to manipulate PubSub node schema Goffi <goffi@goffi.org> parents: 2339 diff changeset	696
3040 fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	697 async def start(self):
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	698 try:
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	699 schema = await self.host.bridge.psSchemaGet(
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	700 self.args.service,
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	701 self.args.node,
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	702 self.profile,
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	703 )
3453 864485605d12 jp (pubsub/node/schema): handle not-found conditions properly Goffi <goffi@goffi.org> parents: 3413 diff changeset	704 except BridgeException as e:
864485605d12 jp (pubsub/node/schema): handle not-found conditions properly Goffi <goffi@goffi.org> parents: 3413 diff changeset	705 if e.condition == 'item-not-found' or e.classname=="NotFound":
864485605d12 jp (pubsub/node/schema): handle not-found conditions properly Goffi <goffi@goffi.org> parents: 3413 diff changeset	706 schema = None
864485605d12 jp (pubsub/node/schema): handle not-found conditions properly Goffi <goffi@goffi.org> parents: 3413 diff changeset	707 else:
864485605d12 jp (pubsub/node/schema): handle not-found conditions properly Goffi <goffi@goffi.org> parents: 3413 diff changeset	708 self.disp(f"can't get schema: {e}", error=True)
864485605d12 jp (pubsub/node/schema): handle not-found conditions properly Goffi <goffi@goffi.org> parents: 3413 diff changeset	709 self.host.quit(C.EXIT_BRIDGE_ERRBACK)
864485605d12 jp (pubsub/node/schema): handle not-found conditions properly Goffi <goffi@goffi.org> parents: 3413 diff changeset	710
864485605d12 jp (pubsub/node/schema): handle not-found conditions properly Goffi <goffi@goffi.org> parents: 3413 diff changeset	711 if schema:
864485605d12 jp (pubsub/node/schema): handle not-found conditions properly Goffi <goffi@goffi.org> parents: 3413 diff changeset	712 await self.output(schema)
864485605d12 jp (pubsub/node/schema): handle not-found conditions properly Goffi <goffi@goffi.org> parents: 3413 diff changeset	713 self.host.quit()
3040 fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	714 else:
3453 864485605d12 jp (pubsub/node/schema): handle not-found conditions properly Goffi <goffi@goffi.org> parents: 3413 diff changeset	715 self.disp(_("no schema found"), 1)
864485605d12 jp (pubsub/node/schema): handle not-found conditions properly Goffi <goffi@goffi.org> parents: 3413 diff changeset	716 self.host.quit(C.EXIT_NOT_FOUND)
2351 3c0a3fae1862 jp (pubsub/node): added schema (set/edit/get) commands to manipulate PubSub node schema Goffi <goffi@goffi.org> parents: 2339 diff changeset	717
3c0a3fae1862 jp (pubsub/node): added schema (set/edit/get) commands to manipulate PubSub node schema Goffi <goffi@goffi.org> parents: 2339 diff changeset	718
3c0a3fae1862 jp (pubsub/node): added schema (set/edit/get) commands to manipulate PubSub node schema Goffi <goffi@goffi.org> parents: 2339 diff changeset	719 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	720 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	721
3c0a3fae1862 jp (pubsub/node): added schema (set/edit/get) commands to manipulate PubSub node schema Goffi <goffi@goffi.org> parents: 2339 diff changeset	722 def __init__(self, host):
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	723 super(NodeSchema, self).__init__(
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2910 diff changeset	724 host, "schema", use_profile=False, help=_("data schema manipulation")
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	725 )
2351 3c0a3fae1862 jp (pubsub/node): added schema (set/edit/get) commands to manipulate PubSub node schema Goffi <goffi@goffi.org> parents: 2339 diff changeset	726
3c0a3fae1862 jp (pubsub/node): added schema (set/edit/get) commands to manipulate PubSub node schema Goffi <goffi@goffi.org> parents: 2339 diff changeset	727
2197 e0e06391ce91 jp (pubsub): added pubsub/node/info command to retrieve node configuration Goffi <goffi@goffi.org> parents: 2195 diff changeset	728 class Node(base.CommandBase):
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	729 subcommands = (
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	730 NodeInfo,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	731 NodeCreate,
2802 f61a50790fae jp (pubsub/node): added purge command Goffi <goffi@goffi.org> parents: 2777 diff changeset	732 NodePurge,
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	733 NodeDelete,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	734 NodeSet,
2804 710de41da2f2 jp (pubsub/node): new "import" command, to publish many nodes from an XML file Goffi <goffi@goffi.org> parents: 2803 diff changeset	735 NodeImport,
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	736 NodeAffiliations,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	737 NodeSubscriptions,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	738 NodeSchema,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	739 )
2197 e0e06391ce91 jp (pubsub): added pubsub/node/info command to retrieve node configuration Goffi <goffi@goffi.org> parents: 2195 diff changeset	740
e0e06391ce91 jp (pubsub): added pubsub/node/info command to retrieve node configuration Goffi <goffi@goffi.org> parents: 2195 diff changeset	741 def __init__(self, host):
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	742 super(Node, self).__init__(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	743 host, "node", use_profile=False, help=_("node handling")
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	744 )
2197 e0e06391ce91 jp (pubsub): added pubsub/node/info command to retrieve node configuration Goffi <goffi@goffi.org> parents: 2195 diff changeset	745
e0e06391ce91 jp (pubsub): added pubsub/node/info command to retrieve node configuration Goffi <goffi@goffi.org> parents: 2195 diff changeset	746
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	747 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	748 def __init__(self, host):
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	749 base.CommandBase.__init__(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	750 self,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	751 host,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	752 "set",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	753 use_pubsub=True,
3412 7b4ae3dbc041 jp (base, pubsub/set): new `--quiet` argument: Goffi <goffi@goffi.org> parents: 3411 diff changeset	754 use_quiet=True,
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	755 pubsub_flags={C.NODE},
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2910 diff changeset	756 help=_("publish a new item or update an existing one"),
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	757 )
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	758
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	759 def add_parser_options(self):
3100 cea52c9ddfd9 plugin XEP-0060, jp (pubsub/set): publish-options implementation: Goffi <goffi@goffi.org> parents: 3040 diff changeset	760 NodeCreate.add_node_config_options(self.parser)
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	761 self.parser.add_argument(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	762 "item",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	763 nargs="?",
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2910 diff changeset	764 default="",
ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2910 diff changeset	765 help=_("id, URL of the item to update, keyword, or nothing for new item"),
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	766 )
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	767
3040 fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	768 async def start(self):
2777 ff1b40823b07 jp (pubsub): new "transform" command: Goffi <goffi@goffi.org> parents: 2773 diff changeset	769 element, etree = xml_tools.etreeParse(self, sys.stdin)
ff1b40823b07 jp (pubsub): new "transform" command: Goffi <goffi@goffi.org> parents: 2773 diff changeset	770 element = xml_tools.getPayload(self, element)
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	771 payload = etree.tostring(element, encoding="unicode")
3100 cea52c9ddfd9 plugin XEP-0060, jp (pubsub/set): publish-options implementation: Goffi <goffi@goffi.org> parents: 3040 diff changeset	772 extra = {}
cea52c9ddfd9 plugin XEP-0060, jp (pubsub/set): publish-options implementation: Goffi <goffi@goffi.org> parents: 3040 diff changeset	773 publish_options = NodeCreate.get_config_options(self.args)
cea52c9ddfd9 plugin XEP-0060, jp (pubsub/set): publish-options implementation: Goffi <goffi@goffi.org> parents: 3040 diff changeset	774 if publish_options:
cea52c9ddfd9 plugin XEP-0060, jp (pubsub/set): publish-options implementation: Goffi <goffi@goffi.org> parents: 3040 diff changeset	775 extra['publish_options'] = publish_options
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	776
3040 fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	777 try:
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	778 published_id = await self.host.bridge.psItemSend(
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	779 self.args.service,
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	780 self.args.node,
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	781 payload,
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	782 self.args.item,
3100 cea52c9ddfd9 plugin XEP-0060, jp (pubsub/set): publish-options implementation: Goffi <goffi@goffi.org> parents: 3040 diff changeset	783 data_format.serialise(extra),
3040 fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	784 self.profile,
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	785 )
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	786 except Exception as e:
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	787 self.disp(_(f"can't send item: {e}"), error=True)
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	788 self.host.quit(C.EXIT_BRIDGE_ERRBACK)
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	789 else:
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	790 if published_id:
3412 7b4ae3dbc041 jp (base, pubsub/set): new `--quiet` argument: Goffi <goffi@goffi.org> parents: 3411 diff changeset	791 if self.args.quiet:
7b4ae3dbc041 jp (base, pubsub/set): new `--quiet` argument: Goffi <goffi@goffi.org> parents: 3411 diff changeset	792 self.disp(published_id, end='')
7b4ae3dbc041 jp (base, pubsub/set): new `--quiet` argument: Goffi <goffi@goffi.org> parents: 3411 diff changeset	793 else:
7b4ae3dbc041 jp (base, pubsub/set): new `--quiet` argument: Goffi <goffi@goffi.org> parents: 3411 diff changeset	794 self.disp(f"Item published at {published_id}")
3040 fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	795 else:
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	796 self.disp("Item published")
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	797 self.host.quit(C.EXIT_OK)
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	798
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	799
2235 4db836386641 jp: added use_pubsub and use_pubsub_node_req arguments to CommandBase Goffi <goffi@goffi.org> parents: 2224 diff changeset	800 class Get(base.CommandBase):
2191 a1a8233f89e8 jp(pubsub/get): pubsub/get command, first draft Goffi <goffi@goffi.org> parents: diff changeset	801 def __init__(self, host):
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	802 base.CommandBase.__init__(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	803 self,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	804 host,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	805 "get",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	806 use_output=C.OUTPUT_LIST_XML,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	807 use_pubsub=True,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	808 pubsub_flags={C.NODE, C.MULTI_ITEMS},
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2910 diff changeset	809 help=_("get pubsub item(s)"),
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	810 )
2191 a1a8233f89e8 jp(pubsub/get): pubsub/get command, first draft Goffi <goffi@goffi.org> parents: diff changeset	811
a1a8233f89e8 jp(pubsub/get): pubsub/get command, first draft Goffi <goffi@goffi.org> parents: diff changeset	812 def add_parser_options(self):
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	813 self.parser.add_argument(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	814 "-S",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	815 "--sub-id",
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2910 diff changeset	816 default="",
ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2910 diff changeset	817 help=_("subscription id"),
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	818 )
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	819 # 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	820 # TODO: add MAM filters
a1a8233f89e8 jp(pubsub/get): pubsub/get command, first draft Goffi <goffi@goffi.org> parents: diff changeset	821
3040 fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	822 async def start(self):
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	823 try:
3312 77177b13ff54 plugin XEP-0060: serialise psItemsGet result with data_format Goffi <goffi@goffi.org> parents: 3275 diff changeset	824 ps_result = data_format.deserialise(
77177b13ff54 plugin XEP-0060: serialise psItemsGet result with data_format Goffi <goffi@goffi.org> parents: 3275 diff changeset	825 await self.host.bridge.psItemsGet(
77177b13ff54 plugin XEP-0060: serialise psItemsGet result with data_format Goffi <goffi@goffi.org> parents: 3275 diff changeset	826 self.args.service,
77177b13ff54 plugin XEP-0060: serialise psItemsGet result with data_format Goffi <goffi@goffi.org> parents: 3275 diff changeset	827 self.args.node,
77177b13ff54 plugin XEP-0060: serialise psItemsGet result with data_format Goffi <goffi@goffi.org> parents: 3275 diff changeset	828 self.args.max,
77177b13ff54 plugin XEP-0060: serialise psItemsGet result with data_format Goffi <goffi@goffi.org> parents: 3275 diff changeset	829 self.args.items,
77177b13ff54 plugin XEP-0060: serialise psItemsGet result with data_format Goffi <goffi@goffi.org> parents: 3275 diff changeset	830 self.args.sub_id,
77177b13ff54 plugin XEP-0060: serialise psItemsGet result with data_format Goffi <goffi@goffi.org> parents: 3275 diff changeset	831 self.getPubsubExtra(),
77177b13ff54 plugin XEP-0060: serialise psItemsGet result with data_format Goffi <goffi@goffi.org> parents: 3275 diff changeset	832 self.profile,
77177b13ff54 plugin XEP-0060: serialise psItemsGet result with data_format Goffi <goffi@goffi.org> parents: 3275 diff changeset	833 )
3040 fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	834 )
3413 f4d417593e77 jp (pubsub/node/info, pubsub/get): better exit code: Goffi <goffi@goffi.org> parents: 3412 diff changeset	835 except BridgeException as e:
f4d417593e77 jp (pubsub/node/info, pubsub/get): better exit code: Goffi <goffi@goffi.org> parents: 3412 diff changeset	836 if e.condition == 'item-not-found' or e.classname=="NotFound":
f4d417593e77 jp (pubsub/node/info, pubsub/get): better exit code: Goffi <goffi@goffi.org> parents: 3412 diff changeset	837 self.disp(
f4d417593e77 jp (pubsub/node/info, pubsub/get): better exit code: Goffi <goffi@goffi.org> parents: 3412 diff changeset	838 f"The node {self.args.node} doesn't exist on {self.args.service}",
f4d417593e77 jp (pubsub/node/info, pubsub/get): better exit code: Goffi <goffi@goffi.org> parents: 3412 diff changeset	839 error=True
f4d417593e77 jp (pubsub/node/info, pubsub/get): better exit code: Goffi <goffi@goffi.org> parents: 3412 diff changeset	840 )
f4d417593e77 jp (pubsub/node/info, pubsub/get): better exit code: Goffi <goffi@goffi.org> parents: 3412 diff changeset	841 self.host.quit(C.EXIT_NOT_FOUND)
f4d417593e77 jp (pubsub/node/info, pubsub/get): better exit code: Goffi <goffi@goffi.org> parents: 3412 diff changeset	842 else:
f4d417593e77 jp (pubsub/node/info, pubsub/get): better exit code: Goffi <goffi@goffi.org> parents: 3412 diff changeset	843 self.disp(f"can't get pubsub items: {e}", error=True)
f4d417593e77 jp (pubsub/node/info, pubsub/get): better exit code: Goffi <goffi@goffi.org> parents: 3412 diff changeset	844 self.host.quit(C.EXIT_BRIDGE_ERRBACK)
3040 fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	845 except Exception as e:
3413 f4d417593e77 jp (pubsub/node/info, pubsub/get): better exit code: Goffi <goffi@goffi.org> parents: 3412 diff changeset	846 self.disp(f"Internal error: {e}", error=True)
f4d417593e77 jp (pubsub/node/info, pubsub/get): better exit code: Goffi <goffi@goffi.org> parents: 3412 diff changeset	847 self.host.quit(C.EXIT_INTERNAL_ERROR)
3040 fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	848 else:
3312 77177b13ff54 plugin XEP-0060: serialise psItemsGet result with data_format Goffi <goffi@goffi.org> parents: 3275 diff changeset	849 await self.output(ps_result['items'])
3040 fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	850 self.host.quit(C.EXIT_OK)
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	851
2191 a1a8233f89e8 jp(pubsub/get): pubsub/get command, first draft Goffi <goffi@goffi.org> parents: diff changeset	852
2281 4af1805cc6df jp (pubsub/delete): delete command implementation (to delete an item) Goffi <goffi@goffi.org> parents: 2280 diff changeset	853 class Delete(base.CommandBase):
4af1805cc6df jp (pubsub/delete): delete command implementation (to delete an item) Goffi <goffi@goffi.org> parents: 2280 diff changeset	854 def __init__(self, host):
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	855 base.CommandBase.__init__(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	856 self,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	857 host,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	858 "delete",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	859 use_pubsub=True,
2910 b2f323237fce jp, plugin merge-requests: used u'' as default for item id in pubsub arguments + fixed some required arguments: Goffi <goffi@goffi.org> parents: 2805 diff changeset	860 pubsub_flags={C.NODE, C.ITEM, C.SINGLE_ITEM},
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2910 diff changeset	861 help=_("delete an item"),
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	862 )
2281 4af1805cc6df jp (pubsub/delete): delete command implementation (to delete an item) Goffi <goffi@goffi.org> parents: 2280 diff changeset	863
4af1805cc6df jp (pubsub/delete): delete command implementation (to delete an item) Goffi <goffi@goffi.org> parents: 2280 diff changeset	864 def add_parser_options(self):
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	865 self.parser.add_argument(
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2910 diff changeset	866 "-f", "--force", action="store_true", help=_("delete without confirmation")
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	867 )
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	868 self.parser.add_argument(
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2910 diff changeset	869 "-N", "--notify", action="store_true", help=_("notify deletion")
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	870 )
2281 4af1805cc6df jp (pubsub/delete): delete command implementation (to delete an item) Goffi <goffi@goffi.org> parents: 2280 diff changeset	871
3040 fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	872 async def start(self):
2281 4af1805cc6df jp (pubsub/delete): delete command implementation (to delete an item) Goffi <goffi@goffi.org> parents: 2280 diff changeset	873 if not self.args.item:
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2910 diff changeset	874 self.parser.error(_("You need to specify an item to delete"))
2281 4af1805cc6df jp (pubsub/delete): delete command implementation (to delete an item) Goffi <goffi@goffi.org> parents: 2280 diff changeset	875 if not self.args.force:
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2910 diff changeset	876 message = _("Are you sure to delete item {item_id} ?").format(
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	877 item_id=self.args.item
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	878 )
3040 fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	879 await self.host.confirmOrQuit(message, _("item deletion cancelled"))
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	880 try:
3467 5901a7170528 plugin XEP-0060: renamed `psRetractItem(s)` to `psItem(s)Retract` for consistency. Goffi <goffi@goffi.org> parents: 3453 diff changeset	881 await self.host.bridge.psItemRetract(
3040 fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	882 self.args.service,
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	883 self.args.node,
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	884 self.args.item,
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	885 self.args.notify,
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	886 self.profile,
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	887 )
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	888 except Exception as e:
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	889 self.disp(_(f"can't delete item: {e}"), error=True)
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	890 self.host.quit(C.EXIT_BRIDGE_ERRBACK)
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	891 else:
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	892 self.disp(_(f"item {self.args.item} has been deleted"))
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	893 self.host.quit(C.EXIT_OK)
2281 4af1805cc6df jp (pubsub/delete): delete command implementation (to delete an item) Goffi <goffi@goffi.org> parents: 2280 diff changeset	894
2191 a1a8233f89e8 jp(pubsub/get): pubsub/get command, first draft Goffi <goffi@goffi.org> parents: diff changeset	895
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	896 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	897 def __init__(self, host):
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	898 base.CommandBase.__init__(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	899 self,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	900 host,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	901 "edit",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	902 use_verbose=True,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	903 use_pubsub=True,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	904 pubsub_flags={C.NODE, C.SINGLE_ITEM},
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	905 use_draft=True,
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2910 diff changeset	906 help=_("edit an existing or new pubsub item"),
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	907 )
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	908 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	909
64e99bf0dfa2 jp (pubsub/edit): new edit subcommand, which work in a same way as for blog Goffi <goffi@goffi.org> parents: 2274 diff changeset	910 def add_parser_options(self):
2532 772447ec070f jp: pubsub options refactoring: Goffi <goffi@goffi.org> parents: 2483 diff changeset	911 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	912
3040 fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	913 async def publish(self, content):
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	914 published_id = await self.host.bridge.psItemSend(
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	915 self.pubsub_service,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	916 self.pubsub_node,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	917 content,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	918 self.pubsub_item or "",
3100 cea52c9ddfd9 plugin XEP-0060, jp (pubsub/set): publish-options implementation: Goffi <goffi@goffi.org> parents: 3040 diff changeset	919 "",
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	920 self.profile,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	921 )
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	922 if published_id:
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2910 diff changeset	923 self.disp("Item published at {pub_id}".format(pub_id=published_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	924 else:
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2910 diff changeset	925 self.disp("Item published")
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	926
3040 fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	927 async def getItemData(self, service, node, item):
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	928 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	929 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	930 except ImportError:
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2910 diff changeset	931 self.disp('lxml module must be installed to use edit, please install it '
ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2910 diff changeset	932 'with "pip install lxml"',
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	933 error=True,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	934 )
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	935 self.host.quit(1)
2910 b2f323237fce jp, plugin merge-requests: used u'' as default for item id in pubsub arguments + fixed some required arguments: Goffi <goffi@goffi.org> parents: 2805 diff changeset	936 items = [item] if item else []
3312 77177b13ff54 plugin XEP-0060: serialise psItemsGet result with data_format Goffi <goffi@goffi.org> parents: 3275 diff changeset	937 ps_result = data_format.deserialise(
77177b13ff54 plugin XEP-0060: serialise psItemsGet result with data_format Goffi <goffi@goffi.org> parents: 3275 diff changeset	938 await self.host.bridge.psItemsGet(
77177b13ff54 plugin XEP-0060: serialise psItemsGet result with data_format Goffi <goffi@goffi.org> parents: 3275 diff changeset	939 service, node, 1, items, "", {}, self.profile
77177b13ff54 plugin XEP-0060: serialise psItemsGet result with data_format Goffi <goffi@goffi.org> parents: 3275 diff changeset	940 )
77177b13ff54 plugin XEP-0060: serialise psItemsGet result with data_format Goffi <goffi@goffi.org> parents: 3275 diff changeset	941 )
77177b13ff54 plugin XEP-0060: serialise psItemsGet result with data_format Goffi <goffi@goffi.org> parents: 3275 diff changeset	942 item_raw = ps_result['items'][0]
3040 fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	943 parser = etree.XMLParser(remove_blank_text=True, recover=True)
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	944 item_elt = etree.fromstring(item_raw, parser)
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	945 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	946 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	947 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	948 except IndexError:
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2910 diff changeset	949 self.disp(_("Item has not payload"), 1)
3275 55eeb0dfd313 jp (pubsub): fixed arguments returned when item has no payload Goffi <goffi@goffi.org> parents: 3137 diff changeset	950 return "", item_id
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	951 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	952
3040 fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	953 async def start(self):
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	954 (self.pubsub_service,
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	955 self.pubsub_node,
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	956 self.pubsub_item,
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	957 content_file_path,
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	958 content_file_obj) = await self.getItemPath()
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	959 await self.runEditor("pubsub_editor_args", content_file_path, content_file_obj)
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	960 self.host.quit()
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	961
64e99bf0dfa2 jp (pubsub/edit): new edit subcommand, which work in a same way as for blog Goffi <goffi@goffi.org> parents: 2274 diff changeset	962
2353 ebc0dfe9c0ca jp (pubsub): added subscribe, unsubscribe and subscriptions methods: Goffi <goffi@goffi.org> parents: 2351 diff changeset	963 class Subscribe(base.CommandBase):
ebc0dfe9c0ca jp (pubsub): added subscribe, unsubscribe and subscriptions methods: Goffi <goffi@goffi.org> parents: 2351 diff changeset	964 def __init__(self, host):
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	965 base.CommandBase.__init__(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	966 self,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	967 host,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	968 "subscribe",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	969 use_pubsub=True,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	970 pubsub_flags={C.NODE},
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	971 use_verbose=True,
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2910 diff changeset	972 help=_("subscribe to a node"),
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	973 )
2353 ebc0dfe9c0ca jp (pubsub): added subscribe, unsubscribe and subscriptions methods: Goffi <goffi@goffi.org> parents: 2351 diff changeset	974
ebc0dfe9c0ca jp (pubsub): added subscribe, unsubscribe and subscriptions methods: Goffi <goffi@goffi.org> parents: 2351 diff changeset	975 def add_parser_options(self):
ebc0dfe9c0ca jp (pubsub): added subscribe, unsubscribe and subscriptions methods: Goffi <goffi@goffi.org> parents: 2351 diff changeset	976 pass
ebc0dfe9c0ca jp (pubsub): added subscribe, unsubscribe and subscriptions methods: Goffi <goffi@goffi.org> parents: 2351 diff changeset	977
3040 fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	978 async def start(self):
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	979 try:
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	980 sub_id = await self.host.bridge.psSubscribe(
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	981 self.args.service,
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	982 self.args.node,
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	983 {},
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	984 self.profile,
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	985 )
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	986 except Exception as e:
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	987 self.disp(_(f"can't subscribe to node: {e}"), error=True)
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	988 self.host.quit(C.EXIT_BRIDGE_ERRBACK)
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	989 else:
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	990 self.disp(_("subscription done"), 1)
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	991 if sub_id:
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	992 self.disp(_("subscription id: {sub_id}").format(sub_id=sub_id))
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	993 self.host.quit()
2353 ebc0dfe9c0ca jp (pubsub): added subscribe, unsubscribe and subscriptions methods: Goffi <goffi@goffi.org> parents: 2351 diff changeset	994
ebc0dfe9c0ca jp (pubsub): added subscribe, unsubscribe and subscriptions methods: Goffi <goffi@goffi.org> parents: 2351 diff changeset	995
ebc0dfe9c0ca jp (pubsub): added subscribe, unsubscribe and subscriptions methods: Goffi <goffi@goffi.org> parents: 2351 diff changeset	996 class Unsubscribe(base.CommandBase):
3040 fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	997 # FIXME: check why we get a a NodeNotFound on subscribe just after unsubscribe
2353 ebc0dfe9c0ca jp (pubsub): added subscribe, unsubscribe and subscriptions methods: Goffi <goffi@goffi.org> parents: 2351 diff changeset	998
ebc0dfe9c0ca jp (pubsub): added subscribe, unsubscribe and subscriptions methods: Goffi <goffi@goffi.org> parents: 2351 diff changeset	999 def __init__(self, host):
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1000 base.CommandBase.__init__(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1001 self,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1002 host,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1003 "unsubscribe",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1004 use_pubsub=True,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1005 pubsub_flags={C.NODE},
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1006 use_verbose=True,
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2910 diff changeset	1007 help=_("unsubscribe from a node"),
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1008 )
2353 ebc0dfe9c0ca jp (pubsub): added subscribe, unsubscribe and subscriptions methods: Goffi <goffi@goffi.org> parents: 2351 diff changeset	1009
ebc0dfe9c0ca jp (pubsub): added subscribe, unsubscribe and subscriptions methods: Goffi <goffi@goffi.org> parents: 2351 diff changeset	1010 def add_parser_options(self):
ebc0dfe9c0ca jp (pubsub): added subscribe, unsubscribe and subscriptions methods: Goffi <goffi@goffi.org> parents: 2351 diff changeset	1011 pass
ebc0dfe9c0ca jp (pubsub): added subscribe, unsubscribe and subscriptions methods: Goffi <goffi@goffi.org> parents: 2351 diff changeset	1012
3040 fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	1013 async def start(self):
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	1014 try:
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	1015 await self.host.bridge.psUnsubscribe(
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	1016 self.args.service,
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	1017 self.args.node,
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	1018 self.profile,
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	1019 )
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	1020 except Exception as e:
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	1021 self.disp(_(f"can't unsubscribe from node: {e}"), error=True)
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	1022 self.host.quit(C.EXIT_BRIDGE_ERRBACK)
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	1023 else:
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	1024 self.disp(_("subscription removed"), 1)
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	1025 self.host.quit()
2353 ebc0dfe9c0ca jp (pubsub): added subscribe, unsubscribe and subscriptions methods: Goffi <goffi@goffi.org> parents: 2351 diff changeset	1026
ebc0dfe9c0ca jp (pubsub): added subscribe, unsubscribe and subscriptions methods: Goffi <goffi@goffi.org> parents: 2351 diff changeset	1027
ebc0dfe9c0ca jp (pubsub): added subscribe, unsubscribe and subscriptions methods: Goffi <goffi@goffi.org> parents: 2351 diff changeset	1028 class Subscriptions(base.CommandBase):
ebc0dfe9c0ca jp (pubsub): added subscribe, unsubscribe and subscriptions methods: Goffi <goffi@goffi.org> parents: 2351 diff changeset	1029 def __init__(self, host):
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1030 base.CommandBase.__init__(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1031 self,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1032 host,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1033 "subscriptions",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1034 use_output=C.OUTPUT_LIST_DICT,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1035 use_pubsub=True,
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2910 diff changeset	1036 help=_("retrieve all subscriptions on a service"),
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1037 )
2353 ebc0dfe9c0ca jp (pubsub): added subscribe, unsubscribe and subscriptions methods: Goffi <goffi@goffi.org> parents: 2351 diff changeset	1038
ebc0dfe9c0ca jp (pubsub): added subscribe, unsubscribe and subscriptions methods: Goffi <goffi@goffi.org> parents: 2351 diff changeset	1039 def add_parser_options(self):
ebc0dfe9c0ca jp (pubsub): added subscribe, unsubscribe and subscriptions methods: Goffi <goffi@goffi.org> parents: 2351 diff changeset	1040 pass
ebc0dfe9c0ca jp (pubsub): added subscribe, unsubscribe and subscriptions methods: Goffi <goffi@goffi.org> parents: 2351 diff changeset	1041
3040 fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	1042 async def start(self):
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	1043 try:
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	1044 subscriptions = await self.host.bridge.psSubscriptionsGet(
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	1045 self.args.service,
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	1046 self.args.node,
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	1047 self.profile,
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	1048 )
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	1049 except Exception as e:
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	1050 self.disp(_(f"can't retrieve subscriptions: {e}"), error=True)
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	1051 self.host.quit(C.EXIT_BRIDGE_ERRBACK)
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	1052 else:
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	1053 await self.output(subscriptions)
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	1054 self.host.quit()
2353 ebc0dfe9c0ca jp (pubsub): added subscribe, unsubscribe and subscriptions methods: Goffi <goffi@goffi.org> parents: 2351 diff changeset	1055
ebc0dfe9c0ca jp (pubsub): added subscribe, unsubscribe and subscriptions methods: Goffi <goffi@goffi.org> parents: 2351 diff changeset	1056
2235 4db836386641 jp: added use_pubsub and use_pubsub_node_req arguments to CommandBase Goffi <goffi@goffi.org> parents: 2224 diff changeset	1057 class Affiliations(base.CommandBase):
2204 afc703419186 jp (pubsub): added affiliations and node/affiliations/get: Goffi <goffi@goffi.org> parents: 2199 diff changeset	1058 def __init__(self, host):
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1059 base.CommandBase.__init__(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1060 self,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1061 host,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1062 "affiliations",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1063 use_output=C.OUTPUT_DICT,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1064 use_pubsub=True,
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2910 diff changeset	1065 help=_("retrieve all affiliations on a service"),
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1066 )
2204 afc703419186 jp (pubsub): added affiliations and node/affiliations/get: Goffi <goffi@goffi.org> parents: 2199 diff changeset	1067
afc703419186 jp (pubsub): added affiliations and node/affiliations/get: Goffi <goffi@goffi.org> parents: 2199 diff changeset	1068 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	1069 pass
2204 afc703419186 jp (pubsub): added affiliations and node/affiliations/get: Goffi <goffi@goffi.org> parents: 2199 diff changeset	1070
3040 fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	1071 async def start(self):
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	1072 try:
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	1073 affiliations = await self.host.bridge.psAffiliationsGet(
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	1074 self.args.service,
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	1075 self.args.node,
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	1076 self.profile,
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	1077 )
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	1078 except Exception as e:
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	1079 self.disp(
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	1080 f"can't get node affiliations: {e}", error=True
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	1081 )
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	1082 self.host.quit(C.EXIT_BRIDGE_ERRBACK)
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	1083 else:
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	1084 await self.output(affiliations)
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	1085 self.host.quit()
2204 afc703419186 jp (pubsub): added affiliations and node/affiliations/get: Goffi <goffi@goffi.org> parents: 2199 diff changeset	1086
afc703419186 jp (pubsub): added affiliations and node/affiliations/get: Goffi <goffi@goffi.org> parents: 2199 diff changeset	1087
2316 7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1088 class Search(base.CommandBase):
3040 fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	1089 """This command do a search without using MAM
2764 92af49cde255 jp (base): MAM and RSM arguments can now be used for pubsub commands: Goffi <goffi@goffi.org> parents: 2761 diff changeset	1090
92af49cde255 jp (base): MAM and RSM arguments can now be used for pubsub commands: Goffi <goffi@goffi.org> parents: 2761 diff changeset	1091 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	1092 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	1093 """
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1094
2316 7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1095 RE_FLAGS = re.MULTILINE \| re.UNICODE
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2910 diff changeset	1096 EXEC_ACTIONS = ("exec", "external")
2316 7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1097
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1098 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	1099 # 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	1100 # 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	1101 # 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	1102 base.CommandBase.__init__(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1103 self,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1104 host,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1105 "search",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1106 use_output=C.OUTPUT_XML,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1107 use_pubsub=True,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1108 pubsub_flags={C.MULTI_ITEMS, C.NO_MAX},
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1109 use_verbose=True,
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2910 diff changeset	1110 help=_("search items corresponding to filters"),
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1111 )
2316 7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1112
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1113 @property
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1114 def etree(self):
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1115 """load lxml.etree only if needed"""
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1116 if self._etree is None:
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1117 from lxml import etree
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1118
2316 7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1119 self._etree = etree
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1120 return self._etree
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1121
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1122 def filter_opt(self, value, type_):
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1123 return (type_, value)
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1124
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1125 def filter_flag(self, value, type_):
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1126 value = C.bool(value)
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1127 return (type_, value)
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1128
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1129 def add_parser_options(self):
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1130 self.parser.add_argument(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1131 "-D",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1132 "--max-depth",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1133 type=int,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1134 default=0,
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2910 diff changeset	1135 help=_("maximum depth of recursion (will search linked nodes if > 0, "
ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2910 diff changeset	1136 "DEFAULT: 0)"),
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1137 )
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1138 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	1139 "-M",
92af49cde255 jp (base): MAM and RSM arguments can now be used for pubsub commands: Goffi <goffi@goffi.org> parents: 2761 diff changeset	1140 "--node-max",
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1141 type=int,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1142 default=30,
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2910 diff changeset	1143 help=_("maximum number of items to get per node ({} to get all items, "
ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2910 diff changeset	1144 "DEFAULT: 30)".format( C.NO_LIMIT)),
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1145 )
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1146 self.parser.add_argument(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1147 "-N",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1148 "--namespace",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1149 action="append",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1150 nargs=2,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1151 default=[],
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1152 metavar="NAME NAMESPACE",
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2910 diff changeset	1153 help=_("namespace to use for xpath"),
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1154 )
2316 7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1155
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1156 # filters
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2910 diff changeset	1157 filter_text = partial(self.filter_opt, type_="text")
ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2910 diff changeset	1158 filter_re = partial(self.filter_opt, type_="regex")
ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2910 diff changeset	1159 filter_xpath = partial(self.filter_opt, type_="xpath")
ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2910 diff changeset	1160 filter_python = partial(self.filter_opt, type_="python")
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1161 filters = self.parser.add_argument_group(
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2910 diff changeset	1162 _("filters"),
ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2910 diff changeset	1163 _("only items corresponding to following filters will be kept"),
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1164 )
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1165 filters.add_argument(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1166 "-t",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1167 "--text",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1168 action="append",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1169 dest="filters",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1170 type=filter_text,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1171 metavar="TEXT",
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2910 diff changeset	1172 help=_("full text filter, item must contain this string (XML included)"),
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1173 )
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1174 filters.add_argument(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1175 "-r",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1176 "--regex",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1177 action="append",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1178 dest="filters",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1179 type=filter_re,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1180 metavar="EXPRESSION",
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2910 diff changeset	1181 help=_("like --text but using a regular expression"),
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1182 )
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1183 filters.add_argument(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1184 "-x",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1185 "--xpath",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1186 action="append",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1187 dest="filters",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1188 type=filter_xpath,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1189 metavar="XPATH",
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2910 diff changeset	1190 help=_("filter items which has elements matching this xpath"),
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1191 )
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1192 filters.add_argument(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1193 "-P",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1194 "--python",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1195 action="append",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1196 dest="filters",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1197 type=filter_python,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1198 metavar="PYTHON_CODE",
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2910 diff changeset	1199 help=_('Python expression which much return a bool (True to keep item, '
ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2910 diff changeset	1200 'False to reject it). "item" is raw text item, "item_xml" is '
ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2910 diff changeset	1201 'lxml\'s etree.Element'
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1202 ),
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
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1205 # filters flags
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2910 diff changeset	1206 flag_case = partial(self.filter_flag, type_="ignore-case")
ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2910 diff changeset	1207 flag_invert = partial(self.filter_flag, type_="invert")
ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2910 diff changeset	1208 flag_dotall = partial(self.filter_flag, type_="dotall")
ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2910 diff changeset	1209 flag_matching = partial(self.filter_flag, type_="only-matching")
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1210 flags = self.parser.add_argument_group(
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2910 diff changeset	1211 _("filters flags"),
ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2910 diff changeset	1212 _("filters modifiers (change behaviour of following filters)"),
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1213 )
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1214 flags.add_argument(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1215 "-C",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1216 "--ignore-case",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1217 action="append",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1218 dest="filters",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1219 type=flag_case,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1220 const=("ignore-case", True),
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1221 nargs="?",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1222 metavar="BOOLEAN",
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2910 diff changeset	1223 help=_("(don't) ignore case in following filters (DEFAULT: case sensitive)"),
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1224 )
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1225 flags.add_argument(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1226 "-I",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1227 "--invert",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1228 action="append",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1229 dest="filters",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1230 type=flag_invert,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1231 const=("invert", True),
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1232 nargs="?",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1233 metavar="BOOLEAN",
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2910 diff changeset	1234 help=_("(don't) invert effect of following filters (DEFAULT: don't invert)"),
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1235 )
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1236 flags.add_argument(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1237 "-A",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1238 "--dot-all",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1239 action="append",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1240 dest="filters",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1241 type=flag_dotall,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1242 const=("dotall", True),
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1243 nargs="?",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1244 metavar="BOOLEAN",
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2910 diff changeset	1245 help=_("(don't) use DOTALL option for regex (DEFAULT: don't use)"),
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1246 )
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1247 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	1248 "-k",
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1249 "--only-matching",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1250 action="append",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1251 dest="filters",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1252 type=flag_matching,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1253 const=("only-matching", True),
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1254 nargs="?",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1255 metavar="BOOLEAN",
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2910 diff changeset	1256 help=_("keep only the matching part of the item"),
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1257 )
2316 7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1258
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1259 # action
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1260 self.parser.add_argument(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1261 "action",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1262 default="print",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1263 nargs="?",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1264 choices=("print", "exec", "external"),
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2910 diff changeset	1265 help=_("action to do on found items (DEFAULT: print)"),
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1266 )
2316 7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1267 self.parser.add_argument("command", nargs=argparse.REMAINDER)
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1268
3040 fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	1269 async def getItems(self, depth, service, node, items):
2316 7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1270 self.to_get += 1
3040 fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	1271 try:
3312 77177b13ff54 plugin XEP-0060: serialise psItemsGet result with data_format Goffi <goffi@goffi.org> parents: 3275 diff changeset	1272 ps_result = data_format.deserialise(
77177b13ff54 plugin XEP-0060: serialise psItemsGet result with data_format Goffi <goffi@goffi.org> parents: 3275 diff changeset	1273 await self.host.bridge.psItemsGet(
77177b13ff54 plugin XEP-0060: serialise psItemsGet result with data_format Goffi <goffi@goffi.org> parents: 3275 diff changeset	1274 service,
77177b13ff54 plugin XEP-0060: serialise psItemsGet result with data_format Goffi <goffi@goffi.org> parents: 3275 diff changeset	1275 node,
77177b13ff54 plugin XEP-0060: serialise psItemsGet result with data_format Goffi <goffi@goffi.org> parents: 3275 diff changeset	1276 self.args.node_max,
77177b13ff54 plugin XEP-0060: serialise psItemsGet result with data_format Goffi <goffi@goffi.org> parents: 3275 diff changeset	1277 items,
77177b13ff54 plugin XEP-0060: serialise psItemsGet result with data_format Goffi <goffi@goffi.org> parents: 3275 diff changeset	1278 "",
77177b13ff54 plugin XEP-0060: serialise psItemsGet result with data_format Goffi <goffi@goffi.org> parents: 3275 diff changeset	1279 self.getPubsubExtra(),
77177b13ff54 plugin XEP-0060: serialise psItemsGet result with data_format Goffi <goffi@goffi.org> parents: 3275 diff changeset	1280 self.profile,
77177b13ff54 plugin XEP-0060: serialise psItemsGet result with data_format Goffi <goffi@goffi.org> parents: 3275 diff changeset	1281 )
3040 fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	1282 )
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	1283 except Exception as e:
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	1284 self.disp(
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	1285 f"can't get pubsub items at {service} (node: {node}): {e}",
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	1286 error=True,
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	1287 )
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	1288 self.to_get -= 1
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	1289 else:
3312 77177b13ff54 plugin XEP-0060: serialise psItemsGet result with data_format Goffi <goffi@goffi.org> parents: 3275 diff changeset	1290 await self.search(ps_result, depth)
2316 7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1291
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1292 def _checkPubsubURL(self, match, found_nodes):
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1293 """check that the matched URL is an xmpp: one
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1294
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1295 @param found_nodes(list[unicode]): found_nodes
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1296 this list will be filled while xmpp: URIs are discovered
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1297 """
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1298 url = match.group(0)
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2910 diff changeset	1299 if url.startswith("xmpp"):
2316 7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1300 try:
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1301 url_data = uri.parseXMPPUri(url)
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1302 except ValueError:
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1303 return
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2910 diff changeset	1304 if url_data["type"] == "pubsub":
ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2910 diff changeset	1305 found_node = {"service": url_data["path"], "node": url_data["node"]}
ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2910 diff changeset	1306 if "item" in url_data:
ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2910 diff changeset	1307 found_node["item"] = url_data["item"]
2316 7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1308 found_nodes.append(found_node)
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1309
3040 fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	1310 async def getSubNodes(self, item, depth):
2316 7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1311 """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	1312 found_nodes = []
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1313 checkURI = partial(self._checkPubsubURL, found_nodes=found_nodes)
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1314 strings.RE_URL.sub(checkURI, item)
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1315 for data in found_nodes:
3040 fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	1316 await self.getItems(
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1317 depth + 1,
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2910 diff changeset	1318 data["service"],
ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2910 diff changeset	1319 data["node"],
ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2910 diff changeset	1320 [data["item"]] if "item" in data else [],
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1321 )
2316 7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1322
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1323 def parseXml(self, item):
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1324 try:
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1325 return self.etree.fromstring(item)
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1326 except self.etree.XMLSyntaxError:
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1327 self.disp(
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2910 diff changeset	1328 _("item doesn't looks like XML, you have probably used --only-matching "
ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2910 diff changeset	1329 "somewhere before and we have no more XML"),
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1330 error=True,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1331 )
2316 7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1332 self.host.quit(C.EXIT_BAD_ARG)
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1333
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1334 def filter(self, item):
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1335 """apply filters given on command line
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1336
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1337 if only-matching is used, item may be modified
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1338 @return (tuple[bool, unicode]): a tuple with:
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1339 - keep: True if item passed the filters
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1340 - item: it is returned in case of modifications
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1341 """
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1342 ignore_case = False
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1343 invert = False
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1344 dotall = False
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1345 only_matching = False
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1346 item_xml = None
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1347 for type_, value in self.args.filters:
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1348 keep = True
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1349
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1350 ## filters
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1351
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2910 diff changeset	1352 if type_ == "text":
2316 7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1353 if ignore_case:
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1354 if value.lower() not in item.lower():
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1355 keep = False
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1356 else:
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1357 if value not in item:
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1358 keep = False
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1359 if keep and only_matching:
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1360 # doesn't really make sens to keep a fixed string
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1361 # so we raise an error
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1362 self.host.disp(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1363 _(
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2910 diff changeset	1364 "--only-matching used with fixed --text string, are you sure?"
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1365 ),
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1366 error=True,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1367 )
2316 7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1368 self.host.quit(C.EXIT_BAD_ARG)
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2910 diff changeset	1369 elif type_ == "regex":
2316 7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1370 flags = self.RE_FLAGS
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1371 if ignore_case:
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1372 flags \|= re.IGNORECASE
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1373 if dotall:
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1374 flags \|= re.DOTALL
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1375 match = re.search(value, item, flags)
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1376 keep = match != None
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1377 if keep and only_matching:
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1378 item = match.group()
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1379 item_xml = None
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2910 diff changeset	1380 elif type_ == "xpath":
2316 7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1381 if item_xml is None:
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1382 item_xml = self.parseXml(item)
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1383 try:
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1384 elts = item_xml.xpath(value, namespaces=self.args.namespace)
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1385 except self.etree.XPathEvalError as e:
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1386 self.disp(
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2910 diff changeset	1387 _("can't use xpath: {reason}").format(reason=e), error=True
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1388 )
2316 7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1389 self.host.quit(C.EXIT_BAD_ARG)
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1390 keep = bool(elts)
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1391 if keep and only_matching:
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1392 item_xml = elts[0]
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1393 try:
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1394 item = self.etree.tostring(item_xml, encoding="unicode")
2316 7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1395 except TypeError:
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1396 # we have a string only, not an element
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2910 diff changeset	1397 item = str(item_xml)
2316 7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1398 item_xml = None
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2910 diff changeset	1399 elif type_ == "python":
2316 7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1400 if item_xml is None:
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1401 item_xml = self.parseXml(item)
3040 fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	1402 cmd_ns = {
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	1403 "etree": self.etree,
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	1404 "item": item,
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	1405 "item_xml": item_xml
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	1406 }
2316 7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1407 try:
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1408 keep = eval(value, cmd_ns)
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1409 except SyntaxError as e:
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2910 diff changeset	1410 self.disp(str(e), error=True)
2316 7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1411 self.host.quit(C.EXIT_BAD_ARG)
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1412
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1413 ## flags
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1414
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2910 diff changeset	1415 elif type_ == "ignore-case":
2316 7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1416 ignore_case = value
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2910 diff changeset	1417 elif type_ == "invert":
2316 7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1418 invert = value
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1419 # we need to continue, else loop would end here
2316 7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1420 continue
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2910 diff changeset	1421 elif type_ == "dotall":
2316 7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1422 dotall = value
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2910 diff changeset	1423 elif type_ == "only-matching":
2316 7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1424 only_matching = value
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1425 else:
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1426 raise exceptions.InternalError(
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2910 diff changeset	1427 _("unknown filter type {type}").format(type=type_)
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1428 )
2316 7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1429
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1430 if invert:
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1431 keep = not keep
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1432 if not keep:
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1433 return False, item
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 return True, item
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1436
3040 fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	1437 async def doItemAction(self, item, metadata):
2316 7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1438 """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	1439
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1440 @param item(unicode): accepted item
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1441 """
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1442 action = self.args.action
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2910 diff changeset	1443 if action == "print" or self.host.verbosity > 0:
2316 7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1444 try:
3040 fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	1445 await self.output(item)
2316 7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1446 except self.etree.XMLSyntaxError:
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1447 # item is not valid XML, but a string
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1448 # can happen when --only-matching is used
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1449 self.disp(item)
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1450 if action in self.EXEC_ACTIONS:
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1451 item_elt = self.parseXml(item)
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2910 diff changeset	1452 if action == "exec":
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1453 use = {
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2910 diff changeset	1454 "service": metadata["service"],
ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2910 diff changeset	1455 "node": metadata["node"],
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1456 "item": item_elt.get("id"),
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1457 "profile": self.profile,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1458 }
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	1459 # 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	1460 # else it would be modified
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1461 parser_args, use_args = arg_tools.get_use_args(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1462 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	1463 )
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	1464 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	1465 else:
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1466 cmd_args = self.args.command
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1467
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1468 self.disp(
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2910 diff changeset	1469 "COMMAND: {command}".format(
ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2910 diff changeset	1470 command=" ".join([arg_tools.escape(a) for a in cmd_args])
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1471 ),
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1472 2,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1473 )
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2910 diff changeset	1474 if action == "exec":
3040 fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	1475 p = await asyncio.create_subprocess_exec(*cmd_args)
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	1476 ret = await p.wait()
2316 7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1477 else:
3040 fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	1478 p = await asyncio.create_subprocess_exec(*cmd_args,
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	1479 stdin=subprocess.PIPE)
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	1480 await p.communicate(item.encode(sys.getfilesystemencoding()))
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	1481 ret = p.returncode
2316 7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1482 if ret != 0:
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1483 self.disp(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1484 A.color(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1485 C.A_FAILURE,
3040 fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	1486 _(f"executed command failed with exit code {ret}"),
2624 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 )
2316 7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1489
3312 77177b13ff54 plugin XEP-0060: serialise psItemsGet result with data_format Goffi <goffi@goffi.org> parents: 3275 diff changeset	1490 async def search(self, ps_result, depth):
2316 7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1491 """callback of getItems
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1492
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1493 this method filters items, get sub nodes if needed,
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1494 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	1495 @param items_data(tuple): result of getItems
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1496 @param depth(int): current depth level
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1497 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	1498 """
3312 77177b13ff54 plugin XEP-0060: serialise psItemsGet result with data_format Goffi <goffi@goffi.org> parents: 3275 diff changeset	1499 for item in ps_result['items']:
2316 7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1500 if depth < self.args.max_depth:
3040 fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	1501 await self.getSubNodes(item, depth)
2316 7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1502 keep, item = self.filter(item)
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1503 if not keep:
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1504 continue
3312 77177b13ff54 plugin XEP-0060: serialise psItemsGet result with data_format Goffi <goffi@goffi.org> parents: 3275 diff changeset	1505 await self.doItemAction(item, ps_result)
2316 7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1506
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1507 # we check if we got all getItems results
2316 7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1508 self.to_get -= 1
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1509 if self.to_get == 0:
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1510 # yes, we can quit
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1511 self.host.quit()
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1512 assert self.to_get > 0
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1513
3040 fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	1514 async def start(self):
2316 7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1515 if self.args.command:
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1516 if self.args.action not in self.EXEC_ACTIONS:
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1517 self.parser.error(
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2910 diff changeset	1518 _("Command can only be used with {actions} actions").format(
ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2910 diff changeset	1519 actions=", ".join(self.EXEC_ACTIONS)
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1520 )
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1521 )
2316 7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1522 else:
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1523 if self.args.action in self.EXEC_ACTIONS:
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2910 diff changeset	1524 self.parser.error(_("you need to specify a command to execute"))
2316 7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1525 if not self.args.node:
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1526 # TODO: handle get service affiliations when node is not set
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2910 diff changeset	1527 self.parser.error(_("empty node is not handled yet"))
2316 7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1528 # 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	1529 # when it reach 0 again, the command is finished
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1530 self.to_get = 0
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1531 self._etree = None
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1532 if self.args.filters is None:
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1533 self.args.filters = []
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1534 self.args.namespace = dict(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1535 self.args.namespace + [("pubsub", "http://jabber.org/protocol/pubsub")]
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1536 )
3040 fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	1537 await self.getItems(0, self.args.service, self.args.node, self.args.items)
2316 7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1538
7b448ac50a69 jp (pubsub): new search command: Goffi <goffi@goffi.org> parents: 2308 diff changeset	1539
2777 ff1b40823b07 jp (pubsub): new "transform" command: Goffi <goffi@goffi.org> parents: 2773 diff changeset	1540 class Transform(base.CommandBase):
ff1b40823b07 jp (pubsub): new "transform" command: Goffi <goffi@goffi.org> parents: 2773 diff changeset	1541 def __init__(self, host):
ff1b40823b07 jp (pubsub): new "transform" command: Goffi <goffi@goffi.org> parents: 2773 diff changeset	1542 base.CommandBase.__init__(
ff1b40823b07 jp (pubsub): new "transform" command: Goffi <goffi@goffi.org> parents: 2773 diff changeset	1543 self,
ff1b40823b07 jp (pubsub): new "transform" command: Goffi <goffi@goffi.org> parents: 2773 diff changeset	1544 host,
ff1b40823b07 jp (pubsub): new "transform" command: Goffi <goffi@goffi.org> parents: 2773 diff changeset	1545 "transform",
ff1b40823b07 jp (pubsub): new "transform" command: Goffi <goffi@goffi.org> parents: 2773 diff changeset	1546 use_pubsub=True,
ff1b40823b07 jp (pubsub): new "transform" command: Goffi <goffi@goffi.org> parents: 2773 diff changeset	1547 pubsub_flags={C.NODE, C.MULTI_ITEMS},
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2910 diff changeset	1548 help=_("modify items of a node using an external command/script"),
2777 ff1b40823b07 jp (pubsub): new "transform" command: Goffi <goffi@goffi.org> parents: 2773 diff changeset	1549 )
ff1b40823b07 jp (pubsub): new "transform" command: Goffi <goffi@goffi.org> parents: 2773 diff changeset	1550
ff1b40823b07 jp (pubsub): new "transform" command: Goffi <goffi@goffi.org> parents: 2773 diff changeset	1551 def add_parser_options(self):
ff1b40823b07 jp (pubsub): new "transform" command: Goffi <goffi@goffi.org> parents: 2773 diff changeset	1552 self.parser.add_argument(
ff1b40823b07 jp (pubsub): new "transform" command: Goffi <goffi@goffi.org> parents: 2773 diff changeset	1553 "--apply",
ff1b40823b07 jp (pubsub): new "transform" command: Goffi <goffi@goffi.org> parents: 2773 diff changeset	1554 action="store_true",
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2910 diff changeset	1555 help=_("apply transformation (DEFAULT: do a dry run)"),
2777 ff1b40823b07 jp (pubsub): new "transform" command: Goffi <goffi@goffi.org> parents: 2773 diff changeset	1556 )
ff1b40823b07 jp (pubsub): new "transform" command: Goffi <goffi@goffi.org> parents: 2773 diff changeset	1557 self.parser.add_argument(
ff1b40823b07 jp (pubsub): new "transform" command: Goffi <goffi@goffi.org> parents: 2773 diff changeset	1558 "--admin",
ff1b40823b07 jp (pubsub): new "transform" command: Goffi <goffi@goffi.org> parents: 2773 diff changeset	1559 action="store_true",
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2910 diff changeset	1560 help=_("do a pubsub admin request, needed to change publisher"),
2777 ff1b40823b07 jp (pubsub): new "transform" command: Goffi <goffi@goffi.org> parents: 2773 diff changeset	1561 )
ff1b40823b07 jp (pubsub): new "transform" command: Goffi <goffi@goffi.org> parents: 2773 diff changeset	1562 self.parser.add_argument(
ff1b40823b07 jp (pubsub): new "transform" command: Goffi <goffi@goffi.org> parents: 2773 diff changeset	1563 "-I",
3411 f30b238d9c45 jp: follow best practices and use a dash (`-`) instead of underscore (`_`) for long options. Goffi <goffi@goffi.org> parents: 3325 diff changeset	1564 "--ignore-errors",
2777 ff1b40823b07 jp (pubsub): new "transform" command: Goffi <goffi@goffi.org> parents: 2773 diff changeset	1565 action="store_true",
ff1b40823b07 jp (pubsub): new "transform" command: Goffi <goffi@goffi.org> parents: 2773 diff changeset	1566 help=_(
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2910 diff changeset	1567 "if command return a non zero exit code, ignore the item and continue"),
2777 ff1b40823b07 jp (pubsub): new "transform" command: Goffi <goffi@goffi.org> parents: 2773 diff changeset	1568 )
ff1b40823b07 jp (pubsub): new "transform" command: Goffi <goffi@goffi.org> parents: 2773 diff changeset	1569 self.parser.add_argument(
ff1b40823b07 jp (pubsub): new "transform" command: Goffi <goffi@goffi.org> parents: 2773 diff changeset	1570 "-A",
ff1b40823b07 jp (pubsub): new "transform" command: Goffi <goffi@goffi.org> parents: 2773 diff changeset	1571 "--all",
ff1b40823b07 jp (pubsub): new "transform" command: Goffi <goffi@goffi.org> parents: 2773 diff changeset	1572 action="store_true",
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2910 diff changeset	1573 help=_("get all items by looping over all pages using RSM")
2777 ff1b40823b07 jp (pubsub): new "transform" command: Goffi <goffi@goffi.org> parents: 2773 diff changeset	1574 )
ff1b40823b07 jp (pubsub): new "transform" command: Goffi <goffi@goffi.org> parents: 2773 diff changeset	1575 self.parser.add_argument(
ff1b40823b07 jp (pubsub): new "transform" command: Goffi <goffi@goffi.org> parents: 2773 diff changeset	1576 "command_path",
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2910 diff changeset	1577 help=_("path to the command to use. Will be called repetitivly with an "
ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2910 diff changeset	1578 "item as input. Output (full item XML) will be used as new one. "
ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2910 diff changeset	1579 'Return "DELETE" string to delete the item, and "SKIP" to ignore it'),
2777 ff1b40823b07 jp (pubsub): new "transform" command: Goffi <goffi@goffi.org> parents: 2773 diff changeset	1580 )
ff1b40823b07 jp (pubsub): new "transform" command: Goffi <goffi@goffi.org> parents: 2773 diff changeset	1581
3040 fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	1582 async def psItemsSendCb(self, item_ids, metadata):
2803 d4a9a60bc650 jp (pubsub/transform): use new psItemsSend method, it is not needed anymore to send items one by one when --admin is not used Goffi <goffi@goffi.org> parents: 2802 diff changeset	1583 if item_ids:
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2910 diff changeset	1584 self.disp(_('items published with ids {item_ids}').format(
ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2910 diff changeset	1585 item_ids=', '.join(item_ids)))
2803 d4a9a60bc650 jp (pubsub/transform): use new psItemsSend method, it is not needed anymore to send items one by one when --admin is not used Goffi <goffi@goffi.org> parents: 2802 diff changeset	1586 else:
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2910 diff changeset	1587 self.disp(_('items published'))
2777 ff1b40823b07 jp (pubsub): new "transform" command: Goffi <goffi@goffi.org> parents: 2773 diff changeset	1588 if self.args.all:
3040 fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	1589 return await self.handleNextPage(metadata)
2777 ff1b40823b07 jp (pubsub): new "transform" command: Goffi <goffi@goffi.org> parents: 2773 diff changeset	1590 else:
ff1b40823b07 jp (pubsub): new "transform" command: Goffi <goffi@goffi.org> parents: 2773 diff changeset	1591 self.host.quit()
ff1b40823b07 jp (pubsub): new "transform" command: Goffi <goffi@goffi.org> parents: 2773 diff changeset	1592
3040 fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	1593 async def handleNextPage(self, metadata):
2777 ff1b40823b07 jp (pubsub): new "transform" command: Goffi <goffi@goffi.org> parents: 2773 diff changeset	1594 """Retrieve new page through RSM or quit if we're in the last page
ff1b40823b07 jp (pubsub): new "transform" command: Goffi <goffi@goffi.org> parents: 2773 diff changeset	1595
ff1b40823b07 jp (pubsub): new "transform" command: Goffi <goffi@goffi.org> parents: 2773 diff changeset	1596 use to handle --all option
ff1b40823b07 jp (pubsub): new "transform" command: Goffi <goffi@goffi.org> parents: 2773 diff changeset	1597 @param metadata(dict): metadata as returned by psItemsGet
ff1b40823b07 jp (pubsub): new "transform" command: Goffi <goffi@goffi.org> parents: 2773 diff changeset	1598 """
ff1b40823b07 jp (pubsub): new "transform" command: Goffi <goffi@goffi.org> parents: 2773 diff changeset	1599 try:
3312 77177b13ff54 plugin XEP-0060: serialise psItemsGet result with data_format Goffi <goffi@goffi.org> parents: 3275 diff changeset	1600 last = metadata['rsm']['last']
77177b13ff54 plugin XEP-0060: serialise psItemsGet result with data_format Goffi <goffi@goffi.org> parents: 3275 diff changeset	1601 index = int(metadata['rsm']['index'])
77177b13ff54 plugin XEP-0060: serialise psItemsGet result with data_format Goffi <goffi@goffi.org> parents: 3275 diff changeset	1602 count = int(metadata['rsm']['count'])
2777 ff1b40823b07 jp (pubsub): new "transform" command: Goffi <goffi@goffi.org> parents: 2773 diff changeset	1603 except KeyError:
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2910 diff changeset	1604 self.disp(_("Can't retrieve all items, RSM metadata not available"),
2777 ff1b40823b07 jp (pubsub): new "transform" command: Goffi <goffi@goffi.org> parents: 2773 diff changeset	1605 error=True)
ff1b40823b07 jp (pubsub): new "transform" command: Goffi <goffi@goffi.org> parents: 2773 diff changeset	1606 self.host.quit(C.EXIT_MISSING_FEATURE)
ff1b40823b07 jp (pubsub): new "transform" command: Goffi <goffi@goffi.org> parents: 2773 diff changeset	1607 except ValueError as e:
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2910 diff changeset	1608 self.disp(_("Can't retrieve all items, bad RSM metadata: {msg}")
2777 ff1b40823b07 jp (pubsub): new "transform" command: Goffi <goffi@goffi.org> parents: 2773 diff changeset	1609 .format(msg=e), error=True)
ff1b40823b07 jp (pubsub): new "transform" command: Goffi <goffi@goffi.org> parents: 2773 diff changeset	1610 self.host.quit(C.EXIT_ERROR)
ff1b40823b07 jp (pubsub): new "transform" command: Goffi <goffi@goffi.org> parents: 2773 diff changeset	1611
ff1b40823b07 jp (pubsub): new "transform" command: Goffi <goffi@goffi.org> parents: 2773 diff changeset	1612 if index + self.args.rsm_max >= count:
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2910 diff changeset	1613 self.disp(_('All items transformed'))
2777 ff1b40823b07 jp (pubsub): new "transform" command: Goffi <goffi@goffi.org> parents: 2773 diff changeset	1614 self.host.quit(0)
ff1b40823b07 jp (pubsub): new "transform" command: Goffi <goffi@goffi.org> parents: 2773 diff changeset	1615
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2910 diff changeset	1616 self.disp(_('Retrieving next page ({page_idx}/{page_total})').format(
2777 ff1b40823b07 jp (pubsub): new "transform" command: Goffi <goffi@goffi.org> parents: 2773 diff changeset	1617 page_idx = int(index/self.args.rsm_max) + 1,
ff1b40823b07 jp (pubsub): new "transform" command: Goffi <goffi@goffi.org> parents: 2773 diff changeset	1618 page_total = int(count/self.args.rsm_max),
ff1b40823b07 jp (pubsub): new "transform" command: Goffi <goffi@goffi.org> parents: 2773 diff changeset	1619 )
ff1b40823b07 jp (pubsub): new "transform" command: Goffi <goffi@goffi.org> parents: 2773 diff changeset	1620 )
ff1b40823b07 jp (pubsub): new "transform" command: Goffi <goffi@goffi.org> parents: 2773 diff changeset	1621
ff1b40823b07 jp (pubsub): new "transform" command: Goffi <goffi@goffi.org> parents: 2773 diff changeset	1622 extra = self.getPubsubExtra()
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2910 diff changeset	1623 extra['rsm_after'] = last
3040 fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	1624 try:
3312 77177b13ff54 plugin XEP-0060: serialise psItemsGet result with data_format Goffi <goffi@goffi.org> parents: 3275 diff changeset	1625 ps_result = await data_format.deserialise(
77177b13ff54 plugin XEP-0060: serialise psItemsGet result with data_format Goffi <goffi@goffi.org> parents: 3275 diff changeset	1626 self.host.bridge.psItemsGet(
77177b13ff54 plugin XEP-0060: serialise psItemsGet result with data_format Goffi <goffi@goffi.org> parents: 3275 diff changeset	1627 self.args.service,
77177b13ff54 plugin XEP-0060: serialise psItemsGet result with data_format Goffi <goffi@goffi.org> parents: 3275 diff changeset	1628 self.args.node,
77177b13ff54 plugin XEP-0060: serialise psItemsGet result with data_format Goffi <goffi@goffi.org> parents: 3275 diff changeset	1629 self.args.rsm_max,
77177b13ff54 plugin XEP-0060: serialise psItemsGet result with data_format Goffi <goffi@goffi.org> parents: 3275 diff changeset	1630 self.args.items,
77177b13ff54 plugin XEP-0060: serialise psItemsGet result with data_format Goffi <goffi@goffi.org> parents: 3275 diff changeset	1631 "",
77177b13ff54 plugin XEP-0060: serialise psItemsGet result with data_format Goffi <goffi@goffi.org> parents: 3275 diff changeset	1632 extra,
77177b13ff54 plugin XEP-0060: serialise psItemsGet result with data_format Goffi <goffi@goffi.org> parents: 3275 diff changeset	1633 self.profile,
77177b13ff54 plugin XEP-0060: serialise psItemsGet result with data_format Goffi <goffi@goffi.org> parents: 3275 diff changeset	1634 )
3040 fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	1635 )
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	1636 except Exception as e:
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	1637 self.disp(
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	1638 f"can't retrieve items: {e}", error=True
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	1639 )
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	1640 self.host.quit(C.EXIT_BRIDGE_ERRBACK)
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	1641 else:
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	1642 await self.psItemsGetCb(ps_result)
2777 ff1b40823b07 jp (pubsub): new "transform" command: Goffi <goffi@goffi.org> parents: 2773 diff changeset	1643
3040 fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	1644 async def psItemsGetCb(self, ps_result):
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	1645 encoding = 'utf-8'
2803 d4a9a60bc650 jp (pubsub/transform): use new psItemsSend method, it is not needed anymore to send items one by one when --admin is not used Goffi <goffi@goffi.org> parents: 2802 diff changeset	1646 new_items = []
2777 ff1b40823b07 jp (pubsub): new "transform" command: Goffi <goffi@goffi.org> parents: 2773 diff changeset	1647
3312 77177b13ff54 plugin XEP-0060: serialise psItemsGet result with data_format Goffi <goffi@goffi.org> parents: 3275 diff changeset	1648 for item in ps_result['items']:
2777 ff1b40823b07 jp (pubsub): new "transform" command: Goffi <goffi@goffi.org> parents: 2773 diff changeset	1649 if self.check_duplicates:
ff1b40823b07 jp (pubsub): new "transform" command: Goffi <goffi@goffi.org> parents: 2773 diff changeset	1650 # this is used when we are not ordering by creation
ff1b40823b07 jp (pubsub): new "transform" command: Goffi <goffi@goffi.org> parents: 2773 diff changeset	1651 # to avoid infinite loop
ff1b40823b07 jp (pubsub): new "transform" command: Goffi <goffi@goffi.org> parents: 2773 diff changeset	1652 item_elt, __ = xml_tools.etreeParse(self, item)
ff1b40823b07 jp (pubsub): new "transform" command: Goffi <goffi@goffi.org> parents: 2773 diff changeset	1653 item_id = item_elt.get('id')
ff1b40823b07 jp (pubsub): new "transform" command: Goffi <goffi@goffi.org> parents: 2773 diff changeset	1654 if item_id in self.items_ids:
ff1b40823b07 jp (pubsub): new "transform" command: Goffi <goffi@goffi.org> parents: 2773 diff changeset	1655 self.disp(_(
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2910 diff changeset	1656 "Duplicate found on item {item_id}, we have probably handled "
ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2910 diff changeset	1657 "all items.").format(item_id=item_id))
2777 ff1b40823b07 jp (pubsub): new "transform" command: Goffi <goffi@goffi.org> parents: 2773 diff changeset	1658 self.host.quit()
ff1b40823b07 jp (pubsub): new "transform" command: Goffi <goffi@goffi.org> parents: 2773 diff changeset	1659 self.items_ids.append(item_id)
ff1b40823b07 jp (pubsub): new "transform" command: Goffi <goffi@goffi.org> parents: 2773 diff changeset	1660
ff1b40823b07 jp (pubsub): new "transform" command: Goffi <goffi@goffi.org> parents: 2773 diff changeset	1661 # we launch the command to filter the item
ff1b40823b07 jp (pubsub): new "transform" command: Goffi <goffi@goffi.org> parents: 2773 diff changeset	1662 try:
3040 fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	1663 p = await asyncio.create_subprocess_exec(
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	1664 self.args.command_path,
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	1665 stdin=subprocess.PIPE,
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	1666 stdout=subprocess.PIPE)
2777 ff1b40823b07 jp (pubsub): new "transform" command: Goffi <goffi@goffi.org> parents: 2773 diff changeset	1667 except OSError as e:
ff1b40823b07 jp (pubsub): new "transform" command: Goffi <goffi@goffi.org> parents: 2773 diff changeset	1668 exit_code = C.EXIT_CMD_NOT_FOUND if e.errno == 2 else C.EXIT_ERROR
3040 fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	1669 self.disp(f"Can't execute the command: {e}", error=True)
2777 ff1b40823b07 jp (pubsub): new "transform" command: Goffi <goffi@goffi.org> parents: 2773 diff changeset	1670 self.host.quit(exit_code)
3040 fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	1671 encoding = "utf-8"
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	1672 cmd_std_out, cmd_std_err = await p.communicate(item.encode(encoding))
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	1673 ret = p.returncode
2777 ff1b40823b07 jp (pubsub): new "transform" command: Goffi <goffi@goffi.org> parents: 2773 diff changeset	1674 if ret != 0:
3040 fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	1675 self.disp(f"The command returned a non zero status while parsing the "
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	1676 f"following item:\n\n{item}", error=True)
2777 ff1b40823b07 jp (pubsub): new "transform" command: Goffi <goffi@goffi.org> parents: 2773 diff changeset	1677 if self.args.ignore_errors:
ff1b40823b07 jp (pubsub): new "transform" command: Goffi <goffi@goffi.org> parents: 2773 diff changeset	1678 continue
ff1b40823b07 jp (pubsub): new "transform" command: Goffi <goffi@goffi.org> parents: 2773 diff changeset	1679 else:
ff1b40823b07 jp (pubsub): new "transform" command: Goffi <goffi@goffi.org> parents: 2773 diff changeset	1680 self.host.quit(C.EXIT_CMD_ERROR)
ff1b40823b07 jp (pubsub): new "transform" command: Goffi <goffi@goffi.org> parents: 2773 diff changeset	1681 if cmd_std_err is not None:
3040 fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	1682 cmd_std_err = cmd_std_err.decode(encoding, errors='ignore')
2777 ff1b40823b07 jp (pubsub): new "transform" command: Goffi <goffi@goffi.org> parents: 2773 diff changeset	1683 self.disp(cmd_std_err, error=True)
3040 fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	1684 cmd_std_out = cmd_std_out.decode(encoding).strip()
2777 ff1b40823b07 jp (pubsub): new "transform" command: Goffi <goffi@goffi.org> parents: 2773 diff changeset	1685 if cmd_std_out == "DELETE":
ff1b40823b07 jp (pubsub): new "transform" command: Goffi <goffi@goffi.org> parents: 2773 diff changeset	1686 item_elt, __ = xml_tools.etreeParse(self, item)
ff1b40823b07 jp (pubsub): new "transform" command: Goffi <goffi@goffi.org> parents: 2773 diff changeset	1687 item_id = item_elt.get('id')
3040 fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	1688 self.disp(_(f"Deleting item {item_id}"))
2777 ff1b40823b07 jp (pubsub): new "transform" command: Goffi <goffi@goffi.org> parents: 2773 diff changeset	1689 if self.args.apply:
3040 fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	1690 try:
3467 5901a7170528 plugin XEP-0060: renamed `psRetractItem(s)` to `psItem(s)Retract` for consistency. Goffi <goffi@goffi.org> parents: 3453 diff changeset	1691 await self.host.bridge.psItemRetract(
3040 fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	1692 self.args.service,
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	1693 self.args.node,
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	1694 item_id,
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	1695 False,
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	1696 self.profile,
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	1697 )
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	1698 except Exception as e:
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	1699 self.disp(
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	1700 f"can't delete item {item_id}: {e}", error=True
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	1701 )
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	1702 self.host.quit(C.EXIT_BRIDGE_ERRBACK)
2777 ff1b40823b07 jp (pubsub): new "transform" command: Goffi <goffi@goffi.org> parents: 2773 diff changeset	1703 continue
ff1b40823b07 jp (pubsub): new "transform" command: Goffi <goffi@goffi.org> parents: 2773 diff changeset	1704 elif cmd_std_out == "SKIP":
ff1b40823b07 jp (pubsub): new "transform" command: Goffi <goffi@goffi.org> parents: 2773 diff changeset	1705 item_elt, __ = xml_tools.etreeParse(self, item)
ff1b40823b07 jp (pubsub): new "transform" command: Goffi <goffi@goffi.org> parents: 2773 diff changeset	1706 item_id = item_elt.get('id')
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2910 diff changeset	1707 self.disp(_("Skipping item {item_id}").format(item_id=item_id))
2777 ff1b40823b07 jp (pubsub): new "transform" command: Goffi <goffi@goffi.org> parents: 2773 diff changeset	1708 continue
ff1b40823b07 jp (pubsub): new "transform" command: Goffi <goffi@goffi.org> parents: 2773 diff changeset	1709 element, etree = xml_tools.etreeParse(self, cmd_std_out)
ff1b40823b07 jp (pubsub): new "transform" command: Goffi <goffi@goffi.org> parents: 2773 diff changeset	1710
ff1b40823b07 jp (pubsub): new "transform" command: Goffi <goffi@goffi.org> parents: 2773 diff changeset	1711 # at this point command has been run and we have a etree.Element object
ff1b40823b07 jp (pubsub): new "transform" command: Goffi <goffi@goffi.org> parents: 2773 diff changeset	1712 if element.tag not in ("item", "{http://jabber.org/protocol/pubsub}item"):
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2910 diff changeset	1713 self.disp("your script must return a whole item, this is not:\n{xml}"
2777 ff1b40823b07 jp (pubsub): new "transform" command: Goffi <goffi@goffi.org> parents: 2773 diff changeset	1714 .format(xml=etree.tostring(element, encoding="unicode")), error=True)
ff1b40823b07 jp (pubsub): new "transform" command: Goffi <goffi@goffi.org> parents: 2773 diff changeset	1715 self.host.quit(C.EXIT_DATA_ERROR)
ff1b40823b07 jp (pubsub): new "transform" command: Goffi <goffi@goffi.org> parents: 2773 diff changeset	1716
ff1b40823b07 jp (pubsub): new "transform" command: Goffi <goffi@goffi.org> parents: 2773 diff changeset	1717 if not self.args.apply:
ff1b40823b07 jp (pubsub): new "transform" command: Goffi <goffi@goffi.org> parents: 2773 diff changeset	1718 # we have a dry run, we just display filtered items
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2910 diff changeset	1719 serialised = etree.tostring(element, encoding='unicode',
2777 ff1b40823b07 jp (pubsub): new "transform" command: Goffi <goffi@goffi.org> parents: 2773 diff changeset	1720 pretty_print=True)
ff1b40823b07 jp (pubsub): new "transform" command: Goffi <goffi@goffi.org> parents: 2773 diff changeset	1721 self.disp(serialised)
ff1b40823b07 jp (pubsub): new "transform" command: Goffi <goffi@goffi.org> parents: 2773 diff changeset	1722 else:
2803 d4a9a60bc650 jp (pubsub/transform): use new psItemsSend method, it is not needed anymore to send items one by one when --admin is not used Goffi <goffi@goffi.org> parents: 2802 diff changeset	1723 new_items.append(etree.tostring(element, encoding="unicode"))
2777 ff1b40823b07 jp (pubsub): new "transform" command: Goffi <goffi@goffi.org> parents: 2773 diff changeset	1724
ff1b40823b07 jp (pubsub): new "transform" command: Goffi <goffi@goffi.org> parents: 2773 diff changeset	1725 if not self.args.apply:
ff1b40823b07 jp (pubsub): new "transform" command: Goffi <goffi@goffi.org> parents: 2773 diff changeset	1726 # on dry run we have nothing to wait for, we can quit
ff1b40823b07 jp (pubsub): new "transform" command: Goffi <goffi@goffi.org> parents: 2773 diff changeset	1727 if self.args.all:
3312 77177b13ff54 plugin XEP-0060: serialise psItemsGet result with data_format Goffi <goffi@goffi.org> parents: 3275 diff changeset	1728 return await self.handleNextPage(ps_result)
2777 ff1b40823b07 jp (pubsub): new "transform" command: Goffi <goffi@goffi.org> parents: 2773 diff changeset	1729 self.host.quit()
2803 d4a9a60bc650 jp (pubsub/transform): use new psItemsSend method, it is not needed anymore to send items one by one when --admin is not used Goffi <goffi@goffi.org> parents: 2802 diff changeset	1730 else:
d4a9a60bc650 jp (pubsub/transform): use new psItemsSend method, it is not needed anymore to send items one by one when --admin is not used Goffi <goffi@goffi.org> parents: 2802 diff changeset	1731 if self.args.admin:
3040 fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	1732 bridge_method = self.host.bridge.psAdminItemsSend
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	1733 else:
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	1734 bridge_method = self.host.bridge.psItemsSend
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	1735
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	1736 try:
3312 77177b13ff54 plugin XEP-0060: serialise psItemsGet result with data_format Goffi <goffi@goffi.org> parents: 3275 diff changeset	1737 ps_items_send_result = await bridge_method(
2803 d4a9a60bc650 jp (pubsub/transform): use new psItemsSend method, it is not needed anymore to send items one by one when --admin is not used Goffi <goffi@goffi.org> parents: 2802 diff changeset	1738 self.args.service,
d4a9a60bc650 jp (pubsub/transform): use new psItemsSend method, it is not needed anymore to send items one by one when --admin is not used Goffi <goffi@goffi.org> parents: 2802 diff changeset	1739 self.args.node,
d4a9a60bc650 jp (pubsub/transform): use new psItemsSend method, it is not needed anymore to send items one by one when --admin is not used Goffi <goffi@goffi.org> parents: 2802 diff changeset	1740 new_items,
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2910 diff changeset	1741 "",
2803 d4a9a60bc650 jp (pubsub/transform): use new psItemsSend method, it is not needed anymore to send items one by one when --admin is not used Goffi <goffi@goffi.org> parents: 2802 diff changeset	1742 self.profile,
d4a9a60bc650 jp (pubsub/transform): use new psItemsSend method, it is not needed anymore to send items one by one when --admin is not used Goffi <goffi@goffi.org> parents: 2802 diff changeset	1743 )
3040 fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	1744 except Exception as e:
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	1745 self.disp(f"can't send item: {e}", error=True)
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	1746 self.host.quit(C.EXIT_BRIDGE_ERRBACK)
2803 d4a9a60bc650 jp (pubsub/transform): use new psItemsSend method, it is not needed anymore to send items one by one when --admin is not used Goffi <goffi@goffi.org> parents: 2802 diff changeset	1747 else:
3312 77177b13ff54 plugin XEP-0060: serialise psItemsGet result with data_format Goffi <goffi@goffi.org> parents: 3275 diff changeset	1748 await self.psItemsSendCb(ps_items_send_result, metadata=ps_result)
2777 ff1b40823b07 jp (pubsub): new "transform" command: Goffi <goffi@goffi.org> parents: 2773 diff changeset	1749
3040 fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	1750 async def start(self):
2777 ff1b40823b07 jp (pubsub): new "transform" command: Goffi <goffi@goffi.org> parents: 2773 diff changeset	1751 if self.args.all and self.args.order_by != C.ORDER_BY_CREATION:
ff1b40823b07 jp (pubsub): new "transform" command: Goffi <goffi@goffi.org> parents: 2773 diff changeset	1752 self.check_duplicates = True
ff1b40823b07 jp (pubsub): new "transform" command: Goffi <goffi@goffi.org> parents: 2773 diff changeset	1753 self.items_ids = []
ff1b40823b07 jp (pubsub): new "transform" command: Goffi <goffi@goffi.org> parents: 2773 diff changeset	1754 self.disp(A.color(
ff1b40823b07 jp (pubsub): new "transform" command: Goffi <goffi@goffi.org> parents: 2773 diff changeset	1755 A.FG_RED, A.BOLD,
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2910 diff changeset	1756 '/!\\ "--all" should be used with "--order-by creation" /!\\\n',
2777 ff1b40823b07 jp (pubsub): new "transform" command: Goffi <goffi@goffi.org> parents: 2773 diff changeset	1757 A.RESET,
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2910 diff changeset	1758 "We'll update items, so order may change during transformation,\n"
ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2910 diff changeset	1759 "we'll try to mitigate that by stopping on first duplicate,\n"
ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2910 diff changeset	1760 "but this method is not safe, and some items may be missed.\n---\n"))
2777 ff1b40823b07 jp (pubsub): new "transform" command: Goffi <goffi@goffi.org> parents: 2773 diff changeset	1761 else:
ff1b40823b07 jp (pubsub): new "transform" command: Goffi <goffi@goffi.org> parents: 2773 diff changeset	1762 self.check_duplicates = False
3040 fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	1763
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	1764 try:
3312 77177b13ff54 plugin XEP-0060: serialise psItemsGet result with data_format Goffi <goffi@goffi.org> parents: 3275 diff changeset	1765 ps_result = data_format.deserialise(
77177b13ff54 plugin XEP-0060: serialise psItemsGet result with data_format Goffi <goffi@goffi.org> parents: 3275 diff changeset	1766 await self.host.bridge.psItemsGet(
77177b13ff54 plugin XEP-0060: serialise psItemsGet result with data_format Goffi <goffi@goffi.org> parents: 3275 diff changeset	1767 self.args.service,
77177b13ff54 plugin XEP-0060: serialise psItemsGet result with data_format Goffi <goffi@goffi.org> parents: 3275 diff changeset	1768 self.args.node,
77177b13ff54 plugin XEP-0060: serialise psItemsGet result with data_format Goffi <goffi@goffi.org> parents: 3275 diff changeset	1769 self.args.max,
77177b13ff54 plugin XEP-0060: serialise psItemsGet result with data_format Goffi <goffi@goffi.org> parents: 3275 diff changeset	1770 self.args.items,
77177b13ff54 plugin XEP-0060: serialise psItemsGet result with data_format Goffi <goffi@goffi.org> parents: 3275 diff changeset	1771 "",
77177b13ff54 plugin XEP-0060: serialise psItemsGet result with data_format Goffi <goffi@goffi.org> parents: 3275 diff changeset	1772 self.getPubsubExtra(),
77177b13ff54 plugin XEP-0060: serialise psItemsGet result with data_format Goffi <goffi@goffi.org> parents: 3275 diff changeset	1773 self.profile,
77177b13ff54 plugin XEP-0060: serialise psItemsGet result with data_format Goffi <goffi@goffi.org> parents: 3275 diff changeset	1774 )
3040 fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	1775 )
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	1776 except Exception as e:
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	1777 self.disp(f"can't retrieve items: {e}", error=True)
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	1778 self.host.quit(C.EXIT_BRIDGE_ERRBACK)
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	1779 else:
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	1780 await self.psItemsGetCb(ps_result)
2777 ff1b40823b07 jp (pubsub): new "transform" command: Goffi <goffi@goffi.org> parents: 2773 diff changeset	1781
ff1b40823b07 jp (pubsub): new "transform" command: Goffi <goffi@goffi.org> parents: 2773 diff changeset	1782
2235 4db836386641 jp: added use_pubsub and use_pubsub_node_req arguments to CommandBase Goffi <goffi@goffi.org> parents: 2224 diff changeset	1783 class Uri(base.CommandBase):
2224 87fcd4a7c7e4 jp (pubsub): added uri command to build pubsub URI Goffi <goffi@goffi.org> parents: 2221 diff changeset	1784 def __init__(self, host):
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1785 base.CommandBase.__init__(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1786 self,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1787 host,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1788 "uri",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1789 use_profile=False,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1790 use_pubsub=True,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1791 pubsub_flags={C.NODE, C.SINGLE_ITEM},
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2910 diff changeset	1792 help=_("build URI"),
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1793 )
2224 87fcd4a7c7e4 jp (pubsub): added uri command to build pubsub URI Goffi <goffi@goffi.org> parents: 2221 diff changeset	1794
87fcd4a7c7e4 jp (pubsub): added uri command to build pubsub URI Goffi <goffi@goffi.org> parents: 2221 diff changeset	1795 def add_parser_options(self):
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1796 self.parser.add_argument(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1797 "-p",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1798 "--profile",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1799 default=C.PROF_KEY_DEFAULT,
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2910 diff changeset	1800 help=_("profile (used when no server is specified)"),
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1801 )
2224 87fcd4a7c7e4 jp (pubsub): added uri command to build pubsub URI Goffi <goffi@goffi.org> parents: 2221 diff changeset	1802
87fcd4a7c7e4 jp (pubsub): added uri command to build pubsub URI Goffi <goffi@goffi.org> parents: 2221 diff changeset	1803 def display_uri(self, jid_):
2239 17502e74c046 jp (pubsub/uri): fixed URI generation Goffi <goffi@goffi.org> parents: 2235 diff changeset	1804 uri_args = {}
2224 87fcd4a7c7e4 jp (pubsub): added uri command to build pubsub URI Goffi <goffi@goffi.org> parents: 2221 diff changeset	1805 if not self.args.service:
87fcd4a7c7e4 jp (pubsub): added uri command to build pubsub URI Goffi <goffi@goffi.org> parents: 2221 diff changeset	1806 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	1807
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1808 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	1809 value = getattr(self.args, key)
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1810 if key == "service":
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1811 key = "path"
2224 87fcd4a7c7e4 jp (pubsub): added uri command to build pubsub URI Goffi <goffi@goffi.org> parents: 2221 diff changeset	1812 if value:
87fcd4a7c7e4 jp (pubsub): added uri command to build pubsub URI Goffi <goffi@goffi.org> parents: 2221 diff changeset	1813 uri_args[key] = value
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2910 diff changeset	1814 self.disp(uri.buildXMPPUri("pubsub", **uri_args))
2224 87fcd4a7c7e4 jp (pubsub): added uri command to build pubsub URI Goffi <goffi@goffi.org> parents: 2221 diff changeset	1815 self.host.quit()
87fcd4a7c7e4 jp (pubsub): added uri command to build pubsub URI Goffi <goffi@goffi.org> parents: 2221 diff changeset	1816
3040 fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	1817 async def start(self):
2224 87fcd4a7c7e4 jp (pubsub): added uri command to build pubsub URI Goffi <goffi@goffi.org> parents: 2221 diff changeset	1818 if not self.args.service:
3040 fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	1819 try:
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	1820 jid_ = await self.host.bridge.asyncGetParamA(
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	1821 "JabberID",
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	1822 "Connection",
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	1823 profile_key=self.args.profile
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	1824 )
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	1825 except Exception as e:
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	1826 self.disp(f"can't retrieve jid: {e}", error=True)
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	1827 self.host.quit(C.EXIT_BRIDGE_ERRBACK)
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	1828 else:
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	1829 self.display_uri(jid_)
2224 87fcd4a7c7e4 jp (pubsub): added uri command to build pubsub URI Goffi <goffi@goffi.org> parents: 2221 diff changeset	1830 else:
87fcd4a7c7e4 jp (pubsub): added uri command to build pubsub URI Goffi <goffi@goffi.org> parents: 2221 diff changeset	1831 self.display_uri(None)
87fcd4a7c7e4 jp (pubsub): added uri command to build pubsub URI Goffi <goffi@goffi.org> parents: 2221 diff changeset	1832
87fcd4a7c7e4 jp (pubsub): added uri command to build pubsub URI Goffi <goffi@goffi.org> parents: 2221 diff changeset	1833
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	1834 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	1835 def __init__(self, host):
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1836 base.CommandBase.__init__(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1837 self,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1838 host,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1839 "create",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1840 use_pubsub=True,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1841 pubsub_flags={C.NODE},
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2910 diff changeset	1842 help=_("create a Pubsub hook"),
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1843 )
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	1844
0b21d87c91cf jp (pubsub/hook): added create/delete/list hook command to handle new Pubsub hook feature Goffi <goffi@goffi.org> parents: 2292 diff changeset	1845 def add_parser_options(self):
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1846 self.parser.add_argument(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1847 "-t",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1848 "--type",
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2910 diff changeset	1849 default="python",
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1850 choices=("python", "python_file", "python_code"),
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2910 diff changeset	1851 help=_("hook type"),
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1852 )
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1853 self.parser.add_argument(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1854 "-P",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1855 "--persistent",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1856 action="store_true",
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2910 diff changeset	1857 help=_("make hook persistent across restarts"),
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1858 )
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1859 self.parser.add_argument(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1860 "hook_arg",
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2910 diff changeset	1861 help=_("argument of the hook (depend of the type)"),
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1862 )
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	1863
0b21d87c91cf jp (pubsub/hook): added create/delete/list hook command to handle new Pubsub hook feature Goffi <goffi@goffi.org> parents: 2292 diff changeset	1864 @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	1865 def checkArgs(self):
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2910 diff changeset	1866 if self.args.type == "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	1867 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	1868 if not os.path.isfile(self.args.hook_arg):
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1869 self.parser.error(
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2910 diff changeset	1870 _("{path} is not a file").format(path=self.args.hook_arg)
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1871 )
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	1872
3040 fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	1873 async def start(self):
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	1874 self.checkArgs(self)
3040 fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	1875 try:
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	1876 await self.host.bridge.psHookAdd(
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	1877 self.args.service,
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	1878 self.args.node,
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	1879 self.args.type,
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	1880 self.args.hook_arg,
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	1881 self.args.persistent,
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	1882 self.profile,
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	1883 )
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	1884 except Exception as e:
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	1885 self.disp(f"can't create hook: {e}", error=True)
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	1886 self.host.quit(C.EXIT_BRIDGE_ERRBACK)
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	1887 else:
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	1888 self.host.quit()
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	1889
0b21d87c91cf jp (pubsub/hook): added create/delete/list hook command to handle new Pubsub hook feature Goffi <goffi@goffi.org> parents: 2292 diff changeset	1890
0b21d87c91cf jp (pubsub/hook): added create/delete/list hook command to handle new Pubsub hook feature Goffi <goffi@goffi.org> parents: 2292 diff changeset	1891 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	1892 def __init__(self, host):
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1893 base.CommandBase.__init__(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1894 self,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1895 host,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1896 "delete",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1897 use_pubsub=True,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1898 pubsub_flags={C.NODE},
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2910 diff changeset	1899 help=_("delete a Pubsub hook"),
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1900 )
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	1901
0b21d87c91cf jp (pubsub/hook): added create/delete/list hook command to handle new Pubsub hook feature Goffi <goffi@goffi.org> parents: 2292 diff changeset	1902 def add_parser_options(self):
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1903 self.parser.add_argument(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1904 "-t",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1905 "--type",
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2910 diff changeset	1906 default="",
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1907 choices=("", "python", "python_file", "python_code"),
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2910 diff changeset	1908 help=_("hook type to remove, empty to remove all (DEFAULT: remove all)"),
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1909 )
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1910 self.parser.add_argument(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1911 "-a",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1912 "--arg",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1913 dest="hook_arg",
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2910 diff changeset	1914 default="",
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1915 help=_(
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2910 diff changeset	1916 "argument of the hook to remove, empty to remove all (DEFAULT: remove all)"
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1917 ),
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1918 )
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	1919
3040 fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	1920 async def start(self):
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	1921 HookCreate.checkArgs(self)
3040 fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	1922 try:
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	1923 nb_deleted = await self.host.bridge.psHookRemove(
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	1924 self.args.service,
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	1925 self.args.node,
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	1926 self.args.type,
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	1927 self.args.hook_arg,
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	1928 self.profile,
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	1929 )
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	1930 except Exception as e:
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	1931 self.disp(f"can't delete hook: {e}", error=True)
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	1932 self.host.quit(C.EXIT_BRIDGE_ERRBACK)
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	1933 else:
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	1934 self.disp(_(f"{nb_deleted} hook(s) have been deleted"))
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	1935 self.host.quit()
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	1936
0b21d87c91cf jp (pubsub/hook): added create/delete/list hook command to handle new Pubsub hook feature Goffi <goffi@goffi.org> parents: 2292 diff changeset	1937
0b21d87c91cf jp (pubsub/hook): added create/delete/list hook command to handle new Pubsub hook feature Goffi <goffi@goffi.org> parents: 2292 diff changeset	1938 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	1939 def __init__(self, host):
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1940 base.CommandBase.__init__(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1941 self,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1942 host,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1943 "list",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1944 use_output=C.OUTPUT_LIST_DICT,
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2910 diff changeset	1945 help=_("list hooks of a profile"),
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1946 )
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	1947
0b21d87c91cf jp (pubsub/hook): added create/delete/list hook command to handle new Pubsub hook feature Goffi <goffi@goffi.org> parents: 2292 diff changeset	1948 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	1949 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	1950
3040 fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	1951 async def start(self):
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	1952 try:
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	1953 data = await self.host.bridge.psHookList(
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	1954 self.profile,
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	1955 )
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	1956 except Exception as e:
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	1957 self.disp(f"can't list hooks: {e}", error=True)
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	1958 self.host.quit(C.EXIT_BRIDGE_ERRBACK)
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	1959 else:
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	1960 if not data:
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	1961 self.disp(_("No hook found."))
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	1962 await self.output(data)
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	1963 self.host.quit()
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	1964
0b21d87c91cf jp (pubsub/hook): added create/delete/list hook command to handle new Pubsub hook feature Goffi <goffi@goffi.org> parents: 2292 diff changeset	1965
0b21d87c91cf jp (pubsub/hook): added create/delete/list hook command to handle new Pubsub hook feature Goffi <goffi@goffi.org> parents: 2292 diff changeset	1966 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	1967 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	1968
0b21d87c91cf jp (pubsub/hook): added create/delete/list hook command to handle new Pubsub hook feature Goffi <goffi@goffi.org> parents: 2292 diff changeset	1969 def __init__(self, host):
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1970 super(Hook, self).__init__(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1971 host,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1972 "hook",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1973 use_profile=False,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1974 use_verbose=True,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1975 help=_("trigger action on Pubsub notifications"),
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1976 )
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	1977
0b21d87c91cf jp (pubsub/hook): added create/delete/list hook command to handle new Pubsub hook feature Goffi <goffi@goffi.org> parents: 2292 diff changeset	1978
2191 a1a8233f89e8 jp(pubsub/get): pubsub/get command, first draft Goffi <goffi@goffi.org> parents: diff changeset	1979 class Pubsub(base.CommandBase):
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1980 subcommands = (
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1981 Set,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1982 Get,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1983 Delete,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1984 Edit,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1985 Subscribe,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1986 Unsubscribe,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1987 Subscriptions,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1988 Node,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1989 Affiliations,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1990 Search,
2777 ff1b40823b07 jp (pubsub): new "transform" command: Goffi <goffi@goffi.org> parents: 2773 diff changeset	1991 Transform,
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1992 Hook,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1993 Uri,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1994 )
2191 a1a8233f89e8 jp(pubsub/get): pubsub/get command, first draft Goffi <goffi@goffi.org> parents: diff changeset	1995
a1a8233f89e8 jp(pubsub/get): pubsub/get command, first draft Goffi <goffi@goffi.org> parents: diff changeset	1996 def __init__(self, host):
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1997 super(Pubsub, self).__init__(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1998 host, "pubsub", use_profile=False, help=_("PubSub nodes/items management")
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	1999 )

Mercurial > libervia-backend

annotate sat_frontends/jp/cmd_pubsub.py @ 3498:d78b5eae912a