Skip to content

Commit 798d3be

Browse files
committed
Support Python 2.7 idioms #361
Do not mix return and generators in functions for Py2 Signed-off-by: Philippe Ombredanne <pombredanne@nexb.com>
1 parent 9d44dca commit 798d3be

1 file changed

Lines changed: 3 additions & 10 deletions

File tree

src/attributecode/transform.py

Lines changed: 3 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -94,7 +94,8 @@ def transform_data(rows, transformer):
9494
if errors:
9595
return column_names, data, errors
9696

97-
data = transformer.filter_rows(data)
97+
if transformer.row_filters:
98+
data = list(transformer.filter_rows(data))
9899

99100
return column_names, data, errors
100101

@@ -253,12 +254,7 @@ def filter_columns(self, data):
253254
columns with a name in the `column_filters`of this Transformer.
254255
Return the data unchanged if no `column_filters` exists.
255256
"""
256-
column_filters = self.column_filters
257-
if not column_filters:
258-
return data
259-
260-
column_filters = set(self.clean_columns(column_filters))
261-
257+
column_filters = set(self.clean_columns(self.column_filters))
262258
for entry in data:
263259
items = ((k, v) for k, v in entry.items() if k in column_filters)
264260
yield OrderedDict(items)
@@ -271,9 +267,6 @@ def filter_rows(self, data):
271267
Transformer.
272268
"""
273269
filters = self.row_filters
274-
if not filters:
275-
return data
276-
277270
for entry in data:
278271
for filt in filters:
279272
for filtered_column_name, filtered_column_value in filt.items():

0 commit comments

Comments
 (0)