Skip to content
Snippets Groups Projects
forms.py 2.95 KiB
Newer Older
# Copyright 2022-2024 ETSI OSG/SDG TeraFlowSDN (TFS) (https://tfs.etsi.org/)
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#      http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.

from flask_wtf import FlaskForm
Lluis Gifre Renom's avatar
Lluis Gifre Renom committed
from wtforms import StringField, SelectField, TextAreaField, SubmitField, BooleanField
from wtforms.validators import DataRequired, Length, NumberRange, ValidationError
longllu's avatar
longllu committed
from common.proto.context_pb2 import DeviceOperationalStatusEnum

class AddDeviceForm(FlaskForm):
    device_id = StringField('ID', validators=[DataRequired(), Length(min=5)])
Lluis Gifre Renom's avatar
Lluis Gifre Renom committed
    device_type = SelectField('Type')
    operational_status = SelectField('Operational Status', coerce=int, validators=[NumberRange(min=0)])
Lluis Gifre Renom's avatar
Lluis Gifre Renom committed
    device_drivers_undefined = BooleanField('UNDEFINED / EMULATED')
longllu's avatar
longllu committed
    device_drivers_openconfig = BooleanField('OPENCONFIG')
    device_drivers_transport_api = BooleanField('TRANSPORT_API')
    device_drivers_p4 = BooleanField('P4')
    device_drivers_ietf_network_topology = BooleanField('IETF_NETWORK_TOPOLOGY')
    device_drivers_onf_tr_532 = BooleanField('ONF_TR_532')
longllu's avatar
longllu committed
    device_drivers_xr = BooleanField('XR')
    device_drivers_ietf_l2vpn = BooleanField('IETF L2VPN')
Lluis Gifre Renom's avatar
Lluis Gifre Renom committed
    device_drivers_gnmi_openconfig = BooleanField('GNMI OPENCONFIG')
    device_drivers_optical_tfs = BooleanField('OPTICAL TFS')
Lluis Gifre Renom's avatar
Lluis Gifre Renom committed
    device_drivers_ietf_actn = BooleanField('IETF ACTN')
    device_drivers_qkd = BooleanField('QKD')
longllu's avatar
longllu committed
    device_config_address = StringField('connect/address',default='127.0.0.1',validators=[DataRequired(), Length(min=5)])
    device_config_port = StringField('connect/port',default='0',validators=[DataRequired(), Length(min=1)])
    device_config_settings = TextAreaField('connect/settings',default='{}',validators=[DataRequired(), Length(min=2)])
    submit = SubmitField('Add')

    def validate_operational_status(form, field):
        if field.data not in DeviceOperationalStatusEnum.DESCRIPTOR.values_by_number:
longllu's avatar
longllu committed
            raise ValidationError('The operational status value selected is incorrect!')

class ConfigForm(FlaskForm):
    device_key_config = StringField('Key configuration')
    device_value_config = StringField('Value configuration')    
    submit = SubmitField('Add')


class UpdateDeviceForm(FlaskForm):
longllu's avatar
longllu committed
    update_operational_status = SelectField('Operational Status',
longllu's avatar
longllu committed
                           choices=[(-1, 'Select...'), (0, 'Undefined'), (1, 'Disabled'), (2, 'Enabled')],
                           coerce=int,
                           validators=[NumberRange(min=0)])