From 9f5871b0251ee9ff5ce250096bd03c54c7611685 Mon Sep 17 00:00:00 2001 From: Jovial Joe Jayarson Date: Thu, 4 Apr 2024 12:45:19 +0530 Subject: [PATCH] fix: reduce memory footprint when loading TLDs --- src/validators/domain.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/src/validators/domain.py b/src/validators/domain.py index 9051462b..0e8c0c48 100644 --- a/src/validators/domain.py +++ b/src/validators/domain.py @@ -9,9 +9,11 @@ def _iana_tld(): - """Load IANA TLDs.""" + """Load IANA TLDs as a Generator.""" with Path(__file__).parent.joinpath("_tld.txt").open() as tld_f: - return (line.strip() for line in tld_f.readlines()[1:]) + _ = next(tld_f) # ignore the first line + for line in tld_f: + yield line.strip() @validator