Mercurial > libervia-backend
annotate frontends/src/jp/cmd_pipe.py @ 1596:b7ee113183fc
jp: better profile commands:
- new "profile/default" command
- info doesn't show password anymore by default, need to be explicitly requested
- info and modify don't need to connect anymore
- modify can now set default profile. As use_profile is set, at least a profile session need to be started when it would not be mandatory technicaly (if just setting the profile as default is needed). But this option should not be used often, and it's not a big side effect, so I don't feel the need to create a new dedicated command, or to do complicated checks to avoid the session start.
| author | Goffi <goffi@goffi.org> |
|---|---|
| date | Sat, 14 Nov 2015 19:18:10 +0100 |
| parents | 069ad98b360d |
| children | 3690b4d4157e |
| rev | line source |
|---|---|
| 815 | 1 #! /usr/bin/python |
| 2 # -*- coding: utf-8 -*- | |
| 3 | |
| 4 # jp: a SAT command line tool | |
| 1396 | 5 # Copyright (C) 2009, 2010, 2011, 2012, 2013, 2014, 2015 Jérôme Poisson (goffi@goffi.org) |
| 815 | 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 | |
| 817 | 20 from sat_frontends.jp import base |
| 0 | 21 |
| 814 | 22 import tempfile |
| 817 | 23 import sys |
| 0 | 24 import os |
| 817 | 25 import os.path |
|
401
b2caa2615c4c
jp roster name manegement + Pipe transfer
Goffi <goffi@goffi.org>
parents:
393
diff
changeset
|
26 import shutil |
| 814 | 27 from sat.core.i18n import _ |
| 0 | 28 |
| 817 | 29 __commands__ = ["Pipe"] |
| 0 | 30 |
| 817 | 31 class PipeOut(base.CommandBase): |
| 657 | 32 |
| 817 | 33 def __init__(self, host): |
| 34 super(PipeOut, self).__init__(host, 'out', help=_('Pipe a stream out')) | |
| 393 | 35 |
| 817 | 36 def add_parser_options(self): |
| 37 self.parser.add_argument("jid", type=base.unicode_decoder, help=_("The destination jid")) | |
|
401
b2caa2615c4c
jp roster name manegement + Pipe transfer
Goffi <goffi@goffi.org>
parents:
393
diff
changeset
|
38 |
| 817 | 39 def pipe_out(self): |
| 40 """ Create named pipe, and send stdin to it """ | |
|
401
b2caa2615c4c
jp roster name manegement + Pipe transfer
Goffi <goffi@goffi.org>
parents:
393
diff
changeset
|
41 tmp_dir = tempfile.mkdtemp() |
|
b2caa2615c4c
jp roster name manegement + Pipe transfer
Goffi <goffi@goffi.org>
parents:
393
diff
changeset
|
42 fifopath = os.path.join(tmp_dir,"pipe_out") |
|
b2caa2615c4c
jp roster name manegement + Pipe transfer
Goffi <goffi@goffi.org>
parents:
393
diff
changeset
|
43 os.mkfifo(fifopath) |
| 817 | 44 self.host.bridge.pipeOut(self.host.get_full_jid(self.args.jid), fifopath, {}, self.profile) |
|
402
f03688bdb858
jp: use with statement to open fifo
Goffi <goffi@goffi.org>
parents:
401
diff
changeset
|
45 with open(fifopath, 'w') as f: |
|
f03688bdb858
jp: use with statement to open fifo
Goffi <goffi@goffi.org>
parents:
401
diff
changeset
|
46 shutil.copyfileobj(sys.stdin, f) |
|
401
b2caa2615c4c
jp roster name manegement + Pipe transfer
Goffi <goffi@goffi.org>
parents:
393
diff
changeset
|
47 shutil.rmtree(tmp_dir) |
| 817 | 48 self.host.quit() |
|
401
b2caa2615c4c
jp roster name manegement + Pipe transfer
Goffi <goffi@goffi.org>
parents:
393
diff
changeset
|
49 |
| 817 | 50 def connected(self): |
| 51 # TODO: check_jids | |
| 52 self.need_loop = True | |
| 53 super(PipeOut, self).connected() | |
| 54 self.pipe_out() | |
| 55 | |
| 56 | |
| 57 class PipeIn(base.CommandAnswering): | |
| 58 confirm_type = "PIPE_TRANSFER" | |
| 0 | 59 |
| 817 | 60 def __init__(self, host): |
| 61 super(PipeIn, self).__init__(host, 'in', help=_('Wait for the reception of a pipe stream')) | |
| 0 | 62 |
| 817 | 63 @property |
| 814 | 64 def dest_jids(self): |
| 817 | 65 return self.args.jids |
|
587
952322b1d490
Remove trailing whitespaces.
Emmanuel Gil Peyrot <linkmauve@linkmauve.fr>
parents:
572
diff
changeset
|
66 |
| 817 | 67 def add_parser_options(self): |
| 68 self.parser.add_argument("jids", type=base.unicode_decoder, nargs="*", help=_('Jids accepted (none means "accept everything")')) | |
| 0 | 69 |
| 817 | 70 def ask(self, data, confirm_id): |
| 814 | 71 answer_data = {} |
| 72 tmp_dir = tempfile.mkdtemp() | |
| 73 fifopath = os.path.join(tmp_dir,"pipe_in") | |
| 74 answer_data["dest_path"] = fifopath | |
| 75 os.mkfifo(fifopath) | |
| 817 | 76 self.host.bridge.confirmationAnswer(confirm_id, True, answer_data, self.profile) |
| 814 | 77 with open(fifopath, 'r') as f: |
| 78 shutil.copyfileobj(f, sys.stdout) | |
| 79 shutil.rmtree(tmp_dir) | |
| 817 | 80 self.host.quit() |
| 81 | |
| 82 | |
| 83 class Pipe(base.CommandBase): | |
| 84 subcommands = (PipeOut, PipeIn) | |
| 85 | |
| 86 def __init__(self, host): | |
| 87 super(Pipe, self).__init__(host, 'pipe', use_profile=False, help=_('Stream piping through XMPP')) |