Add Progress bar to RedhatImporter

Signed-off-by: Harsh Mishra <hmisraji07@gmail.com>

Author: harsh098

vulnerabilities/importers/redhat.py

@@ -15,6 +15,7 @@
 
 import requests
 from packageurl import PackageURL
+from progress.bar import ChargingBar
 from univers.version_range import RpmVersionRange
 
 from vulnerabilities import severity_systems
@@ -33,7 +34,6 @@
 # FIXME: we should use a centralized retry
 requests_session = requests_with_5xx_retry(max_retries=5, backoff_factor=1)
 
-
 def fetch_cves() -> Iterable[List[Dict]]:
     page_no = 1
     cve_data = None
@@ -68,9 +68,17 @@ class RedhatImporter(Importer):
     importer_name = "RedHat Importer"
 
     def advisory_data(self) -> Iterable[AdvisoryData]:
-        for redhat_cves in fetch_cves():
-            for redhat_cve in redhat_cves:
-                yield to_advisory(redhat_cve)
+        try:
+            cves = list(fetch_cves())
+            progress_bar_for_package_fetch = ChargingBar("\tFetching CVEs", max=cves)
+            try:
+                for redhat_cves in cves:
+                    for redhat_cve in redhat_cves:
+                        yield to_advisory(redhat_cve)
+            finally:
+                progress_bar_for_package_fetch.next()
+        finally:
+            progress_bar_for_package_fetch.finish()
 
 
 def to_advisory(advisory_data):