]> jfr.im git - erebus.git/blobdiff - modules/trivia.py
update modules to use default command name (based on function name)
[erebus.git] / modules / trivia.py
index bf31f1c1fb75349f2684eefa9267e006e6c233d1..3f541a55bc64b997c1dd287ed22270ae437d0830 100644 (file)
@@ -7,7 +7,7 @@ modinfo = {
        'author': 'Erebus Team',
        'license': 'public domain',
        'compatible': [1], # compatible module API versions
-       'depends': [], # other modules required to work properly?
+       'depends': ['userinfo'], # other modules required to work properly?
 }
 
 # preamble
@@ -19,9 +19,11 @@ def modstart(parent, *args, **kwargs):
        return lib.modstart(parent, *args, **kwargs)
 def modstop(*args, **kwargs):
        global state
-       stop()
-       state.closeshop()
-       del state
+       try:
+               stop()
+               state.closeshop()
+               del state
+       except Exception: pass
        return lib.modstop(*args, **kwargs)
 
 # module code
@@ -37,6 +39,10 @@ def findnth(haystack, needle, n): #http://stackoverflow.com/a/1884151
                return -1
        return len(haystack)-len(parts[-1])-len(needle)
 
+def person(num): return state.db['users'][state.db['ranks'][num]]['realnick']
+def pts(num): return str(state.db['users'][state.db['ranks'][num]]['points'])
+def country(num, default="??"): return lib.mod('userinfo')._get(person(num), 'country', default=default)
+
 class TriviaState(object):
        def __init__(self, parent=None, pointvote=False):
                if parent is not None:
@@ -44,7 +50,7 @@ class TriviaState(object):
 
        def gotParent(self, parent, pointvote=False):
                self.parent = parent
-               self.questionfile = self.parent.cfg.get('trivia', 'jsonpath')
+               self.questionfile = self.parent.cfg.get('trivia', 'jsonpath', default="./modules/trivia.json")
                self.db = json.load(open(self.questionfile, "r"))
                self.chan = self.db['chan']
                self.curq = None
@@ -113,8 +119,6 @@ class TriviaState(object):
 
        def doGameOver(self):
                msg = self.getchan().msg
-               def person(num): return self.db['users'][self.db['ranks'][num]]['realnick']
-               def pts(num): return str(self.db['users'][self.db['ranks'][num]]['points'])
                winner = person(0)
                try:
                        msg("\00312THE GAME IS OVER!!!")
@@ -191,6 +195,7 @@ class TriviaState(object):
                self.nextquestion() #start the game!
 
        def nextquestion(self, qskipped=False, iteration=0, skipwait=False):
+               self.curq = None
                if self.gameover == True:
                        return self.doGameOver()
                if qskipped:
@@ -198,6 +203,8 @@ class TriviaState(object):
                        self.missedquestions += 1
                else:
                        self.missedquestions = 0
+                       if 'topicformat' in self.db and self.db['topicformat'] is not None:
+                               self.getbot().conn.send("TOPIC %s" % (self.db['chan']))
 
                if isinstance(self.steptimer, threading._Timer):
                        self.steptimer.cancel()
@@ -212,24 +219,26 @@ class TriviaState(object):
 
                if self.missedquestions > self.db['maxmissedquestions']:
                        stop()
-                       self.getbot().msg(self.getchan(), "%d questions unanswered! Stopping the game.")
+                       self.getbot().msg(self.getchan(), "%d questions unanswered! Stopping the game." % (self.missedquestions))
+                       return
 
                if skipwait:
-                       self._nextquestion(qskipped, iteration)
+                       self._nextquestion(iteration)
                else:
-                       print "making timer"
-                       self.nextquestiontimer = threading.Timer(self.db['questionpause'], self._nextquestion, args=[qskipped, iteration])
+                       self.nextquestiontimer = threading.Timer(self.db['questionpause'], self._nextquestion, args=[iteration])
                        self.nextquestiontimer.start()
 
-       def _nextquestion(self, qskipped, iteration):
-               print "_"
+       def _nextquestion(self, iteration):
                if self.nextq is not None:
+                       nextqid = None
                        nextq = self.nextq
                        self.nextq = None
                else:
-                       nextq = random.choice(self.db['questions'])
+                       nextqid = random.randrange(0, len(self.db['questions']))
+                       nextq = self.db['questions'][nextqid]
 
                if nextq[0][0] == "!":
