/
player.py
114 lines (92 loc) · 2.91 KB
/
player.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
import trueskill
from store import Store
__author__ = 'mocsar'
class Player(object):
def __init__(self, name, rating):
"""
:type ratings: trueskill.Rating
:type name: str
"""
self.rating = rating
self.name = name
def get_score(self):
"""
:rtype : float
"""
return (trueskill.expose(self.rating) + trueskill.global_env().mu) * 3
def __repr__(self):
c = type(self)
args = ('.'.join([c.__module__, c.__name__]), repr(self.name), repr(self.rating))
return '%s(%s, %s)' % args
@classmethod
def create_player(cls, name):
"""
:type name: str
"""
if cls.get_player(name):
raise RuntimeError('Player already exists: %s' % name)
Store.get_store().set_player(Player(name, trueskill.Rating()))
@classmethod
def get_all_names(cls):
"""
:rtype : list[str]
"""
res = None
try:
res = [player.name for player in cls.get_all_players()]
except:
pass
return res
@classmethod
def get_player(cls, name):
"""
:rtype : Player
"""
player = Store.get_store().get_player(name)
return player
@classmethod
def get_all_players(cls):
return Store.get_store().get_all_players()
@classmethod
def rate(cls, names):
"""
:type names: list[str]
"""
if not isinstance(names, list):
raise TypeError("Argument must be a list")
if len(names) != 4:
raise ValueError("Please specify 4 players")
players = []
missing = []
for name in names:
player = cls.get_player(name)
if not player:
missing.append(name)
else:
players.append(player)
if missing:
raise RuntimeError("Players not registered: %s" % missing)
winner1 = players[0]
winner2 = players[1]
loser1 = players[2]
loser2 = players[3]
winners = [winner1.rating, winner2.rating]
losers = [loser1.rating, loser2.rating]
winners, losers = trueskill.rate([winners, losers])
winner1.rating = winners[0]
winner2.rating = winners[1]
loser1.rating = losers[0]
loser2.rating = losers[1]
Store.get_store().set_player(winner1)
Store.get_store().set_player(winner2)
Store.get_store().set_player(loser1)
Store.get_store().set_player(loser2)
Store.get_store().log_results(names)
@classmethod
def get_leaderboard(cls):
"""
Returns with sorted list of the players. The best player is the first in the list.
:rtype : list[Player]
"""
players = cls.get_all_players()
return sorted(players, key=lambda player: trueskill.global_env().expose(player.rating), reverse=True)