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

add record counter

Browse source
This commit is contained in:
zandercymatics 2025-03-17 13:38:21 -06:00
parent e4c2511ef3
commit c7146601d3
No known key found for this signature in database
GPG key ID: FF4636ABEC9682B7
2 changed files with 9 additions and 7 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -87,7 +87,7 @@ class PopulateScriptTemplate(ABC):
raise NotImplementedError
def mass_update_records(
self, object_class, filter_conditions, fields_to_update, debug=True, verbose=False
self, object_class, filter_conditions, fields_to_update, debug=True, verbose=False, show_record_count=False
):
"""Loops through each valid "object_class" object - specified by filter_conditions - and
updates fields defined by fields_to_update using update_record.
@ -117,13 +117,14 @@ class PopulateScriptTemplate(ABC):
# apply custom filter
records = self.custom_filter(records)
records_length = len(records)
readable_class_name = self.get_class_name(object_class)
# for use in the execution prompt.
proposed_changes = (
"==Proposed Changes==\n"
f"Number of {readable_class_name} objects to change: {len(records)}\n"
f"Number of {readable_class_name} objects to change: {len(records_length)}\n"
f"These fields will be updated on each record: {fields_to_update}"
)
@ -143,9 +144,11 @@ class PopulateScriptTemplate(ABC):
to_update: List[object_class] = []
to_skip: List[object_class] = []
failed_to_update: List[object_class] = []
for record in records:
for i, record in enumerate(records, start=1):
try:
if not self.should_skip_record(record):
if show_record_count:
logger.info(f"{TerminalColors.BOLD}Record {i}/{records_length}{TerminalColors.ENDC}")
self.update_record(record)
to_update.append(record)
else: