441 lines
18 KiB
Python
441 lines
18 KiB
Python
from toxcore_enums_and_consts import *
|
|
try:
|
|
from PySide import QtCore, QtGui
|
|
except ImportError:
|
|
from PyQt4 import QtCore, QtGui
|
|
import profile
|
|
from file_transfers import TOX_FILE_TRANSFER_STATE, PAUSED_FILE_TRANSFERS, DO_NOT_SHOW_ACCEPT_BUTTON, ACTIVE_FILE_TRANSFERS, SHOW_PROGRESS_BAR
|
|
from util import curr_directory, convert_time
|
|
from widgets import DataLabel, create_menu
|
|
import cgi
|
|
import smileys
|
|
|
|
|
|
class MessageEdit(QtGui.QTextBrowser):
|
|
|
|
def __init__(self, text, width, parent=None):
|
|
super(MessageEdit, self).__init__(parent)
|
|
self.urls = {}
|
|
self.setHorizontalScrollBarPolicy(QtCore.Qt.ScrollBarAlwaysOff)
|
|
self.setVerticalScrollBarPolicy(QtCore.Qt.ScrollBarAlwaysOff)
|
|
self.setWordWrapMode(QtGui.QTextOption.WrapAtWordBoundaryOrAnywhere)
|
|
self.document().setTextWidth(width)
|
|
self.setOpenExternalLinks(True)
|
|
self.setAcceptRichText(True)
|
|
self.setOpenLinks(False)
|
|
self.setSearchPaths([smileys.SmileyLoader.get_instance().get_smileys_path()])
|
|
self.document().setDefaultStyleSheet('a { color: #306EFF; }')
|
|
self.setDecoratedText(text)
|
|
font = QtGui.QFont()
|
|
font.setFamily("Times New Roman")
|
|
font.setPixelSize(14)
|
|
font.setBold(False)
|
|
self.setFont(font)
|
|
self.setFixedHeight(self.document().size().height())
|
|
self.setTextInteractionFlags(QtCore.Qt.TextSelectableByMouse | QtCore.Qt.LinksAccessibleByMouse)
|
|
self.anchorClicked.connect(self.on_anchor_clicked)
|
|
|
|
def contextMenuEvent(self, event):
|
|
menu = create_menu(self.createStandardContextMenu(event.pos()))
|
|
menu.popup(event.globalPos())
|
|
menu.exec_(event.globalPos())
|
|
del menu
|
|
|
|
def on_anchor_clicked(self, url):
|
|
text = str(url.toString())
|
|
if text.startswith('tox:'):
|
|
import menu
|
|
self.add_contact = menu.AddContact(text[4:])
|
|
self.add_contact.show()
|
|
else:
|
|
QtGui.QDesktopServices.openUrl(url)
|
|
self.clearFocus()
|
|
|
|
def addAnimation(self, url, fileName):
|
|
movie = QtGui.QMovie(self)
|
|
movie.setFileName(fileName)
|
|
self.urls[movie] = url
|
|
movie.frameChanged[int].connect(lambda x: self.animate(movie))
|
|
movie.start()
|
|
|
|
def animate(self, movie):
|
|
self.document().addResource(QtGui.QTextDocument.ImageResource,
|
|
self.urls[movie],
|
|
movie.currentPixmap())
|
|
self.setLineWrapColumnOrWidth(self.lineWrapColumnOrWidth())
|
|
|
|
def setDecoratedText(self, text):
|
|
text = cgi.escape(text) # replace < and >
|
|
exp = QtCore.QRegExp(
|
|
'('
|
|
'(?:\\b)((www\\.)|(http[s]?|ftp)://)'
|
|
'\\w+\\S+)'
|
|
'|(?:\\b)(file:///)([\\S| ]*)'
|
|
'|(?:\\b)(tox:[a-zA-Z\\d]{76}$)'
|
|
'|(?:\\b)(mailto:\\S+@\\S+\\.\\S+)'
|
|
'|(?:\\b)(tox:\\S+@\\S+)')
|
|
offset = exp.indexIn(text, 0)
|
|
while offset != -1: # add links
|
|
url = exp.cap()
|
|
if exp.cap(2) == 'www.':
|
|
html = '<a href="http://{0}">{0}</a>'.format(url)
|
|
else:
|
|
html = '<a href="{0}">{0}</a>'.format(url)
|
|
text = text[:offset] + html + text[offset + len(exp.cap()):]
|
|
offset += len(html)
|
|
offset = exp.indexIn(text, offset)
|
|
arr = text.split('\n')
|
|
for i in range(len(arr)): # quotes
|
|
if arr[i].startswith('>'):
|
|
arr[i] = '<font color="green"><b>' + arr[i][4:] + '</b></font>'
|
|
text = '<br>'.join(arr)
|
|
text = smileys.SmileyLoader.get_instance().add_smileys_to_text(text, self) # smileys
|
|
self.setHtml(text)
|
|
|
|
|
|
class MessageItem(QtGui.QWidget):
|
|
"""
|
|
Message in messages list
|
|
"""
|
|
def __init__(self, text, time, user='', sent=True, message_type=TOX_MESSAGE_TYPE['NORMAL'], parent=None):
|
|
QtGui.QWidget.__init__(self, parent)
|
|
self.name = DataLabel(self)
|
|
self.name.setGeometry(QtCore.QRect(2, 2, 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.time = QtGui.QLabel(self)
|
|
self.time.setGeometry(QtCore.QRect(parent.width() - 50, 0, 50, 20))
|
|
font = QtGui.QFont()
|
|
font.setFamily("Times New Roman")
|
|
font.setPointSize(10)
|
|
font.setBold(False)
|
|
self.time.setFont(font)
|
|
|
|
if not sent:
|
|
movie = QtGui.QMovie(curr_directory() + '/images/spinner.gif')
|
|
self.time.setMovie(movie)
|
|
movie.start()
|
|
self.t = time
|
|
else:
|
|
self.time.setText(time)
|
|
|
|
self.message = MessageEdit(text, parent.width() - 150, self)
|
|
self.message.setGeometry(QtCore.QRect(100, 0, parent.width() - 150, self.message.height()))
|
|
self.setFixedHeight(self.message.height())
|
|
|
|
if message_type != TOX_MESSAGE_TYPE['NORMAL']:
|
|
self.name.setStyleSheet("QLabel { color: #5CB3FF; }")
|
|
self.message.setStyleSheet("QTextEdit { color: #5CB3FF; font: italic; font-size: 20px; }")
|
|
self.message.setAlignment(QtCore.Qt.AlignCenter)
|
|
self.time.setStyleSheet("QLabel { color: #5CB3FF; }")
|
|
|
|
def mark_as_sent(self):
|
|
if hasattr(self, 't'):
|
|
self.time.setText(self.t)
|
|
del self.t
|
|
return True
|
|
return False
|
|
|
|
|
|
class ContactItem(QtGui.QWidget):
|
|
"""
|
|
Contact in friends list
|
|
"""
|
|
|
|
def __init__(self, parent=None):
|
|
QtGui.QWidget.__init__(self, parent)
|
|
self.setBaseSize(QtCore.QSize(250, 70))
|
|
self.avatar_label = QtGui.QLabel(self)
|
|
self.avatar_label.setGeometry(QtCore.QRect(3, 3, 64, 64))
|
|
self.avatar_label.setScaledContents(True)
|
|
self.name = DataLabel(self)
|
|
self.name.setGeometry(QtCore.QRect(75, 10, 150, 25))
|
|
font = QtGui.QFont()
|
|
font.setFamily("Times New Roman")
|
|
font.setPointSize(12)
|
|
font.setBold(True)
|
|
self.name.setFont(font)
|
|
self.status_message = DataLabel(self)
|
|
self.status_message.setGeometry(QtCore.QRect(75, 30, 170, 20))
|
|
font.setPointSize(10)
|
|
font.setBold(False)
|
|
self.status_message.setFont(font)
|
|
self.connection_status = StatusCircle(self)
|
|
self.connection_status.setGeometry(QtCore.QRect(230, 5, 32, 32))
|
|
self.messages = UnreadMessagesCount(self)
|
|
self.messages.setGeometry(QtCore.QRect(52, 50, 30, 20))
|
|
|
|
|
|
class StatusCircle(QtGui.QWidget):
|
|
"""
|
|
Connection status
|
|
"""
|
|
def __init__(self, parent):
|
|
QtGui.QWidget.__init__(self, parent)
|
|
self.setGeometry(0, 0, 32, 32)
|
|
self.label = QtGui.QLabel(self)
|
|
self.label.setGeometry(QtCore.QRect(0, 0, 32, 32))
|
|
self.unread = False
|
|
|
|
def update(self, status, unread_messages=None):
|
|
if unread_messages is None:
|
|
unread_messages = self.unread
|
|
else:
|
|
self.unread = unread_messages
|
|
if status == TOX_USER_STATUS['NONE']:
|
|
name = 'online'
|
|
elif status == TOX_USER_STATUS['AWAY']:
|
|
name = 'idle'
|
|
elif status == TOX_USER_STATUS['BUSY']:
|
|
name = 'busy'
|
|
else:
|
|
name = 'offline'
|
|
if unread_messages:
|
|
name += '_notification'
|
|
pixmap = QtGui.QPixmap(curr_directory() + '/images/{}.png'.format(name))
|
|
self.label.setPixmap(pixmap)
|
|
|
|
|
|
class UnreadMessagesCount(QtGui.QWidget):
|
|
|
|
def __init__(self, parent=None):
|
|
super(UnreadMessagesCount, self).__init__(parent)
|
|
self.resize(30, 20)
|
|
self.label = QtGui.QLabel(self)
|
|
self.label.setGeometry(QtCore.QRect(0, 0, 30, 20))
|
|
self.label.setVisible(False)
|
|
font = QtGui.QFont()
|
|
font.setFamily("Times New Roman")
|
|
font.setPointSize(12)
|
|
font.setBold(True)
|
|
self.label.setFont(font)
|
|
self.label.setAlignment(QtCore.Qt.AlignVCenter | QtCore.Qt.AlignCenter)
|
|
self.label.setStyleSheet('QLabel { color: white; background-color: red; border-radius: 10; }')
|
|
|
|
def update(self, messages_count):
|
|
if messages_count:
|
|
self.label.setVisible(True)
|
|
self.label.setText(str(messages_count))
|
|
else:
|
|
self.label.setVisible(False)
|
|
|
|
|
|
class FileTransferItem(QtGui.QListWidget):
|
|
|
|
def __init__(self, file_name, size, time, user, friend_number, file_number, state, width, parent=None):
|
|
|
|
QtGui.QListWidget.__init__(self, parent)
|
|
self.resize(QtCore.QSize(width, 34))
|
|
if state == TOX_FILE_TRANSFER_STATE['CANCELLED']:
|
|
self.setStyleSheet('QListWidget { border: 1px solid #B40404; }')
|
|
elif state in PAUSED_FILE_TRANSFERS:
|
|
self.setStyleSheet('QListWidget { border: 1px solid #FF8000; }')
|
|
else:
|
|
self.setStyleSheet('QListWidget { border: 1px solid green; }')
|
|
self.state = state
|
|
|
|
self.name = DataLabel(self)
|
|
self.name.setGeometry(QtCore.QRect(3, 7, 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.time = QtGui.QLabel(self)
|
|
self.time.setGeometry(QtCore.QRect(width - 53, 7, 50, 20))
|
|
font.setPointSize(10)
|
|
font.setBold(False)
|
|
self.time.setFont(font)
|
|
self.time.setText(convert_time(time))
|
|
|
|
self.cancel = QtGui.QPushButton(self)
|
|
self.cancel.setGeometry(QtCore.QRect(width - 120, 2, 30, 30))
|
|
pixmap = QtGui.QPixmap(curr_directory() + '/images/decline.png')
|
|
icon = QtGui.QIcon(pixmap)
|
|
self.cancel.setIcon(icon)
|
|
self.cancel.setIconSize(QtCore.QSize(30, 30))
|
|
self.cancel.setVisible(state in ACTIVE_FILE_TRANSFERS)
|
|
self.cancel.clicked.connect(lambda: self.cancel_transfer(friend_number, file_number))
|
|
self.cancel.setStyleSheet('QPushButton:hover { border: 1px solid #3A3939; background-color: none;}')
|
|
|
|
self.accept_or_pause = QtGui.QPushButton(self)
|
|
self.accept_or_pause.setGeometry(QtCore.QRect(width - 170, 2, 30, 30))
|
|
if state == TOX_FILE_TRANSFER_STATE['INCOMING_NOT_STARTED']:
|
|
self.accept_or_pause.setVisible(True)
|
|
self.button_update('accept')
|
|
elif state in DO_NOT_SHOW_ACCEPT_BUTTON:
|
|
self.accept_or_pause.setVisible(False)
|
|
elif state == TOX_FILE_TRANSFER_STATE['PAUSED_BY_USER']: # setup for continue
|
|
self.accept_or_pause.setVisible(True)
|
|
self.button_update('resume')
|
|
else: # pause
|
|
self.accept_or_pause.setVisible(True)
|
|
self.button_update('pause')
|
|
self.accept_or_pause.clicked.connect(lambda: self.accept_or_pause_transfer(friend_number, file_number, size))
|
|
|
|
self.accept_or_pause.setStyleSheet('QPushButton:hover { border: 1px solid #3A3939; background-color: none}')
|
|
|
|
self.pb = QtGui.QProgressBar(self)
|
|
self.pb.setGeometry(QtCore.QRect(100, 7, 100, 20))
|
|
self.pb.setValue(0)
|
|
self.pb.setStyleSheet('QProgressBar { background-color: #302F2F; }')
|
|
self.pb.setVisible(state in SHOW_PROGRESS_BAR)
|
|
|
|
self.file_name = DataLabel(self)
|
|
self.file_name.setGeometry(QtCore.QRect(210, 7, width - 440, 20))
|
|
font.setPointSize(12)
|
|
self.file_name.setFont(font)
|
|
file_size = size / 1024
|
|
if not file_size:
|
|
file_size = '{}B'.format(size)
|
|
elif file_size >= 1024:
|
|
file_size = '{}MB'.format(file_size / 1024)
|
|
else:
|
|
file_size = '{}KB'.format(file_size)
|
|
file_data = u'{} {}'.format(file_size, file_name)
|
|
self.file_name.setText(file_data)
|
|
self.file_name.setToolTip(file_name)
|
|
self.saved_name = file_name
|
|
self.time_left = QtGui.QLabel(self)
|
|
self.time_left.setGeometry(QtCore.QRect(width - 83, 7, 30, 20))
|
|
font.setPointSize(10)
|
|
self.time_left.setFont(font)
|
|
self.time_left.setVisible(state == TOX_FILE_TRANSFER_STATE['RUNNING'])
|
|
self.setFocusPolicy(QtCore.Qt.NoFocus)
|
|
self.paused = False
|
|
|
|
def cancel_transfer(self, friend_number, file_number):
|
|
pr = profile.Profile.get_instance()
|
|
pr.cancel_transfer(friend_number, file_number)
|
|
self.setStyleSheet('QListWidget { border: 1px solid #B40404; }')
|
|
self.cancel.setVisible(False)
|
|
self.accept_or_pause.setVisible(False)
|
|
self.pb.setVisible(False)
|
|
|
|
def accept_or_pause_transfer(self, friend_number, file_number, size):
|
|
if self.state == TOX_FILE_TRANSFER_STATE['INCOMING_NOT_STARTED']:
|
|
directory = QtGui.QFileDialog.getExistingDirectory(self,
|
|
QtGui.QApplication.translate("MainWindow", 'Choose folder', None, QtGui.QApplication.UnicodeUTF8),
|
|
curr_directory(),
|
|
QtGui.QFileDialog.ShowDirsOnly)
|
|
self.pb.setVisible(True)
|
|
if directory:
|
|
pr = profile.Profile.get_instance()
|
|
pr.accept_transfer(self, directory + '/' + self.saved_name, friend_number, file_number, size)
|
|
self.button_update('pause')
|
|
elif self.state == TOX_FILE_TRANSFER_STATE['PAUSED_BY_USER']: # resume
|
|
self.paused = False
|
|
profile.Profile.get_instance().resume_transfer(friend_number, file_number)
|
|
self.button_update('pause')
|
|
self.state = TOX_FILE_TRANSFER_STATE['RUNNING']
|
|
else: # pause
|
|
self.paused = True
|
|
self.state = TOX_FILE_TRANSFER_STATE['PAUSED_BY_USER']
|
|
profile.Profile.get_instance().pause_transfer(friend_number, file_number)
|
|
self.button_update('resume')
|
|
self.accept_or_pause.clearFocus()
|
|
|
|
def button_update(self, path):
|
|
pixmap = QtGui.QPixmap(curr_directory() + '/images/{}.png'.format(path))
|
|
icon = QtGui.QIcon(pixmap)
|
|
self.accept_or_pause.setIcon(icon)
|
|
self.accept_or_pause.setIconSize(QtCore.QSize(30, 30))
|
|
|
|
@QtCore.Slot(int, float, int)
|
|
def update(self, state, progress, time):
|
|
self.pb.setValue(int(progress * 100))
|
|
if time + 1:
|
|
m, s = divmod(time, 60)
|
|
self.time_left.setText('{0:02d}:{0:02d}'.format(m, s))
|
|
if self.state != state:
|
|
if state == TOX_FILE_TRANSFER_STATE['CANCELLED']:
|
|
self.setStyleSheet('QListWidget { border: 1px solid #B40404; }')
|
|
self.cancel.setVisible(False)
|
|
self.accept_or_pause.setVisible(False)
|
|
self.pb.setVisible(False)
|
|
self.state = state
|
|
self.time_left.setVisible(False)
|
|
elif state == TOX_FILE_TRANSFER_STATE['FINISHED']:
|
|
self.accept_or_pause.setVisible(False)
|
|
self.pb.setVisible(False)
|
|
self.cancel.setVisible(False)
|
|
self.setStyleSheet('QListWidget { border: 1px solid green; }')
|
|
self.state = state
|
|
self.time_left.setVisible(False)
|
|
elif state == TOX_FILE_TRANSFER_STATE['PAUSED_BY_FRIEND']:
|
|
self.accept_or_pause.setVisible(False)
|
|
self.setStyleSheet('QListWidget { border: 1px solid #FF8000; }')
|
|
self.state = state
|
|
self.time_left.setVisible(False)
|
|
elif state == TOX_FILE_TRANSFER_STATE['PAUSED_BY_USER']:
|
|
self.button_update('resume') # setup button continue
|
|
self.setStyleSheet('QListWidget { border: 1px solid green; }')
|
|
self.state = state
|
|
self.time_left.setVisible(False)
|
|
elif state == TOX_FILE_TRANSFER_STATE['OUTGOING_NOT_STARTED']:
|
|
self.setStyleSheet('QListWidget { border: 1px solid #FF8000; }')
|
|
self.accept_or_pause.setVisible(False)
|
|
self.time_left.setVisible(False)
|
|
self.pb.setVisible(False)
|
|
elif not self.paused: # active
|
|
self.pb.setVisible(True)
|
|
self.accept_or_pause.setVisible(True) # setup to pause
|
|
self.button_update('pause')
|
|
self.setStyleSheet('QListWidget { border: 1px solid green; }')
|
|
self.state = state
|
|
self.time_left.setVisible(True)
|
|
|
|
def mark_as_sent(self):
|
|
return False
|
|
|
|
|
|
class UnsentFileItem(FileTransferItem):
|
|
|
|
def __init__(self, file_name, size, user, time, width, parent=None):
|
|
super(UnsentFileItem, self).__init__(file_name, size, time, user, -1, -1,
|
|
TOX_FILE_TRANSFER_STATE['PAUSED_BY_FRIEND'], width, parent)
|
|
self._time = time
|
|
self.pb.setVisible(False)
|
|
|
|
def cancel_transfer(self, *args):
|
|
pr = profile.Profile.get_instance()
|
|
pr.cancel_not_started_transfer(self._time)
|
|
|
|
|
|
class InlineImageItem(QtGui.QWidget):
|
|
|
|
def __init__(self, data, width, parent=None):
|
|
|
|
QtGui.QWidget.__init__(self, parent)
|
|
self.resize(QtCore.QSize(width, 500))
|
|
self._image_label = QtGui.QLabel(self)
|
|
self._image_label.raise_()
|
|
self._image_label.setAutoFillBackground(True)
|
|
self._image_label.setScaledContents(False)
|
|
self.pixmap = QtGui.QPixmap()
|
|
self.pixmap.loadFromData(QtCore.QByteArray(data), "PNG")
|
|
max_size = width - 40
|
|
if self.pixmap.width() <= max_size:
|
|
self._image_label.setPixmap(self.pixmap)
|
|
self.resize(QtCore.QSize(max_size, self.pixmap.height()))
|
|
else:
|
|
pixmap = self.pixmap.scaled(max_size, max_size, QtCore.Qt.KeepAspectRatio)
|
|
self._image_label.setPixmap(pixmap)
|
|
self.resize(QtCore.QSize(max_size, pixmap.height()))
|
|
|
|
def mark_as_sent(self):
|
|
return False
|
|
|
|
|
|
|
|
|