]> de.git.xonotic.org Git - xonotic/xonstat.git/blobdiff - xonstat/views/server.py
Convert the server info to a class-based view callable. Add a dummy JSON view.
[xonotic/xonstat.git] / xonstat / views / server.py
old mode 100755 (executable)
new mode 100644 (file)
index a706dc2..b914d5d
-import logging\r
-import sqlalchemy.sql.functions as func\r
-import sqlalchemy.sql.expression as expr\r
-import time\r
-from datetime import datetime, timedelta\r
-from pyramid.config import get_current_registry\r
-from pyramid.response import Response\r
-from sqlalchemy import desc\r
-from webhelpers.paginate import Page, PageURL\r
-from xonstat.models import *\r
-from xonstat.util import page_url, html_colors\r
-\r
-log = logging.getLogger(__name__)\r
-\r
-def server_index(request):\r
-    """\r
-    Provides a list of all the current servers. \r
-    """\r
-    if 'page' in request.matchdict:\r
-        current_page = request.matchdict['page']\r
-    else:\r
-        current_page = 1\r
-\r
-    try:\r
-        server_q = DBSession.query(Server).\\r
-                order_by(Server.name)\r
-\r
-        servers = Page(server_q, current_page, url=page_url)\r
-\r
-        \r
-    except Exception as e:\r
-        servers = None\r
-\r
-    return {'servers':servers, }\r
-\r
-\r
-def server_info(request):\r
-    """\r
-    List the stored information about a given server.\r
-    """\r
-    server_id = request.matchdict['id']\r
-\r
-    # get settings specific to this view\r
-    settings = get_current_registry().settings\r
-    try: \r
-        leaderboard_lifetime = int(\r
-                settings['xonstat.leaderboard_lifetime'])\r
-    except:\r
-        leaderboard_lifetime = 30\r
-\r
-    leaderboard_count = 10\r
-    recent_games_count = 20\r
-\r
-    try:\r
-        server = DBSession.query(Server).filter_by(server_id=server_id).one()\r
-\r
-        # top maps by total times played\r
-        top_maps = DBSession.query(Game.map_id, Map.name, \r
-                func.count()).\\r
-                filter(Map.map_id==Game.map_id).\\r
-                filter(Game.server_id==server.server_id).\\r
-                filter(Game.create_dt > \r
-                    (datetime.now() - timedelta(days=leaderboard_lifetime))).\\r
-                order_by(expr.desc(func.count())).\\r
-                group_by(Game.map_id).\\r
-                group_by(Map.name).all()[0:10]\r
-\r
-        for i in range(leaderboard_count-len(top_maps)):\r
-            top_maps.append(('-', '-', '-'))\r
-\r
-        # top players by score\r
-        top_scorers = DBSession.query(Player.player_id, Player.nick, \r
-                func.sum(PlayerGameStat.score)).\\r
-                filter(Player.player_id == PlayerGameStat.player_id).\\r
-                filter(Game.game_id == PlayerGameStat.game_id).\\r
-                filter(Game.server_id == server.server_id).\\r
-                filter(Player.player_id > 2).\\r
-                filter(PlayerGameStat.create_dt > \r
-                        (datetime.now() - timedelta(days=leaderboard_lifetime))).\\r
-                order_by(expr.desc(func.sum(PlayerGameStat.score))).\\r
-                group_by(Player.nick).\\r
-                group_by(Player.player_id).all()[0:10]\r
-\r
-        top_scorers = [(player_id, html_colors(nick), score) \\r
-                for (player_id, nick, score) in top_scorers]\r
-\r
-        for i in range(leaderboard_count-len(top_scorers)):\r
-            top_scorers.append(('-', '-', '-'))\r
-\r
-        # top players by playing time\r
-        top_players = DBSession.query(Player.player_id, Player.nick, \r
-                func.sum(PlayerGameStat.alivetime)).\\r
-                filter(Player.player_id == PlayerGameStat.player_id).\\r
-                filter(Game.game_id == PlayerGameStat.game_id).\\r
-                filter(Game.server_id == server.server_id).\\r
-                filter(Player.player_id > 2).\\r
-                filter(PlayerGameStat.create_dt > \r
-                        (datetime.now() - timedelta(days=leaderboard_lifetime))).\\r
-                order_by(expr.desc(func.sum(PlayerGameStat.alivetime))).\\r
-                group_by(Player.nick).\\r
-                group_by(Player.player_id).all()[0:10]\r
-\r
-        top_players = [(player_id, html_colors(nick), score) \\r
-                for (player_id, nick, score) in top_players]\r
-\r
-        for i in range(leaderboard_count-len(top_players)):\r
-            top_players.append(('-', '-', '-'))\r
-\r
-        # recent games played in descending order\r
-        recent_games = DBSession.query(Game, Server, Map, PlayerGameStat).\\r
-            filter(Game.server_id==Server.server_id).\\r
-            filter(Game.map_id==Map.map_id).\\r
-            filter(PlayerGameStat.game_id==Game.game_id).\\r
-            filter(PlayerGameStat.rank==1).\\r
-            filter(Server.server_id==server.server_id).\\r
-            order_by(expr.desc(Game.start_dt)).all()[0:recent_games_count]\r
-\r
-        for i in range(recent_games_count-len(recent_games)):\r
-            recent_games.append(('-', '-', '-', '-'))\r
-\r
-    except Exception as e:\r
-        server = None\r
-        recent_games = None\r
-        top_players = None\r
-        raise e\r
-    return {'server':server,\r
-            'recent_games':recent_games,\r
-            'top_players': top_players,\r
-            'top_scorers': top_scorers,\r
-            'top_maps': top_maps,\r
-            }\r
-\r
-\r
-def server_game_index(request):\r
-    """\r
-    List the games played on a given server. Paginated.\r
-    """\r
-    server_id = request.matchdict['server_id']\r
-    current_page = request.matchdict['page']\r
-\r
-    try:\r
-        server = DBSession.query(Server).filter_by(server_id=server_id).one()\r
-\r
-        games_q = DBSession.query(Game, Server, Map).\\r
-                filter(Game.server_id == server_id).\\r
-                filter(Game.server_id == Server.server_id).\\r
-                filter(Game.map_id == Map.map_id).\\r
-                order_by(Game.game_id.desc())\r
-\r
-        games = Page(games_q, current_page, url=page_url)\r
-    except Exception as e:\r
-        server = None\r
-        games = None\r
-        raise e\r
-\r
-    return {'games':games,\r
-            'server':server}\r
+import logging
+import sqlalchemy.sql.functions as func
+import sqlalchemy.sql.expression as expr
+from datetime import datetime, timedelta
+from pyramid.httpexceptions import HTTPNotFound
+from webhelpers.paginate import Page
+from xonstat.models import DBSession, Player, Server, Map, Game, PlayerGameStat
+from xonstat.util import page_url, html_colors
+from xonstat.views.helpers import RecentGame, recent_games_q
+
+log = logging.getLogger(__name__)
+
+
+# Defaults
+LEADERBOARD_LIFETIME = 30
+LEADERBOARD_COUNT = 10
+RECENT_GAMES_COUNT = 20
+
+
+class ServerIndex(object):
+    """Returns a list of servers."""
+
+    def __init__(self, request):
+        """Common parameter parsing."""
+        self.request = request
+        self.page = request.params.get("page", 1)
+        self.servers = self._data()
+
+    def _data(self):
+        """Returns the data shared by all renderers."""
+        try:
+            server_q = DBSession.query(Server).order_by(Server.server_id.desc())
+            servers = Page(server_q, self.page, items_per_page=25, url=page_url)
+
+        except:
+            servers = None
+
+        return servers
+
+    def html(self):
+        """For rendering this data using something HTML-based."""
+        return {
+            'servers': self.servers,
+        }
+
+    def json(self):
+        """For rendering this data using JSON."""
+        return {
+            'servers': [s.to_dict() for s in self.servers],
+        }
+
+
+class ServerInfo(object):
+    """Returns detailed information about a particular server."""
+
+    def __init__(self, request):
+        """Common parameter parsing."""
+        self.request = request
+        self.server_id = request.matchdict.get("id", None)
+
+        raw_lifetime = request.registry.settings.get('xonstat.leaderboard_lifetime',
+                                                     LEADERBOARD_LIFETIME)
+        self.lifetime = int(raw_lifetime)
+
+        self.now = datetime.utcnow()
+        self.server_info = self._data()
+
+    def _top_maps(self):
+        """Top maps on this server by total times played."""
+        try:
+            top_maps = DBSession.query(Game.map_id, Map.name, func.count())\
+                .filter(Map.map_id==Game.map_id)\
+                .filter(Game.server_id==self.server_id)\
+                .filter(Game.create_dt > (self.now - timedelta(days=self.lifetime)))\
+                .group_by(Game.map_id)\
+                .group_by(Map.name) \
+                .order_by(expr.desc(func.count()))\
+                .limit(LEADERBOARD_COUNT)
+        except:
+            top_maps = None
+
+        return top_maps
+
+    def _top_scorers(self):
+        """Top scorers on this server by total score."""
+        try:
+            top_scorers = DBSession.query(Player.player_id, Player.nick,
+                                          func.sum(PlayerGameStat.score))\
+                .filter(Player.player_id == PlayerGameStat.player_id)\
+                .filter(Game.game_id == PlayerGameStat.game_id)\
+                .filter(Game.server_id == self.server_id)\
+                .filter(Player.player_id > 2)\
+                .filter(PlayerGameStat.create_dt >
+                        (self.now - timedelta(days=LEADERBOARD_LIFETIME)))\
+                .order_by(expr.desc(func.sum(PlayerGameStat.score)))\
+                .group_by(Player.nick)\
+                .group_by(Player.player_id)\
+                .limit(LEADERBOARD_COUNT)
+
+            # We can't call a Python function directly in our query, so we have to convert
+            # it out-of-band.
+            top_scorers = [(player_id, html_colors(nick), score)
+                           for (player_id, nick, score) in top_scorers]
+        except:
+            top_scorers = None
+
+        return top_scorers
+
+    def _top_players(self):
+        """Top players on this server by total playing time."""
+        try:
+            top_players = DBSession.query(Player.player_id, Player.nick,
+                                          func.sum(PlayerGameStat.alivetime))\
+                .filter(Player.player_id == PlayerGameStat.player_id)\
+                .filter(Game.game_id == PlayerGameStat.game_id)\
+                .filter(Game.server_id == self.server_id)\
+                .filter(Player.player_id > 2)\
+                .filter(PlayerGameStat.create_dt > (self.now - timedelta(days=self.lifetime)))\
+                .order_by(expr.desc(func.sum(PlayerGameStat.alivetime)))\
+                .group_by(Player.nick)\
+                .group_by(Player.player_id)\
+                .limit(LEADERBOARD_COUNT)
+
+            # We can't call a Python function directly in our query, so we have to convert
+            # it out-of-band.
+            top_players = [(player_id, html_colors(nick), score) \
+                    for (player_id, nick, score) in top_players]
+
+        except:
+            top_players = None
+
+        return top_players
+
+    def _data(self):
+        """Returns the data shared by all renderers."""
+        try:
+            server = DBSession.query(Server).filter_by(server_id=self.server_id).one()
+
+            top_maps = self._top_maps()
+            top_scorers = self._top_scorers()
+            top_players = self._top_players()
+
+            rgs = recent_games_q(server_id=self.server_id).limit(RECENT_GAMES_COUNT).all()
+            recent_games = [RecentGame(row) for row in rgs]
+        except:
+            raise HTTPNotFound
+
+        return {
+            'server': server,
+            'recent_games': recent_games,
+            'top_players': top_players,
+            'top_scorers': top_scorers,
+            'top_maps': top_maps,
+        }
+
+    def html(self):
+        """For rendering this data using something HTML-based."""
+        return self.server_info
+
+    def json(self):
+        """For rendering this data using JSON."""
+        return {"status":"Not implemented"}