+                       nextqid = None
                        nextq = specialQuestion(nextq)
 
                if len(nextq) > 2 and nextq[2] - time.time() < 7*24*60*60 and iteration < 10:
@@ -242,6 +251,8 @@ class TriviaState(object):
                nextq[1] = nextq[1].lower()
 
                qtext = "\00304,01Next up: "
+               if nextqid is not None:
+                       qtext += "(%d) " % (nextqid)
                qary = nextq[0].split(None)
                for qword in qary:
                        qtext += "\00304,01"+qword+"\00301,01"+chr(random.randrange(0x61,0x7A)) #a-z
@@ -327,14 +338,15 @@ state = TriviaState()
 def trivia_checkanswer(bot, user, chan, *args):
        line = ' '.join([str(arg) for arg in args])
        if state.checkanswer(line):
+               state.curq = None
                bot.msg(chan, "\00312%s\003 has it! The answer was \00312%s\003. New score: %d. Rank: %d. Target: %s (%s)." % (user, line, state.addpoint(user), state.rank(user), state.targetuser(user), state.targetpoints(user)))
                if state.hintsgiven == 0:
                        bot.msg(chan, "\00312%s\003 got an extra point for getting it before the hints! New score: %d." % (user, state.addpoint(user)))
                state.nextquestion()
 
-@lib.hook('points', needchan=False)
-def cmd_points(bot, user, chan, realtarget, *args):
-       if chan == realtarget: replyto = chan
+@lib.hook(needchan=False)
+def points(bot, user, chan, realtarget, *args):
+       if chan is not None and realtarget == chan.name: replyto = chan
        else: replyto = user
 
        if len(args) != 0: who = args[0]
@@ -342,9 +354,9 @@ def cmd_points(bot, user, chan, realtarget, *args):
 
        bot.msg(replyto, "%s has %d points." % (who, state.points(who)))
 
-@lib.hook('give', glevel=lib.STAFF, needchan=False)
+@lib.hook(glevel=lib.STAFF, needchan=False)
 @lib.argsGE(1)
-def cmd_give(bot, user, chan, realtarget, *args):
+def give(bot, user, chan, realtarget, *args):
        whoto = args[0]
        if len(args) > 1:
                numpoints = int(args[1])
@@ -354,8 +366,8 @@ def cmd_give(bot, user, chan, realtarget, *args):
 
        bot.msg(chan, "%s gave %s %d points. New balance: %d" % (user, whoto, numpoints, balance))
 
-@lib.hook('setnextid', glevel=1, needchan=False)
-def cmd_setnextid(bot, user, chan, realtarget, *args):
+@lib.hook(glevel=1, needchan=False)
+def setnextid(bot, user, chan, realtarget, *args):
        try:
                qid = int(args[0])
                state.nextq = state.db['questions'][qid]
@@ -363,9 +375,9 @@ def cmd_setnextid(bot, user, chan, realtarget, *args):
        except Exception as e:
                bot.msg(user, "Error: %s" % (e))
 
-@lib.hook('setnext', glevel=lib.STAFF, needchan=False)
+@lib.hook(glevel=lib.STAFF, needchan=False)
 @lib.argsGE(1)
-def cmd_setnext(bot, user, chan, realtarget, *args):
+def setnext(bot, user, chan, realtarget, *args):
        line = ' '.join([str(arg) for arg in args])
        linepieces = line.split('*')
        if len(linepieces) < 2:
@@ -376,16 +388,21 @@ def cmd_setnext(bot, user, chan, realtarget, *args):
        state.nextq = [question, answer]
        bot.msg(user, "Done.")
 
-@lib.hook('skip', glevel=1, needchan=False)
-def cmd_skip(bot, user, chan, realtarget, *args):
+@lib.hook(glevel=1, needchan=False)
+def skip(bot, user, chan, realtarget, *args):
        state.nextquestion(qskipped=True, skipwait=True)
 
-@lib.hook('start', needchan=False)
-def cmd_start(bot, user, chan, realtarget, *args):
-       if chan == realtarget: replyto = chan
+@lib.hook(needchan=False)
+def start(bot, user, chan, realtarget, *args):
+       if chan is not None and realtarget == chan.name: replyto = chan
        else: replyto = user
 
