Mercurial > libervia-backend
annotate frontends/src/jp/cmd_pipe.py @ 935:5b2d2f1f05d0
memory: bug fix the current local_dir after sat.conf is auto-updated
| author | souliane <souliane@mailoo.org> |
|---|---|
| date | Mon, 24 Mar 2014 16:43:08 +0100 |
| parents | c39117d00f35 |
| children | 069ad98b360d |
| rev | line source |
|---|---|
| 815 | 1 #! /usr/bin/python |
| 2 # -*- coding: utf-8 -*- | |
| 3 | |
| 4 # jp: a SAT command line tool | |
| 5 # Copyright (C) 2009, 2010, 2011, 2012, 2013, 2014 Jérôme Poisson (goffi@goffi.org) | |
| 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')) |