From 34c6faf6307000e0a63bdf81b0bab03a46437001 Mon Sep 17 00:00:00 2001
From: Wagner Augusto Andreoli <wagnerandreoli@gmail.com>
Date: Fri, 6 Nov 2015 01:42:50 +0100
Subject: [PATCH] Add a lint command with flake8 and isort

---
 {{cookiecutter.app_name}}/.isort.cfg |  2 ++
 {{cookiecutter.app_name}}/manage.py  | 33 ++++++++++++++++++++++------
 {{cookiecutter.app_name}}/setup.cfg  |  2 ++
 3 files changed, 30 insertions(+), 7 deletions(-)
 create mode 100644 {{cookiecutter.app_name}}/.isort.cfg
 create mode 100644 {{cookiecutter.app_name}}/setup.cfg

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