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

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 )