diff options
author | Boris Bobrov <breton@cynicmansion.ru> | 2016-03-08 17:12:49 +0300 |
---|---|---|
committer | Boris Bobrov <breton@cynicmansion.ru> | 2016-03-08 18:23:45 +0300 |
commit | 17aaab69075f33e8e31f5fbf5bcad54882aa244b (patch) | |
tree | cc0baca0d256b91e36f6b766336431a5587c677b /mediagoblin/gmg_commands/alembic_commands.py | |
parent | fec916dfbeedd8798c67a7512147b343dd046f90 (diff) | |
download | mediagoblin-17aaab69075f33e8e31f5fbf5bcad54882aa244b.tar.lz mediagoblin-17aaab69075f33e8e31f5fbf5bcad54882aa244b.tar.xz mediagoblin-17aaab69075f33e8e31f5fbf5bcad54882aa244b.zip |
Fix module name for `gmg alembic` command
gmg_commands.alembic attempted to import from `alembic`. On python2.7
it resulted in import from itself.
Rename alembic.py to alembic_commands.py and fix references to
gmg_commands.alembic.
Diffstat (limited to 'mediagoblin/gmg_commands/alembic_commands.py')
-rw-r--r-- | mediagoblin/gmg_commands/alembic_commands.py | 56 |
1 files changed, 56 insertions, 0 deletions
diff --git a/mediagoblin/gmg_commands/alembic_commands.py b/mediagoblin/gmg_commands/alembic_commands.py new file mode 100644 index 00000000..f255af73 --- /dev/null +++ b/mediagoblin/gmg_commands/alembic_commands.py @@ -0,0 +1,56 @@ +# 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) + # This code is inspired by a hack in Alembic, but isn't the same really. + # Regardless, Alembic is Expat licensed. + 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) |