From 844fbc3c1a8372ed3889f73d55e505c76f59c8b0 Mon Sep 17 00:00:00 2001
From: David-Wobrock <david.wobrock@gmail.com>
Date: Wed, 22 Nov 2017 17:07:03 +0100
Subject: [PATCH] Add pycodestyle and flake8 tests to Travis

Refactor style issues for pycodestyle tests to pass
---
 .travis.yml                                 | 2 ++
 django_migration_linter/migration_linter.py | 6 ++++--
 setup.py                                    | 4 +++-
 3 files changed, 9 insertions(+), 3 deletions(-)

diff --git a/.travis.yml b/.travis.yml
index 10973e3..110fb48 100644
--- a/.travis.yml
+++ b/.travis.yml
@@ -5,6 +5,8 @@ python:
   - "3.5"
   - "3.6"
 script:
+  - pycodestyle django_migration_linter
+  - flake8 django_migration_linter
   - py.test
 install:
   - pip install -e ".[test]"
diff --git a/django_migration_linter/migration_linter.py b/django_migration_linter/migration_linter.py
index 40f388f..38de743 100644
--- a/django_migration_linter/migration_linter.py
+++ b/django_migration_linter/migration_linter.py
@@ -109,7 +109,8 @@ class MigrationLinter:
         diff_process.wait()
         if diff_process.returncode != 0:
             output = []
-            for line in map(clean_bytes_to_str, diff_process.stderr.readlines()):
+            for line in map(
+                    clean_bytes_to_str, diff_process.stderr.readlines()):
                 output.append(line)
             log.info("Error while git diff command:\n{}".format(
                 "".join(output)))
@@ -187,7 +188,8 @@ class MigrationLinter:
         sqlmigrate_process = Popen(
             git_diff_command, shell=True, stdout=PIPE, stderr=PIPE)
         sql_statements = []
-        for line in map(clean_bytes_to_str, sqlmigrate_process.stdout.readlines()):
+        for line in map(
+                clean_bytes_to_str, sqlmigrate_process.stdout.readlines()):
             if not line.startswith('--'):  # Ignore comments
                 sql_statements.append(line)
         sqlmigrate_process.wait()
diff --git a/setup.py b/setup.py
index 7fa637c..f5a7f85 100644
--- a/setup.py
+++ b/setup.py
@@ -28,7 +28,9 @@ install_requirements = [
 test_requirements = [
     'pytest',
     'django',
-    'django-fake-database-backends'
+    'django-fake-database-backends',
+    'pycodestyle',
+    'flake8'
 ]
 
 setup(