From 4b98486fc70a685dd4d8cae6a8197904365c9c8c Mon Sep 17 00:00:00 2001 From: Christian Franke Date: Sun, 17 Mar 2013 12:43:30 +0100 Subject: Update subbot.py to allow light control --- subbot.py | 147 +++++++++++++++++++++++++++++++++++++++++++++++++++++++------- 1 file changed, 131 insertions(+), 16 deletions(-) diff --git a/subbot.py b/subbot.py index 7462bce..c860c17 100644 --- a/subbot.py +++ b/subbot.py @@ -3,54 +3,169 @@ import irc.bot import json, time import traceback +import liblo +import urllib2 import settings svcpw = settings.bot_svcpw +# Known lights, terminal symbols +lights_terminal = {} + +for i in range(1,5): + ident = '%02d' % i + lights_terminal[ident] = 'osc.udp://172.22.83.5:4243/dali/lamps/%s/bright' % ident + +# Production Rules/Aliases +lights_production = { + u'lounge-stage': [u'01'], + u'lounge-office': [u'02'], + u'lounge-bar': [u'03'], + u'lounge-clock': [u'04'], + u'lounge-west': [u'lounge-bar', u'lounge-clock'], + u'lounge-east': [u'lounge-stage', u'lounge-office'], + u'lounge': [u'lounge-east', u'lounge-west'], +} + +class OSCMessage(object): + def __init__(self, url): + parsed = urllib2.urlparse.urlparse(url) + self.path = parsed.path + self.url = urllib2.urlparse.urlunparse((parsed.scheme, parsed.netloc, '', '', '', '')) + + self.msg = liblo.Message(self.path) + + def __getattr__(self, name): + return getattr(self.msg, name) + + def send(self): + liblo.send(self.url, self.msg) + class TestBot(irc.bot.SingleServerIRCBot): def __init__(self, channel, nickname, server, port=6667): - irc.bot.SingleServerIRCBot.__init__(self, [(server, port)], nickname, nickname) + self.nickname_base = nickname + self.status = 'closed' + self.update_nick(False) + + irc.bot.SingleServerIRCBot.__init__(self, [(server, port)], self.nickname, self.nickname) self.channel = channel def on_nicknameinuse(self, c, e): - c.nick(c.get_nickname() + "_") + self.nickname_base += '_' + self.update_nick() def on_welcome(self, c, e): - c.privmsg('NickServ', 'identify sublab|open %s' % (svcpw, )) + c.privmsg('NickServ', 'identify %s %s' % (self.nickname, svcpw)) self.refresh() c.join(self.channel) - def refresh(self): + def on_pubmsg(self, c, e): + message = e.arguments()[0] + + light_command_prefix = '!light' + if message.startswith(light_command_prefix): + nick = e.source().nick + self.on_light_command(nick, message[len(light_command_prefix) + 1:]) + + def on_light_command(self, nick, commandline): + tokens = commandline.split(' ') + if len(tokens) < 1: + self.light_usage() + return + + if tokens[0].lower() == 'set': + self.on_light_set(tokens[1:]) + elif tokens[0].lower() == 'list': + self.on_light_list(tokens[1:]) + else: + self.light_usage() + + def on_light_list(self, tokens): + self.connection.privmsg(self.channel, + "The following ids are currently known: %s" % ', '.join(sorted(lights_production.keys()))) + + def on_light_set(self, tokens): + if len(tokens) != 2: + return False + + lights = tokens[0].split(',') + for i in range(32): + for i in range(len(lights)): + light = lights[i] + if light in lights_terminal: + continue + if light not in lights_production: + self.connection.privmsg(self.channel, "Unknown Alias: '%s'" % light) + return + lights = lights[:i] + lights_production[light] + lights[i+1:] + break + + status = tokens[1] + if status.lower() == 'on': + status = 255 + elif status.lower() == 'off': + status = 0 + else: + try: + status = int(status, 0) + if status not in range(256): + raise ValueError + except ValueError: + self.connection.privmsg(self.channel, "Invalid light value: '%s'" % status) + return + + for light in lights: + msg = OSCMessage(lights_terminal[light]) + msg.add(status) + msg.send() + + def light_usage(self): + self.connection.privmsg(self.channel, "Usage: !light list") + self.connection.privmsg(self.channel, "Usage: !light set [,...] ") + + def update_nick(self, server_interaction=True): + """Set new nickname from nickname base and current status""" + wantnick = self.nickname_base + '|' + self.status + + if not server_interaction: + self.nickname = wantnick + return + + curtime = time.time() c = self.connection havenick = c.get_nickname() - curtime = time.time() + if wantnick != havenick: + print int(curtime), 'nick:', havenick, '->', wantnick + c.privmsg('NickServ', 'ghost %s %s' % (wantnick, svcpw)) + c.privmsg('NickServ', 'release %s %s' % (wantnick, svcpw)) + c.nick(wantnick) + self.nickname = wantnick + + def refresh(self): + curtime = time.time() try: sdata = json.load(file('/home/services/http/subcan.json', 'r')) door = sdata[u'door.lock'] if door[u'ts'] < curtime - 120: - wantnick = 'sublab|error' + self.status = 'error' elif door[u'text'] == u'closed': - wantnick = 'sublab|closed' + self.status = 'closed' elif door[u'text'] == u'open': - wantnick = 'sublab|open' + self.status = 'open' else: - wantnick = 'sublab|error' + self.status = 'error' except Exception, e: traceback.print_exc() - wantnick = 'sublab|error' + self.status = 'error' - if wantnick != havenick: - print int(curtime), 'nick:', havenick, '->', wantnick - c.privmsg('NickServ', 'ghost %s %s' % (wantnick, svcpw)) - c.privmsg('NickServ', 'release %s %s' % (wantnick, svcpw)) - c.nick(wantnick) + self.update_nick() self.ircobj.execute_delayed(5, self.refresh) def main(): - bot = TestBot('#sublab', 'sublab|closed', '172.22.24.1', 6667) + bot = TestBot('#sublab', 'sublab', '172.22.24.1', 6667) bot.start() if __name__ == "__main__": -- cgit v1.2.1