Coverage for airflow/www/forms.py : 100%

from __future__ import absolute_import 

from __future__ import division 

from __future__ import print_function 

from __future__ import unicode_literals 

from datetime import datetime 

from flask.ext.admin.form import DateTimePickerWidget 

from wtforms import DateTimeField, SelectField 

from flask_wtf import Form 

class DateTimeForm(Form): 

    # Date filter form needed for gantt and graph view 

    execution_date = DateTimeField( 

        "Execution date", widget=DateTimePickerWidget()) 

class DateTimeWithNumRunsForm(Form): 

    # Date time and number of runs form for tree view, task duration 

    # and landing times 

    base_date = DateTimeField( 

        "Anchor date", widget=DateTimePickerWidget(), default=datetime.now()) 

    num_runs = SelectField("Number of runs", default=25, choices=( 

        (5, "5"), 

        (25, "25"), 

        (50, "50"), 

        (100, "100"), 

        (365, "365"), 

    ))