1 # GNU MediaGoblin -- federated, autonomous media hosting
2 # Copyright (C) 2011 Free Software Foundation, Inc
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 mediagoblin
.db
import util
as db_util
20 from mediagoblin
.db
.open import setup_connection_and_db_from_config
21 from mediagoblin
.init
.config
import read_mediagoblin_config
23 # This MUST be imported so as to set up the appropriate migrations!
24 from mediagoblin
.db
import migrations
27 def migrate_parser_setup(subparser
):
28 subparser
.add_argument(
29 '-cf', '--conf_file', default
='mediagoblin.ini',
30 help="Config file used to set up environment")
33 def _print_started_migration(migration_number
, migration_func
):
35 "Running migration %s, '%s'... " % (
36 migration_number
, migration_func
.func_name
))
40 def _print_finished_migration(migration_number
, migration_func
):
41 sys
.stdout
.write("done.\n")
46 config
, validation_result
= read_mediagoblin_config(args
.conf_file
)
47 connection
, db
= setup_connection_and_db_from_config(
48 config
['mediagoblin'], use_pymongo
=True)
49 migration_manager
= db_util
.MigrationManager(db
)
52 print "== Clearing old indexes... =="
53 removed_indexes
= db_util
.remove_deprecated_indexes(db
)
55 for collection
, index_name
in removed_indexes
:
56 print "Removed index '%s' in collection '%s'" % (
57 index_name
, collection
)
60 print "\n== Applying migrations... =="
61 migration_manager
.migrate_new(
62 pre_callback
=_print_started_migration
,
63 post_callback
=_print_finished_migration
)
66 print "\n== Adding new indexes... =="
67 new_indexes
= db_util
.add_new_indexes(db
)
69 for collection
, index_name
in new_indexes
:
70 print "Added index '%s' to collection '%s'" % (
71 index_name
, collection
)