X-Git-Url: https://vcs.fsf.org/?a=blobdiff_plain;f=mediagoblin%2Finit%2F__init__.py;h=44f604b1c95f5b89fe45fc58fafb8191d74dbe5b;hb=9f552f449bf9b7d12a4754a4ab1a63829c66e8d7;hp=4a14fdf8b83c71986aa9f9eea4004a58cd4ff822;hpb=fe289be4c85774b1d48f9db1ef644ee88b672e08;p=mediagoblin.git

diff --git a/mediagoblin/init/__init__.py b/mediagoblin/init/__init__.py
index 4a14fdf8..44f604b1 100644
--- a/mediagoblin/init/__init__.py
+++ b/mediagoblin/init/__init__.py
@@ -18,6 +18,12 @@ import jinja2
 from mediagoblin import staticdirect
 from mediagoblin.init.config import (
     read_mediagoblin_config, generate_validation_report)
+from mediagoblin import mg_globals
+from mediagoblin.mg_globals import setup_globals
+from mediagoblin.db.open import setup_connection_and_db_from_config
+from mediagoblin.db.util import MigrationManager
+from mediagoblin.workbench import WorkbenchManager
+from mediagoblin.storage import storage_system_from_config
 
 
 class Error(Exception): pass
@@ -33,8 +39,46 @@ def setup_global_and_app_config(config_path):
     if validation_report:
         raise ImproperlyConfigured(validation_report)
 
+    setup_globals(
+        app_config=app_config,
+        global_config=global_config)
+
     return global_config, app_config
 
+
+def setup_database():
+    app_config = mg_globals.app_config
+
+    # This MUST be imported so as to set up the appropriate migrations!
+    from mediagoblin.db import migrations
+
+    # Set up the database
+    connection, db = setup_connection_and_db_from_config(app_config)
+
+    # Init the migration number if necessary
+    migration_manager = MigrationManager(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():
+        db_migration_num = migration_manager.database_current_migration()
+        latest_migration_num = migration_manager.latest_migration()
+        if db_migration_num < latest_migration_num:
+            print (
+                "*WARNING:* Your migrations are out of date, "
+                "maybe run ./bin/gmg migrate?")
+        elif db_migration_num > latest_migration_num:
+            print (
+                "*WARNING:* Your migrations are out of date... "
+                "in fact they appear to be from the future?!")
+
+    setup_globals(
+        db_connection = connection,
+        database = db)
+
+    return connection, db
+
+
 def get_jinja_loader(user_template_path=None):
     """
     Set up the Jinja template loaders, possibly allowing for user
@@ -65,3 +109,29 @@ def get_staticdirector(app_config):
         raise ImproperlyConfigured(
             "One of direct_remote_path or "
             "direct_remote_paths must be provided")
+
+
+def setup_storage():
+    global_config = mg_globals.global_config
+
+    key_short = 'publicstore'
+    key_long = "storage:" + key_short
+    public_store = storage_system_from_config(global_config[key_long])
+
+    key_short = 'queuestore'
+    key_long = "storage:" + key_short
+    queue_store = storage_system_from_config(global_config[key_long])
+
+    setup_globals(
+        public_store = public_store,
+        queue_store = queue_store)
+
+    return public_store, queue_store
+
+
+def setup_workbench():
+    app_config = mg_globals.app_config
+
+    workbench_manager = WorkbenchManager(app_config['workbench_path'])
+
+    setup_globals(workbench_manager = workbench_manager)