diff --git a/{{cookiecutter.app_name}}/.isort.cfg b/{{cookiecutter.app_name}}/.isort.cfg new file mode 100644 index 0000000..8913d96 --- /dev/null +++ b/{{cookiecutter.app_name}}/.isort.cfg @@ -0,0 +1,2 @@ +[settings] +line_length=120 diff --git a/{{cookiecutter.app_name}}/manage.py b/{{cookiecutter.app_name}}/manage.py index ed8ad09..6fbb022 100644 --- a/{{cookiecutter.app_name}}/manage.py +++ b/{{cookiecutter.app_name}}/manage.py @@ -1,7 +1,10 @@ #!/usr/bin/env python # -*- coding: utf-8 -*- +"""Management script.""" import os -from flask_script import Manager, Shell, Server +from subprocess import call + +from flask_script import Manager, Shell, Server, Command from flask_script.commands import Clean, ShowUrls from flask_migrate import MigrateCommand @@ -10,7 +13,7 @@ from {{cookiecutter.app_name}}.user.models import User from {{cookiecutter.app_name}}.settings import DevConfig, ProdConfig from {{cookiecutter.app_name}}.database import db -if os.environ.get("{{cookiecutter.app_name | upper}}_ENV") == 'prod': +if os.environ.get('{{cookiecutter.app_name | upper}}_ENV') == 'prod': app = create_app(ProdConfig) else: app = create_app(DevConfig) @@ -22,9 +25,7 @@ manager = Manager(app) def _make_context(): - """Return context dict for a shell session so you can access - app, db, and the User model by default. - """ + """Return context dict for a shell session so you can access app, db, and the User model by default.""" return {'app': app, 'db': db, 'User': User} @@ -36,11 +37,29 @@ def test(): return exit_code +class Lint(Command): + """Lint and check code style.""" + + def run(self): + """Run command.""" + skip = ['requirements'] + files = [name for name in next(os.walk('.'))[1] if not name.startswith('.') and name not in skip] + + command_line = ['isort', '-rc'] + files + print('Fixing import order: %s' % ' '.join(command_line)) + call(command_line) + + command_line = ['flake8'] + files + print('Checking code style: %s' % ' '.join(command_line)) + exit(call(command_line)) + + manager.add_command('server', Server()) manager.add_command('shell', Shell(make_context=_make_context)) manager.add_command('db', MigrateCommand) -manager.add_command("urls", ShowUrls()) -manager.add_command("clean", Clean()) +manager.add_command('urls', ShowUrls()) +manager.add_command('clean', Clean()) +manager.add_command('lint', Lint()) if __name__ == '__main__': manager.run() diff --git a/{{cookiecutter.app_name}}/setup.cfg b/{{cookiecutter.app_name}}/setup.cfg new file mode 100644 index 0000000..aa079ec --- /dev/null +++ b/{{cookiecutter.app_name}}/setup.cfg @@ -0,0 +1,2 @@ +[flake8] +max-line-length=120