# HG changeset patch # User Goffi # Date 1517296637 -3600 # Node ID 512c443a58bacaa43c3c09b1d505a3369ec600b1 # Parent 785b6a1cef0aa20856664d7d6b504c4a9b6802cd jp (forums): forums handling commands, first draft diff -r 785b6a1cef0a -r 512c443a58ba frontends/src/jp/cmd_blog.py --- a/frontends/src/jp/cmd_blog.py Tue Jan 30 08:17:08 2018 +0100 +++ b/frontends/src/jp/cmd_blog.py Tue Jan 30 08:17:17 2018 +0100 @@ -175,7 +175,7 @@ def __init__(self, host): extra_outputs = {'default': self.default_output, 'fancy': self.fancy_output} - base.CommandBase.__init__(self, host, 'get', use_verbose=True, use_output='complex', extra_outputs=extra_outputs, help=_(u'get blog item(s)')) + base.CommandBase.__init__(self, host, 'get', use_verbose=True, use_output=C.OUTPUT_COMPLEX, extra_outputs=extra_outputs, help=_(u'get blog item(s)')) BlogCommon.__init__(self, self.host) self.need_loop=True diff -r 785b6a1cef0a -r 512c443a58ba frontends/src/jp/cmd_forums.py --- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/frontends/src/jp/cmd_forums.py Tue Jan 30 08:17:17 2018 +0100 @@ -0,0 +1,166 @@ +#!/usr/bin/env python2 +# -*- coding: utf-8 -*- + +# jp: a SàT command line tool +# Copyright (C) 2009-2018 Jérôme Poisson (goffi@goffi.org) + +# This program is free software: you can redistribute it and/or modify +# it under the terms of the GNU Affero General Public License as published by +# the Free Software Foundation, either version 3 of the License, or +# (at your option) any later version. + +# This program is distributed in the hope that it will be useful, +# but WITHOUT ANY WARRANTY; without even the implied warranty of +# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the +# GNU Affero General Public License for more details. + +# You should have received a copy of the GNU Affero General Public License +# along with this program. If not, see . + + +import base +from sat.core.i18n import _ +from sat_frontends.jp.constants import Const as C +from sat_frontends.jp import common +from sat.tools.common.ansi import ANSI as A +from functools import partial +import json + +__commands__ = ["Forums"] + +FORUMS_TMP_DIR = u"forums" + + +class Edit(base.CommandBase, common.BaseEdit): + use_items=False + + def __init__(self, host): + base.CommandBase.__init__(self, host, 'edit', use_pubsub=True, use_verbose=True, help=_(u'edit forums')) + common.BaseEdit.__init__(self, self.host, FORUMS_TMP_DIR) + self.need_loop=True + + def add_parser_options(self): + common.BaseEdit.add_parser_options(self) + self.parser.add_argument("-k", "--key", type=base.unicode_decoder, default=u'', + help=_(u"forum key (DEFAULT: default forums)")) + + def getTmpSuff(self): + """return suffix used for content file""" + return u'json' + + def forumsSetCb(self): + self.disp(_(u'forums have been edited'), 1) + self.host.quit() + + def publish(self, forums_raw): + self.host.bridge.forumsSet( + forums_raw, + self.args.service, + self.args.node, + self.args.key, + self.profile, + callback=self.forumsSetCb, + errback=partial(self.errback, + msg=_(u"can't set forums: {}"), + exit_code=C.EXIT_BRIDGE_ERRBACK)) + + def forumsGetCb(self, forums_json): + content_file_obj, content_file_path = self.getTmpFile() + forums_json = forums_json.strip() + if forums_json: + # we loads and dumps to have pretty printed json + forums = json.loads(forums_json) + json.dump(forums, content_file_obj, ensure_ascii=False, indent=4) + content_file_obj.seek(0) + self.runEditor("forums_editor_args", content_file_path, content_file_obj) + + def forumsGetEb(self, failure_): + # FIXME: error handling with bridge is broken, need to be properly fixed + if failure_.condition == u'item-not-found': + self.forumsGetCb(u'') + else: + self.errback(failure_, + msg=_(u"can't get forums structure: {}"), + exit_code=C.EXIT_BRIDGE_ERRBACK) + + def start(self): + common.checkURI(self.args) + self.host.bridge.forumsGet( + self.args.service, + self.args.node, + self.args.key, + self.profile, + callback=self.forumsGetCb, + errback=self.forumsGetEb) + + +class Get(base.CommandBase): + + def __init__(self, host): + extra_outputs = {'default': self.default_output} + base.CommandBase.__init__(self, host, 'get', use_output=C.OUTPUT_COMPLEX, extra_outputs=extra_outputs, use_pubsub=True, use_verbose=True, help=_(u'get forums structure')) + self.need_loop=True + + def add_parser_options(self): + self.parser.add_argument("-k", "--key", type=base.unicode_decoder, default=u'', + help=_(u"forum key (DEFAULT: default forums)")) + + def default_output(self, forums, level=0): + for forum in forums: + keys = list(forum.keys()) + keys.sort() + try: + keys.remove(u'title') + except ValueError: + pass + else: + keys.insert(0, u'title') + try: + keys.remove(u'sub-forums') + except ValueError: + pass + else: + keys.append(u'sub-forums') + + for key in keys: + value = forum[key] + if key == 'sub-forums': + self.default_output(value, level+1) + else: + if self.host.verbosity < 1 and key != u'title': + continue + head_color = C.A_LEVEL_COLORS[level % len(C.A_LEVEL_COLORS)] + self.disp(A.color(level * 4 * u' ', + head_color, + key, + A.RESET, + u': ', + value)) + + def forumsGetCb(self, forums_raw): + if not forums_raw: + self.disp(_(u'no schema found'), 1) + self.host.quit(1) + forums = json.loads(forums_raw) + self.output(forums) + self.host.quit() + + def start(self): + common.checkURI(self.args) + self.host.bridge.forumsGet( + self.args.service, + self.args.node, + self.args.key, + self.profile, + callback=self.forumsGetCb, + errback=partial(self.errback, + msg=_(u"can't get forums: {}"), + exit_code=C.EXIT_BRIDGE_ERRBACK)) + + + +class Forums(base.CommandBase): + subcommands = (Get, Edit) + + def __init__(self, host): + super(Forums, self).__init__(host, 'forums', use_profile=False, help=_(u'Forums structure edition')) diff -r 785b6a1cef0a -r 512c443a58ba frontends/src/jp/constants.py --- a/frontends/src/jp/constants.py Tue Jan 30 08:17:08 2018 +0100 +++ b/frontends/src/jp/constants.py Tue Jan 30 08:17:17 2018 +0100 @@ -39,6 +39,8 @@ # ANSI A_HEADER = A.BOLD + A.FG_YELLOW A_SUBHEADER = A.BOLD + A.FG_RED + # A_LEVEL_COLORS may be used to cycle on colors according to depth of data + A_LEVEL_COLORS = (A_HEADER, A.BOLD + A.FG_BLUE, A.FG_MAGENTA, A.FG_CYAN) A_SUCCESS = A.BOLD + A.FG_GREEN A_FAILURE = A.BOLD + A.FG_RED # A_PROMPT_* is for shell