+       if chan is not None and chan.name != state.db['chan']:
+               bot.msg(replyto, "That command isn't valid here.")
+               return
+
        if state.curq is None and state.pointvote is None and state.nextquestiontimer is None:
+               bot.msg(state.db['chan'], "%s has started the game!" % (user))
                state.nextquestion(skipwait=True)
        elif state.pointvote is not None:
                bot.msg(replyto, "There's a vote in progress!")
@@ -401,8 +418,9 @@ def cmd_stop(bot, user, chan, realtarget, *args):
 
 def stop():
        try:
-               if state.curq is not None:
+               if state.curq is not None or state.nextq is not None:
                        state.curq = None
+                       state.nextq = None
                        try:
                                state.steptimer.cancel()
                        except Exception as e:
@@ -418,9 +436,44 @@ def stop():
        except NameError:
                pass
 
-@lib.hook('rank', needchan=False)
-def cmd_rank(bot, user, chan, realtarget, *args):
-       if chan == realtarget: replyto = chan
+@lib.hook(needchan=False)
+@lib.argsGE(2)
+def badq(bot, user, chan, realtarget, *args):
+       qid = int(args[0])
+       reason = ' '.join(args[1:])
+       state.db['badqs'].append([qid, reason])
+       bot.msg(user, "Reported bad question #%d" % (qid))
+
+@lib.hook(glevel=lib.STAFF, needchan=False)
+def badqs(bot, user, chan, realtarget, *args):
+       if len(state.db['badqs']) == 0:
+               bot.msg(user, "No reports.")
+
+       for i in range(len(state.db['badqs'])):
+               try:
+                       report = state.db['badqs'][i]
+                       bot.msg(user, "Report #%d: Q%d: %s" % (i, report[0], report[1]))
+                       question = state.db['questions'][int(report[0])]
+                       bot.msg(user, "- Question is: %s" % (question[0]))
+                       bot.msg(user, "- Answer is: %s" % (question[1]))
+               except Exception as e:
+                       bot.msg(user, "- Exception: %r" % (e))
+
+@lib.hook(glevel=lib.STAFF, needchan=False)
+def clearbadqs(bot, user, chan, realtarget, *args):
+       state.db['badqs'] = []
+       bot.msg(user, "Cleared reports.")
+
+@lib.hook(glevel=lib.STAFF, needchan=False)
+@lib.argsEQ(1)
+def delbadq(bot, user, chan, realtarget, *args):
+       qid = int(args[0])
+       del state.db['badqs'][qid]
+       bot.msg(user, "Removed report #%d" % (qid))
+
+@lib.hook(needchan=False)
+def rank(bot, user, chan, realtarget, *args):
+       if chan is not None and realtarget == chan.name: replyto = chan
        else: replyto = user
 
        if len(args) != 0: who = args[0]
@@ -428,19 +481,19 @@ def cmd_rank(bot, user, chan, realtarget, *args):
 
        bot.msg(replyto, "%s is in %d place (%s points). Target is: %s (%s points)." % (who, state.rank(who), state.points(who), state.targetuser(who), state.targetpoints(who)))
 
-@lib.hook('top10', needchan=False)
-def cmd_top10(bot, user, chan, realtarget, *args):
+@lib.hook(needchan=False)
+def top10(bot, user, chan, realtarget, *args):
        if len(state.db['ranks']) == 0:
                return bot.msg(state.db['chan'], "No one is ranked!")
 
-       replylist = []
-       for nick in state.db['ranks'][0:10]:
-               user = state.db['users'][nick]
-               replylist.append("%s (%s)" % (user['realnick'], user['points']))
-       bot.msg(state.db['chan'], ', '.join(replylist))
+       max = len(state.db['ranks'])
+       if max > 10:
+               max = 10
+       replylist = ', '.join(["%s (%s) %s" % (person(x), country(x, "unknown"), pts(x)) for x in range(max)])
+       bot.msg(state.db['chan'], "Top %d: %s" % (max, replylist))
 
-@lib.hook('settarget', glevel=lib.ADMIN, needchan=False)
-def cmd_settarget(bot, user, chan, realtarget, *args):
+@lib.hook(glevel=lib.ADMIN, needchan=False)
+def settarget(bot, user, chan, realtarget, *args):
        try:
                state.db['target'] = int(args[0])
                bot.msg(state.db['chan'], "Target has been changed to %s points!" % (state.db['target']))
