Skip to content

Commit 56bfb4e

Browse files
committed
Change Number middleware to regex
This change has been made to detect words that contain numbers with themselves, because these words are not meaningful.
1 parent 521e46a commit 56bfb4e

File tree

1 file changed

+7
-10
lines changed

1 file changed

+7
-10
lines changed

middlewares/numbers.py

+7-10
Original file line numberDiff line numberDiff line change
@@ -1,3 +1,4 @@
1+
import re
12
from typing import List
23

34
from models.middleware import Middleware
@@ -6,17 +7,13 @@
67
class Number(Middleware):
78
def validate(self, data: List[str]) -> (List[str], int):
89
"""
9-
Validate the data and exclude only numbers data.
10+
validate data and exclude tokens that contain numbers with regex
1011
1112
:param data: The data to validate.
1213
:return: The validated data and the number of errors.
1314
"""
14-
errors = 0
15-
validated_data = []
16-
for item in data:
17-
try:
18-
int(item)
19-
errors += 1
20-
except ValueError:
21-
validated_data.append(item)
22-
return validated_data, errors
15+
regex: re.Pattern = re.compile(r'\d')
16+
17+
result = [token for token in data if not regex.match(token)]
18+
19+
return result, len(data) - len(result)

0 commit comments

Comments
 (0)