1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
|
# -*- coding: utf-8 -*-
# This file is part of emesene.
#
# Emesene is free software; you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation; either version 2 of the License, or
# (at your option) any later version.
#
# emesene is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with emesene; if not, write to the Free Software
# Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA
import os
import time
import gtk
import gobject
import emesenelib.common
import desktop
class FileTransfer:
''' this class represents one filetransfer'''
def __init__(self, controller, p2p, conversation, session, context, sender):
'''Constructor
The sender parameter accepts the special value "Me", which is not
a valid mail, and is used for sent files'''
#constants
#self.state should always be one of those states
self.WAITING = 0
self.TRANSFERING = 1
self.RECEIVED = 2
self.FAILED = 3
self.controller = controller
self.p2p = p2p
self.conversation = conversation
self.session = int(session)
self.context = context
self.sender = sender
self.receivedBytes = 0
self.state = self.WAITING
self.timeAccepted = None
self.previewImage = None
if context.preview != '':
try:
loader = gtk.gdk.PixbufLoader()
loader.write(context.preview)
loader.close()
self.previewImage = loader.get_pixbuf()
del loader
except gobject.GError:
del loader
self.localPath = None
if sender == 'Me':
self.conversation.appendOutputText(None, \
_('Sending %s') % self.context.filename, 'information')
else:
args = {'mail': self.p2p.mail, 'file': self.context.filename}
self.conversation.appendOutputText(None, \
_('%(mail)s is sending you %(file)s') % args, \
'information')
self.signals = []
sap = self.signals.append
sap(self.p2p.connect('transfer-progress', self.onFtProgress))
sap(self.p2p.connect('transfer-failed', self.onFtFailed))
sap(self.p2p.connect('file-transfer-complete', self.onFtReceived))
if sender == 'Me':
sap(self.p2p.connect('file-transfer-accepted', self.onFtAccepted))
def disconnect(self):
for identifier in self.signals:
self.p2p.disconnect(identifier)
def getPreviewImage(self):
return self.previewImage
def getBytes(self):
'''returns a tuple with received and total bytes'''
return int(self.receivedBytes), int(self.context.file_size)
def accept(self):
'''accept this transfer'''
self.conversation.appendOutputText(None, \
_('You have accepted %s') % self.context.filename, \
'information')
self.p2p.emit('file-transfer-accepted', self.session, self.context,
self.sender)
self.timeAccepted = time.time()
self.state = self.TRANSFERING
self.ui.stateChanged()
def cancel(self):
'''cancel this transfer'''
self.conversation.appendOutputText(None,
_('You have canceled the transfer of %s') % self.context.filename,
'information')
self.p2p.emit('file-transfer-canceled', self.session, self.context,
self.sender)
self.state = self.FAILED
self.ui.stateChanged()
self.remove()
def getElapsedTime(self):
'''return the time elapsed since this transfer was accepted'''
if not self.timeAccepted:
return 0
else:
return int(time.time()) - int(self.timeAccepted)
def getAverageSpeed(self):
'''get average file transfer speed (bytes per second)'''
time = self.getElapsedTime()
if time == 0: # prevent division by zero
return 0
else:
return self.receivedBytes / time
def remove(self):
'''remove this transfer'''
self.conversation.transfers.remove(self)
self.disconnect()
def onFtAccepted(self, p2p, session, context, sender):
if session != self.session: return
self.conversation.appendOutputText(None, \
_('Starting transfer of %s') % self.context.filename, \
'information')
self.timeAccepted = time.time()
self.state = self.TRANSFERING
self.ui.stateChanged()
def onFtProgress(self, switchboard, session, bytes):
if session != self.session: return
self.receivedBytes = int(bytes)
self.ui.updateProgress()
def onFtFailed(self, switchboard, session, reason):
if session != self.session: return
output = _('Transfer of %s failed.') % self.context.filename + ' '
if reason == 'corrupt':
output += _('Some parts of the file are missing')
elif reason == 'cancelled':
output += _('Interrupted by user')
elif reason == 'error':
output += _('Transfer error')
self.conversation.appendOutputText(None, output, 'error')
self.state = self.FAILED
self.ui.stateChanged()
def onFtReceived(self, p2p, session, context, src, sender):
'''called when filetransfer is finished'''
if session != self.session: return
if sender == 'Me':
self.conversation.appendOutputText(None, \
_('%s sent successfully') % context.filename, 'information')
else:
self.conversation.appendOutputText(None, \
_('%s received successfully.') % context.filename, \
'information')
self.conversation.doMessageWaiting(sender)
self.receivedBytes = context.file_size
self.state = self.RECEIVED
self.ui.stateChanged()
if sender == 'Me' or src.closed:
return
# get file dir
config = self.controller.config
receivedFilesDir = os.path.expanduser(config.user['receivedFilesDir'])
# if directory is invalid, save to home directory
if not os.path.exists(receivedFilesDir):
print receivedFilesDir + ' does not exist. ' \
'Saving files to home directory.'
receivedFilesDir = os.path.expanduser('~/')
# separated folders between contacts
if config.user['receivedFilesSortedByUser']:
receivedFilesDirSub = os.path.join(receivedFilesDir, p2p.mail)
if os.path.exists(receivedFilesDirSub):
receivedFilesDir = receivedFilesDirSub
else:
os.mkdir(receivedFilesDirSub)
# ugly, but "if os.mkdir(..)" doesn't work :S
if os.path.exists(receivedFilesDirSub):
receivedFilesDir = receivedFilesDirSub
# if we can't create the dir, we put the file directly
# in received files dir
name = os.path.join(receivedFilesDir, self.getFilename())
num = 0
numstr = ''
while os.path.exists(name + numstr):
num += 1
numstr = '.' + str(num)
self.localPath = name + numstr
# we don't copy because the permissions may be too restrictive
# and not match the user preferred umask blah blah
dest = open(self.localPath, 'wb')
src.seek(0)
buffer = src.read(32 * 1024)
while buffer:
dest.write(buffer)
buffer = src.read(32 * 1024)
dest.close()
src.close()
def open(self):
'''open received file'''
if self.localPath != None:
try:
desktop.open(self.localPath)
except OSError:
pass
def getFilename(self):
return self.context.filename
def setBytesReceived(self, bytes):
self.receivedBytes = bytes
def getFraction(self):
'''return the progress of this transfer as a float from 0 to 1'''
if self.context.file_size:
return float(self.receivedBytes) / float(self.context.file_size)
else:
return 1
|
