+@lib.hook(needchan=False)
+@lib.help(None, "show current category")
+def showcat(bot, user, chan, realtarget, *args):
+ bot.msg(user, "Current category: %s" % (state.db['category']))
+
+@lib.hook(glevel=1, needchan=False)
+@lib.help("<category>", "change category")
+def setcat(bot, user, chan, realtarget, *args):
+ category = args[0].lower()
+ if category in state.db['questions']:
+ state.db['category'] = category
+ state.questions = state.db['questions'][category]
+ bot.msg(user, "Changed category to %s" % (category))
+ else:
+ bot.msg(user, "That category doesn't exist.")
+
+@lib.hook(needchan=False)
+@lib.help(None, "list categories", "the current category will be marked with a *")
+def listcats(bot, user, chan, realtarget, *args):
+ cats = ["%s%s (%d)" % ("*" if c == state.db['category'] else "", c, len(state.db['questions'][c])) for c in state.db['questions'].keys()]
+ bot.msg(user, "Categories: %s" % (', '.join(cats)))
+
+@lib.hook(glevel=lib.STAFF, needchan=False)
+@lib.help("<category>", "adds an empty category")
+def addcat(bot, user, chan, realtarget, *args):
+ category = args[0].lower()
+ if category not in state.db['questions']:
+ state.db['questions'][category] = []
+ bot.msg(user, "Added category %s" % (category))
+ else:
+ bot.msg(user, "Category already exists.")
+
+@lib.hook(glevel=lib.MANAGER, needchan=False)
+@lib.help("<category>", "deletes an entire category")
+def delcat(bot, user, chan, realtarget, *args):
+ category = args[0].lower()
+ if category == state.db['category']:
+ bot.msg(user, "Category currently in use!")
+ elif category in state.db['questions']:
+ length = len(state.db['questions'][category])
+ del state.db['questions'][category]
+ bot.msg(user, "Deleted category %s (%d questions)" % (category, length))
+ 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
+
+ streak = state.get_streak(who)
+ bot.msg(replyto, "%s's highest streak is %d, set %s" % (who, streak[0], time.strftime("%d %b %Y %H:%M:%S %Z", time.gmtime(streak[1]))))
+
+@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 = [(nick, db[nick][0], db[nick][1]) for nick 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> [<timestamp>]", "set a user's max streak")
+@lib.argsGE(2)
+def setstreak(bot, user, chan, realtarget, *args):
+ temp = 0
+ target = args[0].lower()
+ try:
+ newstreak = int(args[1])
+ except ValueError:
+ return "Error: <new streak> must be integer"
+ timestamp = time.time()
+ if len(args) > 2:
+ try:
+ timestamp = int(args[2])
+ except ValueError:
+ return "Error: <timestamp> must be integer (unix timestamp)"
+
+ 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, timestamp]
+ return "Done. Streak used to be %d set %d" % (temp[0], temp[1])