Skip to content

Commit 51cd275

Browse files
authored
Merge branch 'main' into update-readme
2 parents 6f14358 + d41e021 commit 51cd275

File tree

1 file changed

+9
-2
lines changed

1 file changed

+9
-2
lines changed

vulnerabilities/migrations/0071_auto_20241007_1044.py

Lines changed: 9 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -16,21 +16,28 @@ def split_packagerelatedvulnerability(apps, schema_editor):
1616
progress_step=1,
1717
logger=print,
1818
)
19+
fixing_packages = []
20+
affected_packages = []
1921
for prv in progress.iter(obsolete_package_relation_query.iterator(chunk_size=10000)):
2022
if prv.fix:
21-
FixingPackageRelatedVulnerability.objects.create(
23+
fp = FixingPackageRelatedVulnerability(
2224
package=prv.package,
2325
vulnerability=prv.vulnerability,
2426
created_by=prv.created_by,
2527
confidence=prv.confidence,
2628
)
29+
fixing_packages.append(fp)
2730
else:
28-
AffectedByPackageRelatedVulnerability.objects.create(
31+
ap = AffectedByPackageRelatedVulnerability(
2932
package=prv.package,
3033
vulnerability=prv.vulnerability,
3134
created_by=prv.created_by,
3235
confidence=prv.confidence,
3336
)
37+
affected_packages.append(ap)
38+
39+
FixingPackageRelatedVulnerability.objects.bulk_create(fixing_packages, batch_size=10000)
40+
AffectedByPackageRelatedVulnerability.objects.bulk_create(affected_packages, batch_size=10000)
3441

3542
def reverse_migration(apps, schema_editor):
3643
FixingPackageRelatedVulnerability = apps.get_model('vulnerabilities', 'FixingPackageRelatedVulnerability')

0 commit comments

Comments
 (0)