diff sat_frontends/jp/cmd_debug.py @ 2562:26edcf3a30eb

core, setup: huge cleaning: - moved directories from src and frontends/src to sat and sat_frontends, which is the recommanded naming convention - move twisted directory to root - removed all hacks from setup.py, and added missing dependencies, it is now clean - use https URL for website in setup.py - removed "Environment :: X11 Applications :: GTK", as wix is deprecated and removed - renamed sat.sh to sat and fixed its installation - added python_requires to specify Python version needed - replaced glib2reactor which use deprecated code by gtk3reactor sat can now be installed directly from virtualenv without using --system-site-packages anymore \o/
author Goffi <goffi@goffi.org>
date Mon, 02 Apr 2018 19:44:50 +0200
parents frontends/src/jp/cmd_debug.py@0046283a285d
children 56f94936df1e
line wrap: on
line diff
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/sat_frontends/jp/cmd_debug.py	Mon Apr 02 19:44:50 2018 +0200
@@ -0,0 +1,160 @@
+#!/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 <http://www.gnu.org/licenses/>.
+
+
+import base
+from sat.core.i18n import _
+from sat_frontends.jp.constants import Const as C
+from sat.tools.common.ansi import ANSI as A
+import json
+
+__commands__ = ["Debug"]
+
+
+class BridgeCommon(object):
+
+    def evalArgs(self):
+        if self.args.arg:
+            try:
+                return eval(u'[{}]'.format(u",".join(self.args.arg)))
+            except SyntaxError as e:
+                self.disp(u"Can't evaluate arguments: {mess}\n{text}\n{offset}^".format(
+                    mess=e,
+                    text=e.text.decode('utf-8'),
+                    offset=u" "*(e.offset-1)
+                    ), error=True)
+                self.host.quit(C.EXIT_BAD_ARG)
+        else:
+            return []
+
+
+class Method(base.CommandBase, BridgeCommon):
+
+    def __init__(self, host):
+        base.CommandBase.__init__(self, host, 'method', help=_(u'call a bridge method'))
+        BridgeCommon.__init__(self)
+        self.need_loop=True
+
+    def add_parser_options(self):
+        self.parser.add_argument("method", type=str, help=_(u"name of the method to execute"))
+        self.parser.add_argument("arg", type=base.unicode_decoder, nargs="*", help=_(u"argument of the method"))
+
+    def method_cb(self, ret=None):
+        if ret is not None:
+            self.disp(unicode(ret))
+        self.host.quit()
+
+    def method_eb(self, failure):
+        self.disp(_(u"Error while executing {}: {}".format(self.args.method, failure)), error=True)
+        self.host.quit(C.EXIT_ERROR)
+
+    def start(self):
+        method = getattr(self.host.bridge, self.args.method)
+        args = self.evalArgs()
+        try:
+            method(*args, profile=self.profile, callback=self.method_cb, errback=self.method_eb)
+        except TypeError:
+            # maybe the method doesn't need a profile ?
+            try:
+                method(*args, callback=self.method_cb, errback=self.method_eb)
+            except TypeError:
+                self.method_eb(_(u"bad arguments"))
+
+
+class Signal(base.CommandBase, BridgeCommon):
+
+    def __init__(self, host):
+        base.CommandBase.__init__(self, host, 'signal', help=_(u'send a fake signal from backend'))
+        BridgeCommon.__init__(self)
+
+    def add_parser_options(self):
+        self.parser.add_argument("signal", type=str, help=_(u"name of the signal to send"))
+        self.parser.add_argument("arg", type=base.unicode_decoder, nargs="*", help=_(u"argument of the signal"))
+
+    def start(self):
+        args = self.evalArgs()
+        json_args = json.dumps(args)
+        # XXX: we use self.args.profile and not self.profile
+        #      because we want the raw profile_key (so plugin handle C.PROF_KEY_NONE)
+        self.host.bridge.debugFakeSignal(self.args.signal, json_args, self.args.profile)
+
+
+class Bridge(base.CommandBase):
+    subcommands = (Method, Signal)
+
+    def __init__(self, host):
+        super(Bridge, self).__init__(host, 'bridge', use_profile=False, help=_('bridge s(t)imulation'))
+
+
+class Monitor(base.CommandBase):
+
+    def __init__(self, host):
+        super(Monitor, self).__init__(host,
+                                      'monitor',
+                                      use_verbose=True,
+                                      use_profile=False,
+                                      use_output=C.OUTPUT_XML,
+                                      help=_('monitor XML stream'))
+        self.need_loop = True
+
+    def add_parser_options(self):
+        self.parser.add_argument("-d", "--direction", choices=('in', 'out', 'both'), default='both', help=_(u"stream direction filter"))
+
+    def printXML(self, direction, xml_data, profile):
+        if self.args.direction == 'in' and direction != 'IN':
+            return
+        if self.args.direction == 'out' and direction != 'OUT':
+            return
+        verbosity = self.host.verbosity
+        if not xml_data.strip():
+            if verbosity <= 2:
+                return
+            whiteping = True
+        else:
+            whiteping = False
+
+        if verbosity:
+            profile_disp = u' ({})'.format(profile) if verbosity>1 else u''
+            if direction == 'IN':
+                self.disp(A.color(A.BOLD, A.FG_YELLOW, '<<<===== IN ====', A.FG_WHITE, profile_disp))
+            else:
+                self.disp(A.color(A.BOLD, A.FG_CYAN, '==== OUT ====>>>', A.FG_WHITE, profile_disp))
+        if whiteping:
+            self.disp('[WHITESPACE PING]')
+        else:
+            try:
+                self.output(xml_data)
+            except Exception:
+                # initial stream is not valid XML,
+                # in this case we print directly to data
+                # FIXME: we should test directly lxml.etree.XMLSyntaxError
+                #        but importing lxml directly here is not clean
+                #        should be wrapped in a custom Exception
+                self.disp(xml_data)
+                self.disp(u'')
+
+    def start(self):
+        self.host.bridge.register_signal('xmlLog', self.printXML, 'plugin')
+
+
+class Debug(base.CommandBase):
+    subcommands = (Bridge, Monitor)
+
+    def __init__(self, host):
+        super(Debug, self).__init__(host, 'debug', use_profile=False, help=_('debugging tools'))