zydronium/manage.get.gov
1
0
Fork 0
mirror of https://github.com/cisagov/manage.get.gov.git synced 2025-07-30 22:46:30 +02:00
Code Issues Projects Releases Packages Wiki Activity

Add unit test

Browse source
This commit is contained in:
zandercymatics 2024-01-08 11:52:09 -07:00
parent fe29eaa9fb
commit 0f92f588a8
No known key found for this signature in database
GPG key ID: FF4636ABEC9682B7
3 changed files with 34 additions and 17 deletions
Download patch file Download diff file Expand all files Collapse all files

8
src/registrar/management/commands/utility/terminal_helper.py
View file

@ -48,7 +48,7 @@ class ScriptDataHelper:
"""Helper method with utilities to speed up development of scripts that do DB operations"""
@staticmethod
def bulk_update_fields(model_class, update_list, batch_size=1000):
def bulk_update_fields(model_class, update_list, fields_to_update, batch_size=1000):
"""
This function performs a bulk update operation on a specified Django model class in batches.
It uses Django's Paginator to handle large datasets in a memory-efficient manner.
@ -64,6 +64,8 @@ class ScriptDataHelper:
Defaults to 1000. If you're dealing with models that have a large number of fields,
or large field values, you may need to decrease this value to prevent out-of-memory errors.
fields_to_update:
Usage:
bulk_update_fields(Domain, page.object_list, ["first_ready"])
"""
@ -72,10 +74,10 @@ class ScriptDataHelper:
paginator = Paginator(update_list, batch_size)
for page_num in paginator.page_range:
page = paginator.page(page_num)
model_class.objects.bulk_update(page.object_list, update_list)
model_class.objects.bulk_update(page.object_list, fields_to_update)
class TerminalHelper:
@staticmethod
def log_script_run_summary(to_update, failed_to_update, skipped, debug: bool):
"""Prints success, failed, and skipped counts, as well as