class DagRunEditForm(DynamicForm): """Form for editing DAG Run. We don't actually want to allow editing, so everything is read-only here. """ dag_id = StringField(lazy_gettext('Dag Id'), 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'), widget=BS3TextFieldROWidget()) state = StringField(lazy_gettext('State'), widget=BS3TextFieldROWidget()) execution_date = DateTimeWithTimezoneField( lazy_gettext('Execution Date'), widget=AirflowDateTimePickerROWidget(), ) conf = TextAreaField(lazy_gettext('Conf'), widget=BS3TextAreaROWidget()) def populate_obj(self, item): """Populates the attributes of the passed obj with data from the form’s fields.""" super().populate_obj(item) item.run_type = DagRunType.from_run_id(item.run_id) if item.conf: item.conf = json.loads(item.conf)
class TaskInstanceEditForm(DynamicForm): """Form for editing TaskInstance""" dag_id = StringField(lazy_gettext('Dag Id'), validators=[InputRequired()], widget=BS3TextFieldROWidget()) task_id = StringField(lazy_gettext('Task Id'), validators=[InputRequired()], widget=BS3TextFieldROWidget()) start_date = DateTimeWithTimezoneField( lazy_gettext('Start Date'), widget=AirflowDateTimePickerROWidget()) end_date = DateTimeWithTimezoneField( lazy_gettext('End Date'), widget=AirflowDateTimePickerROWidget()) state = SelectField( lazy_gettext('State'), choices=( ('success', 'success'), ('running', 'running'), ('failed', 'failed'), ('up_for_retry', 'up_for_retry'), ), widget=Select2Widget(), validators=[InputRequired()], ) execution_date = DateTimeWithTimezoneField( lazy_gettext('Execution Date'), widget=AirflowDateTimePickerROWidget(), validators=[InputRequired()], )
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() )