Example #1
0
File: amqp.py Project: M-o-a-T/moat
	def __init__(self,conn,parent):
		self.parent = conn
		self.exchange=parent.exchange
		self.strip=parent.strip
		self.prefix = tuple(parent.prefix)
		self._direct = parent.shunt
		self.filter = parent.filter

		if self._direct:
			try: i = self.filter.index('*')
			except ValueError: i = 999
			try: i = min(self.filter.index('*'),i)
			except ValueError: pass
			if i < self.strip or self.strip and not self.filter:
				raise RuntimeError("You can't use 'shunt' if you strip elements you can't restore!")
		for k in self.filter:
			if hasattr(k,'startswith') and k.startswith('*'):
				self._simple = False
				break

		name = parent.dest
		if name is None:
			global _seq
			_seq += 1
			name = SName(conn.name+("f"+str(_seq),))
		self.channel = self.parent.conn.channel()
		self.channel.exchange_declare(exchange=self.exchange, type='topic', auto_delete=False, passive=False)
		super(EventCallback,self).__init__(name)
		register_worker(self, self._direct)
Example #2
0
 def exposed_monitor(self, callback, *args):
     try:
         w = EventCallback(self, callback, *args)
         self.workers.add(w)
         register_worker(w)
         return w
     except Exception as ex:
         fix_exception(ex)
         process_failure(ex)
Example #3
0
	def __setitem__(self,key,val):
		assert val.name==key, repr(val.name)+" != "+repr(key)
		onHandlers[val.id] = val
		try:
			onHandlers2[val.args].append(val)
		except KeyError:
			onHandlers2[val.args] = [val]
		super(_OnHandlers,self).__setitem__(key,val)
		register_worker(val)
Example #4
0
File: rpc.py Project: smurfix/MoaT
	def exposed_monitor(self,callback,*args):
		try:
			w = EventCallback(self,callback,*args)
			self.workers.add(w)
			register_worker(w)
			return w
		except Exception as ex:
			fix_exception(ex)
			process_failure(ex)
Example #5
0
 def __setitem__(self, key, val):
     assert val.name == key, repr(val.name) + " != " + repr(key)
     onHandlers[val.id] = val
     try:
         onHandlers2[val.args].append(val)
     except KeyError:
         onHandlers2[val.args] = [val]
     super(_OnHandlers, self).__setitem__(key, val)
     register_worker(val)
Example #6
0
	def __init__(self,name, host,port,vhost,app,codec, username,password):
		self.name = name
		self.host=host
		self.port=port
		self.vhost=vhost
		self.app=app
		self.codec=codec
		self.username=username
		self.password=password
		super().__init__()
		self.server = qbroker.Unit(self.app, amqp=dict(codec=self.codec, server={'host':self.host,'port':self.port,'virtualhost':self.vhost,'login':self.username,'password':self.password}), loop=qbroker.loop)
		self._rpc_connect()
		self.evt = EventCallback(self)
		register_worker(self.evt)
Example #7
0
def start_up():
	"""\
		Code to be called first. The main loop is NOT running.
		"""
	register_worker(Shutdown_Worker("shutdown handler"))

	global running
	if not running:
		running = True
		try:
			process_event(startup_event)
		except Exception as e:
			fix_exception(e)
			process_failure(e)
Example #8
0
def start_up():
    """\
		Code to be called first. The main loop is NOT running.
		"""
    register_worker(Shutdown_Worker("shutdown handler"))

    global running
    if not running:
        running = True
        try:
            process_event(startup_event)
        except Exception as e:
            fix_exception(e)
            process_failure(e)
Example #9
0
    def __init__(self, conn, parent):
        self.parent = conn
        self.exchange = parent.exchange
        self.strip = parent.strip
        self.prefix = tuple(parent.prefix)
        self._direct = parent.shunt
        self.filter = parent.filter

        if self._direct:
            try:
                i = self.filter.index('*')
            except ValueError:
                i = 999
            try:
                i = min(self.filter.index('*'), i)
            except ValueError:
                pass
            if i < self.strip or self.strip and not self.filter:
                raise RuntimeError(
                    "You can't use 'shunt' if you strip elements you can't restore!"
                )
        for k in self.filter:
            if hasattr(k, 'startswith') and k.startswith('*'):
                self._simple = False
                break

        name = parent.dest
        if name is None:
            global _seq
            _seq += 1
            name = SName(conn.name + ("f" + str(_seq), ))
        self.channel = self.parent.conn.channel()
        self.channel.exchange_declare(exchange=self.exchange,
                                      type='topic',
                                      auto_delete=False,
                                      passive=False,
                                      durable=True)
        super(EventCallback, self).__init__(name)
        register_worker(self, self._direct)
