toxygen/src/file_transfers.py

76 lines
2.3 KiB
Python
Raw Normal View History

2016-03-16 16:54:46 +00:00
# TODO: add support of file transfers
# TODO: add support of avatars
2016-03-16 20:11:18 +00:00
from toxcore_enums_and_consts import TOX_FILE_KIND
from os.path import basename, getsize
2016-03-17 17:46:18 +00:00
TOX_FILE_TRANSFER_STATE = {
'RUNNING': 0,
'PAUSED': 1,
'CANCELED': 2,
'FINISHED': 3,
}
2016-03-16 20:11:18 +00:00
class FileTransfer(object):
2016-03-17 17:46:18 +00:00
def __init__(self, path, tox, friend_number, file_number = None):
2016-03-16 20:11:18 +00:00
self._path = path
self._tox = tox
self._friend_number = friend_number
2016-03-17 17:46:18 +00:00
self.state = TOX_FILE_TRANSFER_STATE['RUNNING']
self._file_number = file_number
2016-03-16 20:11:18 +00:00
def set_tox(self, tox):
self._tox = tox
2016-03-17 17:46:18 +00:00
def get_file_number(self):
return self._file_number
def get_friend_number(self):
return self._friend_number
def send_control(self, control):
if self._tox.file_control(self._friend_number, self._file_number, control):
self.state = control
def get_file_id(self):
return self._tox.file_get_file_id(self._friend_number, self._file_number)
def file_seek(self):
# TODO implement
pass
2016-03-16 20:11:18 +00:00
class SendTransfer(FileTransfer):
def __init__(self, path, tox, friend_number):
super(self.__class__, self).__init__(path, tox, friend_number)
self._file_number = tox.file_send(friend_number, TOX_FILE_KIND['DATA'], getsize(path), None, basename(path))
2016-03-16 21:07:09 +00:00
self._file = open(path, 'rb')
2016-03-16 20:11:18 +00:00
def send_chunk(self, position, size):
2016-03-16 21:07:09 +00:00
self._file.seek(position)
data = self._file.read(size)
2016-03-16 20:11:18 +00:00
return self._tox.file_send_chunk(self._friend_number, self._file_number, position, data)
class ReceiveTransfer(FileTransfer):
2016-03-17 17:46:18 +00:00
def __init__(self, path, tox, friend_number, file_number):
super(self.__class__, self).__init__(path, tox, friend_number, file_number)
2016-03-16 21:07:09 +00:00
self._file = open(self._path, 'wb')
2016-03-17 16:28:52 +00:00
self._file.truncate(0)
2016-03-17 17:46:18 +00:00
self._size = 0
2016-03-16 20:11:18 +00:00
def write_chunk(self, position, data):
2016-03-17 15:55:18 +00:00
if data is not None:
2016-03-17 17:46:18 +00:00
data = ''.join(chr(x) for x in data)
if self._size < position:
2016-03-17 15:55:18 +00:00
self._file.seek(0, 2)
2016-03-17 17:46:18 +00:00
self._file.write('\0' * (position - self._size))
2016-03-16 21:07:09 +00:00
self._file.seek(position)
2016-03-17 17:46:18 +00:00
self._file.write(data)
2016-03-16 21:07:09 +00:00
self._file.flush()
2016-03-17 17:46:18 +00:00
if position + len(data) > self._size:
self._size = position + len(data)
2016-03-16 20:11:18 +00:00
else:
2016-03-17 15:55:18 +00:00
self._file.close()