243 lines
10 KiB
Python
243 lines
10 KiB
Python
from contacts.friend import *
|
|
from user_data.settings import *
|
|
from wrapper.toxcore_enums_and_consts import *
|
|
from util.util import log
|
|
from history.database import *
|
|
from file_transfers.file_transfers import *
|
|
import time
|
|
from contacts import basecontact
|
|
from contacts.group_chat import *
|
|
import util.ui as util_ui
|
|
|
|
|
|
class Profile(basecontact.BaseContact):
|
|
"""
|
|
Profile of current toxygen user. Contains friends list, tox instance
|
|
"""
|
|
def __init__(self, profile_manager, tox, screen, file_transfer_handler):
|
|
"""
|
|
:param tox: tox instance
|
|
:param screen: ref to main screen
|
|
"""
|
|
basecontact.BaseContact.__init__(self,
|
|
profile_manager,
|
|
tox.self_get_name(),
|
|
tox.self_get_status_message(),
|
|
screen.user_info,
|
|
tox.self_get_address())
|
|
self._file_transfer_handler = file_transfer_handler
|
|
self._screen = screen
|
|
self._messages = screen.messages
|
|
self._tox = tox
|
|
self._file_transfers = {} # dict of file transfers. key - tuple (friend_number, file_number)
|
|
self._load_history = True
|
|
self._waiting_for_reconnection = False
|
|
self._contacts_manager = None
|
|
|
|
# -----------------------------------------------------------------------------------------------------------------
|
|
# Edit current user's data
|
|
# -----------------------------------------------------------------------------------------------------------------
|
|
|
|
def change_status(self):
|
|
"""
|
|
Changes status of user (online, away, busy)
|
|
"""
|
|
if self._status is not None:
|
|
self.set_status((self._status + 1) % 3)
|
|
|
|
def set_status(self, status):
|
|
super().set_status(status)
|
|
if status is not None:
|
|
self._tox.self_set_status(status)
|
|
elif not self._waiting_for_reconnection:
|
|
self._waiting_for_reconnection = True
|
|
QtCore.QTimer.singleShot(50000, self.reconnect)
|
|
|
|
def set_name(self, value):
|
|
if self.name == value:
|
|
return
|
|
tmp = self.name
|
|
super().set_name(value.encode('utf-8'))
|
|
self._tox.self_set_name(self._name.encode('utf-8'))
|
|
message = util_ui.tr('User {} is now known as {}')
|
|
message = message.format(tmp, value)
|
|
for friend in self._contacts:
|
|
friend.append_message(InfoMessage(message, time.time()))
|
|
if self._active_friend + 1:
|
|
self.create_message_item(message, time.time(), '', MESSAGE_TYPE['INFO_MESSAGE'])
|
|
self._messages.scrollToBottom()
|
|
|
|
def set_status_message(self, value):
|
|
super().set_status_message(value)
|
|
self._tox.self_set_status_message(self._status_message.encode('utf-8'))
|
|
|
|
def new_nospam(self):
|
|
"""Sets new nospam part of tox id"""
|
|
import random
|
|
self._tox.self_set_nospam(random.randint(0, 4294967295)) # no spam - uint32
|
|
self._tox_id = self._tox.self_get_address()
|
|
|
|
return self._tox_id
|
|
|
|
# -----------------------------------------------------------------------------------------------------------------
|
|
# Friend connection status callbacks
|
|
# -----------------------------------------------------------------------------------------------------------------
|
|
|
|
def send_files(self, friend_number):
|
|
friend = self.get_friend_by_number(friend_number)
|
|
friend.remove_invalid_unsent_files()
|
|
files = friend.get_unsent_files()
|
|
try:
|
|
for fl in files:
|
|
data = fl.get_data()
|
|
if data[1] is not None:
|
|
self.send_inline(data[1], data[0], friend_number, True)
|
|
else:
|
|
self.send_file(data[0], friend_number, True)
|
|
friend.clear_unsent_files()
|
|
for key in list(self._paused_file_transfers.keys()):
|
|
data = self._paused_file_transfers[key]
|
|
if not os.path.exists(data[0]):
|
|
del self._paused_file_transfers[key]
|
|
elif data[1] == friend_number and not data[2]:
|
|
self.send_file(data[0], friend_number, True, key)
|
|
del self._paused_file_transfers[key]
|
|
if friend_number == self.get_active_number() and self.is_active_a_friend():
|
|
self.update()
|
|
except Exception as ex:
|
|
print('Exception in file sending: ' + str(ex))
|
|
|
|
def friend_exit(self, friend_number):
|
|
"""
|
|
Friend with specified number quit
|
|
"""
|
|
self.get_friend_by_number(friend_number).status = None
|
|
self.friend_typing(friend_number, False)
|
|
if friend_number in self._call:
|
|
self._call.finish_call(friend_number, True)
|
|
for friend_num, file_num in list(self._file_transfers.keys()):
|
|
if friend_num == friend_number:
|
|
ft = self._file_transfers[(friend_num, file_num)]
|
|
if type(ft) is SendTransfer:
|
|
self._paused_file_transfers[ft.get_id()] = [ft.get_path(), friend_num, False, -1]
|
|
elif type(ft) is ReceiveTransfer and ft.state != TOX_FILE_TRANSFER_STATE['INCOMING_NOT_STARTED']:
|
|
self._paused_file_transfers[ft.get_id()] = [ft.get_path(), friend_num, True, ft.total_size()]
|
|
self.cancel_transfer(friend_num, file_num, True)
|
|
|
|
# -----------------------------------------------------------------------------------------------------------------
|
|
# Private messages
|
|
# -----------------------------------------------------------------------------------------------------------------
|
|
|
|
def receipt(self):
|
|
i = 0
|
|
while i < self._messages.count() and not self._messages.itemWidget(self._messages.item(i)).mark_as_sent():
|
|
i += 1
|
|
|
|
def send_messages(self, friend_number):
|
|
"""
|
|
Send 'offline' messages to friend
|
|
"""
|
|
friend = self.get_friend_by_number(friend_number)
|
|
friend.load_corr()
|
|
messages = friend.get_unsent_messages()
|
|
try:
|
|
for message in messages:
|
|
self.split_and_send(friend_number, message.get_data()[-1], message.get_data()[0].encode('utf-8'))
|
|
friend.inc_receipts()
|
|
except Exception as ex:
|
|
log('Sending pending messages failed with ' + str(ex))
|
|
|
|
def delete_message(self, message_id):
|
|
friend = self.get_curr_friend()
|
|
friend.delete_message(time)
|
|
self._history.delete_message(friend.tox_id, message_id)
|
|
self.update()
|
|
|
|
# -----------------------------------------------------------------------------------------------------------------
|
|
# Friend, message and file transfer items creation
|
|
# -----------------------------------------------------------------------------------------------------------------
|
|
|
|
def create_message_item(self, text, time, owner, message_type, append=True):
|
|
if message_type == MESSAGE_TYPE['INFO_MESSAGE']:
|
|
name = ''
|
|
elif owner == MESSAGE_OWNER['FRIEND']:
|
|
name = self.get_active_name()
|
|
else:
|
|
name = self._name
|
|
pixmap = None
|
|
if self._show_avatars:
|
|
if owner == MESSAGE_OWNER['FRIEND']:
|
|
pixmap = self.get_curr_friend().get_pixmap()
|
|
else:
|
|
pixmap = self.get_pixmap()
|
|
return self._factory.message_item(text, time, name, owner != MESSAGE_OWNER['NOT_SENT'],
|
|
message_type, append, pixmap)
|
|
|
|
def create_gc_message_item(self, text, time, owner, name, message_type, append=True):
|
|
pixmap = None
|
|
if self._show_avatars:
|
|
if owner == MESSAGE_OWNER['FRIEND']:
|
|
pixmap = self.get_curr_friend().get_pixmap()
|
|
else:
|
|
pixmap = self.get_pixmap()
|
|
return self._factory.message_item(text, time, name, True,
|
|
message_type - 5, append, pixmap)
|
|
|
|
def create_file_transfer_item(self, tm, append=True):
|
|
data = list(tm.get_data())
|
|
data[3] = self.get_friend_by_number(data[4]).name if data[3] else self._name
|
|
return self._factory.file_transfer_item(data, append)
|
|
|
|
def create_unsent_file_item(self, message, append=True):
|
|
data = message.get_data()
|
|
return self._factory.unsent_file_item(os.path.basename(data[0]),
|
|
os.path.getsize(data[0]) if data[1] is None else len(data[1]),
|
|
self.name,
|
|
data[2],
|
|
append)
|
|
|
|
def create_inline_item(self, data, append=True):
|
|
return self._factory.inline_item(data, append)
|
|
|
|
# -----------------------------------------------------------------------------------------------------------------
|
|
# Reset
|
|
# -----------------------------------------------------------------------------------------------------------------
|
|
|
|
def reset(self, restart):
|
|
"""
|
|
Recreate tox instance
|
|
:param restart: method which calls restart and returns new tox instance
|
|
"""
|
|
for friend in self._contacts:
|
|
self.friend_exit(friend.number)
|
|
del self._tox
|
|
self._tox = restart()
|
|
self.status = None
|
|
self._contacts_manager.update_filtration()
|
|
|
|
def reconnect(self):
|
|
self._waiting_for_reconnection = False
|
|
if self.status is None or all(list(map(lambda x: x.status is None, self._contacts))) and len(self._contacts):
|
|
self._waiting_for_reconnection = True
|
|
self.reset(self._screen.reset)
|
|
QtCore.QTimer.singleShot(50000, self.reconnect)
|
|
|
|
def close(self):
|
|
for friend in filter(lambda x: type(x) is Friend, self._contacts):
|
|
self.friend_exit(friend.number)
|
|
for i in range(len(self._contacts)):
|
|
del self._contacts[0]
|
|
if hasattr(self, '_call'):
|
|
self._call.stop()
|
|
del self._call
|
|
|
|
def reset_avatar(self):
|
|
super().reset_avatar()
|
|
for friend in filter(lambda x: x.status is not None, self._contacts):
|
|
self.send_avatar(friend.number)
|
|
|
|
def set_avatar(self, data):
|
|
super().set_avatar(data)
|
|
for friend in filter(lambda x: x.status is not None, self._contacts):
|
|
self.send_avatar(friend.number)
|