zydronium/manage.get.gov
1
0
Fork 0
mirror of https://github.com/cisagov/manage.get.gov.git synced 2025-08-13 04:59:59 +02:00
Code Issues Projects Releases Packages Wiki Activity

build on the terminal helper a bit

Browse source
This commit is contained in:
matthewswspence 2024-08-26 16:14:17 -05:00
parent 4026e9be86
commit 7dafc863c0
No known key found for this signature in database
GPG key ID: FB458202A7852BA4
2 changed files with 6 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

5
src/registrar/management/commands/update_first_ready.py
View file

@ -19,9 +19,10 @@ class Command(BaseCommand, PopulateScriptTemplate):
record.first_ready = record.created_at.date() record.first_ready = record.created_at.date()
logger.info( logger.info(
f"{TerminalColors.OKCYAN}Updating {record} => first_ready: " f"{record.first_ready}{TerminalColors.OKCYAN}" f"{TerminalColors.OKCYAN}Updating {record} => first_ready: " f"{record.first_ready}{TerminalColors.ENDC}"
) )
# check if a transition domain object for this domain name exists, and if so whether # check if a transition domain object for this domain name exists,
# and if so whether its first_ready value matches its created_at date
def should_update(self, record: Domain) -> bool: def should_update(self, record: Domain) -> bool:
return TransitionDomain.objects.filter(domain_name=record.name).exists() and record.first_ready != record.created_at.date() return TransitionDomain.objects.filter(domain_name=record.name).exists() and record.first_ready != record.created_at.date()

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

@ -188,7 +188,9 @@ class PopulateScriptTemplate(ABC):
return f"{TerminalColors.FAIL}" f"Failed to update {record}" f"{TerminalColors.ENDC}" return f"{TerminalColors.FAIL}" f"Failed to update {record}" f"{TerminalColors.ENDC}"
def should_skip_record(self, record) -> bool: # noqa def should_skip_record(self, record) -> bool: # noqa
"""Defines the condition in which we should skip updating a record. Override as needed.""" """Defines the condition in which we should skip updating a record. Override as needed.
The difference between this and Custom_filter is that records matching these conditions
*will* be included in the run but will be skipped (and logged as such)."""
# By default - don't skip # By default - don't skip
return False return False