@@ -453,8 +506,8 @@ def cmd_settarget(bot, user, chan, realtarget, *args):
                print e
                bot.msg(user, "Failed to set target.")
 
-@lib.hook('vote', needchan=False)
-def cmd_vote(bot, user, chan, realtarget, *args):
+@lib.hook(needchan=False)
+def vote(bot, user, chan, realtarget, *args):
        if state.pointvote is not None:
                if int(args[0]) in state.voteamounts:
                        state.voteamounts[int(args[0])] += 1
@@ -464,40 +517,40 @@ def cmd_vote(bot, user, chan, realtarget, *args):
        else:
                bot.msg(user, "There's no vote in progress.")
 
-@lib.hook('maxmissed', glevel=lib.ADMIN, needchan=False)
-def cmd_maxmissed(bot, user, chan, realtarget, *args):
+@lib.hook(glevel=lib.ADMIN, needchan=False)
+def maxmissed(bot, user, chan, realtarget, *args):
        try:
                state.db['maxmissedquestions'] = int(args[0])
                bot.msg(state.db['chan'], "Max missed questions before round ends has been changed to %s." % (state.db['maxmissedquestions']))
        except:
                bot.msg(user, "Failed to set maxmissed.")
 
-@lib.hook('hinttimer', glevel=lib.ADMIN, needchan=False)
-def cmd_hinttimer(bot, user, chan, realtarget, *args):
+@lib.hook(glevel=lib.ADMIN, needchan=False)
+def hinttimer(bot, user, chan, realtarget, *args):
        try:
                state.db['hinttimer'] = float(args[0])
                bot.msg(state.db['chan'], "Time between hints has been changed to %s." % (state.db['hinttimer']))
        except:
                bot.msg(user, "Failed to set hint timer.")
 
-@lib.hook('hintnum', glevel=lib.ADMIN, needchan=False)
-def cmd_hintnum(bot, user, chan, realtarget, *args):
+@lib.hook(glevel=lib.ADMIN, needchan=False)
+def hintnum(bot, user, chan, realtarget, *args):
        try:
                state.db['hintnum'] = int(args[0])
                bot.msg(state.db['chan'], "Max number of hints has been changed to %s." % (state.db['hintnum']))
        except:
                bot.msg(user, "Failed to set hintnum.")
 
-@lib.hook('questionpause', glevel=lib.ADMIN, needchan=False)
-def cmd_questionpause(bot, user, chan, realtarget, *args):
+@lib.hook(glevel=lib.ADMIN, needchan=False)
+def questionpause(bot, user, chan, realtarget, *args):
        try:
                state.db['questionpause'] = float(args[0])
                bot.msg(state.db['chan'], "Pause between questions has been changed to %s." % (state.db['questionpause']))
        except:
                bot.msg(user, "Failed to set questionpause.")
 
-@lib.hook('findq', glevel=1, needchan=False)
-def cmd_findquestion(bot, user, chan, realtarget, *args):
+@lib.hook(glevel=1, needchan=False)
+def findq(bot, user, chan, realtarget, *args):
        matches = [str(i) for i in range(len(state.db['questions'])) if state.db['questions'][i][0] == ' '.join(args)] #TODO looser equality check
        if len(matches) > 1:
                bot.msg(user, "Multiple matches: %s" % (', '.join(matches)))
@@ -506,9 +559,8 @@ def cmd_findquestion(bot, user, chan, realtarget, *args):
        else:
                bot.msg(user, "No match.")
 
-@lib.hook('delq', glevel=lib.STAFF, needchan=False)
-@lib.hook('deleteq', glevel=lib.STAFF, needchan=False)
-def cmd_deletequestion(bot, user, chan, realtarget, *args):
+@lib.hook(('delq', 'deleteq'), glevel=lib.STAFF, needchan=False)
+def delq(bot, user, chan, realtarget, *args):
        try:
                backup = state.db['questions'][int(args[0])]
                del state.db['questions'][int(args[0])]
@@ -516,8 +568,8 @@ def cmd_deletequestion(bot, user, chan, realtarget, *args):
        except:
                bot.msg(user, "Couldn't delete that question.")
 
