from django import forms
from .models import Unit

class UnitForm(forms.ModelForm):
    class Meta:
        model = Unit
        fields = ('location_lat', 'location_lon', 'status')
        widgets = {
            'location_lat': forms.NumberInput(attrs={'class': 'form-control'}),
            'location_lon': forms.NumberInput(attrs={'class': 'form-control'}),
            'status': forms.Select(attrs={'class': 'form-select'}),
        }