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 webhelpers.paginate import Page
   7 from xonstat.models import *
   8 from xonstat.util import page_url, html_colors
   9 from xonstat.views.helpers import RecentGame, recent_games_q
  10
  11 log = logging.getLogger(__name__)
  12
  13 def _map_index_data(request):
  14     if request.params.has_key('page'):
  15         current_page = request.params['page']
  16     else:
  17         current_page = 1
  18
  19     try:
  20         map_q = DBSession.query(Map).\
  21                 order_by(Map.map_id.desc())
  22
  23         maps = Page(map_q, current_page, items_per_page=25, url=page_url)
  24
  25     except Exception as e:
  26         maps = None
  27
  28     return {'maps':maps, }
  29
  30
  31 def map_index(request):
  32     """
  33     Provides a list of all the current maps.
  34     """
  35     return _map_index_data(request)
  36
  37
  38 def map_index_json(request):
  39     """
  40     Provides a JSON-serialized list of all the current maps.
  41     """
  42     view_data = _map_index_data(request)
  43
  44     maps = [m.to_dict() for m in view_data['maps']]
  45
  46     return maps
  47
  48
  49 def _map_info_data(request):
  50     map_id = int(request.matchdict['id'])
  51
  52     try:
  53         leaderboard_lifetime = int(
  54                 request.registry.settings['xonstat.leaderboard_lifetime'])
  55     except:
  56         leaderboard_lifetime = 30
  57
  58     leaderboard_count = 10
  59     recent_games_count = 20
  60
  61     # captime tuples
  62     Captime = namedtuple('Captime', ['player_id', 'nick_html_colors',
  63         'fastest_cap', 'game_id'])
  64
  65     try:
  66         gmap = DBSession.query(Map).filter_by(map_id=map_id).one()
  67
  68         # recent games played in descending order
  69         rgs = recent_games_q(map_id=map_id).limit(recent_games_count).all()
  70         recent_games = [RecentGame(row) for row in rgs]
  71
  72         # top players by score
  73         top_scorers = DBSession.query(Player.player_id, Player.nick,
  74                 func.sum(PlayerGameStat.score)).\
  75                 filter(Player.player_id == PlayerGameStat.player_id).\
  76                 filter(Game.game_id == PlayerGameStat.game_id).\
  77                 filter(Game.map_id == map_id).\
  78                 filter(Player.player_id > 2).\
  79                 filter(PlayerGameStat.create_dt >
  80                         (datetime.utcnow() - timedelta(days=leaderboard_lifetime))).\
  81                 order_by(expr.desc(func.sum(PlayerGameStat.score))).\
  82                 group_by(Player.nick).\
  83                 group_by(Player.player_id).all()[0:leaderboard_count]
  84
  85         top_scorers = [(player_id, html_colors(nick), score) \
  86                 for (player_id, nick, score) in top_scorers]
  87
  88         # top players by playing time
  89         top_players = DBSession.query(Player.player_id, Player.nick,
  90                 func.sum(PlayerGameStat.alivetime)).\
  91                 filter(Player.player_id == PlayerGameStat.player_id).\
  92                 filter(Game.game_id == PlayerGameStat.game_id).\
  93                 filter(Game.map_id == map_id).\
  94                 filter(Player.player_id > 2).\
  95                 filter(PlayerGameStat.create_dt >
  96                         (datetime.utcnow() - timedelta(days=leaderboard_lifetime))).\
  97                 order_by(expr.desc(func.sum(PlayerGameStat.alivetime))).\
  98                 group_by(Player.nick).\
  99                 group_by(Player.player_id).all()[0:leaderboard_count]
 100
 101         top_players = [(player_id, html_colors(nick), score) \
 102                 for (player_id, nick, score) in top_players]
 103
 104         # top servers using/playing this map
 105         top_servers = DBSession.query(Server.server_id, Server.name,
 106                 func.count(Game.game_id)).\
 107                 filter(Game.server_id == Server.server_id).\
 108                 filter(Game.map_id == map_id).\
 109                 filter(Game.create_dt >
 110                         (datetime.utcnow() - timedelta(days=leaderboard_lifetime))).\
 111                 order_by(expr.desc(func.count(Game.game_id))).\
 112                 group_by(Server.name).\
 113                 group_by(Server.server_id).all()[0:leaderboard_count]
 114
 115         # top captimes
 116         captimes_raw = DBSession.query(Player.player_id, Player.nick,
 117             PlayerCaptime.fastest_cap, PlayerCaptime.game_id).\
 118                 filter(PlayerCaptime.map_id == map_id).\
 119                 filter(Player.player_id == PlayerCaptime.player_id).\
 120                 order_by(PlayerCaptime.fastest_cap).\
 121                 limit(25).\
 122                 all()
 123
 124         captimes = [Captime(c.player_id, html_colors(c.nick),
 125             c.fastest_cap, c.game_id) for c in captimes_raw]
 126
 127     except Exception as e:
 128         gmap = None
 129     return {'gmap':gmap,
 130             'recent_games':recent_games,
 131             'top_scorers':top_scorers,
 132             'top_players':top_players,
 133             'top_servers':top_servers,
 134             'captimes':captimes,
 135             }
 136
 137
 138 def map_info(request):
 139     """
 140     List the information stored about a given map.
 141     """
 142     mapinfo_data =  _map_info_data(request)
 143
 144     # FIXME: code clone, should get these from _map_info_data
 145     leaderboard_count = 10
 146     recent_games_count = 20
 147
 148     for i in range(leaderboard_count-len(mapinfo_data['top_scorers'])):
 149         mapinfo_data['top_scorers'].append(('-', '-', '-'))
 150
 151     for i in range(leaderboard_count-len(mapinfo_data['top_players'])):
 152         mapinfo_data['top_players'].append(('-', '-', '-'))
 153
 154     for i in range(leaderboard_count-len(mapinfo_data['top_servers'])):
 155         mapinfo_data['top_servers'].append(('-', '-', '-'))
 156
 157     return mapinfo_data
 158
 159
 160 def map_info_json(request):
 161     """
 162     List the information stored about a given map. JSON.
 163     """
 164     return [{'status':'not implemented'}]
 165
 166
 167 class MapCapTime(object):
 168     def __init__(self, row):
 169         self.fastest_cap          = row.fastest_cap
 170         self.create_dt            = row.create_dt
 171         self.create_dt_epoch      = timegm(row.create_dt.timetuple())
 172         self.create_dt_fuzzy      = pretty_date(row.create_dt)
 173         self.player_id            = row.player_id
 174         self.player_nick          = row.player_nick
 175         self.player_nick_stripped = strip_colors(row.player_nick)
 176         self.player_nick_html     = html_colors(row.player_nick)
 177         self.game_id              = row.game_id
 178         self.server_id            = row.server_id
 179         self.server_name          = row.server_name
 180
 181     def to_dict(self):
 182         return {
 183             "fastest_cap"          : self.fastest_cap.total_seconds(),
 184             "create_dt_epoch"      : self.create_dt_epoch,
 185             "create_dt_fuzzy"      : self.create_dt_fuzzy,
 186             "player_id"            : self.player_id,
 187             "player_nick"          : self.player_nick,
 188             "player_nick_stripped" : self.player_nick_stripped,
 189             "game_id"              : self.game_id,
 190             "server_id"            : self.server_id,
 191             "server_name"          : self.server_name,
 192             }
 193
 194 def map_captimes_data(request):
 195     map_id = int(request.matchdict['id'])
 196
 197     current_page = request.params.get('page', 1)
 198
 199     mmap = DBSession.query(Map).filter_by(map_id=map_id).one()
 200
 201     mct_q = DBSession.query(PlayerCaptime.fastest_cap, PlayerCaptime.create_dt,
 202             PlayerCaptime.player_id, PlayerCaptime.game_id,
 203             Game.server_id, Server.name.label('server_name'),
 204             PlayerGameStat.nick.label('player_nick')).\
 205             filter(PlayerCaptime.map_id==map_id).\
 206             filter(PlayerCaptime.game_id==Game.game_id).\
 207             filter(PlayerCaptime.map_id==Map.map_id).\
 208             filter(Game.server_id==Server.server_id).\
 209             filter(PlayerCaptime.player_id==PlayerGameStat.player_id).\
 210             filter(PlayerCaptime.game_id==PlayerGameStat.game_id).\
 211             order_by(expr.asc(PlayerCaptime.fastest_cap))
 212
 213     map_captimes = Page(mct_q, current_page, items_per_page=20, url=page_url)
 214
 215     map_captimes.items = [MapCapTime(row) for row in map_captimes.items]
 216
 217     return {
 218             'map_id':map_id,
 219             'map':mmap,
 220             'captimes':map_captimes,
 221         }
 222
 223 def map_captimes(request):
 224     return map_captimes_data(request)
 225
 226 def map_captimes_json(request):
 227     current_page = request.params.get('page', 1)
 228     data = map_captimes_data(request)
 229
 230     return {
 231             "map": data["map"].to_dict(),
 232             "captimes": [e.to_dict() for e in data["captimes"].items],
 233             "page": current_page,
 234             }