X-Git-Url: https://jfr.im/git/erebus.git/blobdiff_plain/442ed9233f37873744c289c57bbcf36f194aa3e2..be8072b500f64089b2bc540e81ce90f115231853:/modules/trivia.py diff --git a/modules/trivia.py b/modules/trivia.py index bf31f1c..0343418 100644 --- a/modules/trivia.py +++ b/modules/trivia.py @@ -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,16 +19,18 @@ 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 import json, random, threading, re, time, datetime try: - import twitter + import twitter except: pass # doesn't matter if we don't have twitter, updating the status just will fall through the try-except if so... def findnth(haystack, needle, n): #http://stackoverflow.com/a/1884151 @@ -37,26 +39,37 @@ 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 MyTimer(threading._Timer): + def __init__(self, *args, **kwargs): + threading._Timer.__init__(self, *args, **kwargs) + self.daemon = True + class TriviaState(object): def __init__(self, parent=None, pointvote=False): if parent is not None: self.gotParent(parent, pointvote) def gotParent(self, parent, pointvote=False): - self.parent = parent - self.questionfile = self.parent.cfg.get('trivia', 'jsonpath') - self.db = json.load(open(self.questionfile, "r")) - self.chan = self.db['chan'] - self.curq = None - self.nextq = None - self.nextquestiontimer = None - self.steptimer = None - self.hintstr = None - self.hintanswer = None - self.hintsgiven = 0 + self.parent = parent + 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 + self.nextq = None + self.nextquestiontimer = None + self.steptimer = None + self.hintstr = None + self.hintanswer = None + self.hintsgiven = 0 self.revealpossibilities = None - self.gameover = False - self.missedquestions = 0 + self.gameover = False + self.missedquestions = 0 + self.curqid = None + self.lastqid = None if 'starttime' not in self.db or self.db['starttime'] is None: self.db['starttime'] = time.time() @@ -65,7 +78,7 @@ class TriviaState(object): self.getchan().msg("Vote for the next round target points! Options: %s. Vote using !vote " % (', '.join([str(x) for x in self.db['targetoptions']]))) self.getchan().msg("You have %s seconds." % (self.db['votetimer'])) self.voteamounts = dict([(x, 0) for x in self.db['targetoptions']]) # make a dict {pointsoptionA: 0, pointsoptionB: 0, ...} - self.pointvote = threading.Timer(self.db['votetimer'], self.endPointVote) + self.pointvote = MyTimer(self.db['votetimer'], self.endPointVote) self.pointvote.start() else: self.pointvote = None @@ -100,21 +113,19 @@ class TriviaState(object): revealloc = findnth(''.join(self.hintstr), '*', revealcount) self.revealpossibilities.remove(revealcount) self.hintstr[revealloc] = answer[revealloc] - self.parent.channel(self.chan).bot.msg(self.chan, "\00304,01Here's a hint: %s" % (''.join(self.hintstr))) + self.parent.channel(self.chan).bot.fastmsg(self.chan, "\00304,01Here's a hint: %s" % (''.join(self.hintstr))) self.hintsgiven += 1 if hintnum < self.db['hintnum']: - self.steptimer = threading.Timer(self.db['hinttimer'], self.nexthint, args=[hintnum+1]) + self.steptimer = MyTimer(self.db['hinttimer'], self.nexthint, args=[hintnum+1]) self.steptimer.start() else: - self.steptimer = threading.Timer(self.db['hinttimer'], self.nextquestion, args=[True]) + self.steptimer = MyTimer(self.db['hinttimer'], self.nextquestion, args=[True]) self.steptimer.start() 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!!!") @@ -125,6 +136,9 @@ class TriviaState(object): except IndexError: pass except Exception as e: msg("DERP! %r" % (e)) + self.db['lastwinner'] = winner + self.db['lastwon'] = time.time() + if self.db['hofpath'] is not None and self.db['hofpath'] != '': self.writeHof() @@ -191,13 +205,18 @@ class TriviaState(object): self.nextquestion() #start the game! def nextquestion(self, qskipped=False, iteration=0, skipwait=False): + self.lastqid = self.curqid + self.curq = None + self.curqid = None if self.gameover == True: return self.doGameOver() if qskipped: - self.getchan().msg("\00304Fail! The correct answer was: %s" % (self.hintanswer)) + self.getchan().fastmsg("\00304Fail! The correct answer was: %s" % (self.hintanswer)) 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 +231,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 = MyTimer(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,17 +263,20 @@ class TriviaState(object): nextq[1] = nextq[1].lower() qtext = "\00304,01Next up: " + qtext += "(%5d)" % (random.randint(0,99999)) qary = nextq[0].split(None) + qtext += " " for qword in qary: qtext += "\00304,01"+qword+"\00301,01"+chr(random.randrange(0x61,0x7A)) #a-z - self.getbot().msg(self.chan, qtext) + self.getbot().fastmsg(self.chan, qtext) self.curq = nextq + self.curqid = nextqid if isinstance(self.curq[1], basestring): self.hintanswer = self.curq[1] else: self.hintanswer = random.choice(self.curq[1]) - self.steptimer = threading.Timer(self.db['hinttimer'], self.nexthint, args=[1]) + self.steptimer = MyTimer(self.db['hinttimer'], self.nexthint, args=[1]) self.steptimer.start() def checkanswer(self, answer): @@ -327,14 +351,15 @@ state = TriviaState() def trivia_checkanswer(bot, user, chan, *args): line = ' '.join([str(arg) for arg in args]) if state.checkanswer(line): - 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))) + state.curq = None + bot.fastmsg(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 +367,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,18 +379,22 @@ 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] - bot.msg(user, "Done. Next question is: %s" % (state.nextq[0])) + if user.glevel >= lib.STAFF: + respstr = "Done. Next question is: %s" % (state.nextq[0]) + else: + respstr = "Done." + bot.msg(user, respstr) 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 +405,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!") @@ -399,10 +433,15 @@ def cmd_stop(bot, user, chan, realtarget, *args): else: bot.msg(user, "Game isn't running.") +@lib.hook('exception') +def cmd_exception(*args, **kwargs): + raise Exception() + 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 +457,49 @@ 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(1) +def badq(bot, user, chan, realtarget, *args): + lastqid = state.lastqid + curqid = state.curqid + + reason = ' '.join(args) + state.db['badqs'].append([lastqid, curqid, reason]) + bot.msg(user, "Reported bad question.") + +@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: LastQ=%r CurQ=%r: %s" % (i, report[0], report[1], report[2])) + try: lq = state.db['questions'][int(report[0])] + except Exception as e: lq = (None,None) + try: cq = state.db['questions'][int(report[1])] + except Exception as e: cq = (None, None) + bot.msg(user, "- Last: %s*%s" % (lq[0], lq[1])) + bot.msg(user, "- Curr: %s*%s" % (cq[0], cq[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 +507,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 +532,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,51 +543,57 @@ 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): - 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: +@lib.hook(glevel=1, needchan=False) +def findq(bot, user, chan, realtarget, *args): + if len(args) == 0: + bot.msg(user, "You need to specify a search string.") + return + + searcher = re.compile(' '.join(args)) + matches = [str(i) for i in range(len(state.db['questions'])) if searcher.search(state.db['questions'][i][0]) is not None] + if len(matches) > 25: + bot.msg(user, "Too many matches! (>25)") + elif len(matches) > 1: bot.msg(user, "Multiple matches: %s" % (', '.join(matches))) elif len(matches) == 1: bot.msg(user, "One match: %s" % (matches[0])) 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 +601,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 +614,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 []") - bot.msg(user, "RANK []") - else: - bot.msg(user, "START (ANYONE )") - bot.msg(user, "TOP10 (ANYONE )") - bot.msg(user, "POINTS [] (ANYONE )") - bot.msg(user, "RANK [] (ANYONE )") +@lib.hook(needchan=False) +def triviahelp(bot, user, chan, realtarget, *args): + bot.msg(user, "START") + bot.msg(user, "TOP10") + bot.msg(user, "POINTS []") + bot.msg(user, "RANK []") + bot.msg(user, "BADQ (include info to identify question)") if user.glevel >= 1: bot.msg(user, "SKIP (>=KNOWN)") bot.msg(user, "STOP (>=KNOWN)") @@ -550,6 +630,9 @@ def cmd_triviahelp(bot, user, chan, realtarget, *args): bot.msg(user, "SETNEXT * (>=STAFF)") bot.msg(user, "ADDQ * (>=STAFF)") bot.msg(user, "DELETEQ * (>=STAFF) [aka DELQ]") + bot.msg(user, "BADQS (>=STAFF)") + bot.msg(user, "CLEARBADQS (>=STAFF)") + bot.msg(user, "DELBADQ (>=STAFF)") if user.glevel >= lib.ADMIN: bot.msg(user, "SETTARGET (>=ADMIN)") bot.msg(user, "MAXMISSED (>=ADMIN)") @@ -562,6 +645,40 @@ 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']) + ]), + 'lastwinner': state.db['lastwinner'], + 'lastwon': time.strftime("%b %d", time.gmtime(state.db['lastwon'])), + '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]]