X-Git-Url: https://jfr.im/git/erebus.git/blobdiff_plain/80d02bd8940750b5ac5e61974f9de3b0995ca979..5a81c82b9a94a1a2f66552c78ca53b4e10d38a77:/modules/trivia.py diff --git a/modules/trivia.py b/modules/trivia.py index 673a14e..a1ed94b 100644 --- a/modules/trivia.py +++ b/modules/trivia.py @@ -1,7 +1,16 @@ # Erebus IRC bot - Author: Erebus Team -# simple module example +# trivia module # This file is released into the public domain; see http://unlicense.org/ +#TODO: +# timers (stop game/skip question/hinting) +# bonus points +# ability to REDUCE users points +# dynamic questions +# v2 +# team play +# statistics + # module info modinfo = { 'author': 'Erebus Team', @@ -14,8 +23,7 @@ modinfo = { import modlib lib = modlib.modlib(__name__) def modstart(parent, *args, **kwargs): - global state - state = TriviaState("/home/jrunyon/erebus/modules/trivia.json", parent) #TODO + state.parent = parent return lib.modstart(parent, *args, **kwargs) def modstop(*args, **kwargs): global state @@ -26,69 +34,100 @@ def modstop(*args, **kwargs): import json, random class TriviaState(object): - def __init__(self, questionfile, parent): - self.parent = parent + def __init__(self, questionfile): self.questionfile = questionfile self.db = json.load(open(questionfile, "r")) + self.chan = self.db['chan'] + self.curq = None + self.nextq = None def __del__(self): - json.dump(self.db, open(self.questionfile, "w"), indent=4, separators=(',',': ')) + if json is not None and json.dump is not None: + json.dump(self.db, open(self.questionfile, "w"), indent=4, separators=(',',': ')) def nextquestion(self): - nextq = random.choice(self.db['questions']) - self.curq = nextq - self.parent.msg(self.chan, "Next up: %s" % (nextq['question'])) + if state.nextq is not None: + nextq = state.nextq + self.curq = nextq + state.nextq = None + else: + nextq = random.choice(self.db['questions']) + self.curq = nextq + + qtext = "\00300,01Next up: " + qary = nextq['question'].split(None) + for qword in qary: + qtext += "\00300,01"+qword+"\00301,01"+chr(random.randrange(32,126)) + self.parent.channel(self.chan).bot.msg(self.chan, qtext) def checkanswer(self, answer): - if isinstance(self.curq['answer'], basestring): + if self.curq is None: + return False + elif isinstance(self.curq['answer'], basestring): return answer.lower() == self.curq['answer'] else: # assume it's a list or something. return answer.lower() in self.curq['answer'] def addpoint(self, _user, count=1): + _user = str(_user) user = _user.lower() if user in self.db['users']: self.db['users'][user]['points'] += count else: self.db['users'][user] = {'points': count, 'realnick': _user, 'rank': len(self.db['ranks'])} - self.db['ranks']append(user) + self.db['ranks'].append(user) oldrank = self.db['users'][user]['rank'] while oldrank != 0: nextperson = self.db['ranks'][oldrank-1] if self.db['users'][user]['points'] > self.db['users'][nextperson]['points']: self.db['ranks'][oldrank-1] = user + self.db['users'][user]['rank'] = oldrank-1 self.db['ranks'][oldrank] = nextperson + self.db['users'][nextperson]['rank'] = oldrank oldrank = oldrank-1 else: break return self.db['users'][user]['points'] def points(self, user): - user = user.lower() + user = str(user).lower() if user in self.db['users']: return self.db['users'][user]['points'] else: return 0 def rank(self, user): - return self.db['users'][user]['rank'] + user = str(user).lower() + return self.db['users'][user]['rank']+1 - def targetuser(self, user): return "TODO" #TODO - def targetpoints(self, user): return 0 #TODO + def targetuser(self, user): + user = str(user).lower() + rank = self.db['users'][user]['rank'] + if rank == 0: + return "you're in the lead!" + else: + return self.db['ranks'][rank-1] + def targetpoints(self, user): + user = str(user).lower() + rank = self.db['users'][user]['rank'] + if rank == 0: + return "N/A" + else: + return self.db['users'][self.db['ranks'][rank-1]]['points'] + +state = TriviaState("/home/jrunyon/erebus/modules/trivia.json") #TODO get path from config @lib.hookchan(state.db['chan']) def trivia_checkanswer(bot, user, chan, *args): - global state line = ' '.join([str(arg) for arg in args]) if state.checkanswer(line): - bot.msg(chan, "\00308%s\003 has it! The answer was \00308%s\003. Current points: %d. Rank: %d. Target: %s (%d)." % (user, line, state.addpoint(user), state.rank(user), state.targetuser(user), state.targetpoints(user))) - state.nextquestion() + bot.msg(chan, "\00308%s\003 has it! The answer was \00308%s\003. Current points: %d. Rank: %d. Target: %s (%s)." % (user, line, state.addpoint(user), state.rank(user), state.targetuser(user), state.targetpoints(user))) + state.nextquestion() @lib.hook('points') def cmd_points(bot, user, chan, realtarget, *args): - global state - if chan is not None: replyto = chan + if chan == realtarget: replyto = chan else: replyto = user if len(args) != 0: who = args[0] @@ -99,20 +138,55 @@ def cmd_points(bot, user, chan, realtarget, *args): @lib.hook('give', clevel=lib.OP) @lib.argsGE(1) def cmd_give(bot, user, chan, realtarget, *args): - global state - if len(args) > 1 and args[1] != 1: - bot.msg(user, "Giving more than one point is not yet implemented.") - return NotImplemented - whoto = args[0] - balance = state.addpoint(whoto) - bot.msg(chan, "%s gave %s %d points. New balance: %d" % (user, whoto, 1, balance)) + if len(args) > 1: + numpoints = int(args[1]) + else: + numpoints = 1 + balance = state.addpoint(whoto, numpoints) + bot.msg(chan, "%s gave %s %d points. New balance: %d" % (user, whoto, numpoints, balance)) + +@lib.hook('setnext', clevel=lib.OP) +@lib.argsGE(1) +def cmd_setnext(bot, user, chan, realtarget, *args): + line = ' '.join([str(arg) for arg in args]) + linepieces = line.split('*') + question = linepieces[0].strip() + answer = linepieces[1].strip() + state.nextq = {'question':question,'answer':answer} + bot.msg(user, "Done.") + +@lib.hook('skip', clevel=lib.KNOWN) +def cmd_skip(bot, user, chan, realtarget, *args): + state.nextquestion() + +@lib.hook('start') +def cmd_start(bot, user, chan, realtarget, *args): + if chan == realtarget: replyto = chan + else: replyto = user + + if state.curq is None: + state.nextquestion() + else: + bot.msg(replyto, "Game is already started!") + +@lib.hook('stop', clevel=lib.KNOWN) +def cmd_stop(bot, user, chan, realtarget, *args): + if chan == realtarget: replyto = chan + else: replyto = user + + if state.curq is not None: + state.curq = None + bot.msg(chan, "Game ended by %s" % (user)) + else: + bot.msg(replyto, "Game isn't running.") @lib.hook('rank') -@lib.argsEQ(1) def cmd_rank(bot, user, chan, realtarget, *args): - global state - if chan is not None: replyto = chan + if chan == realtarget: replyto = chan else: replyto = user - bot.msg(replyto, "%s is in %d place." % (args[0], state.rank(args[0])) + if len(args) != 0: who = args[0] + else: who = user + + bot.msg(replyto, "%s is in %d place." % (who, state.rank(who)))