xonstat/views/map.py

   1 import logging
   2 import sqlalchemy.sql.functions as func
   3 import sqlalchemy.sql.expression as expr
   4 from collections import namedtuple
   5 from datetime import datetime, timedelta
   6 from pyramid.response import Response
   7 from sqlalchemy import desc
   8 from webhelpers.paginate import Page, PageURL
   9 from xonstat.models import *
  10 from xonstat.util import page_url, html_colors
  11
  12 log = logging.getLogger(__name__)
  13
  14 def _map_index_data(request):
  15     if request.params.has_key('page'):
  16         current_page = request.params['page']
  17     else:
  18         current_page = 1
  19
  20     try:
  21         map_q = DBSession.query(Map).\
  22                 order_by(Map.map_id.desc())
  23
  24         maps = Page(map_q, current_page, items_per_page=10, url=page_url)
  25
  26     except Exception as e:
  27         maps = None
  28
  29     return {'maps':maps, }
  30
  31
  32 def map_index(request):
  33     """
  34     Provides a list of all the current maps.
  35     """
  36     return _map_index_data(request)
  37
  38
  39 def map_index_json(request):
  40     """
  41     Provides a JSON-serialized list of all the current maps.
  42     """
  43     view_data = _map_index_data(request)
  44
  45     maps = [m.to_dict() for m in view_data['maps']]
  46
  47     return maps
  48
  49
  50 def _map_info_data(request):
  51     map_id = request.matchdict['id']
  52
  53     try:
  54         leaderboard_lifetime = int(
  55                 request.registry.settings['xonstat.leaderboard_lifetime'])
  56     except:
  57         leaderboard_lifetime = 30
  58
  59     leaderboard_count = 10
  60     recent_games_count = 20
  61
  62     # captime tuples
  63     Captime = namedtuple('Captime', ['player_id', 'nick_html_colors',
  64         'fastest_cap', 'game_id'])
  65
  66     try:
  67         gmap = DBSession.query(Map).filter_by(map_id=map_id).one()
  68
  69         # recent games on this map
  70         recent_games = DBSession.query(Game, Server, Map, PlayerGameStat).\
  71             filter(Game.server_id==Server.server_id).\
  72             filter(Game.map_id==Map.map_id).\
  73             filter(Game.map_id==map_id).\
  74             filter(PlayerGameStat.game_id==Game.game_id).\
  75             filter(PlayerGameStat.rank==1).\
  76             order_by(expr.desc(Game.start_dt)).all()[0:recent_games_count]
  77
  78         # top players by score
  79         top_scorers = DBSession.query(Player.player_id, Player.nick,
  80                 func.sum(PlayerGameStat.score)).\
  81                 filter(Player.player_id == PlayerGameStat.player_id).\
  82                 filter(Game.game_id == PlayerGameStat.game_id).\
  83                 filter(Game.map_id == map_id).\
  84                 filter(Player.player_id > 2).\
  85                 filter(PlayerGameStat.create_dt >
  86                         (datetime.utcnow() - timedelta(days=leaderboard_lifetime))).\
  87                 order_by(expr.desc(func.sum(PlayerGameStat.score))).\
  88                 group_by(Player.nick).\
  89                 group_by(Player.player_id).all()[0:leaderboard_count]
  90
  91         top_scorers = [(player_id, html_colors(nick), score) \
  92                 for (player_id, nick, score) in top_scorers]
  93
  94         # top players by playing time
  95         top_players = DBSession.query(Player.player_id, Player.nick,
  96                 func.sum(PlayerGameStat.alivetime)).\
  97                 filter(Player.player_id == PlayerGameStat.player_id).\
  98                 filter(Game.game_id == PlayerGameStat.game_id).\
  99                 filter(Game.map_id == map_id).\
 100                 filter(Player.player_id > 2).\
 101                 filter(PlayerGameStat.create_dt >
 102                         (datetime.utcnow() - timedelta(days=leaderboard_lifetime))).\
 103                 order_by(expr.desc(func.sum(PlayerGameStat.alivetime))).\
 104                 group_by(Player.nick).\
 105                 group_by(Player.player_id).all()[0:leaderboard_count]
 106
 107         top_players = [(player_id, html_colors(nick), score) \
 108                 for (player_id, nick, score) in top_players]
 109
 110         # top servers using/playing this map
 111         top_servers = DBSession.query(Server.server_id, Server.name,
 112                 func.count(Game.game_id)).\
 113                 filter(Game.server_id == Server.server_id).\
 114                 filter(Game.map_id == map_id).\
 115                 filter(Game.create_dt >
 116                         (datetime.utcnow() - timedelta(days=leaderboard_lifetime))).\
 117                 order_by(expr.desc(func.count(Game.game_id))).\
 118                 group_by(Server.name).\
 119                 group_by(Server.server_id).all()[0:leaderboard_count]
 120
 121         # top captimes
 122         captimes_raw = DBSession.query(Player.player_id, Player.nick,
 123             PlayerCaptime.fastest_cap, PlayerCaptime.game_id).\
 124                 filter(PlayerCaptime.map_id == map_id).\
 125                 filter(Player.player_id == PlayerCaptime.player_id).\
 126                 order_by(PlayerCaptime.fastest_cap).\
 127                 limit(10).all()
 128
 129         captimes = [Captime(c.player_id, html_colors(c.nick),
 130             c.fastest_cap, c.game_id) for c in captimes_raw]
 131
 132     except Exception as e:
 133         gmap = None
 134     return {'gmap':gmap,
 135             'recent_games':recent_games,
 136             'top_scorers':top_scorers,
 137             'top_players':top_players,
 138             'top_servers':top_servers,
 139             'captimes':captimes,
 140             }
 141
 142
 143 def map_info(request):
 144     """
 145     List the information stored about a given map.
 146     """
 147     mapinfo_data =  _map_info_data(request)
 148
 149     # FIXME: code clone, should get these from _map_info_data
 150     leaderboard_count = 10
 151     recent_games_count = 20
 152
 153     for i in range(recent_games_count-len(mapinfo_data['recent_games'])):
 154         mapinfo_data['recent_games'].append(('-', '-', '-', '-'))
 155
 156     for i in range(leaderboard_count-len(mapinfo_data['top_scorers'])):
 157         mapinfo_data['top_scorers'].append(('-', '-', '-'))
 158
 159     for i in range(leaderboard_count-len(mapinfo_data['top_players'])):
 160         mapinfo_data['top_players'].append(('-', '-', '-'))
 161
 162     for i in range(leaderboard_count-len(mapinfo_data['top_servers'])):
 163         mapinfo_data['top_servers'].append(('-', '-', '-'))
 164
 165     return mapinfo_data
 166
 167
 168 def map_info_json(request):
 169     """
 170     List the information stored about a given map. JSON.
 171     """
 172     return [{'status':'not implemented'}]