bss/dr.botzo
1
0
Fork 0
Code Issues 3 Pull Requests Releases 2 Wiki Activity

facts: allow adding facts via IRC

Browse Source
This commit is contained in:
Brian S. Stephan 2015-06-20 10:58:40 -05:00
parent 934f83b734
commit 241de26dd5
1 changed files with 18 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
dr_botzo/facts/ircplugin.py
View File

@ -4,8 +4,8 @@ import logging
from irc.client import NickMask from irc.client import NickMask
from ircbot.lib import Plugin from ircbot.lib import Plugin, has_permission
from facts.models import Fact from facts.models import Fact, FactCategory
log = logging.getLogger('facts.ircplugin') log = logging.getLogger('facts.ircplugin')
@ -18,6 +18,8 @@ class Facts(Plugin):
def start(self): def start(self):
"""Set up the handlers.""" """Set up the handlers."""
self.connection.reactor.add_global_regex_handler(['pubmsg', 'privmsg'], r'^!facts\s+add\s+(\S+)\s+(.*)$',
self.handle_add_fact, -20)
self.connection.reactor.add_global_regex_handler(['pubmsg', 'privmsg'], r'^!facts\s+(\S+)(\s+(.*)$|$)', self.connection.reactor.add_global_regex_handler(['pubmsg', 'privmsg'], r'^!facts\s+(\S+)(\s+(.*)$|$)',
self.handle_facts, -20) self.handle_facts, -20)
@ -26,6 +28,7 @@ class Facts(Plugin):
def stop(self): def stop(self):
"""Tear down handlers.""" """Tear down handlers."""
self.connection.reactor.remove_global_regex_handler(['pubmsg', 'privmsg'], self.handle_add_fact)
self.connection.reactor.remove_global_regex_handler(['pubmsg', 'privmsg'], self.handle_facts) self.connection.reactor.remove_global_regex_handler(['pubmsg', 'privmsg'], self.handle_facts)
super(Facts, self).stop() super(Facts, self).stop()
@ -48,5 +51,18 @@ class Facts(Plugin):
return self.bot.reply(event, msg) return self.bot.reply(event, msg)
def handle_add_fact(self, connection, event, match):
"""Add a new fact to the database."""
category_name = match.group(1)
fact_text = match.group(2)
if has_permission(event.source, 'facts.add_fact'):
# create the category
category, created = FactCategory.objects.get_or_create(name=category_name)
fact = Fact.objects.create(fact=fact_text, category=category, nickmask=event.source)
if fact:
return self.bot.reply(event, "fact added to {0:s}".format(category.name))
plugin = Facts plugin = Facts