progress bar in file transfers

This commit is contained in:
Андрей Владимирович 2016-03-21 20:19:13 +03:00
parent 61d4a28d6b
commit 2131fa3868
3 changed files with 46 additions and 45 deletions

View file

@ -15,13 +15,13 @@ TOX_FILE_TRANSFER_STATE = {
} }
class Signal(QtCore.QObject): class StateSignal(QtCore.QObject):
signal = QtCore.Signal(str) signal = QtCore.Signal(int, float)
class FileTransfer(QtCore.QObject): class FileTransfer(QtCore.QObject):
def __init__(self, path, tox, friend_number, file_number=None): def __init__(self, path, tox, friend_number, size, file_number=None):
QtCore.QObject.__init__(self) QtCore.QObject.__init__(self)
self._path = path self._path = path
self._tox = tox self._tox = tox
@ -29,13 +29,15 @@ class FileTransfer(QtCore.QObject):
self.state = TOX_FILE_TRANSFER_STATE['RUNNING'] self.state = TOX_FILE_TRANSFER_STATE['RUNNING']
self._file_number = file_number self._file_number = file_number
self._creation_time = time() self._creation_time = time()
self._signal = Signal() self._size = size
self._done = 0
self._state_changed = StateSignal()
def set_tox(self, tox): def set_tox(self, tox):
self._tox = tox self._tox = tox
def set_event_handler(self, handler): def set_state_changed_handler(self, handler):
self._signal.signal.connect(handler) self._state_changed.signal.connect(handler)
def get_file_number(self): def get_file_number(self):
return self._file_number return self._file_number
@ -46,95 +48,95 @@ class FileTransfer(QtCore.QObject):
def cancel(self): def cancel(self):
self.send_control(TOX_FILE_CONTROL['CANCEL']) self.send_control(TOX_FILE_CONTROL['CANCEL'])
self._file.close() self._file.close()
print id(self._signal) self._state_changed.signal.emit(self.state, self._done / self._size)
self._signal.signal.emit('{} {}'.format(self.state, 0))
def send_control(self, control): def send_control(self, control):
if self._tox.file_control(self._friend_number, self._file_number, control): if self._tox.file_control(self._friend_number, self._file_number, control):
self.state = control self.state = control
self._state_changed.signal.emit(self.state, self._done / self._size)
def get_file_id(self): def get_file_id(self):
return self._tox.file_get_file_id(self._friend_number, self._file_number) return self._tox.file_get_file_id(self._friend_number, self._file_number)
def file_seek(self): def file_seek(self):
# TODO implement # TODO implement or not implement
pass pass
class SendTransfer(FileTransfer): class SendTransfer(FileTransfer):
def __init__(self, path, tox, friend_number, kind=TOX_FILE_KIND['DATA'], file_id=None): def __init__(self, path, tox, friend_number, kind=TOX_FILE_KIND['DATA'], file_id=None):
super(SendTransfer, self).__init__(path, tox, friend_number)
self._file_number = tox.file_send(friend_number,
kind,
getsize(path) if path else 0,
file_id,
basename(path).encode('utf-8') if path else '')
if path is not None: if path is not None:
self._file = open(path, 'rb') self._file = open(path, 'rb')
size = getsize(path)
else:
size = 0
super(SendTransfer, self).__init__(path, tox, friend_number, size)
self._file_number = tox.file_send(friend_number, kind, size, file_id,
basename(path).encode('utf-8') if path else '')
def send_chunk(self, position, size): def send_chunk(self, position, size):
if size: if size:
self._file.seek(position) self._file.seek(position)
data = self._file.read(size) data = self._file.read(size)
# TODO: fix bug with wrong data for file_send_chunk
self._tox.file_send_chunk(self._friend_number, self._file_number, position, data) self._tox.file_send_chunk(self._friend_number, self._file_number, position, data)
self._signal.signal.emit('{} {}'.format(self.state, 0)) self._done += size
#print self._friend_number, self._file_number, position, data self._state_changed.signal.emit(self.state, self._done / self._size)
else: else:
self._file.close() self._file.close()
self.state = TOX_FILE_TRANSFER_STATE['FINISHED'] self.state = TOX_FILE_TRANSFER_STATE['FINISHED']
self._signal.signal.emit('{} {}'.format(self.state, 100)) self._state_changed.signal.emit(self.state, self._done / self._size)
class SendAvatar(SendTransfer): class SendAvatar(SendTransfer):
def __init__(self, path, tox, friend_number): def __init__(self, path, tox, friend_number):
if path is None: if path is None:
super(SendAvatar, self).__init__(path, tox, friend_number, TOX_FILE_KIND['AVATAR']) hash = None
else: else:
with open(path, 'rb') as fl: with open(path, 'rb') as fl:
hash = Tox.hash(fl.read()) hash = Tox.hash(fl.read())
super(self.__class__, self).__init__(path, tox, friend_number, TOX_FILE_KIND['AVATAR'], hash) super(SendAvatar, self).__init__(path, tox, friend_number, TOX_FILE_KIND['AVATAR'], hash)
class ReceiveTransfer(FileTransfer): class ReceiveTransfer(FileTransfer):
def __init__(self, path, tox, friend_number, file_number): def __init__(self, path, tox, friend_number, size, file_number):
super(ReceiveTransfer, self).__init__(path, tox, friend_number, file_number) super(ReceiveTransfer, self).__init__(path, tox, friend_number, size, file_number)
self._file = open(self._path, 'wb') self._file = open(self._path, 'wb')
self._file.truncate(0) self._file.truncate(0)
self._size = 0 self._file_size = 0
def cancel(self): def cancel(self):
super(ReceiveTransfer, self).cancel() super(ReceiveTransfer, self).cancel()
remove(self._path) remove(self._path)
def write_chunk(self, position, data): def write_chunk(self, position, data):
if data is not None: if data is None:
self._file.close()
self.state = TOX_FILE_TRANSFER_STATE['FINISHED']
self._state_changed.signal.emit(self.state, self._done / self._size)
else:
data = ''.join(chr(x) for x in data) data = ''.join(chr(x) for x in data)
if self._size < position: if self._file_size < position:
self._file.seek(0, 2) self._file.seek(0, 2)
self._file.write('\0' * (position - self._size)) self._file.write('\0' * (position - self._file_size))
self._file.seek(position) self._file.seek(position)
self._file.write(data) self._file.write(data)
self._file.flush() self._file.flush()
if position + len(data) > self._size: if position + len(data) > self._file_size:
self._size = position + len(data) self._file_size = position + len(data)
self._signal.signal.emit('{} {}'.format(self.state, 0)) self._done += len(data)
else: self._state_changed.signal.emit(self.state, self._done / self._size)
self._file.close()
self.state = TOX_FILE_TRANSFER_STATE['FINISHED']
self._signal.signal.emit('{} {}'.format(self.state, 100))
class ReceiveAvatar(ReceiveTransfer): class ReceiveAvatar(ReceiveTransfer):
def __init__(self, tox, friend_number, file_number, has_size=True): def __init__(self, tox, friend_number, size, file_number):
path = profile.ProfileHelper.get_path() + '/avatars/{}.png'.format(tox.friend_get_public_key(friend_number)) path = profile.ProfileHelper.get_path() + '/avatars/{}.png'.format(tox.friend_get_public_key(friend_number))
super(ReceiveAvatar, self).__init__(path, tox, friend_number, file_number) super(ReceiveAvatar, self).__init__(path, tox, friend_number, size, file_number)
if exists(path): if exists(path):
if not has_size: if not size:
remove(path) remove(path)
self.send_control(TOX_FILE_CONTROL['CANCEL']) self.send_control(TOX_FILE_CONTROL['CANCEL'])
self.state = TOX_FILE_TRANSFER_STATE['CANCELED'] self.state = TOX_FILE_TRANSFER_STATE['CANCELED']

