Slixfeed/slixfeed/xmpp/utility.py

#!/usr/bin/env python3
# -*- coding: utf-8 -*-

from slixmpp.exceptions import IqError, IqTimeout
import logging

# class XmppChat
# class XmppUtility:


# TODO Rename to get_jid_type
async def get_chat_type(self, jid):
    """
    Check chat (i.e. JID) type.

    If iqresult["disco_info"]["features"] contains XML namespace
    of 'http://jabber.org/protocol/muc', then it is a 'groupchat'.

    Unless it has forward slash, which would indicate that it is
    a chat which is conducted through a groupchat.
    
    Otherwise, determine type 'chat'.

    Parameters
    ----------
    jid : str
        Jabber ID.

    Returns
    -------
    result : str
        'chat' or 'groupchat' or 'error'.
    """
    try:
        iqresult = await self["xep_0030"].get_info(jid=jid)
        features = iqresult["disco_info"]["features"]
        # identity = iqresult['disco_info']['identities']
        # if 'account' in indentity:
        # if 'conference' in indentity:
        if ('http://jabber.org/protocol/muc' in features) and not ('/' in jid):
            result = "groupchat"
        # TODO elif <feature var='jabber:iq:gateway'/>
        # NOTE Is it needed? We do not interact with gateways or services
        else:
            result = "chat"
        logging.info('Jabber ID: {}\n'
                     'Chat Type: {}'.format(jid, result))
    except (IqError, IqTimeout) as e:
        logging.warning('Chat type could not be determined for {}'.format(jid))
        logging.error(e)
        result = 'error'
    # except BaseException as e:
    #     logging.error('BaseException', str(e))
    # finally:
    #     logging.info('Chat type is:', chat_type)
    return result