from flask_wtf import FlaskForm
from wtforms import StringField, IntegerField, RadioField, SelectField


class AddServiceForm(FlaskForm):
    name = StringField('name')
    protocol = SelectField('protocol', choices=[('', "Don't Care"), ('tcp', 'TCP'), ('udp', 'UDP')], default='')
    host = StringField('host')
    port = IntegerField('port')