jsb.lib.container

container for bot to bot communication.

class jsb.lib.container.Container(origin=None, payload=None, type='event', key=None, how='direct')

Bases: jsb.lib.gozerevent.GozerEvent

Container for bot to bot communication. Provides a hmac id that can be checked.

makehmac(key)

makeid()

save(attributes=[])

jsb.lib.container.getid(container)

CODE

# jsb/container.py
#
#

""" container for bot to bot communication. """

__version__ = "1"

jsb imports

from jsb.lib.persist import Persist
from jsb.utils.name import stripname
from jsb.lib.gozerevent import GozerEvent
from jsb.imports import getjson

xmpp import

from jsb.contrib.xmlstream import NodeBuilder, XMLescape, XMLunescape

basic imports

import hmac
import uuid
import time
import hashlib

defines

idattributes = ['createtime', 'origin', 'type', 'idtime', 'payload']

functions

def getid(container):
    name = ""
    for attr in idattributes:
        try: name += str(container[attr])
        except KeyError: pass
    return uuid.uuid3(uuid.NAMESPACE_URL, name).hex

classes

class Container(GozerEvent):

    """ Container for bot to bot communication. Provides a hmac id that can be checked. """

    def __init__(self, origin=None, payload=None, type="event", key=None, how="direct"):
        GozerEvent.__init__(self)
        self.createtime = time.time()
        self.origin = origin
        self.type = str(type)
        self.payload = payload
        self.makeid()
        if key: self.makehmac(key)
        else: self.makehmac(self.id)

    def makeid(self):
        self.idtime = time.time()
        self.id = getid(self)

    def makehmac(self, key):
        self.hash = "sha512"
        self.hashkey = key
        self.digest = hmac.new(key, self.payload, hashlib.sha512).hexdigest()

    def save(self, attributes=[]):
        target = {}
        if attributes:
            for key in attributes: target[key] = self[key]
        else: target = cpy(self)
        targetfile = getdatadir() + os.sep + "containers" + os.sep + str(self.createtime) + "_" + stripname(self.origin)
        p = Persist(targetfile)
        p.data = getjson().dumps(target)
        p.save()