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
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))