Skip to content

Commit 33c1533

Browse files
committed
Add Progress bar to OpensslImporter
Signed-off-by: Harsh Mishra <hmisraji07@gmail.com>
1 parent 1c8f6fc commit 33c1533

1 file changed

Lines changed: 15 additions & 5 deletions

File tree

vulnerabilities/importers/openssl.py

Lines changed: 15 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -16,6 +16,7 @@
1616
import requests
1717
from dateutil import parser as dateparser
1818
from packageurl import PackageURL
19+
from progress.bar import ChargingBar
1920
from univers.version_range import OpensslVersionRange
2021
from univers.versions import OpensslVersion
2122

@@ -49,11 +50,20 @@ def advisory_data(self) -> Iterable[AdvisoryData]:
4950

5051
def parse_vulnerabilities(xml_response) -> Iterable[AdvisoryData]:
5152
root = DET.fromstring(xml_response)
52-
for xml_issue in root:
53-
if xml_issue.tag == "issue":
54-
advisory = to_advisory_data(xml_issue)
55-
if advisory:
56-
yield advisory
53+
progress_bar_for_vulnerability_fetch = ChargingBar("\tFetching Vulnerabilities", max=len(root))
54+
progress_bar_for_vulnerability_fetch.start()
55+
try:
56+
for xml_issue in root:
57+
try:
58+
if xml_issue.tag == "issue":
59+
advisory = to_advisory_data(xml_issue)
60+
if advisory:
61+
yield advisory
62+
finally:
63+
progress_bar_for_vulnerability_fetch.next()
64+
finally:
65+
progress_bar_for_vulnerability_fetch.finish()
66+
5767

5868

5969
def to_advisory_data(xml_issue) -> AdvisoryData:

0 commit comments

Comments
 (0)