Merge remote branch 'remotes/aaronw/feature410_markdown_bio'
[mediagoblin.git] / mediagoblin / app.py
index f688b98945ff8b952a0df39852094b9a317d6ddb..7a6a1f33c7c4610a7dcf32dc98d048a35d97d4fa 100644 (file)
-import sys
+# GNU MediaGoblin -- federated, autonomous media hosting
+# Copyright (C) 2011 Free Software Foundation, Inc
+#
+# This program is free software: you can redistribute it and/or modify
+# it under the terms of the GNU Affero General Public License as published by
+# the Free Software Foundation, either version 3 of the License, or
+# (at your option) any later version.
+#
+# This program is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU Affero General Public License for more details.
+#
+# You should have received a copy of the GNU Affero General Public License
+# along with this program.  If not, see <http://www.gnu.org/licenses/>.
+
+import os
 import urllib
 
-from beaker.middleware import SessionMiddleware
 import routes
-import pymongo
 from webob import Request, exc
 
-from mediagoblin import routing, util
+from mediagoblin import routing, util, storage
+from mediagoblin.db.open import setup_connection_and_db_from_config
+from mediagoblin.db.util import MigrationManager
+from mediagoblin.mg_globals import setup_globals
+from mediagoblin.init.celery import setup_celery_from_config
+from mediagoblin.init import get_jinja_loader, get_staticdirector, \
+    setup_global_and_app_config, setup_workbench
 
 
-class Error(Exception): pass
-class ImproperlyConfigured(Error): pass
-
-
-def load_controller(string):
-    module_name, func_name = string.split(':', 1)
-    __import__(module_name)
-    module = sys.modules[module_name]
-    func = getattr(module, func_name)
-    return func
-
-
-class MediagoblinApp(object):
+class MediaGoblinApp(object):
     """
-    Really basic wsgi app using routes and WebOb.
+    WSGI application of MediaGoblin
+
+    ... this is the heart of the program!
     """
-    def __init__(self, database, user_template_path=None):
-        self.template_env = util.get_jinja_env(user_template_path)
-        self.db = database
+    def __init__(self, config_path, setup_celery=True):
+        """
+        Initialize the application based on a configuration file.
+
+        Arguments:
+         - config_path: path to the configuration file we're opening.
+         - setup_celery: whether or not to setup celery during init.
+           (Note: setting 'celery_setup_elsewhere' also disables
+           setting up celery.)
+        """
+        ##############
+        # Setup config
+        ##############
+
+        # Open and setup the config
+        global_config, app_config = setup_global_and_app_config(config_path)
+
+        ##########################################
+        # Setup other connections / useful objects
+        ##########################################
+
+        # Set up the database
+        self.connection, self.db = setup_connection_and_db_from_config(
+            app_config)
+
+        # Init the migration number if necessary
+        migration_manager = MigrationManager(self.db)
+        migration_manager.install_migration_version_if_missing()
+
+        # Tiny hack to warn user if our migration is out of date
+        if not migration_manager.database_at_latest_migration():
+            print (
+                "*WARNING:* Your migrations are out of date, "
+                "maybe run ./bin/gmg migrate?")
+
+        # Get the template environment
+        self.template_loader = get_jinja_loader(
+            app_config.get('user_template_path'))
+        
+        # Set up storage systems
+        self.public_store = storage.storage_system_from_config(
+            app_config, 'publicstore')
+        self.queue_store = storage.storage_system_from_config(
+            app_config, 'queuestore')
+
+        # set up routing
         self.routing = routing.get_mapper()
 
+        # set up staticdirector tool
+        self.staticdirector = get_staticdirector(app_config)
+
+        # Setup celery, if appropriate
+        if setup_celery and not app_config.get('celery_setup_elsewhere'):
+            if os.environ.get('CELERY_ALWAYS_EAGER'):
+                setup_celery_from_config(
+                    app_config, global_config,
+                    force_celery_always_eager=True)
+            else:
+                setup_celery_from_config(app_config, global_config)
+
+        #######################################################
+        # Insert appropriate things into mediagoblin.mg_globals
+        #
+        # certain properties need to be accessed globally eg from
+        # validators, etc, which might not access to the request
+        # object.
+        #######################################################
+
+        setup_globals(
+            app=self,
+            db_connection=self.connection,
+            database=self.db,
+            public_store=self.public_store,
+            queue_store=self.queue_store)
+
+        # Workbench *currently* only used by celery, so this only
+        # matters in always eager mode :)
+        setup_workbench()
+
     def __call__(self, environ, start_response):
         request = Request(environ)
         path_info = request.path_info
+
+        ## Routing / controller loading stuff
         route_match = self.routing.match(path_info)
 
         # No matching page?
@@ -46,33 +132,36 @@ class MediagoblinApp(object):
                 if request.GET:
                     new_path_info = '%s?%s' % (
                         new_path_info, urllib.urlencode(request.GET))
-                redirect = exc.HTTPTemporaryRedirect(location=new_path_info)
+                redirect = exc.HTTPFound(location=new_path_info)
                 return request.get_response(redirect)(environ, start_response)
 
             # Okay, no matches.  404 time!
             return exc.HTTPNotFound()(environ, start_response)
 
-        controller = load_controller(route_match['controller'])
+        controller = util.import_component(route_match['controller'])
         request.start_response = start_response
 
+        ## Attach utilities to the request object
         request.matchdict = route_match
-        request.app = self
-        request.template_env = self.template_env
         request.urlgen = routes.URLGenerator(self.routing, environ)
+        # Do we really want to load this via middleware?  Maybe?
         request.session = request.environ['beaker.session']
+        # Attach self as request.app
+        # Also attach a few utilities from request.app for convenience?
+        request.app = self
+        request.locale = util.get_locale_from_request(request)
+            
+        request.template_env = util.get_jinja_env(
+            self.template_loader, request.locale)
+        request.db = self.db
+        request.staticdirect = self.staticdirector
 
-        return controller(request)(environ, start_response)
+        util.setup_user_in_request(request)
 
+        return controller(request)(environ, start_response)
 
-def paste_app_factory(global_config, **kw):
-    connection = pymongo.Connection()
-    db = connection[kw.get('db_name', 'mediagoblin')]
 
-    mgoblin_app = MediagoblinApp(
-        db, user_template_path=kw.get('local_templates'))
-    beakered_app = SessionMiddleware(
-        mgoblin_app,
-        {'session.type': 'file',
-         'session.cookie_expires': True})
+def paste_app_factory(global_config, **app_config):
+    mgoblin_app = MediaGoblinApp(app_config['config'])
 
-    return beakered_app
+    return mgoblin_app