From 08e34bdd94f582e32ee8903025b43484d0e3cb1d Mon Sep 17 00:00:00 2001 From: khuxkm fbexl Date: Sun, 2 Dec 2018 05:57:43 +0000 Subject: [PATCH] Add roster command --- bot.py | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/bot.py b/bot.py index 68326fe..3f4821e 100644 --- a/bot.py +++ b/bot.py @@ -1,4 +1,4 @@ -import teambot,tasks,rss,time,sys +import teambot,tasks,rss,time,sys,subprocess class CosmicBot(teambot.Handler): def __init__(self,bot): @@ -9,6 +9,7 @@ class CosmicBot(teambot.Handler): self.tasks.load_state(0) self.commands = dict() self.register_command("botlist",self.on_botlist) + self.register_command("roster",self.on_roster) self.register_command("admin",self.on_admin,True) def register_command(self,name,action,is_admin=False): self.commands[name] = dict(action=action,is_admin=is_admin) @@ -46,6 +47,11 @@ class CosmicBot(teambot.Handler): sys.exit() elif subcmd=="check": self.tasks.states[0] = self.check_rss(self.tasks.states[0],self.tasks.base_state) + def on_roster(self,channel,nick,namecnt=""): + output = subprocess.check_output(["/usr/local/bin/roster",namecnt]).decode("ascii").split("\n") + for line in output: + line = re.sub("\s+"," ",line).split() + self.say(channel,"{}: {} (by {})".format(nick,line[1],line[0])) if __name__=="__main__": channels = "#cosmic".split()