-import logging\r
-import sqlalchemy.sql.functions as func\r
-import sqlalchemy.sql.expression as expr\r
-from datetime import datetime, timedelta\r
-from pyramid.response import Response\r
-from sqlalchemy import desc\r
-from webhelpers.paginate import Page, PageURL\r
-from xonstat.models import *\r
-from xonstat.util import page_url\r
-\r
-log = logging.getLogger(__name__)\r
-\r
-def map_index(request):\r
- """\r
- Provides a list of all the current maps. \r
- """\r
- if request.params.has_key('page'):\r
- current_page = request.params['page']\r
- else:\r
- current_page = 1\r
-\r
- try:\r
- map_q = DBSession.query(Map).\\r
- order_by(Map.map_id.desc())\r
-\r
- maps = Page(map_q, current_page, items_per_page=10, url=page_url)\r
-\r
- except Exception as e:\r
- maps = None\r
-\r
- return {'maps':maps, }\r
-\r
-\r
-def map_info(request):\r
- """\r
- List the information stored about a given map. \r
- """\r
- map_id = request.matchdict['id']\r
-\r
- try: \r
- leaderboard_lifetime = int(\r
- request.registry.settings['xonstat.leaderboard_lifetime'])\r
- except:\r
- leaderboard_lifetime = 30\r
-\r
- leaderboard_count = 10\r
- recent_games_count = 20\r
-\r
- try:\r
- gmap = DBSession.query(Map).filter_by(map_id=map_id).one()\r
-\r
- # recent games on this map\r
- recent_games = DBSession.query(Game, Server, Map, PlayerGameStat).\\r
- filter(Game.server_id==Server.server_id).\\r
- filter(Game.map_id==Map.map_id).\\r
- filter(Game.map_id==map_id).\\r
- filter(PlayerGameStat.game_id==Game.game_id).\\r
- filter(PlayerGameStat.rank==1).\\r
- order_by(expr.desc(Game.start_dt)).all()[0:recent_games_count]\r
-\r
- for i in range(recent_games_count-len(recent_games)):\r
- recent_games.append(('-', '-', '-', '-'))\r
-\r
-\r
- # top players by score\r
- top_scorers = DBSession.query(Player.player_id, Player.nick,\r
- func.sum(PlayerGameStat.score)).\\r
- filter(Player.player_id == PlayerGameStat.player_id).\\r
- filter(Game.game_id == PlayerGameStat.game_id).\\r
- filter(Game.map_id == map_id).\\r
- filter(Player.player_id > 2).\\r
- filter(PlayerGameStat.create_dt > \r
- (datetime.utcnow() - timedelta(days=leaderboard_lifetime))).\\r
- order_by(expr.desc(func.sum(PlayerGameStat.score))).\\r
- group_by(Player.nick).\\r
- group_by(Player.player_id).all()[0:10]\r
-\r
- top_scorers = [(player_id, html_colors(nick), score) \\r
- for (player_id, nick, score) in top_scorers]\r
-\r
- for i in range(leaderboard_count-len(top_scorers)):\r
- top_scorers.append(('-', '-', '-'))\r
-\r
- # top players by playing time\r
- top_players = DBSession.query(Player.player_id, Player.nick, \r
- func.sum(PlayerGameStat.alivetime)).\\r
- filter(Player.player_id == PlayerGameStat.player_id).\\r
- filter(Game.game_id == PlayerGameStat.game_id).\\r
- filter(Game.map_id == map_id).\\r
- filter(Player.player_id > 2).\\r
- filter(PlayerGameStat.create_dt > \r
- (datetime.utcnow() - timedelta(days=leaderboard_lifetime))).\\r
- order_by(expr.desc(func.sum(PlayerGameStat.alivetime))).\\r
- group_by(Player.nick).\\r
- group_by(Player.player_id).all()[0:10]\r
-\r
- top_players = [(player_id, html_colors(nick), score) \\r
- for (player_id, nick, score) in top_players]\r
-\r
- for i in range(leaderboard_count-len(top_players)):\r
- top_players.append(('-', '-', '-'))\r
-\r
- # top servers using/playing this map\r
- top_servers = DBSession.query(Server.server_id, Server.name, \r
- func.count(Game.game_id)).\\r
- filter(Game.server_id == Server.server_id).\\r
- filter(Game.map_id == map_id).\\r
- filter(Game.create_dt > \r
- (datetime.utcnow() - timedelta(days=leaderboard_lifetime))).\\r
- order_by(expr.desc(func.count(Game.game_id))).\\r
- group_by(Server.name).\\r
- group_by(Server.server_id).all()[0:10]\r
-\r
- for i in range(leaderboard_count-len(top_servers)):\r
- top_servers.append(('-', '-', '-'))\r
-\r
- except Exception as e:\r
- gmap = None\r
- return {'gmap':gmap,\r
- 'recent_games':recent_games,\r
- 'top_scorers':top_scorers,\r
- 'top_players':top_players,\r
- 'top_servers':top_servers,\r
- }\r
+import logging
+from collections import namedtuple
+from datetime import datetime, timedelta
+
+import sqlalchemy.sql.expression as expr
+import sqlalchemy.sql.functions as func
+from pyramid.httpexceptions import HTTPNotFound
+from sqlalchemy import func as fg
+from webhelpers.paginate import Page
+from xonstat.models import DBSession, Server, Map, Game, PlayerGameStat, Player, PlayerCaptime
+from xonstat.models.map import MapCapTime
+from xonstat.util import page_url, html_colors
+from xonstat.views.helpers import RecentGame, recent_games_q
+
+log = logging.getLogger(__name__)
+
+# Defaults
+INDEX_COUNT = 20
+LEADERBOARD_LIFETIME = 30
+
+
+class MapIndex(object):
+ """Returns a list of maps."""
+
+ def __init__(self, request):
+ """Common parameter parsing."""
+ self.request = request
+ self.last = request.params.get("last", None)
+
+ # all views share this data, so we'll pre-calculate
+ self.maps = self.map_index()
+
+ def map_index(self):
+ """Returns the raw data shared by all renderers."""
+ try:
+ map_q = DBSession.query(Map)
+
+ if self.last:
+ map_q = map_q.filter(Map.map_id < self.last)
+
+ map_q = map_q.order_by(Map.map_id.desc()).limit(INDEX_COUNT)
+ maps = map_q.all()
+
+ except Exception as e:
+ log.debug(e)
+ raise HTTPNotFound
+
+ return maps
+
+ def html(self):
+ """For rendering this data using something HTML-based."""
+ # build the query string
+ query = {}
+ if len(self.maps) > 1:
+ query['last'] = self.maps[-1].map_id
+
+ return {
+ 'maps': self.maps,
+ 'query': query,
+ }
+
+ def json(self):
+ """For rendering this data using JSON."""
+ return {
+ 'maps': [m.to_dict() for m in self.maps],
+ 'last': self.last,
+ }
+
+
+class MapInfoBase(object):
+ """Base class for all map-based views with a map_id parameter in them."""
+
+ def __init__(self, request, limit=None, last=None):
+ """Common parameter parsing."""
+ self.request = request
+ self.map_id = request.matchdict.get("id", None)
+
+ raw_lifetime = request.registry.settings.get('xonstat.leaderboard_lifetime',
+ LEADERBOARD_LIFETIME)
+ self.lifetime = int(raw_lifetime)
+
+ self.limit = request.params.get("limit", limit)
+ self.last = request.params.get("last", last)
+ self.now = datetime.utcnow()
+
+
+class MapTopScorers(MapInfoBase):
+ """Returns the top scorers on a given map."""
+
+ def __init__(self, request, limit=INDEX_COUNT, last=None):
+ """Common parameter parsing."""
+ super(MapTopScorers, self).__init__(request, limit, last)
+ self.top_scorers = self.get_top_scorers()
+
+ def get_top_scorers(self):
+ """Top players by score. Shared by all renderers."""
+ cutoff = self.now - timedelta(days=self.lifetime)
+
+ top_scorers_q = DBSession.query(
+ fg.row_number().over(order_by=expr.desc(func.sum(PlayerGameStat.score))).label("rank"),
+ Player.player_id, Player.nick, func.sum(PlayerGameStat.score).label("total_score"))\
+ .filter(Player.player_id == PlayerGameStat.player_id)\
+ .filter(Game.game_id == PlayerGameStat.game_id)\
+ .filter(Game.map_id == self.map_id)\
+ .filter(Player.player_id > 2)\
+ .filter(PlayerGameStat.create_dt > cutoff)\
+ .order_by(expr.desc(func.sum(PlayerGameStat.score)))\
+ .group_by(Player.nick)\
+ .group_by(Player.player_id)
+
+ if self.last:
+ top_scorers_q = top_scorers_q.offset(self.last)
+
+ if self.limit:
+ top_scorers_q = top_scorers_q.limit(self.limit)
+
+ top_scorers = top_scorers_q.all()
+
+ return top_scorers
+
+ def html(self):
+ """Returns an HTML-ready representation."""
+ TopScorer = namedtuple("TopScorer", ["rank", "player_id", "nick", "total_score"])
+
+ top_scorers = [TopScorer(ts.rank, ts.player_id, html_colors(ts.nick), ts.total_score)
+ for ts in self.top_scorers]
+
+ # build the query string
+ query = {}
+ if len(top_scorers) > 1:
+ query['last'] = top_scorers[-1].rank
+
+ return {
+ "map_id": self.map_id,
+ "top_scorers": top_scorers,
+ "lifetime": self.lifetime,
+ "query": query,
+ }
+
+ def json(self):
+ """For rendering this data using JSON."""
+ top_scorers = [{
+ "rank": ts.rank,
+ "player_id": ts.player_id,
+ "nick": ts.nick,
+ "score": ts.total_score,
+ } for ts in self.top_scorers]
+
+ return {
+ "map_id": self.map_id,
+ "top_scorers": top_scorers,
+ }
+
+
+class MapTopPlayers(MapInfoBase):
+ """Returns the top players by time on a given map."""
+
+ def __init__(self, request, limit=INDEX_COUNT, last=None):
+ """Common parameter parsing."""
+ super(MapTopPlayers, self).__init__(request, limit, last)
+ self.top_players = self.get_top_players()
+
+ def get_top_players(self):
+ """Top players by score. Shared by all renderers."""
+ cutoff = self.now - timedelta(days=self.lifetime)
+
+ top_players_q = DBSession.query(
+ fg.row_number().over(order_by=expr.desc(func.sum(PlayerGameStat.alivetime))).label("rank"),
+ Player.player_id, Player.nick, func.sum(PlayerGameStat.alivetime).label("alivetime"))\
+ .filter(Player.player_id == PlayerGameStat.player_id)\
+ .filter(Game.game_id == PlayerGameStat.game_id)\
+ .filter(Game.map_id == self.map_id)\
+ .filter(Player.player_id > 2)\
+ .filter(PlayerGameStat.create_dt > cutoff)\
+ .order_by(expr.desc(func.sum(PlayerGameStat.alivetime)))\
+ .group_by(Player.nick)\
+ .group_by(Player.player_id)
+
+ if self.last:
+ top_players_q = top_players_q.offset(self.last)
+
+ if self.limit:
+ top_players_q = top_players_q.limit(self.limit)
+
+ top_players = top_players_q.all()
+
+ return top_players
+
+ def html(self):
+ """Returns the HTML-ready representation."""
+ TopPlayer = namedtuple("TopPlayer", ["rank", "player_id", "nick", "alivetime"])
+
+ top_players = [TopPlayer(tp.rank, tp.player_id, html_colors(tp.nick), tp.alivetime)
+ for tp in self.top_players]
+
+ # build the query string
+ query = {}
+ if len(top_players) > 1:
+ query['last'] = top_players[-1].rank
+
+ return {
+ "map_id": self.map_id,
+ "top_players": top_players,
+ "lifetime": self.lifetime,
+ "last": query.get("last", None),
+ "query": query,
+ }
+
+ def json(self):
+ """For rendering this data using JSON."""
+ top_players = [{
+ "rank": ts.rank,
+ "player_id": ts.player_id,
+ "nick": ts.nick,
+ "time": ts.alivetime.total_seconds(),
+ } for ts in self.top_players]
+
+ return {
+ "map_id": self.map_id,
+ "top_players": top_players,
+ }
+
+
+class MapTopServers(MapInfoBase):
+ """Returns the top servers by the number of times they've played a given map."""
+
+ def __init__(self, request, limit=INDEX_COUNT, last=None):
+ """Common parameter parsing."""
+ super(MapTopServers, self).__init__(request, limit, last)
+ self.top_servers = self.get_top_servers()
+
+ def get_top_servers(self):
+ """Top servers by the number of times they have played the map. Shared by all renderers."""
+ cutoff = self.now - timedelta(days=self.lifetime)
+
+ top_servers_q = DBSession.query(
+ fg.row_number().over(order_by=expr.desc(func.count(Game.game_id))).label("rank"),
+ Server.server_id, Server.name, func.count(Game.game_id).label("games"))\
+ .filter(Game.server_id == Server.server_id)\
+ .filter(Game.map_id == self.map_id)\
+ .filter(Game.create_dt > cutoff)\
+ .order_by(expr.desc(func.count(Game.game_id)))\
+ .group_by(Server.name)\
+ .group_by(Server.server_id)
+
+ if self.last:
+ top_servers_q = top_servers_q.offset(self.last)
+
+ if self.limit:
+ top_servers_q = top_servers_q.limit(self.limit)
+
+ top_servers = top_servers_q.all()
+
+ return top_servers
+
+ def html(self):
+ """Returns the HTML-ready representation."""
+ TopServer = namedtuple("TopServer", ["rank", "server_id", "server_name", "games"])
+
+ top_servers = [TopServer(ts.rank, ts.server_id, ts.name, ts.games)
+ for ts in self.top_servers]
+
+ # build the query string
+ query = {}
+ if len(top_servers) > 1:
+ query['last'] = top_servers[-1].rank
+
+ return {
+ "map_id": self.map_id,
+ "top_servers": top_servers,
+ "lifetime": self.lifetime,
+ "last": query.get("last", None),
+ "query": query,
+ }
+
+ def json(self):
+ """For rendering this data using JSON."""
+ top_servers = [{
+ "rank": ts.rank,
+ "server_id": ts.server_id,
+ "server_name": ts.server_name,
+ "games": ts.games,
+ } for ts in self.top_servers]
+
+ return {
+ "map_id": self.map_id,
+ "top_servers": top_servers,
+ }
+
+
+def _map_info_data(request):
+ map_id = int(request.matchdict['id'])
+
+ try:
+ leaderboard_lifetime = int(
+ request.registry.settings['xonstat.leaderboard_lifetime'])
+ except:
+ leaderboard_lifetime = 30
+
+ leaderboard_count = 10
+ recent_games_count = 20
+
+ # captime tuples
+ Captime = namedtuple('Captime', ['player_id', 'nick_html_colors',
+ 'fastest_cap', 'game_id'])
+
+ try:
+ gmap = DBSession.query(Map).filter_by(map_id=map_id).one()
+
+ # recent games played in descending order
+ rgs = recent_games_q(map_id=map_id).limit(recent_games_count).all()
+ recent_games = [RecentGame(row) for row in rgs]
+
+ # top players by score
+ top_scorers = DBSession.query(Player.player_id, Player.nick,
+ func.sum(PlayerGameStat.score)).\
+ filter(Player.player_id == PlayerGameStat.player_id).\
+ filter(Game.game_id == PlayerGameStat.game_id).\
+ filter(Game.map_id == map_id).\
+ filter(Player.player_id > 2).\
+ filter(PlayerGameStat.create_dt >
+ (datetime.utcnow() - timedelta(days=leaderboard_lifetime))).\
+ order_by(expr.desc(func.sum(PlayerGameStat.score))).\
+ group_by(Player.nick).\
+ group_by(Player.player_id).all()[0:leaderboard_count]
+
+ top_scorers = [(player_id, html_colors(nick), score) \
+ for (player_id, nick, score) in top_scorers]
+
+ # top players by playing time
+ top_players = DBSession.query(Player.player_id, Player.nick,
+ func.sum(PlayerGameStat.alivetime)).\
+ filter(Player.player_id == PlayerGameStat.player_id).\
+ filter(Game.game_id == PlayerGameStat.game_id).\
+ filter(Game.map_id == map_id).\
+ filter(Player.player_id > 2).\
+ filter(PlayerGameStat.create_dt >
+ (datetime.utcnow() - timedelta(days=leaderboard_lifetime))).\
+ order_by(expr.desc(func.sum(PlayerGameStat.alivetime))).\
+ group_by(Player.nick).\
+ group_by(Player.player_id).all()[0:leaderboard_count]
+
+ top_players = [(player_id, html_colors(nick), score) \
+ for (player_id, nick, score) in top_players]
+
+ # top servers using/playing this map
+ top_servers = DBSession.query(Server.server_id, Server.name,
+ func.count(Game.game_id)).\
+ filter(Game.server_id == Server.server_id).\
+ filter(Game.map_id == map_id).\
+ filter(Game.create_dt >
+ (datetime.utcnow() - timedelta(days=leaderboard_lifetime))).\
+ order_by(expr.desc(func.count(Game.game_id))).\
+ group_by(Server.name).\
+ group_by(Server.server_id).all()[0:leaderboard_count]
+
+ # TODO make this a configuration parameter to be set in the settings
+ # top captimes
+ captimes_raw = DBSession.query(Player.player_id, Player.nick,
+ PlayerCaptime.fastest_cap, PlayerCaptime.game_id).\
+ filter(PlayerCaptime.map_id == map_id).\
+ filter(Player.player_id == PlayerCaptime.player_id).\
+ order_by(PlayerCaptime.fastest_cap).\
+ limit(10).\
+ all()
+
+ captimes = [Captime(c.player_id, html_colors(c.nick),
+ c.fastest_cap, c.game_id) for c in captimes_raw]
+
+ except Exception as e:
+ gmap = None
+ return {'gmap':gmap,
+ 'recent_games':recent_games,
+ 'top_scorers':top_scorers,
+ 'top_players':top_players,
+ 'top_servers':top_servers,
+ 'captimes':captimes,
+ }
+
+
+def map_info(request):
+ """
+ List the information stored about a given map.
+ """
+ mapinfo_data = _map_info_data(request)
+
+ # FIXME: code clone, should get these from _map_info_data
+ leaderboard_count = 10
+ recent_games_count = 20
+
+ for i in range(leaderboard_count-len(mapinfo_data['top_scorers'])):
+ mapinfo_data['top_scorers'].append(('-', '-', '-'))
+
+ for i in range(leaderboard_count-len(mapinfo_data['top_players'])):
+ mapinfo_data['top_players'].append(('-', '-', '-'))
+
+ for i in range(leaderboard_count-len(mapinfo_data['top_servers'])):
+ mapinfo_data['top_servers'].append(('-', '-', '-'))
+
+ return mapinfo_data
+
+
+def map_info_json(request):
+ """
+ List the information stored about a given map. JSON.
+ """
+ return [{'status':'not implemented'}]
+
+
+def map_captimes_data(request):
+ map_id = int(request.matchdict['id'])
+
+ current_page = request.params.get('page', 1)
+
+ try:
+ mmap = DBSession.query(Map).filter_by(map_id=map_id).one()
+
+ mct_q = DBSession.query(PlayerCaptime.fastest_cap, PlayerCaptime.create_dt,
+ PlayerCaptime.player_id, PlayerCaptime.game_id,
+ Game.server_id, Server.name.label('server_name'),
+ PlayerGameStat.nick.label('player_nick')).\
+ filter(PlayerCaptime.map_id==map_id).\
+ filter(PlayerCaptime.game_id==Game.game_id).\
+ filter(PlayerCaptime.map_id==Map.map_id).\
+ filter(Game.server_id==Server.server_id).\
+ filter(PlayerCaptime.player_id==PlayerGameStat.player_id).\
+ filter(PlayerCaptime.game_id==PlayerGameStat.game_id).\
+ order_by(expr.asc(PlayerCaptime.fastest_cap))
+
+ except Exception as e:
+ raise HTTPNotFound
+
+ map_captimes = Page(mct_q, current_page, items_per_page=20, url=page_url)
+
+ map_captimes.items = [MapCapTime(row) for row in map_captimes.items]
+
+ return {
+ 'map_id':map_id,
+ 'map':mmap,
+ 'captimes':map_captimes,
+ }
+
+def map_captimes(request):
+ return map_captimes_data(request)
+
+def map_captimes_json(request):
+ current_page = request.params.get('page', 1)
+ data = map_captimes_data(request)
+
+ return {
+ "map": data["map"].to_dict(),
+ "captimes": [e.to_dict() for e in data["captimes"].items],
+ "page": current_page,
+ }