inline images #1
This commit is contained in:
ingvar1995
parent
ba3261b76b
commit
df02fd9c4d
@ -143,7 +143,7 @@ class ReceiveTransfer(FileTransfer):
|
||||
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 if self._size else 0)
|
||||
self._state_changed.signal.emit(self.state, 1)
|
||||
else:
|
||||
data = ''.join(chr(x) for x in data)
|
||||
if self._file_size < position:
|
||||
@ -159,6 +159,32 @@ class ReceiveTransfer(FileTransfer):
|
||||
self._state_changed.signal.emit(self.state, self._done / self._size)
|
||||
|
||||
|
||||
class ReceiveToBuffer(FileTransfer):
|
||||
|
||||
def __init__(self, tox, friend_number, size, file_number):
|
||||
super(ReceiveToBuffer, self).__init__(None, tox, friend_number, size, file_number)
|
||||
self._data = ''
|
||||
self._data_size = 0
|
||||
|
||||
def get_data(self):
|
||||
return self._data
|
||||
|
||||
def write_chunk(self, position, data):
|
||||
if data is None:
|
||||
self.state = TOX_FILE_TRANSFER_STATE['FINISHED']
|
||||
self._state_changed.signal.emit(self.state, 1)
|
||||
else:
|
||||
data = ''.join(chr(x) for x in data)
|
||||
l = len(data)
|
||||
if self._data_size < position:
|
||||
self._data += ('\0' * (position - self._data_size))
|
||||
self._data = self._data[:position] + data + self._data[position + l:]
|
||||
if position + l > self._data_size:
|
||||
self._data_size = position + l
|
||||
self._done += l
|
||||
self._state_changed.signal.emit(self.state, self._done / self._size)
|
||||
|
||||
|
||||
class ReceiveAvatar(ReceiveTransfer):
|
||||
MAX_AVATAR_SIZE = 512 * 1024
|
||||
|
||||
|
||||
@ -242,3 +242,40 @@ class FileTransferItem(QtGui.QListWidget):
|
||||
self.pb.setVisible(False)
|
||||
self.cancel.setVisible(False)
|
||||
|
||||
|
||||
class InlineImageItem(QtGui.QWidget):
|
||||
|
||||
def __init__(self, time, user, data, parent=None):
|
||||
|
||||
QtGui.QWidget.__init__(self, parent)
|
||||
self.resize(QtCore.QSize(620, 500))
|
||||
# self.name = DataLabel(self)
|
||||
# self.name.setGeometry(QtCore.QRect(1, 15, 95, 20))
|
||||
# self.name.setTextFormat(QtCore.Qt.PlainText)
|
||||
# font = QtGui.QFont()
|
||||
# font.setFamily("Times New Roman")
|
||||
# font.setPointSize(11)
|
||||
# font.setBold(True)
|
||||
# self.name.setFont(font)
|
||||
# self.name.setText(user)
|
||||
# self.name.setStyleSheet('QLabel { color: black; }')
|
||||
#
|
||||
# self.time = QtGui.QLabel(self)
|
||||
# self.time.setGeometry(QtCore.QRect(570, 2, 50, 46))
|
||||
# font.setPointSize(10)
|
||||
# font.setBold(False)
|
||||
# self.time.setFont(font)
|
||||
# self.time.setText(convert_time(time))
|
||||
# self.time.setStyleSheet('QLabel { color: black; }')
|
||||
# TODO: resize widget to picture size or resize picture
|
||||
self._image_label = QtGui.QLabel(self)
|
||||
self._image_label.raise_()
|
||||
self._image_label.setGeometry(QtCore.QRect(0, 0, 600, 600))
|
||||
self._image_label.setScaledContents(False)
|
||||
self.pixmap = QtGui.QPixmap()
|
||||
print self.pixmap.loadFromData(QtCore.QByteArray(data), "PNG")
|
||||
self._image_label.setPixmap(self.pixmap)
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
@ -75,7 +75,6 @@ class Toxygen(object):
|
||||
self.tox = tox_factory(data, settings)
|
||||
|
||||
if ProfileHelper.is_active_profile(path, name): # profile is in use
|
||||
deactivate = False
|
||||
reply = QtGui.QMessageBox.question(None,
|
||||
'Profile {}'.format(name),
|
||||
QtGui.QApplication.translate("login", 'Looks like other instance of Toxygen uses this profile! Continue?', None, QtGui.QApplication.UnicodeUTF8),
|
||||
@ -85,7 +84,6 @@ class Toxygen(object):
|
||||
return
|
||||
else:
|
||||
settings.set_active_profile()
|
||||
deactivate = True
|
||||
|
||||
lang = filter(lambda x: x[0] == settings['language'], Settings.supported_languages())[0]
|
||||
translator = QtCore.QTranslator()
|
||||
@ -136,7 +134,6 @@ class Toxygen(object):
|
||||
self.init.wait()
|
||||
data = self.tox.get_savedata()
|
||||
ProfileHelper.save_profile(data)
|
||||
if deactivate:
|
||||
settings.close()
|
||||
del self.tox
|
||||
|
||||
|
||||
@ -1,4 +1,4 @@
|
||||
from list_items import MessageItem, ContactItem, FileTransferItem
|
||||
from list_items import MessageItem, ContactItem, FileTransferItem, InlineImageItem
|
||||
from PySide import QtCore, QtGui
|
||||
from tox import Tox
|
||||
import os
|
||||
@ -194,16 +194,19 @@ class Friend(Contact):
|
||||
"""
|
||||
if hasattr(self, '_message_getter'):
|
||||
del self._message_getter
|
||||
self._corr = filter(lambda x: x.get_type() > 1, self._corr)
|
||||
self._corr = filter(lambda x: x.get_type() == 2, self._corr)
|
||||
self._unsaved_messages = 0
|
||||
|
||||
def update_transfer_data(self, file_number, status):
|
||||
def update_transfer_data(self, file_number, status, inline=None):
|
||||
"""
|
||||
Update status of active transfer
|
||||
"""
|
||||
try:
|
||||
tr = filter(lambda x: x.get_type() == 2 and x.is_active(file_number), self._corr)[0]
|
||||
tr = filter(lambda x: x.get_type() >= 2 and x.is_active(file_number), self._corr)[0]
|
||||
tr.set_status(status)
|
||||
if inline: # inline was loaded
|
||||
i = self._corr.index(tr)
|
||||
self._corr.insert(i, inline)
|
||||
except:
|
||||
pass
|
||||
|
||||
@ -397,6 +400,8 @@ class Profile(Contact, Singleton):
|
||||
if message.get_status() in (2, 4): # active file transfer
|
||||
ft = self._file_transfers[(message.get_friend_number(), message.get_file_number())]
|
||||
ft.set_state_changed_handler(item.update)
|
||||
else: # inline
|
||||
self.create_inline_item(message.get_data())
|
||||
self._messages.scrollToBottom()
|
||||
else:
|
||||
friend = self._friends[self._active_friend]
|
||||
@ -593,6 +598,17 @@ class Profile(Contact, Singleton):
|
||||
self._messages.repaint()
|
||||
return item
|
||||
|
||||
def create_inline_item(self, data, append=True):
|
||||
item = InlineImageItem(0, '', data)
|
||||
elem = QtGui.QListWidgetItem()
|
||||
elem.setSizeHint(QtCore.QSize(600, 600))
|
||||
if append:
|
||||
self._messages.addItem(elem)
|
||||
else:
|
||||
self._messages.insertItem(0, elem)
|
||||
self._messages.setItemWidget(elem, item)
|
||||
self._messages.repaint()
|
||||
|
||||
# -----------------------------------------------------------------------------------------------------------------
|
||||
# Work with friends (remove, set alias, get public key)
|
||||
# -----------------------------------------------------------------------------------------------------------------
|
||||
@ -732,7 +748,18 @@ class Profile(Contact, Singleton):
|
||||
settings = Settings.get_instance()
|
||||
friend = self.get_friend_by_number(friend_number)
|
||||
file_name = file_name.decode('utf-8')
|
||||
if settings['allow_auto_accept'] and friend.tox_id in settings['auto_accept_from_friends']:
|
||||
auto = settings['allow_auto_accept'] and friend.tox_id in settings['auto_accept_from_friends']
|
||||
inline = (file_name == 'toxygen_inline.png' or file_name == 'utox-inline.png') and settings['allow_inline']
|
||||
if inline and size < 1024 * 1024:
|
||||
self.accept_transfer(None, '', friend_number, file_number, size, True)
|
||||
tm = TransferMessage(MESSAGE_OWNER['FRIEND'],
|
||||
time.time(),
|
||||
FILE_TRANSFER_MESSAGE_STATUS['INCOMING_STARTED'],
|
||||
size,
|
||||
file_name,
|
||||
friend_number,
|
||||
file_number)
|
||||
elif auto:
|
||||
path = settings['auto_accept_path'] or curr_directory()
|
||||
new_file_name, i = file_name, 1
|
||||
while os.path.isfile(path + '/' + new_file_name): # file with same name already exists
|
||||
@ -784,20 +811,24 @@ class Profile(Contact, Singleton):
|
||||
self.get_friend_by_number(friend_number).update_transfer_data(file_number,
|
||||
FILE_TRANSFER_MESSAGE_STATUS['CANCELLED'])
|
||||
|
||||
def accept_transfer(self, item, path, friend_number, file_number, size):
|
||||
def accept_transfer(self, item, path, friend_number, file_number, size, inline=False):
|
||||
"""
|
||||
:param item: transfer item
|
||||
:param item: transfer item. None if auto accept
|
||||
:param path: path for saving
|
||||
:param friend_number: friend number
|
||||
:param file_number: file number
|
||||
:param size: file size
|
||||
"""
|
||||
if not inline:
|
||||
rt = ReceiveTransfer(path, self._tox, friend_number, size, file_number)
|
||||
else:
|
||||
rt = ReceiveToBuffer(self._tox, friend_number, size, file_number)
|
||||
self._file_transfers[(friend_number, file_number)] = rt
|
||||
self._tox.file_control(friend_number, file_number, TOX_FILE_CONTROL['RESUME'])
|
||||
if item is not None:
|
||||
rt.set_state_changed_handler(item.update)
|
||||
self.get_friend_by_number(friend_number).update_transfer_data(file_number, FILE_TRANSFER_MESSAGE_STATUS['INCOMING_STARTED'])
|
||||
self.get_friend_by_number(friend_number).update_transfer_data(file_number,
|
||||
FILE_TRANSFER_MESSAGE_STATUS['INCOMING_STARTED'])
|
||||
|
||||
def send_screenshot(self, data):
|
||||
"""
|
||||
@ -847,6 +878,13 @@ class Profile(Contact, Singleton):
|
||||
if type(transfer) is ReceiveAvatar:
|
||||
self.get_friend_by_number(friend_number).load_avatar()
|
||||
self.set_active(None)
|
||||
elif type(transfer) is ReceiveToBuffer:
|
||||
inline = InlineImage(0, '', transfer.get_data())
|
||||
self.get_friend_by_number(friend_number).update_transfer_data(file_number,
|
||||
FILE_TRANSFER_MESSAGE_STATUS['FINISHED'],
|
||||
inline
|
||||
)
|
||||
self.set_active(None)
|
||||
else:
|
||||
self.get_friend_by_number(friend_number).update_transfer_data(file_number, FILE_TRANSFER_MESSAGE_STATUS['FINISHED'])
|
||||
del self._file_transfers[(friend_number, file_number)]
|
||||
|
||||
Loading…
Reference in New Issue
Block a user