Mercurial > libervia-backend

comparison sat_frontends/jp/cmd_ticket.py @ 3458:b68346a52920

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
Backed out changeset 6deea0d8d0e7 This patch has been merged by mistake, it will be applied later.
author Goffi <goffi@goffi.org>
date Thu, 04 Feb 2021 21:07:49 +0100
parents
children
comparison
equal deleted inserted replaced
3457:6791103de47d 3458:b68346a52920
1 #!/usr/bin/env python3
2
3
4 # jp: a SàT command line tool
5 # Copyright (C) 2009-2020 Jérôme Poisson (goffi@goffi.org)
6
7 # This program is free software: you can redistribute it and/or modify
8 # it under the terms of the GNU Affero General Public License as published by
9 # the Free Software Foundation, either version 3 of the License, or
10 # (at your option) any later version.
11
12 # This program is distributed in the hope that it will be useful,
13 # but WITHOUT ANY WARRANTY; without even the implied warranty of
14 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
15 # GNU Affero General Public License for more details.
16
17 # You should have received a copy of the GNU Affero General Public License
18 # along with this program. If not, see <http://www.gnu.org/licenses/>.
19
20
21 import json
22 import os
23 from sat.core.i18n import _
24 from sat.tools.common import data_format
25 from sat_frontends.jp import common
26 from sat_frontends.jp.constants import Const as C
27 from . import base
28
29 __commands__ = ["Ticket"]
30
31 FIELDS_MAP = "mapping"
32
33
34 class Get(base.CommandBase):
35 def __init__(self, host):
36 base.CommandBase.__init__(
37 self,
38 host,
39 "get",
40 use_verbose=True,
41 use_pubsub=True,
42 pubsub_flags={C.MULTI_ITEMS},
43 pubsub_defaults={"service": _("auto"), "node": _("auto")},
44 use_output=C.OUTPUT_LIST_XMLUI,
45 help=_("get tickets"),
46 )
47
48 def add_parser_options(self):
49 pass
50
51 async def start(self):
52 await common.fill_well_known_uri(self, os.getcwd(), "tickets", meta_map={})
53 try:
54 tickets_data = data_format.deserialise(
55 await self.host.bridge.ticketsGet(
56 self.args.service,
57 self.args.node,
58 self.args.max,
59 self.args.items,
60 "",
61 self.getPubsubExtra(),
62 self.profile,
63 ),
64 type_check=list
65 )
66 except Exception as e:
67 self.disp(f"can't get tickets: {e}", error=True)
68 self.host.quit(C.EXIT_BRIDGE_ERRBACK)
69 else:
70 await self.output(tickets_data[0])
71 self.host.quit(C.EXIT_OK)
72
73
74 class Import(base.CommandBase):
75 # TODO: factorize with blog/import
76
77 def __init__(self, host):
78 super(Import, self).__init__(
79 host,
80 "import",
81 use_progress=True,
82 use_verbose=True,
83 help=_("import tickets from external software/dataset"),
84 )
85
86 def add_parser_options(self):
87 self.parser.add_argument(
88 "importer",
89 nargs="?",
90 help=_("importer name, nothing to display importers list"),
91 )
92 self.parser.add_argument(
93 "-o",
94 "--option",
95 action="append",
96 nargs=2,
97 default=[],
98 metavar=("NAME", "VALUE"),
99 help=_("importer specific options (see importer description)"),
100 )
101 self.parser.add_argument(
102 "-m",
103 "--map",
104 action="append",
105 nargs=2,
106 default=[],
107 metavar=("IMPORTED_FIELD", "DEST_FIELD"),
108 help=_(
109 "specified field in import data will be put in dest field (default: use "
110 "same field name, or ignore if it doesn't exist)"
111 ),
112 )
113 self.parser.add_argument(
114 "-s",
115 "--service",
116 default="",
117 metavar="PUBSUB_SERVICE",
118 help=_("PubSub service where the items must be uploaded (default: server)"),
119 )
120 self.parser.add_argument(
121 "-n",
122 "--node",
123 default="",
124 metavar="PUBSUB_NODE",
125 help=_(
126 "PubSub node where the items must be uploaded (default: tickets' "
127 "defaults)"
128 ),
129 )
130 self.parser.add_argument(
131 "location",
132 nargs="?",
133 help=_(
134 "importer data location (see importer description), nothing to show "
135 "importer description"
136 ),
137 )
138
139 async def onProgressStarted(self, metadata):
140 self.disp(_("Tickets upload started"), 2)
141
142 async def onProgressFinished(self, metadata):
143 self.disp(_("Tickets uploaded successfully"), 2)
144
145 async def onProgressError(self, error_msg):
146 self.disp(_(f"Error while uploading tickets: {error_msg}"), error=True)
147
148 async def start(self):
149 if self.args.location is None:
150 # no location, the list of importer or description is requested
151 for name in ("option", "service", "node"):
152 if getattr(self.args, name):
153 self.parser.error(
154 _(f"{name} argument can't be used without location argument"))
155 if self.args.importer is None:
156 self.disp(
157 "\n".join(
158 [
159 f"{name}: {desc}"
160 for name, desc in await self.host.bridge.ticketsImportList()
161 ]
162 )
163 )
164 else:
165 try:
166 short_desc, long_desc = await self.host.bridge.ticketsImportDesc(
167 self.args.importer
168 )
169 except Exception as e:
170 self.disp(f"can't get importer description: {e}", error=True)
171 self.host.quit(C.EXIT_BRIDGE_ERRBACK)
172 else:
173 self.disp(f"{name}: {short_desc}\n\n{long_desc}")
174 self.host.quit()
175 else:
176 # we have a location, an import is requested
177
178 if self.args.progress:
179 # we use a custom progress bar template as we want a counter
180 self.pbar_template = [
181 _("Progress: "), ["Percentage"], " ", ["Bar"], " ",
182 ["Counter"], " ", ["ETA"]
183 ]
184
185 options = {key: value for key, value in self.args.option}
186 fields_map = dict(self.args.map)
187 if fields_map:
188 if FIELDS_MAP in options:
189 self.parser.error(
190 _("fields_map must be specified either preencoded in --option or "
191 "using --map, but not both at the same time")
192 )
193 options[FIELDS_MAP] = json.dumps(fields_map)
194
195 try:
196 progress_id = await self.host.bridge.ticketsImport(
197 self.args.importer,
198 self.args.location,
199 options,
200 self.args.service,
201 self.args.node,
202 self.profile,
203 )
204 except Exception as e:
205 self.disp(
206 _(f"Error while trying to import tickets: {e}"),
207 error=True,
208 )
209 self.host.quit(C.EXIT_BRIDGE_ERRBACK)
210 else:
211 await self.set_progress_id(progress_id)
212
213
214 class Ticket(base.CommandBase):
215 subcommands = (Get, Import)
216
217 def __init__(self, host):
218 super(Ticket, self).__init__(
219 host, "ticket", use_profile=False, help=_("tickets handling")
220 )