toxygen/toxygen/groups/groups_service.py

import common.tox_save as tox_save
import utils.ui as util_ui


class GroupsService(tox_save.ToxSave):

    def __init__(self, tox, contacts_manager, contacts_provider):
        super().__init__(tox)
        self._contacts_manager = contacts_manager
        self._contacts_provider = contacts_provider

    # -----------------------------------------------------------------------------------------------------------------
    # Groups creation
    # -----------------------------------------------------------------------------------------------------------------

    def create_new_gc(self, name, privacy_state):
        group_number = self._tox.group_new(privacy_state, name.encode('utf-8'))
        self._add_new_group_by_number(group_number)

    def join_gc_by_id(self, chat_id, password):
        group_number = self._tox.group_join(chat_id, password)
        self._add_new_group_by_number(group_number)

    def join_gc_via_invite(self, invite_data, friend_number, password):
        group_number = self._tox.group_invite_accept(invite_data, friend_number, password)
        self._add_new_group_by_number(group_number)

    # -----------------------------------------------------------------------------------------------------------------
    # Groups reconnect and leaving
    # -----------------------------------------------------------------------------------------------------------------

    def leave_group(self, group_number):
        group = self._get_group(group_number)
        self._tox.group_leave(group_number)
        self._contacts_manager.delete_group(group_number)
        self._contacts_provider.remove_contact_from_cache(group.tox_id)

    # -----------------------------------------------------------------------------------------------------------------
    # Group invites
    # -----------------------------------------------------------------------------------------------------------------

    def invite_friend(self, friend_number, group_number):
        self._tox.group_invite_friend(group_number, friend_number)

    def process_group_invite(self, friend_number, invite_data):
        friend = self._get_friend(friend_number)
        text = util_ui.tr('Friend {} invites you to group. Accept?')
        if util_ui.question(text.format(friend.name), util_ui.tr('Group invite')):
            self.join_gc_via_invite(invite_data, friend_number, None)

    # -----------------------------------------------------------------------------------------------------------------
    # Group info methods
    # -----------------------------------------------------------------------------------------------------------------

    def update_group_info(self, group):
        group.name = self._tox.group_get_name(group.number).encode('utf-8')
        group.status_message = self._tox.group_get_topic(group.number).encode('utf-8')

    # -----------------------------------------------------------------------------------------------------------------
    # Private methods
    # -----------------------------------------------------------------------------------------------------------------

    def _add_new_group_by_number(self, group_number):
        self._contacts_manager.add_group(group_number)

    def _get_group(self, group_number):
        return self._contacts_provider.get_group_by_number(group_number)

    def _get_friend(self, friend_number):
        return self._contacts_provider.get_friend_by_number(friend_number)