emdee/toxygen
1
0
Fork 0
Code Issues Pull requests Projects Releases Wiki Activity

avatars support

Browse source
This commit is contained in:
ingvar1995 2016-03-18 16:20:07 +03:00
parent 6d56b2aa5b
commit 9d5353640b
3 changed files with 66 additions and 28 deletions
Download patch file Download diff file Expand all files Collapse all files

55
src/file_transfers.py
View file

@ -1,8 +1,11 @@
# 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
from toxcore_enums_and_consts import TOX_FILE_KIND, TOX_FILE_CONTROL
from os.path import basename, getsize, exists
from os import remove
from time import time
from tox import Tox
import profile
TOX_FILE_TRANSFER_STATE = {
@ -44,10 +47,15 @@ class FileTransfer(object):
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 __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) if path else '')
if path is not None:
self._file = open(path, 'rb')
def send_chunk(self, position, size):
self._file.seek(position)
@ -55,9 +63,19 @@ class SendTransfer(FileTransfer):
return self._tox.file_send_chunk(self._friend_number, self._file_number, position, data)
class SendAvatar(SendTransfer):
def __init__(self, path, tox, friend_number):
if path is None:
super(SendAvatar, self).__init__(path, tox, friend_number, TOX_FILE_KIND['AVATAR'])
else:
with open(path, 'rb') as fl:
hash = Tox.hash(fl.read())
super(self.__class__, self).__init__(path, tox, friend_number, TOX_FILE_KIND['AVATAR'], hash)
class ReceiveTransfer(FileTransfer):
def __init__(self, path, tox, friend_number, file_number):
super(self.__class__, self).__init__(path, tox, friend_number, file_number)
super(ReceiveTransfer, self).__init__(path, tox, friend_number, file_number)
self._file = open(self._path, 'wb')
self._file.truncate(0)
self._size = 0
@ -75,3 +93,26 @@ class ReceiveTransfer(FileTransfer):
self._size = position + len(data)
else:
self._file.close()
self.state = TOX_FILE_TRANSFER_STATE['FINISHED']
class ReceiveAvatar(ReceiveTransfer):
def __init__(self, tox, friend_number, file_number, has_size=True):
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)
if exists(path):
if not has_size:
remove(path)
self.send_control(TOX_FILE_CONTROL['CANCEL'])
self.state = TOX_FILE_TRANSFER_STATE['CANCELED']
else:
hash = self.get_file_id()
with open(path, 'rb') as fl:
existing_hash = Tox.hash(fl.read())
if hash == existing_hash:
self.send_control(TOX_FILE_CONTROL['CANCEL'])
self.state = TOX_FILE_TRANSFER_STATE['CANCELED']
else:
self.send_control(TOX_FILE_CONTROL['RESUME'])
else:
self.send_control(TOX_FILE_CONTROL['RESUME'])