Skip to content

Commit d95a618

Browse files
committed
Minor style change and code cleanup
Signed-off-by: Shivam Sandbhor <shivam.sandbhor@gmail.com>
1 parent 1857855 commit d95a618

1 file changed

Lines changed: 3 additions & 7 deletions

File tree

vulnerabilities/import_runner.py

Lines changed: 3 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -24,7 +24,6 @@
2424
import dataclasses
2525
import datetime
2626
import logging
27-
from itertools import chain
2827
from typing import Dict
2928
from typing import List
3029
from typing import Set
@@ -114,13 +113,11 @@ def run(self, cutoff_date: datetime.datetime = None) -> None:
114113
from all Linux distributions that package this kernel version.
115114
"""
116115
logger.debug(f'Starting import for {self.importer.name}.')
117-
data_source = self.importer.make_data_source(
118-
self.batch_size, cutoff_date=cutoff_date)
116+
data_source = self.importer.make_data_source(self.batch_size, cutoff_date=cutoff_date)
119117
with data_source:
120118
_process_added_advisories(data_source)
121119
_process_updated_advisories(data_source)
122-
self.importer.last_run = datetime.datetime.now(
123-
tz=datetime.timezone.utc)
120+
self.importer.last_run = datetime.datetime.now(tz=datetime.timezone.utc)
124121
self.importer.data_source_cfg = dataclasses.asdict(data_source.config)
125122
self.importer.save()
126123

@@ -170,8 +167,7 @@ def _process_added_advisories(data_source: DataSource) -> None:
170167

171168
vulnerabilities = _insert_vulnerabilities_and_references(batch)
172169

173-
_bulk_insert_impacted_and_resolved_packages(
174-
batch, vulnerabilities, impacted, resolved)
170+
_bulk_insert_impacted_and_resolved_packages(batch, vulnerabilities, impacted, resolved)
175171
except (DataError, RuntimeError) as e:
176172
# FIXME This exception might happen when the max. length of a DB column is exceeded.
177173
# Skipping an entire batch because one version number might be too long is obviously a

0 commit comments

Comments
 (0)