Current location

narf Source control manager Git

summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorChristopher Allan Webber <cwebber@dustycloud.org>2016-03-02 17:03:04 -0800
committerChristopher Allan Webber <cwebber@dustycloud.org>2016-03-03 09:31:00 -0800
commit399272eea5c36f8337621575a498f4f0f2686f8f (patch)
tree1f2068f8b34e0797ed374547e98e4822e4d62fb2
parenta48736fc09db8ebe2e90a2e0b7bbd5fbf4083f36 (diff)
Add "gmg raw_alembic" command.
This allows you to dispatch to the "alembic" command line tool properly, but properly respecting MediaGoblin's own configuration.
-rw-r--r--mediagoblin/gmg_commands/__init__.py6
-rw-r--r--mediagoblin/gmg_commands/alembic.py54
2 files changed, 60 insertions, 0 deletions
diff --git a/mediagoblin/gmg_commands/__init__.py b/mediagoblin/gmg_commands/__init__.py
index e5f9a395..b2c8cd6b 100644
--- a/mediagoblin/gmg_commands/__init__.py
+++ b/mediagoblin/gmg_commands/__init__.py
@@ -76,6 +76,12 @@ SUBCOMMAND_MAP = {
'setup': 'mediagoblin.gmg_commands.batchaddmedia:parser_setup',
'func': 'mediagoblin.gmg_commands.batchaddmedia:batchaddmedia',
'help': 'Add many media entries at once'},
+ 'alembic': {
+ 'setup': 'mediagoblin.gmg_commands.alembic:parser_setup',
+ 'func': 'mediagoblin.gmg_commands.alembic:raw_alembic_cli',
+ 'help': (
+ 'Run raw alembic commands with our local database. '
+ '(Unless you know what you\'re doing, use dbupdate instead!)')},
# 'theme': {
# 'setup': 'mediagoblin.gmg_commands.theme:theme_parser_setup',
# 'func': 'mediagoblin.gmg_commands.theme:theme',
diff --git a/mediagoblin/gmg_commands/alembic.py b/mediagoblin/gmg_commands/alembic.py
new file mode 100644
index 00000000..76191876
--- /dev/null
+++ b/mediagoblin/gmg_commands/alembic.py
@@ -0,0 +1,54 @@
+# GNU MediaGoblin -- federated, autonomous media hosting
+# Copyright (C) 2011, 2012 MediaGoblin contributors. See AUTHORS.
+#
+# 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 argparse
+import os
+
+from alembic import config
+from sqlalchemy.orm import sessionmaker
+
+from mediagoblin.db.open import setup_connection_and_db_from_config
+from mediagoblin.init import setup_global_and_app_config
+
+
+class FudgedCommandLine(config.CommandLine):
+ def main(self, args, db):
+ options = self.parser.parse_args(args.args_for_alembic)
+ if not hasattr(options, "cmd"):
+ print(
+ "* Only use this command if you know what you are doing! *\n"
+ "If not, use the 'gmg dbupdate' command instead.\n\n"
+ "Alembic help:\n")
+ self.parser.print_help()
+ return
+ else:
+ Session = sessionmaker(bind=db.engine)
+
+ root_dir = os.path.abspath(os.path.dirname(os.path.dirname(
+ os.path.dirname(__file__))))
+ alembic_cfg_path = os.path.join(root_dir, 'alembic.ini')
+ cfg = config.Config(alembic_cfg_path,
+ cmd_opts=options)
+ cfg.attributes["session"] = Session()
+ self.run_cmd(cfg, options)
+
+def parser_setup(subparser):
+ subparser.add_argument("args_for_alembic", nargs=argparse.REMAINDER)
+
+def raw_alembic_cli(args):
+ global_config, app_config = setup_global_and_app_config(args.conf_file)
+ db = setup_connection_and_db_from_config(app_config, migrations=False)
+ FudgedCommandLine().main(args, db)