toxygen/src/file_transfers.py

# TODO: add support of file transfers
# TODO: add support of avatars
from toxcore_enums_and_consts import TOX_FILE_KIND
from os.path import basename, getsize


class FileTransfer(object):
    def __init__(self, path, tox, friend_number):
        self._path = path
        self._tox = tox
        self._friend_number = friend_number

    def set_tox(self, tox):
        self._tox = tox


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))
        self._file = open(path, 'rb')

    def get_file_number(self):
        return self._file_number

    def send_chunk(self, position, size):
        self._file.seek(position)
        data = self._file.read(size)
        return self._tox.file_send_chunk(self._friend_number, self._file_number, position, data)


class ReceiveTransfer(FileTransfer):
    def __init__(self, path, tox, friend_number):
        super(self.__class__, self).__init__(path, tox, friend_number)
        self._file = open(self._path, 'wb')
        self._file.truncate(0)

    def write_chunk(self, position, data):
        if data is not None:
            size = getsize(self._path)
            if size < position:
                self._file.seek(0, 2)
                self._file.write('\0' * (position - size))
            self._file.seek(position)
            self._file.write(''.join(chr(x) for x in data))
            self._file.flush()
        else:
            self._file.close()