mirror of
https://github.com/netbox-community/netbox.git
synced 2024-05-10 07:54:54 +00:00
63 lines
2.0 KiB
Python
63 lines
2.0 KiB
Python
from django import forms
|
|
from django.utils import timezone
|
|
from django.utils.translation import gettext as _
|
|
|
|
from utilities.forms import BootstrapMixin, DateTimePicker, SelectDurationWidget
|
|
|
|
__all__ = (
|
|
'ScriptForm',
|
|
)
|
|
|
|
|
|
class ScriptForm(BootstrapMixin, forms.Form):
|
|
_commit = forms.BooleanField(
|
|
required=False,
|
|
initial=True,
|
|
label=_("Commit changes"),
|
|
help_text=_("Commit changes to the database (uncheck for a dry-run)")
|
|
)
|
|
_schedule_at = forms.DateTimeField(
|
|
required=False,
|
|
widget=DateTimePicker(),
|
|
label=_("Schedule at"),
|
|
help_text=_("Schedule execution of script to a set time"),
|
|
)
|
|
_interval = forms.IntegerField(
|
|
required=False,
|
|
min_value=1,
|
|
label=_("Recurs every"),
|
|
widget=SelectDurationWidget(),
|
|
help_text=_("Interval at which this script is re-run (in minutes)")
|
|
)
|
|
|
|
def __init__(self, *args, **kwargs):
|
|
super().__init__(*args, **kwargs)
|
|
|
|
# Annotate the current system time for reference
|
|
now = timezone.now().strftime('%Y-%m-%d %H:%M:%S')
|
|
self.fields['_schedule_at'].help_text += f' (current time: <strong>{now}</strong>)'
|
|
|
|
# Move _commit and _schedule_at to the end of the form
|
|
schedule_at = self.fields.pop('_schedule_at')
|
|
interval = self.fields.pop('_interval')
|
|
commit = self.fields.pop('_commit')
|
|
self.fields['_schedule_at'] = schedule_at
|
|
self.fields['_interval'] = interval
|
|
self.fields['_commit'] = commit
|
|
|
|
def clean__schedule_at(self):
|
|
scheduled_time = self.cleaned_data['_schedule_at']
|
|
if scheduled_time and scheduled_time < timezone.now():
|
|
raise forms.ValidationError({
|
|
'_schedule_at': _('Scheduled time must be in the future.')
|
|
})
|
|
|
|
return scheduled_time
|
|
|
|
@property
|
|
def requires_input(self):
|
|
"""
|
|
A boolean indicating whether the form requires user input (ignore the built-in fields).
|
|
"""
|
|
return bool(len(self.fields) > 3)
|