78 lines
1.7 KiB
Python
78 lines
1.7 KiB
Python
from __future__ import print_function
|
|
|
|
import os
|
|
from collections import defaultdict
|
|
|
|
class GameState(object):
|
|
""" Represent current gamestate.
|
|
|
|
players: list of player names + current points
|
|
savedir: dir to save gamestate into, should be overwritable
|
|
board: dict of section with list of questions
|
|
"""
|
|
def __init__(self, savedir):
|
|
self.savedir = savedir
|
|
|
|
self._board = defaultdict(lambda: [None]*5)
|
|
self._players = None
|
|
|
|
if not os.path.exists(self.savedir):
|
|
os.mkdir(self.savedir)
|
|
elif not os.path.isdir(self.savedir):
|
|
raise ValueError("'%s' is not a directory but something else!" % self.savedir)
|
|
|
|
def set_answers(self, section, qnumber, answers):
|
|
self._board[section][qnumber-1] = answers
|
|
|
|
def get_answers(self, section, qnumber):
|
|
return self._board[section][qnumber-1]
|
|
|
|
def set_players(self, players):
|
|
self.players = players
|
|
|
|
def save(self):
|
|
pass
|
|
|
|
@classmethod
|
|
def load(clazz, fpath):
|
|
pass
|
|
|
|
class QuestionAnswers(object):
|
|
def __init__(self, section, qnumber, dj_points=None):
|
|
self.section = section
|
|
self.qnumber = qnumber
|
|
self.nobody_answered = False
|
|
self.dj_points = dj_points
|
|
|
|
self.tries = []
|
|
|
|
def set_dj_points(self, points):
|
|
self.dj_points = points
|
|
|
|
def add_try(self, player, correct):
|
|
self.tries.append((player, correct))
|
|
|
|
def get_winner(self):
|
|
for player, correct in self.tries:
|
|
if correct:
|
|
return player
|
|
return None
|
|
|
|
def points(self):
|
|
return self.qnumber*100
|
|
|
|
def get_dj_points(self):
|
|
return self.dj_points
|
|
|
|
def nobody_knew(self):
|
|
self.nobody_answered = True
|
|
|
|
def got_answered(self):
|
|
return not self.nobody_answered
|
|
|
|
def get_tries(self):
|
|
return self.tries
|
|
|
|
def is_answered(self):
|
|
return self.nobody_answered or any([c for (u, c) in self.tries])
|