xonstat/__init__.py

   1 import sqlahelper
   2 from pyramid_beaker import set_cache_regions_from_settings
   3 from pyramid.authentication import AuthTktAuthenticationPolicy
   4 from pyramid.authorization import ACLAuthorizationPolicy
   5 from pyramid.config import Configurator
   6 from pyramid.httpexceptions import HTTPNotFound
   7 from pyramid.renderers import JSONP
   8 from sqlalchemy import engine_from_config
   9 from xonstat.models import initialize_db
  10 from xonstat.views import *
  11
  12 def main(global_config, **settings):
  13     """ This function returns a Pyramid WSGI application.
  14     """
  15     # setup the database engine
  16     engine = engine_from_config(settings, 'sqlalchemy.', pool_size=5)
  17     sqlahelper.add_engine(engine)
  18
  19     # initialize database structures
  20     initialize_db(engine)
  21
  22     # set up beaker cache
  23     set_cache_regions_from_settings(settings)
  24
  25     config = Configurator(settings=settings)
  26
  27     # mako for templating
  28     config.include('pyramid_mako')
  29
  30     # for json-encoded responses
  31     config.add_renderer('jsonp', JSONP(param_name='callback'))
  32
  33     # authentication and authorization policies
  34     authn_policy = AuthTktAuthenticationPolicy('secret', hashalg='sha512')
  35     authz_policy = ACLAuthorizationPolicy()
  36     config.set_authentication_policy(authn_policy)
  37     config.set_authorization_policy(authz_policy)
  38
  39     # for static assets
  40     config.add_static_view('static', 'xonstat:static')
  41
  42     # for 404s
  43     config.add_view(notfound, context=HTTPNotFound, renderer="404.mako")
  44
  45     # ROOT ROUTE
  46     config.add_route("main_index", "/")
  47     config.add_view(main_index, route_name="main_index", renderer="main_index.mako")
  48
  49     # MAIN SUBMISSION ROUTE
  50     config.add_route("submit_stats", "stats/submit")
  51     config.add_view(submit_stats, route_name="submit_stats")
  52
  53     # PLAYER ROUTES
  54     config.add_route("player_game_index",      "/player/{player_id:\d+}/games")
  55     config.add_view(player_game_index,      route_name="player_game_index",      renderer="player_game_index.mako")
  56
  57     config.add_route("player_game_index_json", "/player/{player_id:\d+}/games.json")
  58     config.add_view(player_game_index_json, route_name="player_game_index_json", renderer="jsonp")
  59
  60     config.add_route("player_info",      "/player/{id:\d+}")
  61     config.add_view(player_info,      route_name="player_info",      renderer="player_info.mako")
  62
  63     config.add_route("player_info_json", "/player/{id:\d+}.json")
  64     config.add_view(player_info_json, route_name="player_info_json", renderer="jsonp")
  65
  66     config.add_route("player_hashkey_info_text", "/player/me")
  67     config.add_view(player_hashkey_info_text, route_name="player_hashkey_info_text", renderer="player_hashkey_info_text.mako")
  68
  69     config.add_route("player_hashkey_info_json", "/player/me.json")
  70     config.add_view(player_hashkey_info_json, route_name="player_hashkey_info_json", renderer="jsonp")
  71
  72     config.add_route("player_elo_info_text", "/player/{hashkey}/elo.txt")
  73     config.add_view(player_elo_info_text, route_name="player_elo_info_text", renderer="player_elo_info_text.mako")
  74
  75     # FIXME - needs an additional method to convert to JSON
  76     config.add_route("player_elo_info_json", "/player/{hashkey}/elo.json")
  77     config.add_view(player_elo_info_json, route_name="player_elo_info_json", renderer="jsonp")
  78
  79     config.add_route("player_accuracy",      "/player/{id:\d+}/accuracy")
  80     config.add_view(player_accuracy_json, route_name="player_accuracy",      renderer="jsonp")
  81
  82     config.add_route("player_index",      "/players")
  83     config.add_view(player_index,      route_name="player_index",      renderer="player_index.mako")
  84
  85     config.add_route("player_index_json", "/players.json")
  86     config.add_view(player_index_json, route_name="player_index_json", renderer="jsonp")
  87
  88     config.add_route("player_captimes",      "/player/{player_id:\d+}/captimes")
  89     config.add_view(player_captimes,      route_name="player_captimes",      renderer="player_captimes.mako")
  90
  91     config.add_route("player_captimes_json", "/player/{player_id:\d+}/captimes.json")
  92     config.add_view(player_captimes_json, route_name="player_captimes_json", renderer="jsonp")
  93
  94     config.add_route("player_weaponstats_data_json", "/player/{id:\d+}/weaponstats.json")
  95     config.add_view(player_weaponstats_data_json, route_name="player_weaponstats_data_json", renderer="jsonp")
  96
  97     config.add_route("top_players_by_time", "/topactive")
  98     config.add_view(top_players_by_time, route_name="top_players_by_time", renderer="top_players_by_time.mako")
  99
 100     config.add_route("top_servers_by_players", "/topservers")
 101     config.add_view(top_servers_by_players, route_name="top_servers_by_players", renderer="top_servers_by_players.mako")
 102
 103     config.add_route("top_maps_by_times_played", "/topmaps")
 104     config.add_view(top_maps_by_times_played, route_name="top_maps_by_times_played", renderer="top_maps_by_times_played.mako")
 105
 106     # GAME ROUTES
 107     config.add_route("game_info",      "/game/{id:\d+}")
 108     config.add_view(game_info,      route_name="game_info",      renderer="game_info.mako")
 109
 110     config.add_route("game_info_json", "/game/{id:\d+}.json")
 111     config.add_view(game_info_json, route_name="game_info_json", renderer="jsonp")
 112
 113     config.add_route("rank_index",      "/ranks/{game_type_cd:ctf|dm|tdm|duel|ca|ft}")
 114     config.add_view(rank_index,      route_name="rank_index",      renderer="rank_index.mako")
 115
 116     config.add_route("rank_index_json", "/ranks/{game_type_cd:ctf|dm|tdm|duel|ca|ft}.json")
 117     config.add_view(rank_index_json, route_name="rank_index_json", renderer="jsonp")
 118
 119     config.add_route("game_index", "/games")
 120     config.add_view(game_finder, route_name="game_index", renderer="game_finder.mako")
 121
 122     # SERVER ROUTES
 123     config.add_route("server_index",      "/servers")
 124     config.add_view(server_index,      route_name="server_index",      renderer="server_index.mako")
 125
 126     config.add_route("server_index_json", "/servers.json")
 127     config.add_view(server_index_json, route_name="server_index_json", renderer="jsonp")
 128
 129     config.add_route("server_game_index",      "/server/{server_id:\d+}/games/page/{page:\d+}")
 130     config.add_view(server_game_index,      route_name="server_game_index",      renderer="server_game_index.mako")
 131
 132     config.add_route("server_game_index_json", "/server/{server_id:\d+}/games.json")
 133     config.add_view(server_game_index_json, route_name="server_game_index_json", renderer="jsonp")
 134
 135     config.add_route("server_info",      "/server/{id:\d+}")
 136     config.add_view(server_info,      route_name="server_info",      renderer="server_info.mako")
 137
 138     config.add_route("server_info_json", "/server/{id:\d+}.json")
 139     config.add_view(server_info_json, route_name="server_info_json", renderer="jsonp")
 140
 141     # MAP ROUTES
 142     config.add_route("map_index",      "/maps")
 143     config.add_view(map_index,      route_name="map_index",      renderer="map_index.mako")
 144
 145     config.add_route("map_index_json", "/maps.json")
 146     config.add_view(map_index_json, route_name="map_index_json", renderer="jsonp")
 147
 148     config.add_route("map_info",      "/map/{id:\d+}")
 149     config.add_view(map_info,      route_name="map_info",      renderer="map_info.mako")
 150
 151     config.add_route("map_info_json", "/map/{id:\d+}.json")
 152     config.add_view(map_info_json, route_name="map_info_json", renderer="jsonp")
 153
 154     config.add_route("map_captimes",      "/map/{id:\d+}/captimes")
 155     config.add_view(map_captimes,      route_name="map_captimes",      renderer="map_captimes.mako")
 156
 157     config.add_route("map_captimes_json", "/map/{id:\d+}/captimes.json")
 158     config.add_view(map_captimes_json, route_name="map_captimes_json", renderer="jsonp")
 159
 160     # SEARCH ROUTES
 161     config.add_route("search",      "search")
 162     config.add_view(search,      route_name="search",      renderer="search.mako")
 163
 164     config.add_route("search_json", "search.json")
 165     config.add_view(search_json, route_name="search_json", renderer="jsonp")
 166
 167     # ADMIN ROUTES
 168     config.add_forbidden_view(forbidden, renderer="forbidden.mako")
 169
 170     config.add_route("merge",      "/merge")
 171     config.add_view(route_name="merge", renderer="merge.mako", permission="admin")
 172
 173     return config.make_wsgi_app()