annotate frontends/src/jp/jp @ 401:b2caa2615c4c

jp roster name manegement + Pipe transfer - added experimental pipe transfer protocol (based on xep-0096) - jp now manage roster name to jid conversion (if an argument correspond to a roster name, it replace it with the corresponding jid) - jp now add last known resource of when a resource is needed but we have a bare jid (e.g. for file or pipe transfer)
author Goffi <goffi@goffi.org>
date Fri, 07 Oct 2011 00:25:15 +0200
parents 393b35aa86d2
children f03688bdb858
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
rev   line source
0
goffi@necton2
parents:
diff changeset
1 #! /usr/bin/python
goffi@necton2
parents:
diff changeset
2 # -*- coding: utf-8 -*-
goffi@necton2
parents:
diff changeset
3
goffi@necton2
parents:
diff changeset
4 """
goffi@necton2
parents:
diff changeset
5 jp: a SAT command line tool
228
b1794cbb88e5 2011 copyright upgrade
Goffi <goffi@goffi.org>
parents: 225
diff changeset
6 Copyright (C) 2009, 2010, 2011 Jérôme Poisson (goffi@goffi.org)
0
goffi@necton2
parents:
diff changeset
7
goffi@necton2
parents:
diff changeset
8 This program is free software: you can redistribute it and/or modify
goffi@necton2
parents:
diff changeset
9 it under the terms of the GNU General Public License as published by
goffi@necton2
parents:
diff changeset
10 the Free Software Foundation, either version 3 of the License, or
goffi@necton2
parents:
diff changeset
11 (at your option) any later version.
goffi@necton2
parents:
diff changeset
12
goffi@necton2
parents:
diff changeset
13 This program is distributed in the hope that it will be useful,
goffi@necton2
parents:
diff changeset
14 but WITHOUT ANY WARRANTY; without even the implied warranty of
goffi@necton2
parents:
diff changeset
15 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
goffi@necton2
parents:
diff changeset
16 GNU General Public License for more details.
goffi@necton2
parents:
diff changeset
17
goffi@necton2
parents:
diff changeset
18 You should have received a copy of the GNU General Public License
goffi@necton2
parents:
diff changeset
19 along with this program. If not, see <http://www.gnu.org/licenses/>.
goffi@necton2
parents:
diff changeset
20 """
goffi@necton2
parents:
diff changeset
21
goffi@necton2
parents:
diff changeset
22 #consts
191
1438a1337732 About added/updated for Wix & Jp
Goffi <goffi@goffi.org>
parents: 165
diff changeset
23 name = u"jp"
228
b1794cbb88e5 2011 copyright upgrade
Goffi <goffi@goffi.org>
parents: 225
diff changeset
24 about = name+u""" v%s (c) Jérôme Poisson (aka Goffi) 2009, 2010, 2011
0
goffi@necton2
parents:
diff changeset
25
goffi@necton2
parents:
diff changeset
26 ---
228
b1794cbb88e5 2011 copyright upgrade
Goffi <goffi@goffi.org>
parents: 225
diff changeset
27 """+name+u""" Copyright (C) 2009, 2010, 2011 Jérôme Poisson (aka Goffi)
0
goffi@necton2
parents:
diff changeset
28 This program comes with ABSOLUTELY NO WARRANTY;
goffi@necton2
parents:
diff changeset
29 This is free software, and you are welcome to redistribute it
goffi@necton2
parents:
diff changeset
30 under certain conditions.
goffi@necton2
parents:
diff changeset
31 ---
goffi@necton2
parents:
diff changeset
32
goffi@necton2
parents:
diff changeset
33 This software is a command line tool for jabber
goffi@necton2
parents:
diff changeset
34 Get the latest version at http://www.goffi.org
goffi@necton2
parents:
diff changeset
35 """
goffi@necton2
parents:
diff changeset
36
goffi@necton2
parents:
diff changeset
37 global pbar_available
goffi@necton2
parents:
diff changeset
38 pbar_available = True #checked before using ProgressBar
goffi@necton2
parents:
diff changeset
39
goffi@necton2
parents:
diff changeset
40 ### logging ###
goffi@necton2
parents:
diff changeset
41 import logging
goffi@necton2
parents:
diff changeset
42 from logging import debug, info, error, warning
goffi@necton2
parents:
diff changeset
43 logging.basicConfig(level=logging.DEBUG,
goffi@necton2
parents:
diff changeset
44 format='%(message)s')
goffi@necton2
parents:
diff changeset
45 ###
goffi@necton2
parents:
diff changeset
46
70
Goffi <goffi@goffi.org>
parents: 57
diff changeset
47 import gettext
Goffi <goffi@goffi.org>
parents: 57
diff changeset
48 gettext.install('jp', "i18n", unicode=True)
Goffi <goffi@goffi.org>
parents: 57
diff changeset
49
0
goffi@necton2
parents:
diff changeset
50 import sys
goffi@necton2
parents:
diff changeset
51 import os
goffi@necton2
parents:
diff changeset
52 from os.path import abspath, basename, dirname
goffi@necton2
parents:
diff changeset
53 from optparse import OptionParser
goffi@necton2
parents:
diff changeset
54 import pdb
236
7cf093e138b6 jp: fixed bad sat.tools.jid import
Goffi <goffi@goffi.org>
parents: 228
diff changeset
55 from sat.tools.jid import JID
0
goffi@necton2
parents:
diff changeset
56 import gobject
225
fd9b7834d98a distutils installation script, draft
Goffi <goffi@goffi.org>
parents: 223
diff changeset
57 from sat_frontends.bridge.DBus import DBusBridgeFrontend,BridgeExceptionNoService
0
goffi@necton2
parents:
diff changeset
58 import tarfile
401
b2caa2615c4c jp roster name manegement + Pipe transfer
Goffi <goffi@goffi.org>
parents: 393
diff changeset
59 import tempfile
b2caa2615c4c jp roster name manegement + Pipe transfer
Goffi <goffi@goffi.org>
parents: 393
diff changeset
60 import shutil
0
goffi@necton2
parents:
diff changeset
61 try:
goffi@necton2
parents:
diff changeset
62 from progressbar import ProgressBar, Percentage, Bar, ETA, FileTransferSpeed
goffi@necton2
parents:
diff changeset
63 except ImportError, e:
70
Goffi <goffi@goffi.org>
parents: 57
diff changeset
64 info (_('ProgressBar not available, please download it at http://pypi.python.org/pypi/progressbar'))
Goffi <goffi@goffi.org>
parents: 57
diff changeset
65 info (_('Progress bar deactivated\n--\n'))
0
goffi@necton2
parents:
diff changeset
66 pbar_available=False
goffi@necton2
parents:
diff changeset
67
goffi@necton2
parents:
diff changeset
68
goffi@necton2
parents:
diff changeset
69
goffi@necton2
parents:
diff changeset
70
goffi@necton2
parents:
diff changeset
71 class JP():
goffi@necton2
parents:
diff changeset
72 def __init__(self):
165
8a2053de6f8c Frontends: management of unlaunched SàT Backend (information message and exit)
Goffi <goffi@goffi.org>
parents: 156
diff changeset
73 try:
8a2053de6f8c Frontends: management of unlaunched SàT Backend (information message and exit)
Goffi <goffi@goffi.org>
parents: 156
diff changeset
74 self.bridge=DBusBridgeFrontend()
8a2053de6f8c Frontends: management of unlaunched SàT Backend (information message and exit)
Goffi <goffi@goffi.org>
parents: 156
diff changeset
75 except BridgeExceptionNoService:
8a2053de6f8c Frontends: management of unlaunched SàT Backend (information message and exit)
Goffi <goffi@goffi.org>
parents: 156
diff changeset
76 print(_(u"Can't connect to SàT backend, are you sure it's launched ?"))
8a2053de6f8c Frontends: management of unlaunched SàT Backend (information message and exit)
Goffi <goffi@goffi.org>
parents: 156
diff changeset
77 import sys
8a2053de6f8c Frontends: management of unlaunched SàT Backend (information message and exit)
Goffi <goffi@goffi.org>
parents: 156
diff changeset
78 sys.exit(1)
391
c34fd9d6242e spelling
Goffi <goffi@goffi.org>
parents: 390
diff changeset
79 self.transfer_id = None
0
goffi@necton2
parents:
diff changeset
80
goffi@necton2
parents:
diff changeset
81 def check_options(self):
goffi@necton2
parents:
diff changeset
82 """Check command line options"""
70
Goffi <goffi@goffi.org>
parents: 57
diff changeset
83 usage=_("""
0
goffi@necton2
parents:
diff changeset
84 %prog [options] [FILE1 FILE2 ...] JID
goffi@necton2
parents:
diff changeset
85 %prog -w [options] [JID1 JID2 ...]
goffi@necton2
parents:
diff changeset
86
goffi@necton2
parents:
diff changeset
87 %prog --help for options list
70
Goffi <goffi@goffi.org>
parents: 57
diff changeset
88 """)
191
1438a1337732 About added/updated for Wix & Jp
Goffi <goffi@goffi.org>
parents: 165
diff changeset
89 version = unicode(self.bridge.getVersion())
1438a1337732 About added/updated for Wix & Jp
Goffi <goffi@goffi.org>
parents: 165
diff changeset
90 parser = OptionParser(usage=usage,version=about % version)
0
goffi@necton2
parents:
diff changeset
91
110
cb904fa7de3c jp: profile management (new option: --profile)
Goffi <goffi@goffi.org>
parents: 70
diff changeset
92 parser.add_option("-p", "--profile", action="store", type="string", default='@DEFAULT@',
156
a216dfbb0d50 jp: added default value in --profile option's help
Goffi <goffi@goffi.org>
parents: 110
diff changeset
93 help=_("Use PROFILE profile key (default: %default)"))
0
goffi@necton2
parents:
diff changeset
94 parser.add_option("-b", "--bz2", action="store_true", default=False,
70
Goffi <goffi@goffi.org>
parents: 57
diff changeset
95 help=_("Make a bzip2 tarball"))
0
goffi@necton2
parents:
diff changeset
96 parser.add_option("-w", "--wait-file", action="store_true", default=False,
70
Goffi <goffi@goffi.org>
parents: 57
diff changeset
97 help=_("Wait for a file to be sent by a contact"))
0
goffi@necton2
parents:
diff changeset
98 parser.add_option("-m", "--multiple", action="store_true", default=False,
70
Goffi <goffi@goffi.org>
parents: 57
diff changeset
99 help=_("Accept multiple files (you'll have to stop manually)"))
0
goffi@necton2
parents:
diff changeset
100 parser.add_option("-f", "--force", action="store_true", default=False,
70
Goffi <goffi@goffi.org>
parents: 57
diff changeset
101 help=_("Force overwritting of existing files"))
110
cb904fa7de3c jp: profile management (new option: --profile)
Goffi <goffi@goffi.org>
parents: 70
diff changeset
102 parser.add_option("-g", "--progress", action="store_true", default=False,
70
Goffi <goffi@goffi.org>
parents: 57
diff changeset
103 help=_("Show progress bar"))
0
goffi@necton2
parents:
diff changeset
104 parser.add_option("-s", "--separate", action="store_true", default=False,
70
Goffi <goffi@goffi.org>
parents: 57
diff changeset
105 help=_("Separate xmpp messages: send one message per line instead of one message alone."))
0
goffi@necton2
parents:
diff changeset
106 parser.add_option("-n", "--new-line", action="store_true", default=False,
70
Goffi <goffi@goffi.org>
parents: 57
diff changeset
107 help=_("Add a new line at the beginning of the input (usefull for ascii art ;))"))
393
393b35aa86d2 jp: added --connect option
Goffi <goffi@goffi.org>
parents: 391
diff changeset
108 parser.add_option("--connect", action="store_true", default=False,
393b35aa86d2 jp: added --connect option
Goffi <goffi@goffi.org>
parents: 391
diff changeset
109 help=_("Connect the profile before doing anything else"))
401
b2caa2615c4c jp roster name manegement + Pipe transfer
Goffi <goffi@goffi.org>
parents: 393
diff changeset
110 parser.add_option("--pipe-in", action="store_true", default=False,
b2caa2615c4c jp roster name manegement + Pipe transfer
Goffi <goffi@goffi.org>
parents: 393
diff changeset
111 help=_("Wait for the reception of a pipe stream"))
b2caa2615c4c jp roster name manegement + Pipe transfer
Goffi <goffi@goffi.org>
parents: 393
diff changeset
112 parser.add_option("--pipe-out", action="store_true", default=False,
b2caa2615c4c jp roster name manegement + Pipe transfer
Goffi <goffi@goffi.org>
parents: 393
diff changeset
113 help=_("Pipe a stream out "))
0
goffi@necton2
parents:
diff changeset
114
goffi@necton2
parents:
diff changeset
115 (self.options, args) = parser.parse_args()
goffi@necton2
parents:
diff changeset
116
goffi@necton2
parents:
diff changeset
117 if len(args) < 1 and not self.options.wait_file:
70
Goffi <goffi@goffi.org>
parents: 57
diff changeset
118 parser.error(_("You must specify the destination JID (Jabber ID)").encode('utf-8'))
0
goffi@necton2
parents:
diff changeset
119
401
b2caa2615c4c jp roster name manegement + Pipe transfer
Goffi <goffi@goffi.org>
parents: 393
diff changeset
120 if self.options.wait_file or self.options.pipe_in:
0
goffi@necton2
parents:
diff changeset
121 #several jid
401
b2caa2615c4c jp roster name manegement + Pipe transfer
Goffi <goffi@goffi.org>
parents: 393
diff changeset
122 self.dest_jids = [arg.decode('utf-8') for arg in args]
0
goffi@necton2
parents:
diff changeset
123 else:
goffi@necton2
parents:
diff changeset
124 #one dest_jid, other args are files
401
b2caa2615c4c jp roster name manegement + Pipe transfer
Goffi <goffi@goffi.org>
parents: 393
diff changeset
125 self.dest_jid = JID(args[-1].decode('utf-8'))
0
goffi@necton2
parents:
diff changeset
126 self.files = args[:-1]
goffi@necton2
parents:
diff changeset
127
goffi@necton2
parents:
diff changeset
128 if not pbar_available and self.options.progress:
goffi@necton2
parents:
diff changeset
129 self.options.progress = False
70
Goffi <goffi@goffi.org>
parents: 57
diff changeset
130 error (_("Option progress is not available, deactivated."))
0
goffi@necton2
parents:
diff changeset
131
401
b2caa2615c4c jp roster name manegement + Pipe transfer
Goffi <goffi@goffi.org>
parents: 393
diff changeset
132 if self.options.progress or self.options.wait_file or self.options.connect or self.options.pipe_in:
0
goffi@necton2
parents:
diff changeset
133 self.start_loop = True #We have to use loop for these options
goffi@necton2
parents:
diff changeset
134 else:
goffi@necton2
parents:
diff changeset
135 self.start_loop = False
goffi@necton2
parents:
diff changeset
136
goffi@necton2
parents:
diff changeset
137
goffi@necton2
parents:
diff changeset
138 return args
goffi@necton2
parents:
diff changeset
139
goffi@necton2
parents:
diff changeset
140 def check_jabber_status(self):
goffi@necton2
parents:
diff changeset
141 """Check that jabber status is allright"""
393
393b35aa86d2 jp: added --connect option
Goffi <goffi@goffi.org>
parents: 391
diff changeset
142 def cantConnect():
393b35aa86d2 jp: added --connect option
Goffi <goffi@goffi.org>
parents: 391
diff changeset
143 error(_(u"Can't connect profile"))
393b35aa86d2 jp: added --connect option
Goffi <goffi@goffi.org>
parents: 391
diff changeset
144 exit(1)
393b35aa86d2 jp: added --connect option
Goffi <goffi@goffi.org>
parents: 391
diff changeset
145
110
cb904fa7de3c jp: profile management (new option: --profile)
Goffi <goffi@goffi.org>
parents: 70
diff changeset
146
cb904fa7de3c jp: profile management (new option: --profile)
Goffi <goffi@goffi.org>
parents: 70
diff changeset
147 self.profile = self.bridge.getProfileName(self.options.profile)
cb904fa7de3c jp: profile management (new option: --profile)
Goffi <goffi@goffi.org>
parents: 70
diff changeset
148 if not self.profile:
cb904fa7de3c jp: profile management (new option: --profile)
Goffi <goffi@goffi.org>
parents: 70
diff changeset
149 error(_("The profile asked doesn't exist"))
cb904fa7de3c jp: profile management (new option: --profile)
Goffi <goffi@goffi.org>
parents: 70
diff changeset
150 exit(1)
cb904fa7de3c jp: profile management (new option: --profile)
Goffi <goffi@goffi.org>
parents: 70
diff changeset
151
393
393b35aa86d2 jp: added --connect option
Goffi <goffi@goffi.org>
parents: 391
diff changeset
152 if self.options.connect: #if connection is asked, we connect the profile
393b35aa86d2 jp: added --connect option
Goffi <goffi@goffi.org>
parents: 391
diff changeset
153 self.bridge.asyncConnect(self.profile, self.connected, cantConnect)
393b35aa86d2 jp: added --connect option
Goffi <goffi@goffi.org>
parents: 391
diff changeset
154 return
393b35aa86d2 jp: added --connect option
Goffi <goffi@goffi.org>
parents: 391
diff changeset
155 elif not self.bridge.isConnected(self.profile):
110
cb904fa7de3c jp: profile management (new option: --profile)
Goffi <goffi@goffi.org>
parents: 70
diff changeset
156 error(_(u"SàT is not conneted, please connect before using jp"))
0
goffi@necton2
parents:
diff changeset
157 exit(1)
goffi@necton2
parents:
diff changeset
158
393
393b35aa86d2 jp: added --connect option
Goffi <goffi@goffi.org>
parents: 391
diff changeset
159 self.connected()
393b35aa86d2 jp: added --connect option
Goffi <goffi@goffi.org>
parents: 391
diff changeset
160
401
b2caa2615c4c jp roster name manegement + Pipe transfer
Goffi <goffi@goffi.org>
parents: 393
diff changeset
161 def check_jids(self):
b2caa2615c4c jp roster name manegement + Pipe transfer
Goffi <goffi@goffi.org>
parents: 393
diff changeset
162 """Check jids validity, transform roster name to corresponding jids"""
b2caa2615c4c jp roster name manegement + Pipe transfer
Goffi <goffi@goffi.org>
parents: 393
diff changeset
163 names2jid = {}
393
393b35aa86d2 jp: added --connect option
Goffi <goffi@goffi.org>
parents: 391
diff changeset
164
401
b2caa2615c4c jp roster name manegement + Pipe transfer
Goffi <goffi@goffi.org>
parents: 393
diff changeset
165 for contact in self.bridge.getContacts(self.options.profile):
b2caa2615c4c jp roster name manegement + Pipe transfer
Goffi <goffi@goffi.org>
parents: 393
diff changeset
166 _jid, attr, groups = contact
b2caa2615c4c jp roster name manegement + Pipe transfer
Goffi <goffi@goffi.org>
parents: 393
diff changeset
167 if attr.has_key("name"):
b2caa2615c4c jp roster name manegement + Pipe transfer
Goffi <goffi@goffi.org>
parents: 393
diff changeset
168 names2jid[attr["name"].lower()] = _jid
b2caa2615c4c jp roster name manegement + Pipe transfer
Goffi <goffi@goffi.org>
parents: 393
diff changeset
169
b2caa2615c4c jp roster name manegement + Pipe transfer
Goffi <goffi@goffi.org>
parents: 393
diff changeset
170 def expandJid(jid):
b2caa2615c4c jp roster name manegement + Pipe transfer
Goffi <goffi@goffi.org>
parents: 393
diff changeset
171 _jid = jid.lower()
b2caa2615c4c jp roster name manegement + Pipe transfer
Goffi <goffi@goffi.org>
parents: 393
diff changeset
172 return unicode(names2jid[_jid] if _jid in names2jid else jid)
b2caa2615c4c jp roster name manegement + Pipe transfer
Goffi <goffi@goffi.org>
parents: 393
diff changeset
173
b2caa2615c4c jp roster name manegement + Pipe transfer
Goffi <goffi@goffi.org>
parents: 393
diff changeset
174 def check(jid):
b2caa2615c4c jp roster name manegement + Pipe transfer
Goffi <goffi@goffi.org>
parents: 393
diff changeset
175 if not jid.is_valid:
b2caa2615c4c jp roster name manegement + Pipe transfer
Goffi <goffi@goffi.org>
parents: 393
diff changeset
176 error (_("%s is not a valid JID !"), self.dest_jid)
b2caa2615c4c jp roster name manegement + Pipe transfer
Goffi <goffi@goffi.org>
parents: 393
diff changeset
177 exit(1)
b2caa2615c4c jp roster name manegement + Pipe transfer
Goffi <goffi@goffi.org>
parents: 393
diff changeset
178
b2caa2615c4c jp roster name manegement + Pipe transfer
Goffi <goffi@goffi.org>
parents: 393
diff changeset
179 try:
b2caa2615c4c jp roster name manegement + Pipe transfer
Goffi <goffi@goffi.org>
parents: 393
diff changeset
180 self.dest_jid = expandJid(self.dest_jid)
b2caa2615c4c jp roster name manegement + Pipe transfer
Goffi <goffi@goffi.org>
parents: 393
diff changeset
181 check(self.dest_jid)
b2caa2615c4c jp roster name manegement + Pipe transfer
Goffi <goffi@goffi.org>
parents: 393
diff changeset
182 except AttributeError:
b2caa2615c4c jp roster name manegement + Pipe transfer
Goffi <goffi@goffi.org>
parents: 393
diff changeset
183 pass
b2caa2615c4c jp roster name manegement + Pipe transfer
Goffi <goffi@goffi.org>
parents: 393
diff changeset
184 try:
b2caa2615c4c jp roster name manegement + Pipe transfer
Goffi <goffi@goffi.org>
parents: 393
diff changeset
185 for i in range(len(self.dest_jids)):
b2caa2615c4c jp roster name manegement + Pipe transfer
Goffi <goffi@goffi.org>
parents: 393
diff changeset
186 self.dest_jids[i] = expandJid(self.dest_jids[i])
b2caa2615c4c jp roster name manegement + Pipe transfer
Goffi <goffi@goffi.org>
parents: 393
diff changeset
187 check(self.dest_jids[i])
b2caa2615c4c jp roster name manegement + Pipe transfer
Goffi <goffi@goffi.org>
parents: 393
diff changeset
188 except AttributeError:
b2caa2615c4c jp roster name manegement + Pipe transfer
Goffi <goffi@goffi.org>
parents: 393
diff changeset
189 pass
0
goffi@necton2
parents:
diff changeset
190
goffi@necton2
parents:
diff changeset
191 def send_stdin(self):
goffi@necton2
parents:
diff changeset
192 """Send incomming data on stdin to jabber contact"""
goffi@necton2
parents:
diff changeset
193 header = "\n" if self.options.new_line else ""
goffi@necton2
parents:
diff changeset
194
goffi@necton2
parents:
diff changeset
195 if self.options.separate: #we send stdin in several messages
goffi@necton2
parents:
diff changeset
196 if header:
110
cb904fa7de3c jp: profile management (new option: --profile)
Goffi <goffi@goffi.org>
parents: 70
diff changeset
197 self.bridge.sendMessage(self.dest_jid, header, profile_key=self.profile)
0
goffi@necton2
parents:
diff changeset
198 while (True):
goffi@necton2
parents:
diff changeset
199 line = sys.stdin.readline()
goffi@necton2
parents:
diff changeset
200 if not line:
goffi@necton2
parents:
diff changeset
201 break
110
cb904fa7de3c jp: profile management (new option: --profile)
Goffi <goffi@goffi.org>
parents: 70
diff changeset
202 self.bridge.sendMessage(self.dest_jid, line.replace("\n",""), profile_key=self.profile)
0
goffi@necton2
parents:
diff changeset
203 else:
110
cb904fa7de3c jp: profile management (new option: --profile)
Goffi <goffi@goffi.org>
parents: 70
diff changeset
204 self.bridge.sendMessage(self.dest_jid, header + "".join(sys.stdin.readlines()), profile_key=self.profile)
0
goffi@necton2
parents:
diff changeset
205
401
b2caa2615c4c jp roster name manegement + Pipe transfer
Goffi <goffi@goffi.org>
parents: 393
diff changeset
206
b2caa2615c4c jp roster name manegement + Pipe transfer
Goffi <goffi@goffi.org>
parents: 393
diff changeset
207 def pipe_out(self):
b2caa2615c4c jp roster name manegement + Pipe transfer
Goffi <goffi@goffi.org>
parents: 393
diff changeset
208 """Create named pipe, and send stdin to it"""
b2caa2615c4c jp roster name manegement + Pipe transfer
Goffi <goffi@goffi.org>
parents: 393
diff changeset
209 tmp_dir = tempfile.mkdtemp()
b2caa2615c4c jp roster name manegement + Pipe transfer
Goffi <goffi@goffi.org>
parents: 393
diff changeset
210 fifopath = os.path.join(tmp_dir,"pipe_out")
b2caa2615c4c jp roster name manegement + Pipe transfer
Goffi <goffi@goffi.org>
parents: 393
diff changeset
211 os.mkfifo(fifopath)
b2caa2615c4c jp roster name manegement + Pipe transfer
Goffi <goffi@goffi.org>
parents: 393
diff changeset
212 self.bridge.pipeOut(self._getFullJid(self.dest_jid), fifopath, {}, profile_key=self.profile)
b2caa2615c4c jp roster name manegement + Pipe transfer
Goffi <goffi@goffi.org>
parents: 393
diff changeset
213 f = open(fifopath, 'w+')
b2caa2615c4c jp roster name manegement + Pipe transfer
Goffi <goffi@goffi.org>
parents: 393
diff changeset
214 shutil.copyfileobj(sys.stdin, f)
b2caa2615c4c jp roster name manegement + Pipe transfer
Goffi <goffi@goffi.org>
parents: 393
diff changeset
215 f.close()
b2caa2615c4c jp roster name manegement + Pipe transfer
Goffi <goffi@goffi.org>
parents: 393
diff changeset
216 shutil.rmtree(tmp_dir)
b2caa2615c4c jp roster name manegement + Pipe transfer
Goffi <goffi@goffi.org>
parents: 393
diff changeset
217
b2caa2615c4c jp roster name manegement + Pipe transfer
Goffi <goffi@goffi.org>
parents: 393
diff changeset
218
0
goffi@necton2
parents:
diff changeset
219 def send_files(self):
goffi@necton2
parents:
diff changeset
220 """Send files to jabber contact"""
goffi@necton2
parents:
diff changeset
221
goffi@necton2
parents:
diff changeset
222 for file in self.files:
goffi@necton2
parents:
diff changeset
223 if not os.path.exists(file):
70
Goffi <goffi@goffi.org>
parents: 57
diff changeset
224 error (_("File [%s] doesn't exist !") % file)
0
goffi@necton2
parents:
diff changeset
225 exit(1)
goffi@necton2
parents:
diff changeset
226 if not self.options.bz2 and os.path.isdir(file):
70
Goffi <goffi@goffi.org>
parents: 57
diff changeset
227 error (_("[%s] is a dir ! Please send files inside or use compression") % file)
0
goffi@necton2
parents:
diff changeset
228 exit(1)
goffi@necton2
parents:
diff changeset
229
401
b2caa2615c4c jp roster name manegement + Pipe transfer
Goffi <goffi@goffi.org>
parents: 393
diff changeset
230 full_dest_jid = self._getFullJid(self.dest_jid)
0
goffi@necton2
parents:
diff changeset
231 if self.options.bz2:
goffi@necton2
parents:
diff changeset
232 tmpfile = (basename(self.files[0]) or basename(dirname(self.files[0])) ) + '.tar.bz2' #FIXME: tmp, need an algorithm to find a good name/path
goffi@necton2
parents:
diff changeset
233 if os.path.exists(tmpfile):
70
Goffi <goffi@goffi.org>
parents: 57
diff changeset
234 error (_("tmp file (%s) already exists ! Please remove it"), tmpfile)
0
goffi@necton2
parents:
diff changeset
235 exit(1)
70
Goffi <goffi@goffi.org>
parents: 57
diff changeset
236 warning(_("bz2 is an experimental option at an early dev stage, use with caution"))
0
goffi@necton2
parents:
diff changeset
237 #FIXME: check free space, writting perm, tmp dir, filename (watch for OS used)
70
Goffi <goffi@goffi.org>
parents: 57
diff changeset
238 info(_("Starting compression, please wait..."))
0
goffi@necton2
parents:
diff changeset
239 sys.stdout.flush()
goffi@necton2
parents:
diff changeset
240 bz2=tarfile.open(tmpfile, "w:bz2")
goffi@necton2
parents:
diff changeset
241 for file in self.files:
70
Goffi <goffi@goffi.org>
parents: 57
diff changeset
242 info(_("Adding %s"), file)
0
goffi@necton2
parents:
diff changeset
243 bz2.add(file)
goffi@necton2
parents:
diff changeset
244 bz2.close()
70
Goffi <goffi@goffi.org>
parents: 57
diff changeset
245 info(_("OK !"))
0
goffi@necton2
parents:
diff changeset
246 path = abspath(tmpfile)
401
b2caa2615c4c jp roster name manegement + Pipe transfer
Goffi <goffi@goffi.org>
parents: 393
diff changeset
247 self.transfer_id = self.bridge.sendFile(full_dest_jid, path, {}, profile_key=self.profile)
0
goffi@necton2
parents:
diff changeset
248 else:
goffi@necton2
parents:
diff changeset
249 for file in self.files:
goffi@necton2
parents:
diff changeset
250 path = abspath(file)
401
b2caa2615c4c jp roster name manegement + Pipe transfer
Goffi <goffi@goffi.org>
parents: 393
diff changeset
251 self.transfer_id = self.bridge.sendFile(full_dest_jid, path, {}, profile_key=self.profile) #FIXME: show progress only for last transfer_id
b2caa2615c4c jp roster name manegement + Pipe transfer
Goffi <goffi@goffi.org>
parents: 393
diff changeset
252
0
goffi@necton2
parents:
diff changeset
253
401
b2caa2615c4c jp roster name manegement + Pipe transfer
Goffi <goffi@goffi.org>
parents: 393
diff changeset
254 def _getFullJid(self, param_jid):
b2caa2615c4c jp roster name manegement + Pipe transfer
Goffi <goffi@goffi.org>
parents: 393
diff changeset
255 """Return the full jid if possible (add last resource when find a bare jid"""
b2caa2615c4c jp roster name manegement + Pipe transfer
Goffi <goffi@goffi.org>
parents: 393
diff changeset
256 _jid = JID(param_jid)
b2caa2615c4c jp roster name manegement + Pipe transfer
Goffi <goffi@goffi.org>
parents: 393
diff changeset
257 if not _jid.resource:
b2caa2615c4c jp roster name manegement + Pipe transfer
Goffi <goffi@goffi.org>
parents: 393
diff changeset
258 #if the resource is not given, we try to add the last known resource
b2caa2615c4c jp roster name manegement + Pipe transfer
Goffi <goffi@goffi.org>
parents: 393
diff changeset
259 last_resource = self.bridge.getLastResource(param_jid, self.options.profile)
b2caa2615c4c jp roster name manegement + Pipe transfer
Goffi <goffi@goffi.org>
parents: 393
diff changeset
260 if last_resource:
b2caa2615c4c jp roster name manegement + Pipe transfer
Goffi <goffi@goffi.org>
parents: 393
diff changeset
261 return "%s/%s" % (_jid.short, last_resource)
b2caa2615c4c jp roster name manegement + Pipe transfer
Goffi <goffi@goffi.org>
parents: 393
diff changeset
262 return param_jid
b2caa2615c4c jp roster name manegement + Pipe transfer
Goffi <goffi@goffi.org>
parents: 393
diff changeset
263
0
goffi@necton2
parents:
diff changeset
264
goffi@necton2
parents:
diff changeset
265 def askConfirmation(self, type, id, data):
391
c34fd9d6242e spelling
Goffi <goffi@goffi.org>
parents: 390
diff changeset
266 """CB used for file transfer, accept files depending on parameters"""
0
goffi@necton2
parents:
diff changeset
267 answer_data={}
391
c34fd9d6242e spelling
Goffi <goffi@goffi.org>
parents: 390
diff changeset
268 if type == "FILE_TRANSFER":
401
b2caa2615c4c jp roster name manegement + Pipe transfer
Goffi <goffi@goffi.org>
parents: 393
diff changeset
269 if not self.options.wait_file:
b2caa2615c4c jp roster name manegement + Pipe transfer
Goffi <goffi@goffi.org>
parents: 393
diff changeset
270 return
b2caa2615c4c jp roster name manegement + Pipe transfer
Goffi <goffi@goffi.org>
parents: 393
diff changeset
271 if self.dest_jids and not JID(data['from']).short in [JID(_jid).short for _jid in self.dest_jids]:
0
goffi@necton2
parents:
diff changeset
272 return #file is not sent by a filtered jid
goffi@necton2
parents:
diff changeset
273
goffi@necton2
parents:
diff changeset
274 answer_data["dest_path"] = os.getcwd()+'/'+data['filename']
goffi@necton2
parents:
diff changeset
275
goffi@necton2
parents:
diff changeset
276 if self.options.force or not os.path.exists(answer_data["dest_path"]):
goffi@necton2
parents:
diff changeset
277 self.bridge.confirmationAnswer(id, True, answer_data)
70
Goffi <goffi@goffi.org>
parents: 57
diff changeset
278 info(_("Accepted file [%(filename)s] from %(sender)s") % {'filename':data['filename'], 'sender':data['from']})
391
c34fd9d6242e spelling
Goffi <goffi@goffi.org>
parents: 390
diff changeset
279 self.transfer_id = id
0
goffi@necton2
parents:
diff changeset
280 else:
goffi@necton2
parents:
diff changeset
281 self.bridge.confirmationAnswer(id, False, answer_data)
70
Goffi <goffi@goffi.org>
parents: 57
diff changeset
282 warning(_("Refused file [%(filename)s] from %(sender)s: a file with the same name already exist") % {'filename':data['filename'], 'sender':data['from']})
0
goffi@necton2
parents:
diff changeset
283
goffi@necton2
parents:
diff changeset
284
goffi@necton2
parents:
diff changeset
285 if not self.options.multiple and not self.options.progress:
goffi@necton2
parents:
diff changeset
286 #we just accept one file
goffi@necton2
parents:
diff changeset
287 self.loop.quit()
401
b2caa2615c4c jp roster name manegement + Pipe transfer
Goffi <goffi@goffi.org>
parents: 393
diff changeset
288 elif type == "PIPE_TRANSFER":
b2caa2615c4c jp roster name manegement + Pipe transfer
Goffi <goffi@goffi.org>
parents: 393
diff changeset
289 if not self.options.pipe_in:
b2caa2615c4c jp roster name manegement + Pipe transfer
Goffi <goffi@goffi.org>
parents: 393
diff changeset
290 return
b2caa2615c4c jp roster name manegement + Pipe transfer
Goffi <goffi@goffi.org>
parents: 393
diff changeset
291 if self.dest_jids and not JID(data['from']).short in [JID(_jid).short for _jid in self.dest_jids]:
b2caa2615c4c jp roster name manegement + Pipe transfer
Goffi <goffi@goffi.org>
parents: 393
diff changeset
292 return #pipe stream is not sent by a filtered jid
b2caa2615c4c jp roster name manegement + Pipe transfer
Goffi <goffi@goffi.org>
parents: 393
diff changeset
293
b2caa2615c4c jp roster name manegement + Pipe transfer
Goffi <goffi@goffi.org>
parents: 393
diff changeset
294 tmp_dir = tempfile.mkdtemp()
b2caa2615c4c jp roster name manegement + Pipe transfer
Goffi <goffi@goffi.org>
parents: 393
diff changeset
295 fifopath = os.path.join(tmp_dir,"pipe_in")
b2caa2615c4c jp roster name manegement + Pipe transfer
Goffi <goffi@goffi.org>
parents: 393
diff changeset
296 answer_data["dest_path"] = fifopath
b2caa2615c4c jp roster name manegement + Pipe transfer
Goffi <goffi@goffi.org>
parents: 393
diff changeset
297 os.mkfifo(fifopath)
b2caa2615c4c jp roster name manegement + Pipe transfer
Goffi <goffi@goffi.org>
parents: 393
diff changeset
298 self.bridge.confirmationAnswer(id, True, answer_data)
b2caa2615c4c jp roster name manegement + Pipe transfer
Goffi <goffi@goffi.org>
parents: 393
diff changeset
299 f = open(fifopath, 'r')
b2caa2615c4c jp roster name manegement + Pipe transfer
Goffi <goffi@goffi.org>
parents: 393
diff changeset
300 shutil.copyfileobj(f, sys.stdout)
b2caa2615c4c jp roster name manegement + Pipe transfer
Goffi <goffi@goffi.org>
parents: 393
diff changeset
301 f.close()
b2caa2615c4c jp roster name manegement + Pipe transfer
Goffi <goffi@goffi.org>
parents: 393
diff changeset
302 shutil.rmtree(tmp_dir)
b2caa2615c4c jp roster name manegement + Pipe transfer
Goffi <goffi@goffi.org>
parents: 393
diff changeset
303 self.loop.quit()
b2caa2615c4c jp roster name manegement + Pipe transfer
Goffi <goffi@goffi.org>
parents: 393
diff changeset
304
0
goffi@necton2
parents:
diff changeset
305
22
bb72c29f3432 added action cb mechanism for buttons. Tested with a temporary new user registration button.
Goffi <goffi@goffi.org>
parents: 0
diff changeset
306 def actionResult(self, type, id, data):
bb72c29f3432 added action cb mechanism for buttons. Tested with a temporary new user registration button.
Goffi <goffi@goffi.org>
parents: 0
diff changeset
307 #FIXME
70
Goffi <goffi@goffi.org>
parents: 57
diff changeset
308 info (_("FIXME: actionResult not implemented"))
0
goffi@necton2
parents:
diff changeset
309
401
b2caa2615c4c jp roster name manegement + Pipe transfer
Goffi <goffi@goffi.org>
parents: 393
diff changeset
310 def confirmation_reply(self):
b2caa2615c4c jp roster name manegement + Pipe transfer
Goffi <goffi@goffi.org>
parents: 393
diff changeset
311 """Auto reply to confirmations requests"""
390
8e87af2e2d1f jp: fixed file reception
Goffi <goffi@goffi.org>
parents: 387
diff changeset
312 self.bridge.register("askConfirmation", self.askConfirmation)
0
goffi@necton2
parents:
diff changeset
313
goffi@necton2
parents:
diff changeset
314 def progressCB(self):
391
c34fd9d6242e spelling
Goffi <goffi@goffi.org>
parents: 390
diff changeset
315 if self.transfer_id:
c34fd9d6242e spelling
Goffi <goffi@goffi.org>
parents: 390
diff changeset
316 data = self.bridge.getProgress(self.transfer_id)
0
goffi@necton2
parents:
diff changeset
317 if data:
goffi@necton2
parents:
diff changeset
318 if not data['position']:
goffi@necton2
parents:
diff changeset
319 data['position'] = '0'
goffi@necton2
parents:
diff changeset
320 if not self.pbar:
goffi@necton2
parents:
diff changeset
321 #first answer, we must construct the bar
70
Goffi <goffi@goffi.org>
parents: 57
diff changeset
322 self.pbar = ProgressBar(int(data['size']),[_("Progress: "),Percentage()," ",Bar()," ",FileTransferSpeed()," ",ETA()])
0
goffi@necton2
parents:
diff changeset
323 self.pbar.start()
goffi@necton2
parents:
diff changeset
324
goffi@necton2
parents:
diff changeset
325 self.pbar.update(int(data['position']))
goffi@necton2
parents:
diff changeset
326 elif self.pbar:
goffi@necton2
parents:
diff changeset
327 self.pbar.finish()
goffi@necton2
parents:
diff changeset
328 if not self.options.multiple:
goffi@necton2
parents:
diff changeset
329 self.loop.quit()
goffi@necton2
parents:
diff changeset
330 return False
goffi@necton2
parents:
diff changeset
331
goffi@necton2
parents:
diff changeset
332 return True
goffi@necton2
parents:
diff changeset
333
goffi@necton2
parents:
diff changeset
334 def go(self):
goffi@necton2
parents:
diff changeset
335 self.check_options()
goffi@necton2
parents:
diff changeset
336 self.check_jabber_status()
393
393b35aa86d2 jp: added --connect option
Goffi <goffi@goffi.org>
parents: 391
diff changeset
337 if self.start_loop:
393b35aa86d2 jp: added --connect option
Goffi <goffi@goffi.org>
parents: 391
diff changeset
338 self.loop = gobject.MainLoop()
393b35aa86d2 jp: added --connect option
Goffi <goffi@goffi.org>
parents: 391
diff changeset
339 try:
393b35aa86d2 jp: added --connect option
Goffi <goffi@goffi.org>
parents: 391
diff changeset
340 self.loop.run()
393b35aa86d2 jp: added --connect option
Goffi <goffi@goffi.org>
parents: 391
diff changeset
341 except KeyboardInterrupt:
393b35aa86d2 jp: added --connect option
Goffi <goffi@goffi.org>
parents: 391
diff changeset
342 info(_("User interruption: good bye"))
393b35aa86d2 jp: added --connect option
Goffi <goffi@goffi.org>
parents: 391
diff changeset
343
393b35aa86d2 jp: added --connect option
Goffi <goffi@goffi.org>
parents: 391
diff changeset
344 def connected(self):
393b35aa86d2 jp: added --connect option
Goffi <goffi@goffi.org>
parents: 391
diff changeset
345 """This is called when the profile is connected"""
401
b2caa2615c4c jp roster name manegement + Pipe transfer
Goffi <goffi@goffi.org>
parents: 393
diff changeset
346 self.check_jids()
b2caa2615c4c jp roster name manegement + Pipe transfer
Goffi <goffi@goffi.org>
parents: 393
diff changeset
347 if self.options.wait_file or self.options.pipe_in:
b2caa2615c4c jp roster name manegement + Pipe transfer
Goffi <goffi@goffi.org>
parents: 393
diff changeset
348 self.confirmation_reply()
0
goffi@necton2
parents:
diff changeset
349 else:
401
b2caa2615c4c jp roster name manegement + Pipe transfer
Goffi <goffi@goffi.org>
parents: 393
diff changeset
350 if self.files:
b2caa2615c4c jp roster name manegement + Pipe transfer
Goffi <goffi@goffi.org>
parents: 393
diff changeset
351 self.send_files()
b2caa2615c4c jp roster name manegement + Pipe transfer
Goffi <goffi@goffi.org>
parents: 393
diff changeset
352 elif self.options.pipe_out:
b2caa2615c4c jp roster name manegement + Pipe transfer
Goffi <goffi@goffi.org>
parents: 393
diff changeset
353 self.pipe_out()
b2caa2615c4c jp roster name manegement + Pipe transfer
Goffi <goffi@goffi.org>
parents: 393
diff changeset
354 else:
0
goffi@necton2
parents:
diff changeset
355 self.send_stdin()
goffi@necton2
parents:
diff changeset
356
393
393b35aa86d2 jp: added --connect option
Goffi <goffi@goffi.org>
parents: 391
diff changeset
357 if self.options.progress:
393b35aa86d2 jp: added --connect option
Goffi <goffi@goffi.org>
parents: 391
diff changeset
358 self.pbar = None
393b35aa86d2 jp: added --connect option
Goffi <goffi@goffi.org>
parents: 391
diff changeset
359 gobject.timeout_add(10, self.progressCB)
393b35aa86d2 jp: added --connect option
Goffi <goffi@goffi.org>
parents: 391
diff changeset
360
401
b2caa2615c4c jp roster name manegement + Pipe transfer
Goffi <goffi@goffi.org>
parents: 393
diff changeset
361 if self.start_loop and not self.options.progress and not self.options.wait_file and not self.options.pipe_in:
393
393b35aa86d2 jp: added --connect option
Goffi <goffi@goffi.org>
parents: 391
diff changeset
362 self.loop.quit()
0
goffi@necton2
parents:
diff changeset
363
goffi@necton2
parents:
diff changeset
364
goffi@necton2
parents:
diff changeset
365 if __name__ == "__main__":
goffi@necton2
parents:
diff changeset
366 jp = JP()
goffi@necton2
parents:
diff changeset
367 jp.go()