gozerbot.utils.xmpp

xmpp related helper funcions.

gozerbot.utils.xmpp.make_message(xml)

gozerbot.utils.xmpp.make_presence(xml)

CODE

# gozerbot/utils/xmpp.py
#
#

""" xmpp related helper funcions. """

__status__ = "seen"

basic imports

from xml.etree import cElementTree as ET
import types

make_presence function

def make_presence(xml):
    pres = {'subject': '', 'to': '', 'message': '', 'name': '', 'resource': '', 'jid': ''}
    if type(xml) == types.DictType: pres.update(xml) ; return pres
    et = ET.XMLID(xml)
    pres.update(et[0].items())
    xml = ET.XML(xml)
    try:
        pres['name'] = pres['from'].split('@')[0]
        pres['resource'] = pres['from'].split('/')[1]
        pres['jid'] = pres['from'].split('/')[0]
    except KeyError: pass
    return pres

make_message function

def make_message(xml):
    msg = {'subject': '', 'to': '', 'txt': '', 'message': '', 'from': '', 'jid': '', 'name': '', 'resource': ''}
    if type(xml) == types.DictType: msg.update(xml) ; return msg
    et = ET.XMLID(xml)
    msg.update(et[0].items())
    xml = ET.XML(xml)
    msg['message'] = unicode(xml.findtext('body'))
    msg['name'] = msg['from'].split('@')[0]
    msg['resource'] = msg['from'].split('/')[1]
    msg['jid'] = msg['from'].split('/')[0]
    return msg