xonstat/views/map.py

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