2 from collections import namedtuple
3 from datetime import datetime, timedelta
5 import sqlalchemy.sql.expression as expr
6 import sqlalchemy.sql.functions as func
7 from pyramid.httpexceptions import HTTPNotFound
8 from sqlalchemy import func as fg
9 from webhelpers.paginate import Page
10 from xonstat.models import DBSession, Server, Map, Game, PlayerGameStat, Player, PlayerCaptime
11 from xonstat.models.map import MapCapTime
12 from xonstat.util import page_url, html_colors
13 from xonstat.views.helpers import RecentGame, recent_games_q
15 log = logging.getLogger(__name__)
19 LEADERBOARD_LIFETIME = 30
22 class MapIndex(object):
23 """Returns a list of maps."""
25 def __init__(self, request):
26 """Common parameter parsing."""
27 self.request = request
28 self.last = request.params.get("last", None)
30 # all views share this data, so we'll pre-calculate
31 self.maps = self.map_index()
34 """Returns the raw data shared by all renderers."""
36 map_q = DBSession.query(Map)
39 map_q = map_q.filter(Map.map_id < self.last)
41 map_q = map_q.order_by(Map.map_id.desc()).limit(INDEX_COUNT)
44 except Exception as e:
51 """For rendering this data using something HTML-based."""
52 # build the query string
54 if len(self.maps) > 1:
55 query['last'] = self.maps[-1].map_id
63 """For rendering this data using JSON."""
65 'maps': [m.to_dict() for m in self.maps],
70 class MapInfoBase(object):
71 """Base class for all map-based views with a map_id parameter in them."""
73 def __init__(self, request, limit=None, last=None):
74 """Common parameter parsing."""
75 self.request = request
76 self.map_id = request.matchdict.get("id", None)
78 raw_lifetime = request.registry.settings.get('xonstat.leaderboard_lifetime',
80 self.lifetime = int(raw_lifetime)
82 self.limit = request.params.get("limit", limit)
83 self.last = request.params.get("last", last)
84 self.now = datetime.utcnow()
87 class MapTopScorers(MapInfoBase):
88 """Returns the top scorers on a given map."""
90 def __init__(self, request, limit=INDEX_COUNT, last=None):
91 """Common parameter parsing."""
92 super(MapTopScorers, self).__init__(request, limit, last)
93 self.top_scorers = self.get_top_scorers()
95 def get_top_scorers(self):
96 """Top players by score. Shared by all renderers."""
97 cutoff = self.now - timedelta(days=self.lifetime)
100 top_scorers_q = DBSession.query(
101 fg.row_number().over(order_by=expr.desc(func.sum(PlayerGameStat.score))).label("rank"),
102 Player.player_id, Player.nick, func.sum(PlayerGameStat.score).label("total_score"))\
103 .filter(Player.player_id == PlayerGameStat.player_id)\
104 .filter(Game.game_id == PlayerGameStat.game_id)\
105 .filter(Game.map_id == self.map_id)\
106 .filter(Player.player_id > 2)\
107 .filter(PlayerGameStat.create_dt > cutoff)\
108 .order_by(expr.desc(func.sum(PlayerGameStat.score)))\
109 .group_by(Player.nick)\
110 .group_by(Player.player_id)
113 top_scorers_q = top_scorers_q.offset(self.last)
116 top_scorers_q = top_scorers_q.limit(self.limit)
118 top_scorers = top_scorers_q.all()
121 except Exception as e:
126 """Returns an HTML-ready representation."""
127 TopScorer = namedtuple("TopScorer", ["rank", "player_id", "nick", "total_score"])
129 top_scorers = [TopScorer(ts.rank, ts.player_id, html_colors(ts.nick), ts.total_score)
130 for ts in self.top_scorers]
132 # build the query string
134 if len(top_scorers) > 1:
135 query['last'] = top_scorers[-1].rank
138 "map_id": self.map_id,
139 "top_scorers": top_scorers,
140 "lifetime": self.lifetime,
145 """For rendering this data using JSON."""
148 "player_id": ts.player_id,
150 "score": ts.total_score,
151 } for ts in self.top_scorers]
154 "map_id": self.map_id,
155 "top_scorers": top_scorers,
159 class MapTopPlayers(MapInfoBase):
160 """Returns the top players by time on a given map."""
162 def __init__(self, request, limit=INDEX_COUNT, last=None):
163 """Common parameter parsing."""
164 super(MapTopPlayers, self).__init__(request, limit, last)
165 self.top_players = self.get_top_players()
167 def get_top_players(self):
168 """Top players by score. Shared by all renderers."""
169 cutoff = self.now - timedelta(days=self.lifetime)
172 top_players_q = DBSession.query(
173 fg.row_number().over(order_by=expr.desc(func.sum(PlayerGameStat.alivetime))).label("rank"),
174 Player.player_id, Player.nick, func.sum(PlayerGameStat.alivetime).label("alivetime"))\
175 .filter(Player.player_id == PlayerGameStat.player_id)\
176 .filter(Game.game_id == PlayerGameStat.game_id)\
177 .filter(Game.map_id == self.map_id)\
178 .filter(Player.player_id > 2)\
179 .filter(PlayerGameStat.create_dt > cutoff)\
180 .order_by(expr.desc(func.sum(PlayerGameStat.alivetime)))\
181 .group_by(Player.nick)\
182 .group_by(Player.player_id)
185 top_players_q = top_players_q.offset(self.last)
188 top_players_q = top_players_q.limit(self.limit)
190 top_players = top_players_q.all()
193 except Exception as e:
198 """Returns the HTML-ready representation."""
199 TopPlayer = namedtuple("TopPlayer", ["rank", "player_id", "nick", "alivetime"])
201 top_players = [TopPlayer(tp.rank, tp.player_id, html_colors(tp.nick), tp.alivetime)
202 for tp in self.top_players]
204 # build the query string
206 if len(top_players) > 1:
207 query['last'] = top_players[-1].rank
210 "map_id": self.map_id,
211 "top_players": top_players,
212 "lifetime": self.lifetime,
213 "last": query.get("last", None),
218 """For rendering this data using JSON."""
221 "player_id": ts.player_id,
223 "time": ts.alivetime.total_seconds(),
224 } for ts in self.top_players]
227 "map_id": self.map_id,
228 "top_players": top_players,
232 class MapTopServers(MapInfoBase):
233 """Returns the top servers by the number of times they've played a given map."""
235 def __init__(self, request, limit=INDEX_COUNT, last=None):
236 """Common parameter parsing."""
237 super(MapTopServers, self).__init__(request, limit, last)
238 self.top_servers = self.get_top_servers()
240 def get_top_servers(self):
241 """Top servers by the number of times they have played the map. Shared by all renderers."""
242 cutoff = self.now - timedelta(days=self.lifetime)
245 top_servers_q = DBSession.query(
246 fg.row_number().over(order_by=expr.desc(func.count(Game.game_id))).label("rank"),
247 Server.server_id, Server.name, func.count(Game.game_id).label("games"))\
248 .filter(Game.server_id == Server.server_id)\
249 .filter(Game.map_id == self.map_id)\
250 .filter(Game.create_dt > cutoff)\
251 .order_by(expr.desc(func.count(Game.game_id)))\
252 .group_by(Server.name)\
253 .group_by(Server.server_id)
256 top_servers_q = top_servers_q.offset(self.last)
259 top_servers_q = top_servers_q.limit(self.limit)
261 top_servers = top_servers_q.all()
264 except Exception as e:
269 """Returns the HTML-ready representation."""
270 TopServer = namedtuple("TopServer", ["rank", "server_id", "server_name", "games"])
272 top_servers = [TopServer(ts.rank, ts.server_id, ts.name, ts.games)
273 for ts in self.top_servers]
275 # build the query string
277 if len(top_servers) > 1:
278 query['last'] = top_servers[-1].rank
281 "map_id": self.map_id,
282 "top_servers": top_servers,
283 "lifetime": self.lifetime,
284 "last": query.get("last", None),
289 """For rendering this data using JSON."""
292 "server_id": ts.server_id,
293 "server_name": ts.server_name,
295 } for ts in self.top_servers]
298 "map_id": self.map_id,
299 "top_servers": top_servers,
303 def _map_info_data(request):
304 map_id = int(request.matchdict['id'])
307 leaderboard_lifetime = int(
308 request.registry.settings['xonstat.leaderboard_lifetime'])
310 leaderboard_lifetime = 30
312 leaderboard_count = 10
313 recent_games_count = 20
316 Captime = namedtuple('Captime', ['player_id', 'nick_html_colors',
317 'fastest_cap', 'game_id'])
320 gmap = DBSession.query(Map).filter_by(map_id=map_id).one()
322 # recent games played in descending order
323 rgs = recent_games_q(map_id=map_id).limit(recent_games_count).all()
324 recent_games = [RecentGame(row) for row in rgs]
326 # top players by score
327 top_scorers = DBSession.query(Player.player_id, Player.nick,
328 func.sum(PlayerGameStat.score)).\
329 filter(Player.player_id == PlayerGameStat.player_id).\
330 filter(Game.game_id == PlayerGameStat.game_id).\
331 filter(Game.map_id == map_id).\
332 filter(Player.player_id > 2).\
333 filter(PlayerGameStat.create_dt >
334 (datetime.utcnow() - timedelta(days=leaderboard_lifetime))).\
335 order_by(expr.desc(func.sum(PlayerGameStat.score))).\
336 group_by(Player.nick).\
337 group_by(Player.player_id).all()[0:leaderboard_count]
339 top_scorers = [(player_id, html_colors(nick), score) \
340 for (player_id, nick, score) in top_scorers]
342 # top players by playing time
343 top_players = DBSession.query(Player.player_id, Player.nick,
344 func.sum(PlayerGameStat.alivetime)).\
345 filter(Player.player_id == PlayerGameStat.player_id).\
346 filter(Game.game_id == PlayerGameStat.game_id).\
347 filter(Game.map_id == map_id).\
348 filter(Player.player_id > 2).\
349 filter(PlayerGameStat.create_dt >
350 (datetime.utcnow() - timedelta(days=leaderboard_lifetime))).\
351 order_by(expr.desc(func.sum(PlayerGameStat.alivetime))).\
352 group_by(Player.nick).\
353 group_by(Player.player_id).all()[0:leaderboard_count]
355 top_players = [(player_id, html_colors(nick), score) \
356 for (player_id, nick, score) in top_players]
358 # top servers using/playing this map
359 top_servers = DBSession.query(Server.server_id, Server.name,
360 func.count(Game.game_id)).\
361 filter(Game.server_id == Server.server_id).\
362 filter(Game.map_id == map_id).\
363 filter(Game.create_dt >
364 (datetime.utcnow() - timedelta(days=leaderboard_lifetime))).\
365 order_by(expr.desc(func.count(Game.game_id))).\
366 group_by(Server.name).\
367 group_by(Server.server_id).all()[0:leaderboard_count]
369 # TODO make this a configuration parameter to be set in the settings
371 captimes_raw = DBSession.query(Player.player_id, Player.nick,
372 PlayerCaptime.fastest_cap, PlayerCaptime.game_id).\
373 filter(PlayerCaptime.map_id == map_id).\
374 filter(Player.player_id == PlayerCaptime.player_id).\
375 order_by(PlayerCaptime.fastest_cap).\
379 captimes = [Captime(c.player_id, html_colors(c.nick),
380 c.fastest_cap, c.game_id) for c in captimes_raw]
382 except Exception as e:
385 'recent_games':recent_games,
386 'top_scorers':top_scorers,
387 'top_players':top_players,
388 'top_servers':top_servers,
393 def map_info(request):
395 List the information stored about a given map.
397 mapinfo_data = _map_info_data(request)
399 # FIXME: code clone, should get these from _map_info_data
400 leaderboard_count = 10
401 recent_games_count = 20
403 for i in range(leaderboard_count-len(mapinfo_data['top_scorers'])):
404 mapinfo_data['top_scorers'].append(('-', '-', '-'))
406 for i in range(leaderboard_count-len(mapinfo_data['top_players'])):
407 mapinfo_data['top_players'].append(('-', '-', '-'))
409 for i in range(leaderboard_count-len(mapinfo_data['top_servers'])):
410 mapinfo_data['top_servers'].append(('-', '-', '-'))
415 def map_info_json(request):
417 List the information stored about a given map. JSON.
419 return [{'status':'not implemented'}]
422 def map_captimes_data(request):
423 map_id = int(request.matchdict['id'])
425 current_page = request.params.get('page', 1)
428 mmap = DBSession.query(Map).filter_by(map_id=map_id).one()
430 mct_q = DBSession.query(PlayerCaptime.fastest_cap, PlayerCaptime.create_dt,
431 PlayerCaptime.player_id, PlayerCaptime.game_id,
432 Game.server_id, Server.name.label('server_name'),
433 PlayerGameStat.nick.label('player_nick')).\
434 filter(PlayerCaptime.map_id==map_id).\
435 filter(PlayerCaptime.game_id==Game.game_id).\
436 filter(PlayerCaptime.map_id==Map.map_id).\
437 filter(Game.server_id==Server.server_id).\
438 filter(PlayerCaptime.player_id==PlayerGameStat.player_id).\
439 filter(PlayerCaptime.game_id==PlayerGameStat.game_id).\
440 order_by(expr.asc(PlayerCaptime.fastest_cap))
442 except Exception as e:
445 map_captimes = Page(mct_q, current_page, items_per_page=20, url=page_url)
447 map_captimes.items = [MapCapTime(row) for row in map_captimes.items]
452 'captimes':map_captimes,
455 def map_captimes(request):
456 return map_captimes_data(request)
458 def map_captimes_json(request):
459 current_page = request.params.get('page', 1)
460 data = map_captimes_data(request)
463 "map": data["map"].to_dict(),
464 "captimes": [e.to_dict() for e in data["captimes"].items],
465 "page": current_page,