Exemple #1
0
 def __init__(self, **kwargs):
     super(GenericClient, self).__init__()
     self.port = kwargs.get('port', 4202)
     self.host = kwargs.get('host', '')
     self.stdout = ColoredStreamHandler(self.stdout).stream
Exemple #2
0
import logging
from uliweb.utils.coloredlog import ColoredStreamHandler
import cPickle
from cmd import Cmd

log = logging.getLogger('redbreast.daemon')
log.addHandler(ColoredStreamHandler())
log.setLevel(logging.DEBUG)

__daemon_client_version__ = 'v0.0.1'


class DaemonMsg(object):
    def serialize(self):
        return cPickle.dumps(self, cPickle.HIGHEST_PROTOCOL)

    @classmethod
    def load(cls, s):
        return cPickle.loads(s)

    @classmethod
    def load_from_socket(cls, sock):
        msgstr = []
        while True:
            s = sock.recv(1024)

            if not s:
                break
            msgstr.append(s)
            try:
                msg = DaemonMsg.load("".join(msgstr))