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

Modify terminal helper

Browse source
This commit is contained in:
zandercymatics 2025-01-10 14:03:44 -07:00
parent ac08b17f94
commit 2ca7cd468f
No known key found for this signature in database
GPG key ID: FF4636ABEC9682B7
2 changed files with 9 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

9
src/registrar/management/commands/create_federal_portfolio.py
View file

@ -113,9 +113,12 @@ class Command(BaseCommand):
if parse_requests or both:
domain_requests = self.handle_portfolio_requests(portfolio, federal_agency)
all_suborganizations.extend(suborganizations)
all_domains.extend(domains)
all_domain_requests.extend(domain_requests)
if suborganizations:
all_suborganizations.extend(suborganizations)
if all_domains:
all_domains.extend(domains)
if domain_requests:
all_domain_requests.extend(domain_requests)
except Exception as exec:
self.failed_portfolios.add(federal_agency)
logger.error(exec)

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

@ -442,13 +442,14 @@ class TerminalHelper:
f.write(file_contents)
@staticmethod
def colorful_logger(log_level, color, message):
def colorful_logger(log_level, color, message, exc_info=True):
"""Adds some color to your log output.
Args:
log_level: str | Logger.method -> Desired log level. ex: logger.info or "INFO"
color: str | TerminalColors -> Output color. ex: TerminalColors.YELLOW or "YELLOW"
message: str -> Message to display.
exc_info: bool -> Whether the log should print exc_info or not
"""
if isinstance(log_level, str) and hasattr(logger, log_level.lower()):
@ -462,4 +463,4 @@ class TerminalHelper:
terminal_color = color
colored_message = f"{terminal_color}{message}{TerminalColors.ENDC}"
log_method(colored_message)
log_method(colored_message, exc_info=exc_info)