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 = '{0}'.format(url) else: html = '{0}'.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] = '' + arr[i][4:] + '' text = '
'.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