View file

@ -238,16 +238,15 @@ class FileTransferItem(QtGui.QListWidget):
pr.accept_transfer(self, directory + '/' + self.saved_name, friend_number, file_number) pr.accept_transfer(self, directory + '/' + self.saved_name, friend_number, file_number)
self.accept.setVisible(False) self.accept.setVisible(False)
@QtCore.Slot(str) @QtCore.Slot(int, float)
def update(self, data): def update(self, state, progress):
arr = data.split() self.pb.setValue(int(progress * 100))
self.pb.setValue(int(arr[1])) if state == TOX_FILE_TRANSFER_STATE['CANCELED']:
if int(arr[0]) == TOX_FILE_TRANSFER_STATE['CANCELED']:
self.setStyleSheet('QListWidget { background-color: red; }') self.setStyleSheet('QListWidget { background-color: red; }')
self.cancel.setVisible(False) self.cancel.setVisible(False)
self.accept.setVisible(False) self.accept.setVisible(False)
self.pb.setVisible(False) self.pb.setVisible(False)
elif int(arr[0]) == TOX_FILE_TRANSFER_STATE['FINISHED']: elif state == TOX_FILE_TRANSFER_STATE['FINISHED']:
self.pb.setVisible(False) self.pb.setVisible(False)
self.cancel.setVisible(False) self.cancel.setVisible(False)

View file

@ -733,7 +733,7 @@ class Profile(Contact, Singleton):
rt = ReceiveTransfer(path, self._tox, friend_number, file_number) rt = ReceiveTransfer(path, self._tox, friend_number, file_number)
self._file_transfers[(friend_number, file_number)] = rt self._file_transfers[(friend_number, file_number)] = rt
self._tox.file_control(friend_number, file_number, TOX_FILE_CONTROL['RESUME']) self._tox.file_control(friend_number, file_number, TOX_FILE_CONTROL['RESUME'])
rt.set_event_handler(item.update) rt.set_state_changed_handler(item.update)
def incoming_avatar(self, friend_number, file_number, size): def incoming_avatar(self, friend_number, file_number, size):
""" """
@ -769,7 +769,7 @@ class Profile(Contact, Singleton):
st = SendTransfer(path, self._tox, friend_number) st = SendTransfer(path, self._tox, friend_number)
self._file_transfers[(friend_number, st.get_file_number())] = st self._file_transfers[(friend_number, st.get_file_number())] = st
item = self.create_file_transfer_item(os.path.basename(path), os.path.getsize(path), friend_number, st.get_file_number(), False) item = self.create_file_transfer_item(os.path.basename(path), os.path.getsize(path), friend_number, st.get_file_number(), False)
st.set_event_handler(item.update) st.set_state_changed_handler(item.update)
def outgoing_chunk(self, friend_number, file_number, position, size): def outgoing_chunk(self, friend_number, file_number, position, size):
if (friend_number, file_number) in self._file_transfers: if (friend_number, file_number) in self._file_transfers: