123 lines
4.3 KiB
Python
123 lines
4.3 KiB
Python
from PySide import QtCore
|
|
from notifications import *
|
|
from settings import Settings
|
|
from profile import Profile
|
|
from toxcore_enums_and_consts import *
|
|
# TODO: add all callbacks (remove test callbacks and use wrappers)
|
|
|
|
|
|
class InvokeEvent(QtCore.QEvent):
|
|
EVENT_TYPE = QtCore.QEvent.Type(QtCore.QEvent.registerEventType())
|
|
|
|
def __init__(self, fn, *args, **kwargs):
|
|
QtCore.QEvent.__init__(self, InvokeEvent.EVENT_TYPE)
|
|
self.fn = fn
|
|
self.args = args
|
|
self.kwargs = kwargs
|
|
|
|
|
|
class Invoker(QtCore.QObject):
|
|
|
|
def event(self, event):
|
|
event.fn(*event.args, **event.kwargs)
|
|
return True
|
|
|
|
_invoker = Invoker()
|
|
|
|
|
|
def invoke_in_main_thread(fn, *args, **kwargs):
|
|
QtCore.QCoreApplication.postEvent(_invoker, InvokeEvent(fn, *args, **kwargs))
|
|
|
|
|
|
def self_connection_status(st, tox_link):
|
|
"""
|
|
:param st: widget on mainscreen which shows status
|
|
:param tox_link: tox instance
|
|
:return: function for tox.callback_self_connection_status
|
|
"""
|
|
def wrapped(tox, connection, user_data):
|
|
print 'Connection status: ', str(connection)
|
|
invoke_in_main_thread(st.repaint)
|
|
return wrapped
|
|
|
|
|
|
def friend_status(tox, friend_num, new_status, user_data):
|
|
"""
|
|
Check friend's status (none, busy, away)
|
|
"""
|
|
print "Friend's #{} status changed! New status: ".format(friend_num, new_status)
|
|
profile = Profile.get_instance()
|
|
friend = filter(lambda x: x.number == friend_num, profile.friends)[0]
|
|
invoke_in_main_thread(friend.set_status, new_status)
|
|
|
|
|
|
def friend_connection_status(tox, friend_num, new_status, user_data):
|
|
"""
|
|
Check friend's connection status (offline, udp, tcp)
|
|
"""
|
|
print "Friend #{} connected! Friend's status: ".format(friend_num, new_status)
|
|
profile = Profile.get_instance()
|
|
friend = filter(lambda x: x.number == friend_num, profile.friends)[0]
|
|
if new_status == TOX_CONNECTION['NONE']:
|
|
invoke_in_main_thread(friend.set_status, None)
|
|
elif friend.status is None:
|
|
invoke_in_main_thread(friend.set_status, TOX_USER_STATUS['NONE'])
|
|
|
|
|
|
def friend_name(window):
|
|
"""
|
|
:param window: main window
|
|
:return: function for callback friend_name. It updates friend's name
|
|
and calls window repaint
|
|
"""
|
|
def wrapped(tox, friend_num, name, size, user_data):
|
|
profile = Profile.get_instance()
|
|
friend = filter(lambda x: x.number == friend_num, profile.friends)[0]
|
|
print 'New name: ', str(friend_num), str(name)
|
|
invoke_in_main_thread(friend.set_name, name)
|
|
invoke_in_main_thread(window.update_active_friend)
|
|
return wrapped
|
|
|
|
|
|
def friend_status_message(window):
|
|
"""
|
|
:param window: main window
|
|
:return: function for callback friend_status_message. It updates friend's status message
|
|
and calls window repaint
|
|
"""
|
|
def wrapped(tox, friend_num, status_message, size, user_data):
|
|
profile = Profile.get_instance()
|
|
friend = filter(lambda x: x.number == friend_num, profile.friends)[0]
|
|
invoke_in_main_thread(friend.set_status_message, status_message)
|
|
print 'User #{} has new status: {}'.format(friend_num, status_message)
|
|
invoke_in_main_thread(window.update_active_friend)
|
|
return wrapped
|
|
|
|
|
|
def friend_message(window):
|
|
"""
|
|
:param window: main window
|
|
:return: function for tox.callback_friend_message. Adds new message to list
|
|
"""
|
|
def wrapped(tox, friend_number, message_type, message, size, user_data):
|
|
print 'Message: ', message.decode('utf8')
|
|
if not window.isActiveWindow() and Settings()['notifications']:
|
|
tray_notification('Message', message.decode('utf8'))
|
|
profile = Profile.get_instance()
|
|
invoke_in_main_thread(profile.new_message, friend_number, message_type, message)
|
|
return wrapped
|
|
|
|
|
|
def init_callbacks(tox, window):
|
|
"""
|
|
Initialization of all callbacks.
|
|
:param tox: tox instance
|
|
:param window: main window
|
|
"""
|
|
tox.callback_friend_status(friend_status, 0)
|
|
tox.callback_friend_message(friend_message(window), 0)
|
|
tox.callback_self_connection_status(self_connection_status(window.connection_status, tox), 0)
|
|
tox.callback_friend_connection_status(friend_connection_status, 0)
|
|
tox.callback_friend_name(friend_name(window), 0)
|
|
tox.callback_friend_status_message(friend_status_message(window), 0)
|