From 934f83b734573f741f45eecc0c7f57eca177ba45 Mon Sep 17 00:00:00 2001 From: "Brian S. Stephan" Date: Sat, 20 Jun 2015 10:46:52 -0500 Subject: [PATCH] facts: show the source of facts, if desired --- dr_botzo/facts/ircplugin.py | 10 +++++++++- 1 file changed, 9 insertions(+), 1 deletion(-) diff --git a/dr_botzo/facts/ircplugin.py b/dr_botzo/facts/ircplugin.py index 649bb3f..bd516c0 100644 --- a/dr_botzo/facts/ircplugin.py +++ b/dr_botzo/facts/ircplugin.py @@ -2,6 +2,8 @@ from __future__ import unicode_literals import logging +from irc.client import NickMask + from ircbot.lib import Plugin from facts.models import Fact @@ -38,7 +40,13 @@ class Facts(Plugin): fact = Fact.objects.random_fact(category, regex) if fact: - return self.bot.reply(event, fact.fact) + if fact.category.show_source: + nick = NickMask(fact.nickmask).nick + msg = "{0:s} ({1:s})".format(fact.fact, nick) + else: + msg = fact.fact + + return self.bot.reply(event, msg) plugin = Facts