1 # GNU MediaGoblin -- federated, autonomous media hosting
2 # Copyright (C) 2011, 2012 MediaGoblin contributors. See AUTHORS.
4 # This program is free software: you can redistribute it and/or modify
5 # it under the terms of the GNU Affero General Public License as published by
6 # the Free Software Foundation, either version 3 of the License, or
7 # (at your option) any later version.
9 # This program is distributed in the hope that it will be useful,
10 # but WITHOUT ANY WARRANTY; without even the implied warranty of
11 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
12 # GNU Affero General Public License for more details.
14 # You should have received a copy of the GNU Affero General Public License
15 # along with this program. If not, see <http://www.gnu.org/licenses/>.
20 from mediagoblin
.routing
import get_url_map
21 from mediagoblin
.tools
.routing
import endpoint_to_controller
23 from werkzeug
.wrappers
import Request
24 from werkzeug
.exceptions
import HTTPException
25 from werkzeug
.routing
import RequestRedirect
27 from mediagoblin
import meddleware
, __version__
28 from mediagoblin
.db
.util
import check_db_up_to_date
29 from mediagoblin
.tools
import common
, session
, translate
, template
30 from mediagoblin
.tools
.response
import render_http_exception
31 from mediagoblin
.tools
.theme
import register_themes
32 from mediagoblin
.tools
import request
as mg_request
33 from mediagoblin
.media_types
.tools
import media_type_warning
34 from mediagoblin
.mg_globals
import setup_globals
35 from mediagoblin
.init
.celery
import setup_celery_from_config
36 from mediagoblin
.init
.plugins
import setup_plugins
37 from mediagoblin
.init
import (get_jinja_loader
, get_staticdirector
,
38 setup_global_and_app_config
, setup_locales
, setup_workbench
, setup_database
,
40 from mediagoblin
.tools
.pluginapi
import PluginManager
, hook_transform
41 from mediagoblin
.tools
.crypto
import setup_crypto
42 from mediagoblin
.auth
.tools
import check_auth_enabled
, no_auth_logout
45 _log
= logging
.getLogger(__name__
)
48 class MediaGoblinApp(object):
50 WSGI application of MediaGoblin
52 ... this is the heart of the program!
54 def __init__(self
, config_path
, setup_celery
=True):
56 Initialize the application based on a configuration file.
59 - config_path: path to the configuration file we're opening.
60 - setup_celery: whether or not to setup celery during init.
61 (Note: setting 'celery_setup_elsewhere' also disables
64 _log
.info("GNU MediaGoblin %s main server starting", __version__
)
65 _log
.debug("Using config file %s", config_path
)
70 # Open and setup the config
71 global_config
, app_config
= setup_global_and_app_config(config_path
)
77 ##########################################
78 # Setup other connections / useful objects
79 ##########################################
81 # Setup Session Manager, not needed in celery
82 self
.session_manager
= session
.SessionManager()
84 # load all available locales
87 # Set up plugins -- need to do this early so that plugins can
89 _log
.info("Setting up plugins.")
93 self
.db
= setup_database(app_config
['run_migrations'])
95 # Quit app if need to run dbupdate
99 self
.theme_registry
, self
.current_theme
= register_themes(app_config
)
101 # Get the template environment
102 self
.template_loader
= get_jinja_loader(
103 app_config
.get('local_templates'),
105 PluginManager().get_template_paths()
108 # Check if authentication plugin is enabled and respond accordingly.
109 self
.auth
= check_auth_enabled()
111 app_config
['allow_comments'] = False
113 # Set up storage systems
114 self
.public_store
, self
.queue_store
= setup_storage()
117 self
.url_map
= get_url_map()
119 # set up staticdirector tool
120 self
.staticdirector
= get_staticdirector(app_config
)
122 # Setup celery, if appropriate
123 if setup_celery
and not app_config
.get('celery_setup_elsewhere'):
124 if os
.environ
.get('CELERY_ALWAYS_EAGER', 'false').lower() == 'true':
125 setup_celery_from_config(
126 app_config
, global_config
,
127 force_celery_always_eager
=True)
129 setup_celery_from_config(app_config
, global_config
)
131 #######################################################
132 # Insert appropriate things into mediagoblin.mg_globals
134 # certain properties need to be accessed globally eg from
135 # validators, etc, which might not access to the request
137 #######################################################
139 setup_globals(app
=self
)
141 # Workbench *currently* only used by celery, so this only
142 # matters in always eager mode :)
145 # instantiate application meddleware
146 self
.meddleware
= [common
.import_component(m
)(self
)
147 for m
in meddleware
.ENABLED_MEDDLEWARE
]
149 def call_backend(self
, environ
, start_response
):
150 request
= Request(environ
)
152 # Compatibility with django, use request.args preferrably
153 request
.GET
= request
.args
155 ## Routing / controller loading stuff
156 map_adapter
= self
.url_map
.bind_to_environ(request
.environ
)
158 # By using fcgi, mediagoblin can run under a base path
159 # like /mediagoblin/. request.path_info contains the
160 # path inside mediagoblin. If the something needs the
161 # full path of the current page, that should include
163 # Note: urlgen and routes are fine!
164 request
.full_path
= environ
["SCRIPT_NAME"] + request
.path
165 # python-routes uses SCRIPT_NAME. So let's use that too.
166 # The other option would be:
167 # request.full_path = environ["SCRIPT_URL"]
169 # Fix up environ for urlgen
170 # See bug: https://bitbucket.org/bbangert/routes/issue/55/cache_hostinfo-breaks-on-https-off
171 if environ
.get('HTTPS', '').lower() == 'off':
174 ## Attach utilities to the request object
175 # Do we really want to load this via middleware? Maybe?
176 session_manager
= self
.session_manager
177 request
.session
= session_manager
.load_session_from_cookie(request
)
178 # Attach self as request.app
179 # Also attach a few utilities from request.app for convenience?
183 request
.staticdirect
= self
.staticdirector
185 request
.locale
= translate
.get_locale_from_request(request
)
186 request
.template_env
= template
.get_jinja_env(
187 self
.template_loader
, request
.locale
)
189 def build_proxy(endpoint
, **kw
):
191 qualified
= kw
.pop('qualified')
195 return map_adapter
.build(
198 force_external
=qualified
)
200 request
.urlgen
= build_proxy
202 # Log user out if authentication_disabled
203 no_auth_logout(request
)
205 mg_request
.setup_user_in_request(request
)
207 request
.controller_name
= None
209 found_rule
, url_values
= map_adapter
.match(return_rule
=True)
210 request
.matchdict
= url_values
211 except RequestRedirect
as response
:
212 # Deal with 301 responses eg due to missing final slash
213 return response(environ
, start_response
)
214 except HTTPException
as exc
:
215 # Stop and render exception
216 return render_http_exception(
218 exc
.get_description(environ
))(environ
, start_response
)
220 controller
= endpoint_to_controller(found_rule
)
221 # Make a reference to the controller's symbolic name on the request...
222 # used for lazy context modification
223 request
.controller_name
= found_rule
.endpoint
225 # pass the request through our meddleware classes
227 for m
in self
.meddleware
:
228 response
= m
.process_request(request
, controller
)
229 if response
is not None:
230 return response(environ
, start_response
)
231 except HTTPException
as e
:
232 return render_http_exception(
234 e
.get_description(environ
))(environ
, start_response
)
236 request
= hook_transform("modify_request", request
)
238 request
.start_response
= start_response
240 # get the Http response from the controller
242 response
= controller(request
)
243 except HTTPException
as e
:
244 response
= render_http_exception(
245 request
, e
, e
.get_description(environ
))
247 # pass the response through the meddlewares
249 for m
in self
.meddleware
[::-1]:
250 m
.process_response(request
, response
)
251 except HTTPException
as e
:
252 response
= render_http_exception(
253 request
, e
, e
.get_description(environ
))
255 session_manager
.save_session_to_cookie(request
.session
,
258 return response(environ
, start_response
)
260 def __call__(self
, environ
, start_response
):
261 ## If more errors happen that look like unclean sessions:
262 # self.db.check_session_clean()
265 return self
.call_backend(environ
, start_response
)
267 # Reset the sql session, so that the next request
268 # gets a fresh session
269 self
.db
.reset_after_request()
272 def paste_app_factory(global_config
, **app_config
):
273 configs
= app_config
['config'].split()
274 mediagoblin_config
= None
275 for config
in configs
:
276 if os
.path
.exists(config
) and os
.access(config
, os
.R_OK
):
277 mediagoblin_config
= config
280 if not mediagoblin_config
:
281 raise IOError("Usable mediagoblin config not found.")
283 mgoblin_app
= MediaGoblinApp(mediagoblin_config
)
284 mgoblin_app
= hook_transform('wrap_wsgi', mgoblin_app
)