from __future__ import unicode_literals
from __future__ import absolute_import

from flask.ext.wtf import Form
from wtforms import RadioField, TextAreaField, validators

class QueryForm(Form):
    """The form used to allow freeform queries to be executed against
    PuppetDB."""
    query = TextAreaField('Query', [validators.Required(
      message='A query is required.')])
    endpoints = RadioField('API endpoint', choices = [
      ('nodes', 'Nodes'),
      ('resources', 'Resources'),
      ('facts', 'Facts'),
      ('fact-names', 'Fact Names'),
      ('reports', 'Reports'),
      ('events', 'Events'),
      ])