satnogs-db/db/base/forms.py

"""SatNOGS DB django base Forms class"""
from django import forms
from django.core.exceptions import ValidationError
from django.utils.translation import ugettext_lazy as _

from db.base.models import Transmitter, TransmitterEntry


def existing_uuid(value):
    """ensures the UUID is existing and valid"""
    try:
        Transmitter.objects.get(uuid=value)
    except Transmitter.DoesNotExist:
        raise ValidationError(
            _('%(value)s is not a valid uuid'),
            code='invalid',
            params={'value': value},
        )


class TransmitterEntryForm(forms.ModelForm):
    """Model Form class for TransmitterEntry objects"""

    uuid = forms.CharField(required=False, validators=[existing_uuid])

    class Meta:
        model = TransmitterEntry
        fields = [
            'description', 'status', 'type', 'uplink_low', 'uplink_high', 'downlink_low',
            'downlink_high', 'uplink_drift', 'downlink_drift', 'downlink_mode', 'uplink_mode',
            'invert', 'baud', 'satellite', 'citation', 'service'
        ]