datatracker/ietf/submit/management/commands/purge_yang_checks.py
Robert Sparks 95cb3f041e Merged in [19965] from rjsparks@nostrum.com:
Remove unused imports.
 - Legacy-Id: 19966
Note: SVN reference [19965] has been migrated to Git commit 70cd714fac087b5b5257708541d8ae4de8c5a6a6
2022-02-22 23:12:21 +00:00

29 lines
933 B
Python

# Copyright The IETF Trust 2022 All Rights Reserved
from tqdm import tqdm
from django.core.management.base import BaseCommand
from ietf.submit.models import Submission, SubmissionCheck
class Command(BaseCommand):
help = ("Remove all but the first and last yangchecks for each Submission")
def handle(self, *args, **options):
print("Identifying purgeable SubmissionChecks")
keep = set()
for submission in tqdm(Submission.objects.all()):
qs = submission.checks.filter(checker="yang validation")
if qs.count() == 0:
continue
qs = qs.order_by("time")
keep.add(qs.first().pk)
keep.add(qs.last().pk)
keep.discard(None)
print("Purging SubmissionChecks")
print(
SubmissionCheck.objects.filter(checker="yang validation")
.exclude(pk__in=list(keep))
.delete()
)