Mercurial > libervia-backend
annotate frontends/src/jp/cmd_pipe.py @ 1599:e2ed8009e66e
backend, bridge, frontends: actionNew has now a security_limit argument + added some docstring to explain data argument
author | Goffi <goffi@goffi.org> |
---|---|
date | Sat, 14 Nov 2015 19:21:56 +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')) |