class DagRunForm(DynamicForm):
dag_id = StringField(lazy_gettext('Dag Id'),
validators=[validators.DataRequired()],
widget=BS3TextFieldWidget())
start_date = DateTimeField(lazy_gettext('Start Date'),
widget=AirflowDateTimePickerWidget())
end_date = DateTimeField(lazy_gettext('End Date'),
widget=AirflowDateTimePickerWidget())
run_id = StringField(lazy_gettext('Run Id'),
validators=[validators.DataRequired()],
widget=BS3TextFieldWidget())
state = SelectField(lazy_gettext('State'),
choices=(
('success', 'success'),
('running', 'running'),
('failed', 'failed'),
),
widget=Select2Widget())
execution_date = DateTimeField(lazy_gettext('Execution Date'),
widget=AirflowDateTimePickerWidget())
external_trigger = BooleanField(lazy_gettext('External Trigger'))
conf = TextAreaField(lazy_gettext('Conf'),
validators=[ValidJson(),
validators.Optional()],
widget=BS3TextAreaFieldWidget())
def populate_obj(self, item):
# TODO: This is probably better done as a custom field type so we can
# set TZ at parse time
super().populate_obj(item)
item.execution_date = timezone.make_aware(item.execution_date)
if item.conf:
item.conf = json.loads(item.conf)
class DagRunForm(DynamicForm):
"""Form for editing and adding DAG Run"""
dag_id = StringField(lazy_gettext('Dag Id'),
validators=[DataRequired()],
widget=BS3TextFieldWidget())
start_date = DateTimeWithTimezoneField(
lazy_gettext('Start Date'), widget=AirflowDateTimePickerWidget())
end_date = DateTimeWithTimezoneField(lazy_gettext('End Date'),
widget=AirflowDateTimePickerWidget())
run_id = StringField(lazy_gettext('Run Id'),
validators=[DataRequired()],
widget=BS3TextFieldWidget())
state = SelectField(
lazy_gettext('State'),
choices=(
('success', 'success'),
('running', 'running'),
('failed', 'failed'),
),
widget=Select2Widget(),
)
execution_date = DateTimeWithTimezoneField(
lazy_gettext('Execution Date'), widget=AirflowDateTimePickerWidget())
external_trigger = BooleanField(lazy_gettext('External Trigger'))
conf = TextAreaField(lazy_gettext('Conf'),
validators=[ValidJson(), Optional()],
widget=BS3TextAreaFieldWidget())
def populate_obj(self, item):
"""Populates the attributes of the passed obj with data from the form’s fields."""
super().populate_obj(item) # pylint: disable=no-member
item.run_type = DagRunType.from_run_id(item.run_id)
if item.conf:
item.conf = json.loads(item.conf)
class DagRunForm(DynamicForm):
dag_id = StringField(lazy_gettext('Dag Id'),
validators=[validators.DataRequired()],
widget=BS3TextFieldWidget())
start_date = DateTimeWithTimezoneField(
lazy_gettext('Start Date'), widget=AirflowDateTimePickerWidget())
end_date = DateTimeWithTimezoneField(lazy_gettext('End Date'),
widget=AirflowDateTimePickerWidget())
run_id = StringField(lazy_gettext('Run Id'),
validators=[validators.DataRequired()],
widget=BS3TextFieldWidget())
state = SelectField(lazy_gettext('State'),
choices=(
('success', 'success'),
('running', 'running'),
('failed', 'failed'),
),
widget=Select2Widget())
execution_date = DateTimeWithTimezoneField(
lazy_gettext('Execution Date'), widget=AirflowDateTimePickerWidget())
external_trigger = BooleanField(lazy_gettext('External Trigger'))
conf = TextAreaField(lazy_gettext('Conf'),
validators=[ValidJson(),
validators.Optional()],
widget=BS3TextAreaFieldWidget())
def populate_obj(self, item):
super().populate_obj(item)
item.run_type = DagRunType.from_run_id(item.run_id).value
if item.conf:
item.conf = json.loads(item.conf)
class DagRunEditForm(DagRunForm):
"""Form for editing DAG Run"""
dag_id = StringField(lazy_gettext('Dag Id'), validators=[DataRequired()], widget=BS3TextFieldROWidget())
start_date = DateTimeWithTimezoneField(lazy_gettext('Start Date'), widget=AirflowDateTimePickerROWidget())
end_date = DateTimeWithTimezoneField(lazy_gettext('End Date'), widget=AirflowDateTimePickerROWidget())
run_id = StringField(lazy_gettext('Run Id'), validators=[DataRequired()], widget=BS3TextFieldROWidget())
execution_date = DateTimeWithTimezoneField(
lazy_gettext('Execution Date'), widget=AirflowDateTimePickerROWidget()
)
conf = TextAreaField(
lazy_gettext('Conf'), validators=[ValidJson(), Optional()], widget=BS3TextAreaROWidget()
)
