Skip to content

Commit 16994eb

Browse files
committed
Use iterator in vulnerability_status improver
Signed-off-by: Keshav Priyadarshi <[email protected]>
1 parent 02d75b2 commit 16994eb

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

vulnerabilities/improvers/vulnerability_status.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -10,7 +10,6 @@
1010
from typing import Iterable
1111
from urllib.parse import urljoin
1212

13-
from django.db.models import Q
1413
from django.db.models.query import QuerySet
1514

1615
from vulnerabilities.importer import AdvisoryData
@@ -37,7 +36,9 @@ class VulnerabilityStatusImprover(Improver):
3736

3837
@property
3938
def interesting_advisories(self) -> QuerySet:
40-
return Advisory.objects.filter(Q(created_by=NVDImporterPipeline.pipeline_id)).paginated()
39+
return Advisory.objects.filter(created_by=NVDImporterPipeline.pipeline_id).iterator(
40+
chunk_size=5000
41+
)
4142

4243
def get_inferences(self, advisory_data: AdvisoryData) -> Iterable[Inference]:
4344
"""

0 commit comments

Comments
 (0)