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/>.
19 from sqlalchemy
.orm
import sessionmaker
21 from mediagoblin
.db
.open import setup_connection_and_db_from_config
22 from mediagoblin
.db
.migration_tools
import MigrationManager
23 from mediagoblin
.init
import setup_global_and_app_config
24 from mediagoblin
.tools
.common
import import_component
26 _log
= logging
.getLogger(__name__
)
28 _log
.setLevel(logging
.DEBUG
)
30 def dbupdate_parse_setup(subparser
):
34 class DatabaseData(object):
35 def __init__(self
, name
, models
, foundations
, migrations
):
38 self
.foundations
= foundations
39 self
.migrations
= migrations
41 def make_migration_manager(self
, session
):
42 return MigrationManager(
43 self
.name
, self
.models
, self
.foundations
, self
.migrations
, session
)
46 def gather_database_data(plugins
):
48 Gather all database data relevant to the extensions we have
49 installed so we can do migrations and table initialization.
51 Returns a list of DatabaseData objects.
56 from mediagoblin
.db
.models
import MODELS
as MAIN_MODELS
57 from mediagoblin
.db
.migrations
import MIGRATIONS
as MAIN_MIGRATIONS
58 from mediagoblin
.db
.models
import FOUNDATIONS
as MAIN_FOUNDATIONS
60 managed_dbdata
.append(
62 u
'__main__', MAIN_MODELS
, MAIN_FOUNDATIONS
, MAIN_MIGRATIONS
))
64 for plugin
in plugins
:
66 models
= import_component('{0}.models:MODELS'.format(plugin
))
67 except ImportError as exc
:
68 _log
.debug('No models found for {0}: {1}'.format(
73 except AttributeError as exc
:
74 _log
.warning('Could not find MODELS in {0}.models, have you \
75 forgotten to add it? ({1})'.format(plugin
, exc
))
79 migrations
= import_component('{0}.migrations:MIGRATIONS'.format(
81 except ImportError as exc
:
82 _log
.debug('No migrations found for {0}: {1}'.format(
87 except AttributeError as exc
:
88 _log
.debug('Could not find MIGRATIONS in {0}.migrations, have you \
89 forgotten to add it? ({1})'.format(plugin
, exc
))
93 foundations
= import_component('{0}.models:FOUNDATIONS'.format(plugin
))
94 except ImportError as exc
:
95 _log
.debug('No foundations found for {0}: {1}'.format(
100 except AttributeError as exc
:
101 _log
.debug('Could not find FOUNDATIONS in {0}.models, have you \
102 forgotten to add it? ({1})'.format(plugin
, exc
))
106 managed_dbdata
.append(
107 DatabaseData(plugin
, models
, foundations
, migrations
))
110 return managed_dbdata
113 def run_dbupdate(app_config
, global_config
):
115 Initialize or migrate the database as specified by the config file.
117 Will also initialize or migrate all extensions (media types, and
118 in the future, plugins)
121 # Set up the database
122 db
= setup_connection_and_db_from_config(app_config
, migrations
=True)
124 run_all_migrations(db
, app_config
, global_config
)
127 def run_all_migrations(db
, app_config
, global_config
):
129 Initializes or migrates a database that already has a
130 connection setup and also initializes or migrates all
131 extensions based on the config files.
133 It can be used to initialize an in-memory database for
136 # Gather information from all media managers / projects
137 dbdatas
= gather_database_data(
138 global_config
.get('plugins', {}).keys())
140 Session
= sessionmaker(bind
=db
.engine
)
142 # Setup media managers for all dbdata, run init/migrate and print info
143 # For each component, create/migrate tables
144 for dbdata
in dbdatas
:
145 migration_manager
= dbdata
.make_migration_manager(Session())
146 migration_manager
.init_or_migrate()
150 global_config
, app_config
= setup_global_and_app_config(args
.conf_file
)
151 run_dbupdate(app_config
, global_config
)