diff options
Diffstat (limited to 'animism-align/cli/cli.py')
| -rwxr-xr-x | animism-align/cli/cli.py | 9 |
1 files changed, 8 insertions, 1 deletions
diff --git a/animism-align/cli/cli.py b/animism-align/cli/cli.py index 2158398..0913cd1 100755 --- a/animism-align/cli/cli.py +++ b/animism-align/cli/cli.py @@ -15,7 +15,7 @@ if __name__ == '__main__': # argparse: intercept group argv_tmp = sys.argv sys.argv = sys.argv[:2] - ap = argparse.ArgumentParser('\033[1m\033[94mSwimmer\033[0m') + ap = argparse.ArgumentParser('\033[1m\033[94mAnimism\033[0m') ap.add_argument('group', choices=app_cfg.CLICK_GROUPS.keys()) args = ap.parse_args() sys.argv = argv_tmp @@ -29,6 +29,13 @@ if __name__ == '__main__': cli = FlaskGroup(create_app=create_app) + elif args.group == 'viewer': + + from flask.cli import FlaskGroup + from app.server.viewer import create_app + + cli = FlaskGroup(create_app=create_app) + elif args.group == 'db': import re |