trivia - add max-streak tracking

diff --git a/modules/contrib/trivia/trivia.json.example b/modules/contrib/trivia/trivia.json.example
index d353b0ecbba2dcdab3461a4da63227141a5fe1e2..3aff6e5b17f94ab04da5f8bc763303c12f9f6c38 100644 (file)
--- a/modules/contrib/trivia/trivia.json.example
+++ b/modules/contrib/trivia/trivia.json.example
@@ -27,5 +27,6 @@
     "maxmissedquestions": 25,
     "hinttimer": 10.0,
     "starttime": 0,
-    "target": 500
+    "target": 500,
+    "streaks": {}
 }

diff --git a/modules/trivia.py b/modules/trivia.py
index b716a597672834719e26051a0aa5a2473a6daa38..ff106fab2f6d5f2aab9499ecd4fb3e6d375b7367 100644 (file)
--- a/modules/trivia.py
+++ b/modules/trivia.py
@@ -419,6 +419,13 @@ class TriviaState(object):
                else:
                        return len(self.db['users'])+1
 
+       def get_streak(self, user):
+               user = str(user).lower()
+               if user in self.db['streaks']:
+                       return self.db['streaks'][user]
+               else:
+                       return 0
+
        def targetuser(self, user):
                if len(self.db['ranks']) == 0: return "no one is ranked!"
 
@@ -466,6 +473,12 @@ def trivia_checkanswer(bot, user, chan, *args):
                response += ". New score: %d. Rank: %d. Target: %s %s" % (state.addpoint(user), state.rank(user), state.targetuser(user), state.targetpoints(user))
                bot.fastmsg(chan, response)
 
+               user_lower = str(user).lower()
+               if user_lower not in state.db['streaks']:
+                       state.db['streaks'][user_lower] = 0
+               if state.streak > state.db['streaks'][user_lower]:
+                       state.db['streaks'][user_lower] = state.streak
+
                if state.streak >= 3:
                        bot.msg(chan, "\00312%s\003 is on a streak! \00307%d\003 answers correct in a row!" % (user, state.streak))
 
@@ -488,7 +501,7 @@ def save(bot, user, chan, realtarget, *args):
 @lib.hook(needchan=False, wantchan=True)
 @lib.help("[<user>]", "shows how many points you or someone has")
 def points(bot, user, chan, realtarget, *args):
-       if chan is not None: eplyto = chan
+       if chan is not None: replyto = chan
        else: replyto = user
 
        if len(args) != 0: who = args[0]
@@ -874,12 +887,52 @@ def delcat(bot, user, chan, realtarget, *args):
        else:
                bot.msg(user, "Category does not exist.")
 
+@lib.hook(needchan=False)
+@lib.help("[<nick>]", "shows a user's (or your own) max streak")
+def streak(bot, user, chan, realtarget, *args):
+       if chan is not None: replyto = chan
+       else: replyto = user
+
+       if len(args) != 0: who = args[0]
+       else: who = user
+
+       bot.msg(replyto, "%s's highest streak is %d" % (who, state.get_streak(who)))
+
+@lib.hook(needchan=False)
+@lib.help(None, "shows top streaks of all time")
+def topstreaks(bot, user, chan, realtarget, *args):
+       db = state.db['streaks']
+       streaks = [(k, db[k]) for k in db.keys()]
+       streaks.sort(key=lambda v: v[1], reverse=True)
+       return "Top streaks of all time: %s (%d), %s (%d), %s (%d)." % (streaks[0][0], streaks[0][1], streaks[1][0], streaks[1][1], streaks[2][0], streaks[2][1])
+
+@lib.hook(glevel=lib.MANAGER, needchan=False)
+@lib.help("<nick> <new streak>", "set a user's max streak")
+@lib.argsEQ(2)
+def setstreak(bot, user, chan, realtarget, *args):
+       temp = 0
+       target = args[0].lower()
+       try:
+               newstreak = int(args[1])
+       except ValueError:
+               return "Syntax: SETSTREAK <nick> <new streak>"
+
+       if target in state.db['streaks']:
+               temp = state.db['streaks'][target]
+               if newstreak == 0:
+                       del state.db['streaks'][target]
+       if newstreak > 0:
+               state.db['streaks'][target] = newstreak
+       return "Done. Streak used to be %d" % (temp)
+
 @lib.hook(needchan=False)
 def triviahelp(bot, user, chan, realtarget, *args):
        bot.slowmsg(user,             "START")
        bot.slowmsg(user,             "TOP10")
        bot.slowmsg(user,             "POINTS        [<user>]")
        bot.slowmsg(user,             "RANK          [<user>]")
+       bot.slowmsg(user,             "STREAK        [<user>]")
+       bot.slowmsg(user,             "TOPSTREAKS")
        bot.slowmsg(user,             "BADQ          <reason> (include info to identify question)")
        if user.glevel >= 1:
                bot.slowmsg(user,         "SKIP                            (KNOWN)")
@@ -902,6 +955,7 @@ def triviahelp(bot, user, chan, realtarget, *args):
                                bot.slowmsg(user, "HINTTIMER     <float seconds>   (ADMIN)")
                                bot.slowmsg(user, "HINTNUM       <hints>           (ADMIN)")
                                bot.slowmsg(user, "QUESTIONPAUSE <float seconds>   (ADMIN)")
+                               bot.slowmsg(user, "SETSTREAK     <nick> <streak>   (ADMIN)")
 
 @lib.hooknum(332) # topic is...
 @lib.hooknum(331) # no topic set