avatars rest fix

This commit is contained in:
ingvar1995 2016-05-25 16:05:16 +03:00
parent be4a592599
commit b2fa484bb3

View file

@ -242,13 +242,13 @@ class ReceiveAvatar(ReceiveTransfer):
if size > self.MAX_AVATAR_SIZE: if size > self.MAX_AVATAR_SIZE:
self.send_control(TOX_FILE_CONTROL['CANCEL']) self.send_control(TOX_FILE_CONTROL['CANCEL'])
remove(path + '.tmp') remove(path + '.tmp')
elif exists(path): elif not size:
if not size:
self.send_control(TOX_FILE_CONTROL['CANCEL']) self.send_control(TOX_FILE_CONTROL['CANCEL'])
self._file.close() self._file.close()
if exists(path):
remove(path) remove(path)
remove(path + '.tmp') remove(path + '.tmp')
else: elif exists(path):
hash = self.get_file_id() hash = self.get_file_id()
with open(path) as fl: with open(path) as fl:
data = fl.read() data = fl.read()