-@lib.hook('addq', glevel=lib.STAFF, needchan=False)
-def cmd_addquestion(bot, user, chan, realtarget, *args):
+@lib.hook(glevel=lib.STAFF, needchan=False)
+def addq(bot, user, chan, realtarget, *args):
        line = ' '.join([str(arg) for arg in args])
        linepieces = line.split('*')
        if len(linepieces) < 2:
@@ -529,18 +581,13 @@ def cmd_addquestion(bot, user, chan, realtarget, *args):
        bot.msg(user, "Done. Question is #%s" % (len(state.db['questions'])-1))
 
 
-@lib.hook('triviahelp', needchan=False)
-def cmd_triviahelp(bot, user, chan, realtarget, *args):
-       if user.glevel == 0:
-               bot.msg(user,         "START")
-               bot.msg(user,         "TOP10")
-               bot.msg(user,         "POINTS        [<user>]")
-               bot.msg(user,         "RANK          [<user>]")
-       else:
-               bot.msg(user,         "START                           (ANYONE )")
-               bot.msg(user,         "TOP10                           (ANYONE )")
-               bot.msg(user,         "POINTS        [<user>]          (ANYONE )")
-               bot.msg(user,         "RANK          [<user>]          (ANYONE )")
+@lib.hook(needchan=False)
+def triviahelp(bot, user, chan, realtarget, *args):
+       bot.msg(user,             "START")
+       bot.msg(user,             "TOP10")
+       bot.msg(user,             "POINTS        [<user>]")
+       bot.msg(user,             "RANK          [<user>]")
+       bot.msg(user,             "BADQ          <id> <reason>")
        if user.glevel >= 1:
                bot.msg(user,         "SKIP                            (>=KNOWN)")
                bot.msg(user,         "STOP                            (>=KNOWN)")
@@ -550,6 +597,9 @@ def cmd_triviahelp(bot, user, chan, realtarget, *args):
                        bot.msg(user,     "SETNEXT       <q>*<a>           (>=STAFF)")
                        bot.msg(user,     "ADDQ          <q>*<a>           (>=STAFF)")
                        bot.msg(user,     "DELETEQ       <q>*<a>           (>=STAFF)  [aka DELQ]")
+                       bot.msg(user,     "BADQS                           (>=STAFF)")
+                       bot.msg(user,     "CLEARBADQS                      (>=STAFF)")
+                       bot.msg(user,     "DELBADQ       <reportid>        (>=STAFF)")
                        if user.glevel >= lib.ADMIN:
                                bot.msg(user, "SETTARGET     <points>          (>=ADMIN)")
                                bot.msg(user, "MAXMISSED     <questions>       (>=ADMIN)")
@@ -562,6 +612,38 @@ def num_417(bot, textline):
        bot.msg(state.db['chan'], "Whoops, it looks like that question didn't quite go through! (E:417). Let's try another...")
        state.nextquestion(qskipped=False, skipwait=True)
 
+@lib.hooknum(332)
+def num_TOPIC(bot, textline):
+       pieces = textline.split(None, 4)
+       chan = pieces[3]
+       if chan != state.db['chan']:
+               return
+       gottopic = pieces[4][1:]
+
+       formatted = state.db['topicformat'] % {
+               'chan': state.db['chan'],
+               'top1': "%s (%s)" % (person(0), pts(0)),
+               'top3': '/'.join([
+                       "%s (%s)" % (person(x), pts(x))
+                       for x in range(3) if x < len(state.db['ranks'])
+               ]),
+               'top3c': ' '.join([
+                       "%s (%s, %s)" % (person(x), pts(x), country(x))
+                       for x in range(3) if x < len(state.db['ranks'])
+               ]),
+               'top10': ' '.join([
+                       "%s (%s)" % (person(x), pts(x))
+                       for x in range(10) if x < len(state.db['ranks'])
+               ]),
+               'top10c': ' '.join([
+                       "%s (%s, %s)" % (person(x), pts(x), country(x))
+                       for x in range(10) if x < len(state.db['ranks'])
+               ]),
+               'target': state.db['target'],
+       }
+       if gottopic != formatted:
+               state.getbot().conn.send("TOPIC %s :%s" % (state.db['chan'], formatted))
+
 
 def specialQuestion(oldq):
        newq = [oldq[0], oldq[1]]