add migrate and cli

alice/cli.py

@@ -0,0 +1,83 @@
+import importlib
+import os
+import sys
+from enum import Enum
+
+import click
+from click import BadParameter, ClickException
+
+sys.path.append(os.getcwd())
+
+
+class Color(str, Enum):
+    green = 'green'
+    red = 'red'
+
+
+@click.group(context_settings={'help_option_names': ['-h', '--help']})
+@click.option('-c', '--config', default='settings',
+              help='Tortoise-ORM config module, will read config variable from it, default is `settings`.')
+@click.option('-t', '--tortoise-orm', default='TORTOISE_ORM',
+              help='Tortoise-ORM config dict variable, default is `TORTOISE_ORM`.')
+@click.option('-l', '--location', default='./migrations',
+              help='Migrate store location, default is `./migrations`.')
+@click.option('--connection', default='default', help='Tortoise-ORM connection name, default is `default`.')
+@click.pass_context
+def cli(ctx, config, tortoise_orm, location, connection):
+    ctx.ensure_object(dict)
+    try:
+        config_module = importlib.import_module(config)
+        config = getattr(config_module, tortoise_orm, None)
+        if not config:
+            raise BadParameter(param_hint=['--config'],
+                               message=f'Can\'t get "{tortoise_orm}" from module "{config_module}"')
+        ctx.obj['config'] = config
+        ctx.obj['location'] = location
+        if connection not in config.get('connections').keys():
+            raise BadParameter(param_hint=['--connection'], message=f'No connection found in "{config}"')
+    except ModuleNotFoundError:
+        raise BadParameter(param_hint=['--tortoise-orm'], message=f'No module named "{config}"')
+
+
+@cli.command()
+@click.pass_context
+def migrate(ctx):
+    config = ctx.obj['config']
+
+
+@cli.command()
+@click.pass_context
+def upgrade():
+    pass
+
+
+@cli.command()
+@click.pass_context
+def downgrade():
+    pass
+
+
+@cli.command()
+@click.pass_context
+def initdb():
+    pass
+
+
+@cli.command()
+@click.option('--overwrite', type=bool, default=False, help='Overwrite old_models.py.')
+@click.pass_context
+def init(ctx, overwrite):
+    location = ctx.obj['location']
+    config = ctx.obj['config']
+    if not os.path.isdir(location) or overwrite:
+        os.mkdir(location)
+        connections = config.get('connections').keys()
+        for connection in connections:
+            dirname = os.path.join(location, connection)
+            if not os.path.isdir(dirname):
+                os.mkdir(dirname)
+                click.secho(f'Success create migrate location {dirname}', fg=Color.green)
+            if overwrite:
+                pass
+    else:
+        raise ClickException('Already inited')