1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
|
from flake8_class_newline.__version__ import __version__
new_lines_on = {}
DOUBLE_QUOTE = '"'
SINGLE_QUOTE = "'"
CLASS = 'class'
METHOD = 'def '
DECORATOR = '@'
def new_line_checker(logical_line, line_number, filename):
if filename not in new_lines_on:
new_lines_on[filename] = []
if logical_line.startswith(CLASS):
new_lines_on[filename].append(line_number + 1)
if line_number in new_lines_on[filename]:
# if the line is not only whitespace and not empty
double_quotes = logical_line.strip().startswith(DOUBLE_QUOTE * 3)
quotes = logical_line.strip().startswith(SINGLE_QUOTE * 3)
is_method = logical_line.strip().startswith(METHOD)
is_decorator = logical_line.strip().startswith(DECORATOR)
is_callable = is_method or is_decorator
not_only_space = not logical_line.isspace() and logical_line
not_is_docstring = not (quotes or double_quotes)
if not_only_space and not_is_docstring and is_callable:
# this means there's something on the line
offset = line_number, 0
yield offset, 'CNL100 Class definition does not have a new line.'
new_line_checker.name = 'new_line_checker'
new_line_checker.version = __version__
|