Example #10
0
 def __init__(self, name, host, port, vhost, app, codec, username,
              password):
     self.name = name
     self.host = host
     self.port = port
     self.vhost = vhost
     self.app = app
     self.codec = codec
     self.username = username
     self.password = password
     super().__init__()
     self.server = qbroker.Unit(self.app,
                                amqp=dict(codec=self.codec,
                                          server={
                                              'host': self.host,
                                              'port': self.port,
                                              'virtualhost': self.vhost,
                                              'login': self.username,
                                              'password': self.password
                                          }),
                                loop=qbroker.loop)
     self._rpc_connect()
     self.evt = EventCallback(self)
     register_worker(self.evt)
Example #11
0
            level = a[0]
            level = LogLevels.get(level, level)
            if lim > level:
                return
            a = (b, ) + a[1:]
            level = LogNames[level]

    logger.log(getattr(logging, level, logging.DEBUG), "%s",
               " ".join(str(x) for x in a))

    exc = []
    for l in list(Loggers.values()):
        if not l.ready:
            continue
        try:
            l.log(level, *a)
        except Exception as e:
            fix_exception(e)
            print("LOGGER CRASH 0", file=sys.stderr)
            print_exception(e, file=sys.stderr)
            l.delete()
            exc.append(sys.exc_info())
    if exc:
        for e in exc:
            log_exc(msg="Logging error", err=e)


log_event = LogWorker()
register_worker(log_event)
log_event = log_event.process
Example #12
0
		for w in sorted(collections.values(),key=lambda x:x.prio):
			if not w.can_do("del"):
				continue
			for d in w.values():
				try:
					d.delete(event.ctx)
				except Exception as ex:
					fix_exception(ex)
					print_exception(ex)
					# Logging may not work any more

	def report(self,*a,**k):
		return ()

register_worker(Shutdown_Collections("free all collections"))

class ShutdownHandler(Statement):
	"""A command handler to stop the whole thing."""
	name="shutdown"
	doc="stops executing the program."
	long_doc="""\
shutdown      stops executing the program.
shutdown now  ... but does not wait for active events to terminate.
"""
	def run(self,ctx,**k):
		event = self.params(ctx)
		if len(event):
			if tuple(event) == ("now",):
				_stop_mainloop()
				return
Example #13
0
			b = level
			level = a[0]
			level = LogLevels.get(level,level)
			if lim > level:
				return
			a = (b,)+a[1:]
			level = LogNames[level]
			
	logger.log(getattr(logging,level,logging.DEBUG),"%s"," ".join(str(x) for x in a))

	exc = []
	for l in list(Loggers.values()):
		if not l.ready:
			continue
		try:
			l.log(level, *a)
		except Exception as e:
			fix_exception(e)
			print("LOGGER CRASH 0", file=sys.stderr)
			print_exception(e,file=sys.stderr)
			l.delete()
			exc.append(sys.exc_info())
	if exc:
		for e in exc:
			log_exc(msg="Logging error", err=e)

log_event = LogWorker()
register_worker(log_event)
log_event = log_event.process

Example #14
0
        for w in sorted(collections.values(), key=lambda x: x.prio):
            if not w.can_do("del"):
                continue
            for d in w.values():
                try:
                    d.delete(event.ctx)
                except Exception as ex:
                    fix_exception(ex)
                    print_exception(ex)
                    # Logging may not work any more

    def report(self, *a, **k):
        return ()


register_worker(Shutdown_Collections("free all collections"))


class ShutdownHandler(Statement):
    """A command handler to stop the whole thing."""
    name = "shutdown"
    doc = "stops executing the program."
    long_doc = """\
shutdown      stops executing the program.
shutdown now  ... but does not wait for active events to terminate.
"""

    def run(self, ctx, **k):
        event = self.params(ctx)
        if len(event):
            if tuple(event) == ("now", ):
Example #15
0
class SayMoreWorker(SeqWorker):
	"""A WorkSequence-generating worker which logs something twice."""
	prio = 5
	def does_event(self,e):
		return e[0]=="say more"
	def process(self,event=None,**k):
		super(SayMoreWorker,self).process(event=event,**k)
		w = WorkSequence(event,self)
		w.append(SayWorker("TellOne"))
		w.append(SayWorker("TellTwo"))
		return w

from test import run_logger
run_logger("basic")

hello_ev = Event(Context(), "say","hello")
hello2_ev = Event(Context(), "say more","greeting")
	
register_worker(SayWorker("TellMe"))
register_worker(SayMoreWorker("say something"))

def main():
	try:
		process_event(hello_ev)
		process_event(hello2_ev)
	finally:
		shut_down()

mainloop(main)