def dccloop(self, event, s): sockfile = s.makefile('rw') #s.setblocking(True) self.register_fd(s) while 1: try: res = sockfile.readline() if not res: break res = res.rstrip() logging.info("< %s %s" % (event.origin, res)) e = Event() e.btype = "DCC" e.txt = res e.outer = sockfile e.origin = event.origin super().dispatch(e) except socket.timeout: time.sleep(0.01) except socket.error as ex: if ex.errno in [ socket.EAGAIN, ]: continue else: raise except Exception as ex: logging.error(get_exception()) sockfile.close()
def timer(event): """ echo txt on a given time. """ if not event._parsed.rest: return seconds = 0 line = "" for word in event._parsed.args: if word.startswith("+"): try: seconds = int(word[1:]) except: event.reply("%s is not an integer" % seconds) return else: line += word + " " if seconds: target = time.time() + seconds else: try: target = get_day(event._parsed.rest) except ENODATE: target = to_day(day()) hour = get_hour(event._parsed.rest) if hour: target += hour if not target or time.time() > target: event.reply("already passed given time.") return e = Event() e.services = "clock" e.prefix = "timer" e.txt = event._parsed.rest e.time = target e.done = False e.save() timer = Timer(target, e.reply, e.txt) launcher.launch(timer.start) event.ok(time.ctime(target))
def re(event): try: t = parse_time(event._parsed.rest) except ValueError: t = None if not t: event.reply("since when ?") ; return for obj in storage.since("cblog", t): e = Event(obj) event.reply("%s %s %s" % (e.created.split()[3], e.nick or e.origin, e.txt))
def event(self): event = Event() event.btype = sname(self) event.txt = str(os.read(fd, 512), "utf-8") event.txt = event.txt.rstrip() event.origin = "root@shell" yield event
def cmnd(self, txt): event = Event() event._server = self._cfg.server event.btype = sname(self) event.origin = "user@bot" event.txt = txt self.dispatch(event) return event
def init(event): for fn in storage.prefixed("timer"): e = Event().load(fn) if e.done: continue if "time" not in e: continue if time.time() < int(e.time): timer = Timer(int(e.time), e.announce, e.txt) launcher.launch(timer.start)
def event(self): e = Event() e.server = self._cfg.server e.btype = sname(self) e.txt = input() e.txt = e.txt.rstrip() e.origin = "root@shell" return e
def testcmnds(event): cfg.workdir = "test.data" for cmnd in sorted(kernel._handlers.keys()): e = Event() e.btype = event.btype e.txt = "%s %s" % (cmnd, randomname()) e.origin = "test@bot" e.parse() event.reply("put %s" % e.txt) kernel.put(e)
def scan(self, path, *args, **kwargs): """ scan all files in the cfg.main.workdir directory. """ p = Event(**kwargs) if not path.endswith(os.sep): path += os.sep for root, dirs, files in os.walk(path, topdown=True): if not os.path.isdir(root): continue for fn in files: fnn = j(root, fn) timed = fn_time(fnn) if timed and p.start and timed < p.start: continue if timed and p.end and timed > p.end: continue yield fnn
def since(self, start, *args, **kwargs): """ return all object since a given time. """ e = Event(**kwargs) e.start = start for fn in self.prefixed(*args, **e): try: obj = Object().load(fn) except: logging.warn("fail %s" % fn) continue if "deleted" in obj and obj.deleted: continue yield obj
def last(self, *args, **kwargs): """ return last record with a matching prefix. """ e = Event(**kwargs) for fn in self.prefixed(*args, **kwargs)[::-1]: try: obj = Object().load(fn) except: logging.warn("fail %s" % fn) continue if "deleted" in obj and obj.deleted: continue if len(args) > 1 and obj.get(args[0]) != args[1]: continue return obj
def reload(self, name, force=False, event=None): from meds.event import Event e = event or Event() if name not in self._table: self.load(name, True) n = name.split(".")[-1] if name in self._table and "shutdown" in dir(self._table[name]): self._table[name].shutdown(e) e.ready() if name not in self._table or force: self.load(name, True) else: self._table[name] = importlib.reload(self._table[name]) if "init" in dir(self._table[name]): self._table[name].init(e) e.ready() return self._table[name]
def cmndrun(event): cfg.workdir = "test.data" for name in sorted(kernel.modules()): if name in ["meds.plugs.test", "meds.plugs.rss"]: continue mod = kernel.direct(name) for n in dir(mod): if n in exclude: continue func = getattr(mod, n, None) if func and type(func) in [types.FunctionType, types.MethodType]: if "event" in func.__code__.co_varnames: e = Event() e.btype = event.btype e._funcs.append(func) e.workdir = "test.data" e.origin = "test@bot" kernel.put(e)
def reboot(event): """ reboot the bot, allowing statefull reboot (keeping connections alive). """ if not cfg.reboot: event.reply("reboot is not enabled") return if not users.allowed(event.origin, "REBOOT"): event.reply("you don't have reboot permissions.") return event.announce("rebooting") resume = Object() resume.kernel = kernel._resume resume.fleet = fleet resume.save("resume") e = Event() e._funcs.append(real_reboot) print(e) kernel.put(e)
def initialize(self, name): """ initialze package. """ event = Event() thrs = [] for modname in self.walk(name): n = modname.split(".")[-1] mod = self._table.get(modname, None) if not mod: mod = self.load(modname) if n in self._cfg.default: mod.init(event) if self._cfg.all or n in self._cfg.init.split(","): if n not in self._cfg.exclude.split(','): logging.info("! init %s" % n) thr = self.launch(mod.init, event, name="%s.initialize" % n) thrs.append(thr) event.ready() return thrs
def shutdown(self, close=True): """ stop bots, services and plugins. """ logging.debug("shutdown") event = Event() thrs = [] for key, mod in self._table.items(): if "meds.run" not in key: continue if "shutdown" in dir(mod): thr = self.launch(mod.shutdown, event) thrs.append(thr) for bot in meds.core.fleet: if "stop" in dir(bot): thr = self.launch(bot.stop) thrs.append(thr) if "exit" in dir(bot): thr = self.launch(bot.exit) thrs.append(thr) bot.ready() meds.core.launcher.waiter(thrs) self.killall() self.ready()
def stats(event, **kwargs): e = Event(kwargs) cfg = cfgs.get("stats", None) if not cfg: cfgs.stats = cfg = storage.last("cfg", "stats") name = event._parsed.rest or e.name if "." in name: name = name.split(".")[-1] name = name.lower() delta = time.time() - starttime awake = time.time() - today() needed = seconds(nr(name)) if needed: nrtimes = int(delta / needed) txt = "%s #%s" % (name.upper(), nrtimes) if name in omschrijving: txt += " (%s)" % omschrijving.get(name) txt += " elke %s" % elapsed(seconds(nr(name))) if name in soort: txt += " door een %s" % soort.get(name, "slachtoffer") txt += " bijv. in %s" % random.choice(gemeenten) if name in tags: txt += " %s" % tags.get(name) if cfg and cfg.showurl and name in urls: txt += " - %s" % urls.get(name) event.announce(txt)
def parsing(self, txt): rawstr = str(txt) obj = Event() obj.server = self._cfg.server obj.cc = self.cc obj.btype = sname(self) obj.arguments = [] arguments = rawstr.split() obj.origin = arguments[0] if obj.origin.startswith(":"): obj.origin = obj.origin[1:] if len(arguments) > 1: obj.command = arguments[1] if len(arguments) > 2: txtlist = [] adding = False for arg in arguments[2:]: if arg.startswith(":"): adding = True txtlist.append(arg[1:]) continue if adding: txtlist.append(arg) else: obj.arguments.append(arg) obj.txt = " ".join(txtlist) else: obj.command = obj.origin obj.origin = self._cfg.server try: obj.nick, obj.userhost = obj.origin.split("!") except: pass if obj.arguments: obj.target = obj.arguments[-1] else: obj.target = "" if obj.target.startswith("#"): obj.channel = obj.target if not obj.txt and len(arguments) == 1: obj.txt = arguments[1] if not obj.txt: obj.txt = rawstr.split(":")[-1] return obj
def testcfg(event): cfg.workdir = event._parsed.rest e = Event() path = e.save() event.reply(path)