Skip to content

Commit

Permalink
Remove default option, add Default string to INFO
Browse files Browse the repository at this point in the history
  • Loading branch information
renatoalmeidaoliveira committed Jan 28, 2025
1 parent b53d1f5 commit fd407f9
Show file tree
Hide file tree
Showing 4 changed files with 13 additions and 16 deletions.
6 changes: 2 additions & 4 deletions netbox/extras/choices.py
Original file line number Diff line number Diff line change
Expand Up @@ -155,24 +155,22 @@ class JournalEntryKindChoices(ChoiceSet):
class LogLevelChoices(ChoiceSet):

LOG_DEBUG = 'debug'
LOG_DEFAULT = 'default'
LOG_INFO = 'info'
LOG_SUCCESS = 'success'
LOG_WARNING = 'warning'
LOG_FAILURE = 'failure'

CHOICES = (
(LOG_DEFAULT, _('Default'), 'gray'),
(LOG_DEBUG, _('Debug'), 'teal'),
(LOG_INFO, _('Info'), 'cyan'),
(LOG_SUCCESS, _('Success'), 'green'),
(LOG_WARNING, _('Warning'), 'yellow'),
(LOG_FAILURE, _('Failure'), 'red'),
(LOG_DEBUG, _('Debug'), 'teal'),

)

SYSTEM_LEVELS = {
LOG_DEBUG: logging.DEBUG,
LOG_DEFAULT: logging.INFO,
LOG_INFO: logging.INFO,
LOG_SUCCESS: logging.INFO,
LOG_WARNING: logging.WARNING,
Expand Down
9 changes: 4 additions & 5 deletions netbox/extras/constants.py
Original file line number Diff line number Diff line change
Expand Up @@ -138,9 +138,8 @@

LOG_LEVEL_RANK = {
LogLevelChoices.LOG_DEBUG: 0,
LogLevelChoices.LOG_DEFAULT: 1,
LogLevelChoices.LOG_INFO: 2,
LogLevelChoices.LOG_SUCCESS: 3,
LogLevelChoices.LOG_WARNING: 4,
LogLevelChoices.LOG_FAILURE: 5,
LogLevelChoices.LOG_INFO: 1,
LogLevelChoices.LOG_SUCCESS: 2,
LogLevelChoices.LOG_WARNING: 3,
LogLevelChoices.LOG_FAILURE: 4,
}
12 changes: 6 additions & 6 deletions netbox/extras/views.py
Original file line number Diff line number Diff line change
Expand Up @@ -1315,17 +1315,17 @@ def get_table(self, job, request, bulk_actions=True):
index = 0

try:
log_threshold = LOG_LEVEL_RANK[request.GET.get('log_threshold', LogLevelChoices.LOG_DEFAULT)]
log_threshold = LOG_LEVEL_RANK[request.GET.get('log_threshold', LogLevelChoices.LOG_INFO)]
except KeyError:
log_threshold = LOG_LEVEL_RANK[LogLevelChoices.LOG_DEFAULT]
log_threshold = LOG_LEVEL_RANK[LogLevelChoices.LOG_INFO]
if job.data:

if 'log' in job.data:
if 'tests' in job.data:
tests = job.data['tests']

for log in job.data['log']:
log_level = LOG_LEVEL_RANK.get(log.get('status'), LogLevelChoices.LOG_DEFAULT)
log_level = LOG_LEVEL_RANK.get(log.get('status'), LogLevelChoices.LOG_INFO)
if log_level >= log_threshold:
index += 1
result = {
Expand All @@ -1348,7 +1348,7 @@ def get_table(self, job, request, bulk_actions=True):
for method, test_data in tests.items():
if 'log' in test_data:
for time, status, obj, url, message in test_data['log']:
log_level = LOG_LEVEL_RANK.get(status, LogLevelChoices.LOG_DEFAULT)
log_level = LOG_LEVEL_RANK.get(status, LogLevelChoices.LOG_INFO)
if log_level >= log_threshold:
index += 1
result = {
Expand All @@ -1374,9 +1374,9 @@ def get(self, request, **kwargs):
if job.completed:
table = self.get_table(job, request, bulk_actions=False)

log_threshold = request.GET.get('log_threshold', LogLevelChoices.LOG_DEFAULT)
log_threshold = request.GET.get('log_threshold', LogLevelChoices.LOG_INFO)
if log_threshold not in LOG_LEVEL_RANK:
log_threshold = LogLevelChoices.LOG_DEFAULT
log_threshold = LogLevelChoices.LOG_INFO

context = {
'script': job.object,
Expand Down
2 changes: 1 addition & 1 deletion netbox/templates/extras/script_result.html
Original file line number Diff line number Diff line change
Expand Up @@ -53,7 +53,7 @@
<div class="dropdown-menu">
{% for level, name in log_levels.items %}
<a class="dropdown-item d-flex justify-content-between" href="{% url 'extras:script_result' job_pk=job.pk %}?log_threshold={{ level }}">
{{ name }}{% if forloop.last %} ({% trans "All" %}){% endif %}
{{ name }}{% if forloop.counter == 1 %} ({% trans "All" %}){% elif forloop.counter == 2 %} ({% trans "Default" %}){% endif %}
{% if level == log_threshold %}<span class="badge bg-green ms-auto"></span>{% endif %}
</a>
{% endfor %}
Expand Down

0 comments on commit fd407f9

Please sign in to comment.