xonstat/views/game.py

   1 import datetime
   2 import logging
   3 import re
   4 import time
   5 from collections import OrderedDict
   6 from pyramid import httpexceptions
   7 from pyramid.response import Response
   8 from sqlalchemy import desc, func, over
   9 from webhelpers.paginate import Page, PageURL
  10 from xonstat.models import *
  11 from xonstat.util import page_url
  12 from xonstat.views.helpers import RecentGame, recent_games_q
  13
  14
  15 log = logging.getLogger(__name__)
  16
  17
  18 def _game_info_data(request):
  19     game_id = int(request.matchdict['id'])
  20
  21     show_elo = False
  22     if request.params.has_key('show_elo'):
  23         show_elo = True
  24
  25     show_latency = False
  26
  27     try:
  28         notfound = False
  29
  30         (game, server, map, gametype) = DBSession.query(Game, Server, Map, GameType).\
  31                 filter(Game.game_id == game_id).\
  32                 filter(Game.server_id == Server.server_id).\
  33                 filter(Game.map_id == Map.map_id).\
  34                 filter(Game.game_type_cd == GameType.game_type_cd).one()
  35
  36         pgstats = DBSession.query(PlayerGameStat).\
  37                 filter(PlayerGameStat.game_id == game_id).\
  38                 order_by(PlayerGameStat.scoreboardpos).\
  39                 order_by(PlayerGameStat.score).\
  40                 all()
  41
  42         # if at least one player has a valid latency, we'll show the column
  43         for pgstat in pgstats:
  44             if pgstat.avg_latency is not None:
  45                 show_latency = True
  46
  47         q = DBSession.query(TeamGameStat).\
  48                 filter(TeamGameStat.game_id == game_id)
  49         if game.game_type_cd == 'ctf':
  50             q = q.order_by(TeamGameStat.caps.desc())
  51         elif game.game_type_cd == 'ca':
  52             q = q.order_by(TeamGameStat.rounds.desc())
  53         # dom -> ticks, rc -> laps, nb -> goals, as -> objectives
  54
  55         q = q.order_by(TeamGameStat.score.desc())
  56
  57         tgstats = q.all()
  58
  59         stats_by_team = OrderedDict()
  60         for pgstat in pgstats:
  61             if pgstat.team not in stats_by_team.keys():
  62                 stats_by_team[pgstat.team] = []
  63             stats_by_team[pgstat.team].append(pgstat)
  64
  65         captimes = []
  66         if game.game_type_cd == 'ctf':
  67             for pgstat in pgstats:
  68                 if pgstat.fastest is not None:
  69                     captimes.append(pgstat)
  70             captimes = sorted(captimes, key=lambda x:x.fastest)
  71
  72         pwstats = {}
  73         for (pwstat, weapon) in DBSession.query(PlayerWeaponStat, Weapon).\
  74                 filter(PlayerWeaponStat.game_id == game_id).\
  75                 filter(PlayerWeaponStat.weapon_cd == Weapon.weapon_cd).\
  76                 order_by(PlayerWeaponStat.actual.desc()).\
  77                 all():
  78                     if pwstat.player_game_stat_id not in pwstats:
  79                         pwstats[pwstat.player_game_stat_id] = []
  80
  81                     pwstats[pwstat.player_game_stat_id].append((weapon.descr,
  82                         weapon.weapon_cd, pwstat.actual, pwstat.max,
  83                         pwstat.hit, pwstat.fired, pwstat.frags))
  84
  85     except Exception as inst:
  86         game = None
  87         server = None
  88         map = None
  89         gametype = None
  90         pgstats = None
  91         tgstats = None
  92         pwstats = None
  93         captimes = None
  94         show_elo = False
  95         show_latency = False
  96         stats_by_team = None
  97         raise inst
  98
  99     return {'game':game,
 100             'server':server,
 101             'map':map,
 102             'gametype':gametype,
 103             'pgstats':pgstats,
 104             'tgstats':tgstats,
 105             'pwstats':pwstats,
 106             'captimes':captimes,
 107             'show_elo':show_elo,
 108             'show_latency':show_latency,
 109             'stats_by_team':stats_by_team,
 110             }
 111
 112
 113 def game_info(request):
 114     """
 115     List the game stats (scoreboard) for a particular game. Paginated.
 116     """
 117     return _game_info_data(request)
 118
 119
 120 def game_info_json(request):
 121     """
 122     List the game stats (scoreboard) for a particular game. Paginated. JSON.
 123     """
 124     return [{'status':'not implemented'}]
 125
 126
 127 def _rank_index_data(request):
 128     if request.params.has_key('page'):
 129         current_page = request.params['page']
 130     else:
 131         current_page = 1
 132
 133     game_type_cd = request.matchdict['game_type_cd']
 134
 135     ranks_q = DBSession.query(PlayerRank).\
 136             filter(PlayerRank.game_type_cd==game_type_cd).\
 137             order_by(PlayerRank.rank)
 138
 139     ranks = Page(ranks_q, current_page, url=page_url)
 140
 141     if len(ranks) == 0:
 142         ranks = None
 143
 144     return {
 145             'ranks':ranks,
 146             'game_type_cd':game_type_cd,
 147            }
 148
 149
 150 def rank_index(request):
 151     """
 152     Provide a list of gametype ranks, paginated.
 153     """
 154     return _rank_index_data(request)
 155
 156
 157 def rank_index_json(request):
 158     """
 159     Provide a list of gametype ranks, paginated. JSON.
 160     """
 161     return [{'status':'not implemented'}]
 162
 163
 164 def game_finder_data(request):
 165     if request.params.has_key('page'):
 166         current_page = request.params['page']
 167     else:
 168         current_page = 1
 169
 170     query = {}
 171
 172     server_id, map_id, player_id = None, None, None
 173     game_type_cd, start_game_id, end_game_id = None, None, None
 174     game_type_descr = None
 175
 176     # these become WHERE clauses when present
 177     if request.params.has_key('server_id'):
 178         server_id = request.params['server_id']
 179         query['server_id'] = server_id
 180
 181     if request.params.has_key('map_id'):
 182         map_id = request.params['map_id']
 183         query['map_id'] = map_id
 184
 185     if request.params.has_key('player_id'):
 186         player_id = request.params['player_id']
 187         query['player_id'] = player_id
 188
 189     if request.params.has_key('start_game_id'):
 190         start_game_id = request.params['start_game_id']
 191         query['start_game_id'] = start_game_id
 192
 193     if request.params.has_key('end_game_id'):
 194         end_game_id = request.params['end_game_id']
 195         query['end_game_id'] = end_game_id
 196
 197     if request.params.has_key('type'):
 198         game_type_cd = request.params['type']
 199         query['type'] = game_type_cd
 200         try:
 201             game_type_descr = DBSession.query(GameType.descr).\
 202                 filter(GameType.game_type_cd == game_type_cd).\
 203                 one()[0]
 204         except Exception as e:
 205             game_type_cd = None
 206
 207     rgs_q = recent_games_q(server_id=server_id, map_id=map_id,
 208             player_id=player_id, game_type_cd=game_type_cd,
 209             start_game_id=start_game_id, end_game_id=end_game_id)
 210
 211     try:
 212         recent_games = [RecentGame(row) for row in rgs_q.limit(20).all()]
 213
 214         if len(recent_games) > 0:
 215             query['start_game_id'] = recent_games[-1].game_id + 1
 216
 217         # build the list of links for the stripe across the top
 218         game_type_links = []
 219
 220         # clear out the game_id window
 221         gt_query = query.copy()
 222         if 'start_game_id' in gt_query:
 223             del gt_query['start_game_id']
 224         if 'end_game_id' in gt_query:
 225             del gt_query['end_game_id']
 226
 227         for gt in ('overall','duel','ctf','dm','tdm','ca','kh','ft',
 228                 'lms','as','dom','nb','cts','rc'):
 229             gt_query['type'] = gt
 230             url = request.route_url("game_index", _query=gt_query)
 231             game_type_links.append((gt, url))
 232
 233     except:
 234         raise httpexceptions.HTTPBadRequest("Malformed Query")
 235
 236     return {
 237             'recent_games':recent_games,
 238             'query':query,
 239             'game_type_cd':game_type_cd,
 240             'game_type_links':game_type_links,
 241            }
 242
 243
 244 def game_finder(request):
 245     """
 246     Provide a list of recent games with an advanced filter.
 247     """
 248     return game_finder_data(request)
 249
 250
 251 def game_finder_json(request):
 252     """
 253     Provide a list of recent games in JSON format.
 254     """
 255     data = game_finder_data(request)
 256     return [rg.to_dict() for rg in data["recent_games"]]