Skip to content

Commit ea44255

Browse files
Adding pagination to redissemination cmd
1 parent 087690e commit ea44255

1 file changed

Lines changed: 6 additions & 4 deletions

File tree

backend/dissemination/management/commands/delete_and_regenerate_dissemination_from_intake.py

Lines changed: 6 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -18,6 +18,8 @@
1818

1919
logger = logging.getLogger(__name__)
2020

21+
CHUNK_SIZE = 500
22+
2123

2224
class Command(BaseCommand):
2325
"""
@@ -70,10 +72,8 @@ def handle(self, *args, **_kwargs):
7072
sac = SingleAuditChecklist.objects.get(report_id=report_id)
7173
sac.redisseminate()
7274
logger.info(f"Redisseminated: {report_id}.")
73-
exit(0)
7475
except SingleAuditChecklist.DoesNotExist:
7576
logger.error(f"No report with report_id found: {report_id}. Exiting.")
76-
exit(-1)
7777
else:
7878
if year:
7979
years = [year]
@@ -86,10 +86,12 @@ def handle(self, *args, **_kwargs):
8686
for year in years:
8787
logger.info(f"Working year {year}")
8888

89-
for sac in SingleAuditChecklist.objects.filter(
89+
queryset = SingleAuditChecklist.objects.filter(
9090
submission_status__in=[STATUS.DISSEMINATED, STATUS.RESUBMITTED],
9191
general_information__auditee_fiscal_period_end__startswith=f"{year}",
92-
):
92+
)
93+
94+
for sac in queryset.iterator(chunk_size=CHUNK_SIZE):
9395
logger.info(f"Redisseminating {sac.report_id}.")
9496
sac.redisseminate()
9597
redisseminated[sac.report_id] = True

0 commit comments

Comments
 (0)