X-Git-Url: https://jfr.im/git/erebus.git/blobdiff_plain/8ba566068fc22454dc1ec3d296b85f35aa7aec17..7bd5e7d512da00962d6027348c7325ae1188b40d:/modules/trivia.py?ds=sidebyside

diff --git a/modules/trivia.py b/modules/trivia.py
index 74fd142..7e9283c 100644
--- a/modules/trivia.py
+++ b/modules/trivia.py
@@ -24,7 +24,7 @@ def modstop(*args, **kwargs):
 	return lib.modstop(*args, **kwargs)
 
 # module code
-import json, random, threading, re, time
+import json, random, threading, re, time, datetime
 
 try:
 	import twitter 
@@ -52,6 +52,9 @@ class TriviaState(object):
 		self.gameover = False
 		self.missedquestions = 0
 
+		if 'starttime' not in self.db or self.db['starttime'] is None:
+			self.db['starttime'] = time.time()
+
 		if pointvote:
 			self.getchan().msg("Vote for the next round target points! Options: %s. Vote using !vote <choice>" % (', '.join([str(x) for x in self.db['targetoptions']])))
 			self.getchan().msg("You have %s seconds." % (self.db['votetimer']))
@@ -99,9 +102,9 @@ class TriviaState(object):
 			self.steptimer.start()
 
 	def doGameOver(self):
-		def msg(line): self.getbot().msg(self.getchan(), line)
+		msg = self.getchan().msg
 		def person(num): return self.db['users'][self.db['ranks'][num]]['realnick']
-		def pts(num): return self.db['users'][self.db['ranks'][num]]['points']
+		def pts(num): return str(self.db['users'][self.db['ranks'][num]]['points'])
 		winner = person(0)
 		try:
 			msg("\00312THE GAME IS OVER!!!")
@@ -112,6 +115,9 @@ class TriviaState(object):
 		except IndexError: pass
 		except Exception as e: msg("DERP! %r" % (e))
 
+		if self.db['hofpath'] is not None and self.db['hofpath'] != '':
+			self.writeHof()
+
 		self.db['users'] = {}
 		self.db['ranks'] = []
 		stop()
@@ -127,6 +133,40 @@ class TriviaState(object):
 
 		self.__init__(self.questionfile, self.parent, True)
 
+	def writeHof(self):
+		def person(num):
+			try: return self.db['users'][self.db['ranks'][num]]['realnick']
+			except: return "none"
+		def pts(num):
+			try: return str(self.db['users'][self.db['ranks'][num]]['points'])
+			except: return 0
+
+		try:
+			f = open(self.db['hofpath'], 'rb+')
+			for i in range(self.db['hoflines']): #skip this many lines
+				f.readline()
+			insertpos = f.tell()
+			fcontents = f.read()
+			f.seek(insertpos)
+			f.write((self.db['hofformat']+"\n") % {
+				'date': time.strftime("%F", time.gmtime()),
+				'duration': str(datetime.timedelta(seconds=time.time()-self.db['starttime'])),
+				'targetscore': self.db['target'],
+				'firstperson': person(0),
+				'firstscore': pts(0),
+				'secondperson': person(1),
+				'secondscore': pts(1),
+				'thirdperson': person(2),
+				'thirdscore': pts(2),
+			})
+			f.write(fcontents)
+			return True
+		except Exception as e:
+			raise e
+			return False
+		finally:
+			f.close()
+
 	def endPointVote(self):
 		self.getchan().msg("Voting has ended!")
 		votelist = sorted(self.voteamounts.items(), key=lambda item: item[1]) #sort into list of tuples: [(option, number_of_votes), ...]
@@ -364,16 +404,17 @@ def cmd_settarget(bot, user, chan, realtarget, *args):
 		state.db['target'] = int(args[0])
 		bot.msg(state.db['chan'], "Target has been changed to %s points!" % (state.db['target']))
 
-		if state.votetimer is not None:
-			state.votetimer.cancel()
-			state.votetimer = None
+		if state.pointvote is not None:
+			state.pointvote.cancel()
+			state.pointvote = None
 			bot.msg(state.db['chan'], "Vote has been cancelled!")
-	except:
+	except Exception as e:
+		print e
 		bot.msg(user, "Failed to set target.")
 
 @lib.hook('vote', needchan=False)
 def cmd_vote(bot, user, chan, realtarget, *args):
-	if state.votetimer is not None:
+	if state.pointvote is not None:
 		if int(args[0]) in state.voteamounts:
 			state.voteamounts[int(args[0])] += 1
 			bot.msg(user, "Your vote has been recorded.")
@@ -472,7 +513,7 @@ def specialQuestion(oldq):
 
 	if qtype == "!MONTH":
 		newq['question'] = "What month is it currently (in UTC)?"
-		newq['answer'] = time.strftime("%B").lower()
+		newq['answer'] = time.strftime("%B", time.gmtime()).lower()
 	elif qtype == "!MATH+":
 		randnum1 = random.randrange(0, 11)
 		randnum2 = random.randrange(0, 11)