Skip to content

Commit 6ef62fa

Browse files
committed
Rename DataSource to Improver, blackify, ignore tests
This is required to merge the PR to main for a clean slate for new contributions for importer-improver migration Signed-off-by: Hritik Vijay <hritikxx8@gmail.com>
1 parent 8298b4c commit 6ef62fa

2 files changed

Lines changed: 11 additions & 12 deletions

File tree

pytest.ini

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -32,5 +32,6 @@ addopts =
3232
--ignore=vulnerabilities/importers/ubuntu_usn.py
3333
--ignore=vulnerabilities/importers/mozilla.py
3434
--ignore=vulnerabilities/importers/mattermost.py
35+
--ignore=vulnerabilities/importers/xen.py
3536
--ignore=vulnerabilities/management/commands/create_cpe_to_purl_map.py
3637
--ignore=vulnerabilities/lib_oval.py

vulnerabilities/importers/xen.py

Lines changed: 10 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -27,20 +27,14 @@
2727
import requests
2828
from packageurl import PackageURL
2929

30-
from vulnerabilities.data_source import DataSource
31-
from vulnerabilities.data_source import Advisory
32-
from vulnerabilities.data_source import Reference
30+
from vulnerabilities.importer import Importer
31+
from vulnerabilities.importer import Advisory
32+
from vulnerabilities.importer import Reference
3333
from vulnerabilities.helpers import create_etag
3434
from vulnerabilities.helpers import is_cve
3535

3636

37-
@dataclasses.dataclass
38-
class XenDBConfiguration:
39-
etags: list
40-
db_url: str
41-
42-
43-
class XenDataSource(DataSource):
37+
class XenImporter(Importer):
4438
CONFIG_CLASS = XenDBConfiguration
4539

4640
def updated_advisories(self):
@@ -83,8 +77,12 @@ def to_advisories(xen_db):
8377

8478

8579
def get_xen_references(xsa_id):
86-
return Reference(reference_id="XSA-" + xsa_id, url="https://xenbits.xen.org/xsa/advisory-{}.html".format(xsa_id))
80+
return Reference(
81+
reference_id="XSA-" + xsa_id,
82+
url="https://xenbits.xen.org/xsa/advisory-{}.html".format(xsa_id),
83+
)
84+
8785

8886
def fetch(url):
8987
response = requests.get(url).content
90-
return json.loads(response)
88+
return json.loads(response)

0 commit comments

Comments
 (0)