|
24 | 24 | import dataclasses |
25 | 25 | import datetime |
26 | 26 | import logging |
27 | | -from itertools import chain |
28 | 27 | from typing import Dict |
29 | 28 | from typing import List |
30 | 29 | from typing import Set |
@@ -114,13 +113,11 @@ def run(self, cutoff_date: datetime.datetime = None) -> None: |
114 | 113 | from all Linux distributions that package this kernel version. |
115 | 114 | """ |
116 | 115 | 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) |
119 | 117 | with data_source: |
120 | 118 | _process_added_advisories(data_source) |
121 | 119 | _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) |
124 | 121 | self.importer.data_source_cfg = dataclasses.asdict(data_source.config) |
125 | 122 | self.importer.save() |
126 | 123 |
|
@@ -170,8 +167,7 @@ def _process_added_advisories(data_source: DataSource) -> None: |
170 | 167 |
|
171 | 168 | vulnerabilities = _insert_vulnerabilities_and_references(batch) |
172 | 169 |
|
173 | | - _bulk_insert_impacted_and_resolved_packages( |
174 | | - batch, vulnerabilities, impacted, resolved) |
| 170 | + _bulk_insert_impacted_and_resolved_packages(batch, vulnerabilities, impacted, resolved) |
175 | 171 | except (DataError, RuntimeError) as e: |
176 | 172 | # FIXME This exception might happen when the max. length of a DB column is exceeded. |
177 | 173 | # Skipping an entire batch because one version number might be too long is obviously a |
|
0 commit comments