Skip to content

Commit

Permalink
Add typings
Browse files Browse the repository at this point in the history
  • Loading branch information
yalef committed Nov 21, 2023
1 parent 65f8090 commit 069e0d8
Showing 1 changed file with 4 additions and 4 deletions.
8 changes: 4 additions & 4 deletions import_export_extensions/resources.py
Original file line number Diff line number Diff line change
Expand Up @@ -48,14 +48,14 @@ def __init__(self, *args, **kwargs):
super().__init__()

@property
def has_skipped_errors(self):
def has_skipped_errors(self) -> bool:
"""Return True if row contain any skipped errors."""
if len(self.non_field_skipped_errors) > 0 or len(self.field_skipped_errors) > 0:
return True
return False

@property
def skipped_errors_count(self):
def skipped_errors_count(self) -> int:
"""Return count of skipped errors."""
return (
len(self.non_field_skipped_errors)
Expand All @@ -67,14 +67,14 @@ class SkippedErrorsResult(results.Result):
"""Custom result class with ability to store info about skipped rows."""

@property
def has_skipped_rows(self):
def has_skipped_rows(self) -> bool:
"""Return True if contain any skipped rows."""
if any(row.has_skipped_errors for row in self.rows):
return True
return False

@property
def skipped_rows(self):
def skipped_rows(self) -> list[SkippedErrorsRowResult]:
"""Return all rows with skipped errors."""
return list(
filter(lambda row: row.has_skipped_errors, self.rows),
Expand Down

0 comments on commit 069e0d8

Please sign in to comment.