Mercurial > libervia-backend
diff sat_frontends/jp/cmd_pipe.py @ 2624:56f94936df1e
code style reformatting using black
author | Goffi <goffi@goffi.org> |
---|---|
date | Wed, 27 Jun 2018 20:14:46 +0200 |
parents | 26edcf3a30eb |
children | 003b8b4b56a7 |
line wrap: on
line diff
--- a/sat_frontends/jp/cmd_pipe.py Wed Jun 27 07:51:29 2018 +0200 +++ b/sat_frontends/jp/cmd_pipe.py Wed Jun 27 20:14:46 2018 +0200 @@ -23,7 +23,7 @@ import sys from sat.core.i18n import _ from sat_frontends.tools import jid -import xml.etree.ElementTree as ET # FIXME: used temporarily to manage XMLUI +import xml.etree.ElementTree as ET # FIXME: used temporarily to manage XMLUI from functools import partial import socket import SocketServer @@ -33,18 +33,20 @@ START_PORT = 9999 + class PipeOut(base.CommandBase): - def __init__(self, host): - super(PipeOut, self).__init__(host, 'out', help=_('send a pipe a stream')) + super(PipeOut, self).__init__(host, "out", help=_("send a pipe a stream")) self.need_loop = True def add_parser_options(self): - self.parser.add_argument("jid", type=base.unicode_decoder, help=_("the destination jid")) + self.parser.add_argument( + "jid", type=base.unicode_decoder, help=_("the destination jid") + ) def streamOutCb(self, port): s = socket.socket(socket.AF_INET, socket.SOCK_STREAM) - s.connect(('127.0.0.1', int(port))) + s.connect(("127.0.0.1", int(port))) while True: buf = sys.stdin.read(4096) if not buf: @@ -53,7 +55,7 @@ s.sendall(buf) except socket.error as e: if e.errno == errno.EPIPE: - sys.stderr.write(str(e) + '\n') + sys.stderr.write(str(e) + "\n") self.host.quit(1) else: raise e @@ -65,13 +67,15 @@ self.host.get_full_jid(self.args.jid), self.profile, callback=self.streamOutCb, - errback=partial(self.errback, - msg=_(u"can't start stream: {}"), - exit_code=C.EXIT_BRIDGE_ERRBACK)) + errback=partial( + self.errback, + msg=_(u"can't start stream: {}"), + exit_code=C.EXIT_BRIDGE_ERRBACK, + ), + ) class StreamServer(SocketServer.BaseRequestHandler): - def handle(self): while True: data = self.request.recv(4096) @@ -81,33 +85,37 @@ try: sys.stdout.flush() except IOError as e: - sys.stderr.write(str(e) + '\n') + sys.stderr.write(str(e) + "\n") break - # calling shutdown will do a deadlock as we don't use separate thread + # calling shutdown will do a deadlock as we don't use separate thread # this is a workaround (cf. https://stackoverflow.com/a/36017741) self.server._BaseServer__shutdown_request = True class PipeIn(base.CommandAnswering): - def __init__(self, host): - super(PipeIn, self).__init__(host, 'in', help=_('receive a pipe stream')) + super(PipeIn, self).__init__(host, "in", help=_("receive a pipe stream")) self.action_callbacks = {"STREAM": self.onStreamAction} def add_parser_options(self): - self.parser.add_argument("jids", type=base.unicode_decoder, nargs="*", help=_('Jids accepted (none means "accept everything")')) + self.parser.add_argument( + "jids", + type=base.unicode_decoder, + nargs="*", + help=_('Jids accepted (none means "accept everything")'), + ) def getXmluiId(self, action_data): # FIXME: we temporarily use ElementTree, but a real XMLUI managing module # should be available in the future # TODO: XMLUI module try: - xml_ui = action_data['xmlui'] + xml_ui = action_data["xmlui"] except KeyError: self.disp(_(u"Action has no XMLUI"), 1) else: - ui = ET.fromstring(xml_ui.encode('utf-8')) - xmlui_id = ui.get('submit') + ui = ET.fromstring(xml_ui.encode("utf-8")) + xmlui_id = ui.get("submit") if not xmlui_id: self.disp(_(u"Invalid XMLUI received"), error=True) return xmlui_id @@ -117,7 +125,7 @@ if xmlui_id is None: return self.host.quitFromSignal(1) try: - from_jid = jid.JID(action_data['meta_from_jid']) + from_jid = jid.JID(action_data["meta_from_jid"]) except KeyError: self.disp(_(u"Ignoring action without from_jid data"), 1) return @@ -134,8 +142,7 @@ raise e else: break - xmlui_data = {'answer': C.BOOL_TRUE, - 'port': unicode(port)} + xmlui_data = {"answer": C.BOOL_TRUE, "port": unicode(port)} self.host.bridge.launchAction(xmlui_id, xmlui_data, profile_key=profile) server.serve_forever() self.host.quitFromSignal() @@ -148,4 +155,6 @@ subcommands = (PipeOut, PipeIn) def __init__(self, host): - super(Pipe, self).__init__(host, 'pipe', use_profile=False, help=_('stream piping through XMPP')) + super(Pipe, self).__init__( + host, "pipe", use_profile=False, help=_("stream piping through XMPP") + )