diff --git a/flake8_isort.py b/flake8_isort.py index ab96cce..26936ee 100644 --- a/flake8_isort.py +++ b/flake8_isort.py @@ -24,19 +24,17 @@ def __init__(self, tree, filename, lines): self.filename = filename self.lines = lines - @classmethod - def add_options(cls, parser): - parser.add_option( + def add_options(option_manager): + option_manager.add_option( '--isort-show-traceback', action='store_true', parse_from_config=True, help='Show full traceback with diff from isort', ) - @classmethod - def parse_options(cls, options): - cls.stdin_display_name = options.stdin_display_name - cls.show_traceback = options.isort_show_traceback + def parse_options(self, option_manager, options, args): + self.stdin_display_name = options.stdin_display_name + self.show_traceback = options.isort_show_traceback class Flake8Isort4(Flake8IsortBase): diff --git a/run_tests.py b/run_tests.py index 7ecf98e..38ac9d5 100644 --- a/run_tests.py +++ b/run_tests.py @@ -223,7 +223,7 @@ def test_isort_formatted_output(tmpdir): diff = ' from __future__ import division\n+\n import os' checker = Flake8Isort(None, file_path, lines) - checker.parse_options(options(None, True, 'stdin')) + checker.parse_options(None, options(None, True, 'stdin'), None) ret = list(checker.run()) assert len(ret) == 1 assert ret[0][0] == 3