Implement alembic manage CLI script using click

This way we can make use of the reusable components of the `verdi`
command line module.

.pre-commit-config.yaml

@@ -40,7 +40,6 @@
         aiida/backends/djsite/queries.py|
         aiida/backends/profile.py|
         aiida/backends/general/abstractqueries.py|
-        aiida/backends/sqlalchemy/alembic_manage.py|
         aiida/backends/sqlalchemy/globalsettings.py|
         aiida/backends/sqlalchemy/__init__.py|
         aiida/backends/sqlalchemy/migrations/env.py|

aiida/backends/sqlalchemy/manage.py

@@ -0,0 +1,88 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+###########################################################################
+# Copyright (c), The AiiDA team. All rights reserved.                     #
+# This file is part of the AiiDA code.                                    #
+#                                                                         #
+# The code is hosted on GitHub at https://github.com/aiidateam/aiida_core #
+# For further information on the license, see the LICENSE.txt file        #
+# For further information please visit http://www.aiida.net               #
+###########################################################################
+# pylint:
+"""Simple wrapper around the alembic command line tool that first loads an AiiDA profile."""
+from __future__ import division
+from __future__ import print_function
+from __future__ import absolute_import
+
+import alembic
+import click
+
+from aiida.cmdline.params import options
+
+
+def execute_alembic_command(command_name, **kwargs):
+    """Execute an Alembic CLI command.
+
+    :param command_name: the sub command name
+    :param kwargs: parameters to pass to the command
+    """
+    from aiida.backends import sqlalchemy as sa
+    from aiida.backends.sqlalchemy.utils import get_alembic_conf
+
+    with sa.ENGINE.begin() as connection:
+        alembic_cfg = get_alembic_conf()
+        alembic_cfg.attributes['connection'] = connection  # pylint: disable=unsupported-assignment-operation
+        command = getattr(alembic.command, command_name)
+        command(alembic_cfg, **kwargs)
+
+
+@click.group()
+@options.PROFILE(required=True)
+def alembic_cli(profile):
+    """Simple wrapper around the alembic command line tool that first loads an AiiDA profile."""
+    from aiida.manage.configuration import load_profile
+    from aiida.manage.manager import get_manager
+
+    load_profile(profile=profile.name)
+    manager = get_manager()
+    manager._load_backend(schema_check=False)  # pylint: disable=protected-access
+
+
+@alembic_cli.command()
+@click.argument('message')
+def alembic_revision(message):
+    """Create a new database revision."""
+    execute_alembic_command('revision', message=message, autogenerate=True)
+
+
+@alembic_cli.command()
+@options.VERBOSE()
+def alembic_current(verbose):
+    """Show the current revision."""
+    execute_alembic_command('current', verbose=verbose)
+
+
+@alembic_cli.command()
+@click.option('-r', '--rev-range')
+@options.VERBOSE()
+def alembic_history(rev_range, verbose):
+    """Show the history for the given revision range."""
+    execute_alembic_command('history', rev_range=rev_range, verbose=verbose)
+
+
+@alembic_cli.command()
+@click.argument('revision', type=click.STRING)
+def alembic_upgrade(revision):
+    """Upgrade the database to the given REVISION."""
+    execute_alembic_command('upgrade', revision=revision)
+
+
+@alembic_cli.command()
+@click.argument('revision', type=click.STRING)
+def alembic_downgrade(revision):
+    """Downgrade the database to the given REVISION."""
+    execute_alembic_command('downgrade', revision=revision)
+
+
+if __name__ == '__main__':
+    alembic_cli()  # pylint: disable=no-value-for-parameter