61 lines
1.9 KiB
Python
61 lines
1.9 KiB
Python
|
#!/usr/bin/env python3
|
||
|
|
# -*- coding: utf-8 -*-
|
||
|
|
|
||
|
|
from slixfeed.log import Logger
|
||
|
|
from slixmpp.exceptions import IqError, IqTimeout
|
||
|
|
|
||
|
|
logger = Logger(__name__)
|
||
|
|
|
||
|
|
# class XmppChat
|
||
|
|
# class XmppUtility:
|
||
|
|
|
||
|
|
|
||
|
|
class XmppUtilities:
|
||
|
|
|
||
|
|
|
||
|
|
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"
|
||
|
|
logger.info('Jabber ID: {}\n'
|
||
|
|
'Chat Type: {}'.format(jid, result))
|
||
|
|
except (IqError, IqTimeout) as e:
|
||
|
|
logger.warning('Chat type could not be determined for {}'.format(jid))
|
||
|
|
logger.error(e)
|
||
|
|
result = 'error'
|
||
|
|
# except BaseException as e:
|
||
|
|
# logger.error('BaseException', str(e))
|
||
|
|
# finally:
|
||
|
|
# logger.info('Chat type is:', chat_type)
|
||
|
|
return result
|