toxygen/toxygen/contacts/friend_factory.py

# -*- mode: python; indent-tabs-mode: nil; py-indent-offset: 4; coding: utf-8 -*-

from contacts.friend import Friend
from common.tox_save import ToxSave

class FriendFactory(ToxSave):

    def __init__(self, profile_manager, settings, tox, db, items_factory):
        super().__init__(tox)
        self._profile_manager = profile_manager
        self._settings = settings
        self._db = db
        self._items_factory = items_factory

    def create_friend_by_public_key(self, public_key):
        friend_number = self._tox.friend_by_public_key(public_key)
        return self.create_friend_by_number(friend_number)

    def create_friend_by_number(self, friend_number:int):
        aliases = self._settings['friends_aliases']
        sToxPk = self._tox.friend_get_public_key(friend_number)
        assert sToxPk, sToxPk
        try:
            alias = list(filter(lambda x: x[0] == sToxPk, aliases))[0][1]
        except:
            alias = ''
        item = self._create_friend_item()
        name = alias or self._tox.friend_get_name(friend_number) or sToxPk
        status_message = self._tox.friend_get_status_message(friend_number)
        message_getter = self._db.messages_getter(sToxPk)
        friend = Friend(self._profile_manager, message_getter, friend_number, name, status_message, item, sToxPk)
        friend.set_alias(alias)

        return friend

    # Private methods

    def _create_friend_item(self):
        """
        Method-factory
        :return: new widget for friend instance
        """
        return self._items_factory.create_contact_item()