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.config import Configurator
   5 from pyramid.httpexceptions import HTTPNotFound
   6 from pyramid.renderers import JSONP
   7 from sqlalchemy import engine_from_config
   8 from xonstat.models import initialize_db
   9 from xonstat.views import *
  10 from xonstat.security 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, root_factory=ACLFactory)
  26
  27     # mako for templating
  28     config.include('pyramid_mako')
  29
  30     # Mozilla Persona as the login verifier. It defines default
  31     # authentication and authorization policies.
  32     config.include('pyramid_persona')
  33
  34     # override the authn policy to provide a callback
  35     secret = settings.get('persona.secret', None)
  36     authn_policy = AuthTktAuthenticationPolicy(secret, callback=groupfinder, hashalg='sha512')
  37     config.set_authentication_policy(authn_policy)
  38
  39     # for json-encoded responses
  40     config.add_renderer('jsonp', JSONP(param_name='callback'))
  41
  42     # for static assets
  43     config.add_static_view('static', 'xonstat:static')
  44
  45     # for 404s
  46     config.add_view(notfound, context=HTTPNotFound, renderer="404.mako")
  47
  48     # ROOT ROUTE
  49     config.add_route("main_index", "/")
  50     config.add_view(main_index, route_name="main_index", renderer="main_index.mako")
  51
  52     # MAIN SUBMISSION ROUTE
  53     config.add_route("submit_stats", "stats/submit")
  54     config.add_view(submit_stats, route_name="submit_stats")
  55
  56     # PLAYER ROUTES
  57     config.add_route("player_game_index",      "/player/{player_id:\d+}/games")
  58     config.add_view(player_game_index,      route_name="player_game_index",      renderer="player_game_index.mako")
  59
  60     config.add_route("player_game_index_json", "/player/{player_id:\d+}/games.json")
  61     config.add_view(player_game_index_json, route_name="player_game_index_json", renderer="jsonp")
  62
  63     config.add_route("player_info",      "/player/{id:\d+}")
  64     config.add_view(player_info,      route_name="player_info",      renderer="player_info.mako")
  65
  66     config.add_route("player_info_json", "/player/{id:\d+}.json")
  67     config.add_view(player_info_json, route_name="player_info_json", renderer="jsonp")
  68
  69     config.add_route("player_hashkey_info_text", "/player/me")
  70     config.add_view(player_hashkey_info_text, route_name="player_hashkey_info_text", renderer="player_hashkey_info_text.mako")
  71
  72     config.add_route("player_hashkey_info_json", "/player/me.json")
  73     config.add_view(player_hashkey_info_json, route_name="player_hashkey_info_json", renderer="jsonp")
  74
  75     config.add_route("player_elo_info_text", "/player/{hashkey}/elo.txt")
  76     config.add_view(player_elo_info_text, route_name="player_elo_info_text", renderer="player_elo_info_text.mako")
  77
  78     # FIXME - needs an additional method to convert to JSON
  79     config.add_route("player_elo_info_json", "/player/{hashkey}/elo.json")
  80     config.add_view(player_elo_info_json, route_name="player_elo_info_json", renderer="jsonp")
  81
  82     config.add_route("player_accuracy",      "/player/{id:\d+}/accuracy")
  83     config.add_view(player_accuracy_json, route_name="player_accuracy",      renderer="jsonp")
  84
  85     config.add_route("player_index",      "/players")
  86     config.add_view(player_index,      route_name="player_index",      renderer="player_index.mako")
  87
  88     config.add_route("player_index_json", "/players.json")
  89     config.add_view(player_index_json, route_name="player_index_json", renderer="jsonp")
  90
  91     config.add_route("player_captimes",      "/player/{player_id:\d+}/captimes")
  92     config.add_view(player_captimes,      route_name="player_captimes",      renderer="player_captimes.mako")
  93
  94     config.add_route("player_captimes_json", "/player/{player_id:\d+}/captimes.json")
  95     config.add_view(player_captimes_json, route_name="player_captimes_json", renderer="jsonp")
  96
  97     config.add_route("player_weaponstats_data_json", "/player/{id:\d+}/weaponstats.json")
  98     config.add_view(player_weaponstats_data_json, route_name="player_weaponstats_data_json", renderer="jsonp")
  99
 100     config.add_route("top_players_by_time", "/topactive")
 101     config.add_view(top_players_by_time, route_name="top_players_by_time", renderer="top_players_by_time.mako")
 102
 103     config.add_route("top_servers_by_players", "/topservers")
 104     config.add_view(top_servers_by_players, route_name="top_servers_by_players", renderer="top_servers_by_players.mako")
 105
 106     config.add_route("top_maps_by_times_played", "/topmaps")
 107     config.add_view(top_maps_by_times_played, route_name="top_maps_by_times_played", renderer="top_maps_by_times_played.mako")
 108
 109     # GAME ROUTES
 110     config.add_route("game_info",      "/game/{id:\d+}")
 111     config.add_view(game_info,      route_name="game_info",      renderer="game_info.mako")
 112
 113     config.add_route("game_info_json", "/game/{id:\d+}.json")
 114     config.add_view(game_info_json, route_name="game_info_json", renderer="jsonp")
 115
 116     config.add_route("rank_index",      "/ranks/{game_type_cd:ctf|dm|tdm|duel|ca|ft}")
 117     config.add_view(rank_index,      route_name="rank_index",      renderer="rank_index.mako")
 118
 119     config.add_route("rank_index_json", "/ranks/{game_type_cd:ctf|dm|tdm|duel|ca|ft}.json")
 120     config.add_view(rank_index_json, route_name="rank_index_json", renderer="jsonp")
 121
 122     config.add_route("game_index", "/games")
 123     config.add_view(game_finder, route_name="game_index", renderer="game_finder.mako")
 124
 125     # SERVER ROUTES
 126     config.add_route("server_index",      "/servers")
 127     config.add_view(server_index,      route_name="server_index",      renderer="server_index.mako")
 128
 129     config.add_route("server_index_json", "/servers.json")
 130     config.add_view(server_index_json, route_name="server_index_json", renderer="jsonp")
 131
 132     config.add_route("server_game_index",      "/server/{server_id:\d+}/games/page/{page:\d+}")
 133     config.add_view(server_game_index,      route_name="server_game_index",      renderer="server_game_index.mako")
 134
 135     config.add_route("server_game_index_json", "/server/{server_id:\d+}/games.json")
 136     config.add_view(server_game_index_json, route_name="server_game_index_json", renderer="jsonp")
 137
 138     config.add_route("server_info",      "/server/{id:\d+}")
 139     config.add_view(server_info,      route_name="server_info",      renderer="server_info.mako")
 140
 141     config.add_route("server_info_json", "/server/{id:\d+}.json")
 142     config.add_view(server_info_json, route_name="server_info_json", renderer="jsonp")
 143
 144     # MAP ROUTES
 145     config.add_route("map_index",      "/maps")
 146     config.add_view(map_index,      route_name="map_index",      renderer="map_index.mako")
 147
 148     config.add_route("map_index_json", "/maps.json")
 149     config.add_view(map_index_json, route_name="map_index_json", renderer="jsonp")
 150
 151     config.add_route("map_info",      "/map/{id:\d+}")
 152     config.add_view(map_info,      route_name="map_info",      renderer="map_info.mako")
 153
 154     config.add_route("map_info_json", "/map/{id:\d+}.json")
 155     config.add_view(map_info_json, route_name="map_info_json", renderer="jsonp")
 156
 157     config.add_route("map_captimes",      "/map/{id:\d+}/captimes")
 158     config.add_view(map_captimes,      route_name="map_captimes",      renderer="map_captimes.mako")
 159
 160     config.add_route("map_captimes_json", "/map/{id:\d+}/captimes.json")
 161     config.add_view(map_captimes_json, route_name="map_captimes_json", renderer="jsonp")
 162
 163     # SEARCH ROUTES
 164     config.add_route("search",      "search")
 165     config.add_view(search,      route_name="search",      renderer="search.mako")
 166
 167     config.add_route("search_json", "search.json")
 168     config.add_view(search_json, route_name="search_json", renderer="jsonp")
 169
 170     # ADMIN ROUTES
 171     config.add_forbidden_view(forbidden, renderer="forbidden.mako")
 172
 173     config.add_route("login", "/login")
 174     config.add_view(login, route_name="login", check_csrf=True, renderer="json")
 175
 176     #config.add_route("merge", "/merge")
 177     #config.add_view(merge, route_name="merge", renderer="merge.mako")
 178
 179     return config.make_wsgi_app()