xonstat/views/main.py

   1 import logging
   2 import sqlalchemy.sql.functions as func
   3 import sqlalchemy.sql.expression as expr
   4 from datetime import datetime, timedelta
   5 from pyramid.response import Response
   6 from xonstat.models import *
   7 from xonstat.util import *
   8
   9 log = logging.getLogger(__name__)
  10
  11 def _main_index_data(request):
  12     try:
  13         leaderboard_lifetime = int(
  14                 request.registry.settings['xonstat.leaderboard_lifetime'])
  15     except:
  16         leaderboard_lifetime = 30
  17
  18     leaderboard_count = 10
  19     recent_games_count = 20
  20
  21     # top ranked duelers
  22     duel_ranks = DBSession.query(PlayerRank.player_id, PlayerRank.nick,
  23             PlayerRank.elo).\
  24             filter(PlayerRank.game_type_cd=='duel').\
  25             order_by(PlayerRank.rank).\
  26             limit(leaderboard_count).all()
  27
  28     duel_ranks = [(player_id, html_colors(nick), elo) \
  29             for (player_id, nick, elo) in duel_ranks]
  30
  31     # top ranked CTF-ers
  32     ctf_ranks = DBSession.query(PlayerRank.player_id, PlayerRank.nick,
  33             PlayerRank.elo).\
  34             filter(PlayerRank.game_type_cd=='ctf').\
  35             order_by(PlayerRank.rank).\
  36             limit(leaderboard_count).all()
  37
  38     ctf_ranks = [(player_id, html_colors(nick), elo) \
  39             for (player_id, nick, elo) in ctf_ranks]
  40
  41     # top ranked DM-ers
  42     dm_ranks = DBSession.query(PlayerRank.player_id, PlayerRank.nick,
  43             PlayerRank.elo).\
  44             filter(PlayerRank.game_type_cd=='dm').\
  45             order_by(PlayerRank.rank).\
  46             limit(leaderboard_count).all()
  47
  48     dm_ranks = [(player_id, html_colors(nick), elo) \
  49             for (player_id, nick, elo) in dm_ranks]
  50
  51     right_now = datetime.utcnow()
  52     back_then = datetime.utcnow() - timedelta(days=leaderboard_lifetime)
  53
  54     # top players by playing time
  55     top_players = DBSession.query(Player.player_id, Player.nick,
  56             func.sum(PlayerGameStat.alivetime)).\
  57             filter(Player.player_id == PlayerGameStat.player_id).\
  58             filter(Player.player_id > 2).\
  59             filter(expr.between(PlayerGameStat.create_dt, back_then, right_now)).\
  60             order_by(expr.desc(func.sum(PlayerGameStat.alivetime))).\
  61             group_by(Player.nick).\
  62             group_by(Player.player_id).limit(leaderboard_count).all()
  63
  64     top_players = [(player_id, html_colors(nick), score) \
  65             for (player_id, nick, score) in top_players]
  66
  67     # top servers by number of total players played
  68     top_servers = DBSession.query(Server.server_id, Server.name,
  69             func.count()).\
  70             filter(Game.server_id==Server.server_id).\
  71             filter(expr.between(Game.create_dt, back_then, right_now)).\
  72             order_by(expr.desc(func.count(Game.game_id))).\
  73             group_by(Server.server_id).\
  74             group_by(Server.name).limit(leaderboard_count).all()
  75
  76     # top maps by total times played
  77     top_maps = DBSession.query(Game.map_id, Map.name,
  78             func.count()).\
  79             filter(Map.map_id==Game.map_id).\
  80             filter(expr.between(Game.create_dt, back_then, right_now)).\
  81             order_by(expr.desc(func.count())).\
  82             group_by(Game.map_id).\
  83             group_by(Map.name).limit(leaderboard_count).all()
  84
  85     # recent games played in descending order
  86     recent_games = DBSession.query(Game, Server, Map, PlayerGameStat).\
  87             filter(Game.server_id==Server.server_id).\
  88             filter(Game.map_id==Map.map_id).\
  89             filter(PlayerGameStat.game_id==Game.game_id).\
  90             filter(PlayerGameStat.rank==1).\
  91             filter(expr.between(Game.create_dt, back_then, right_now)).\
  92             order_by(expr.desc(Game.start_dt)).limit(recent_games_count).all()
  93
  94     return {'top_players':top_players,
  95             'top_servers':top_servers,
  96             'top_maps':top_maps,
  97             'recent_games':recent_games,
  98             'duel_ranks':duel_ranks,
  99             'ctf_ranks':ctf_ranks,
 100             'dm_ranks':dm_ranks,
 101             }
 102
 103
 104 def main_index(request):
 105     """
 106     Display the main page information.
 107     """
 108     mainindex_data =  _main_index_data(request)
 109
 110     # FIXME: code clone, should get these from _main_index_data
 111     leaderboard_count = 10
 112     recent_games_count = 20
 113
 114     for i in range(leaderboard_count-len(mainindex_data['duel_ranks'])):
 115         mainindex_data['duel_ranks'].append(('-', '-', '-'))
 116
 117     for i in range(leaderboard_count-len(mainindex_data['ctf_ranks'])):
 118         mainindex_data['ctf_ranks'].append(('-', '-', '-'))
 119
 120     for i in range(leaderboard_count-len(mainindex_data['dm_ranks'])):
 121         mainindex_data['dm_ranks'].append(('-', '-', '-'))
 122
 123     for i in range(leaderboard_count-len(mainindex_data['top_players'])):
 124         mainindex_data['top_players'].append(('-', '-', '-'))
 125
 126     for i in range(leaderboard_count-len(mainindex_data['top_servers'])):
 127         mainindex_data['top_servers'].append(('-', '-', '-'))
 128
 129     for i in range(leaderboard_count-len(mainindex_data['top_maps'])):
 130         mainindex_data['top_maps'].append(('-', '-', '-'))
 131
 132     for i in range(recent_games_count-len(mainindex_data['recent_games'])):
 133         mainindex_data['recent_games'].append(('-', '-', '-', '-'))
 134
 135     return mainindex_data