from django import forms
from .models import Unidad, Articulo, ArticuloUnidad


class FormUnidad(forms.ModelForm):
  class Meta:
    model = Unidad
    fields=['nombreUnidad', 'telefono', 'ubicacion', 'comandante']

class  FormArticulo(forms.ModelForm):

    class Meta:
        model = Articulo
        fields = ["articulo", "cantidad", "descripcion"]


class FormEnvio(forms.ModelForm):
    cantidad = forms.IntegerField(required=True)
    class Meta:
        model = ArticuloUnidad
        fields = ['articulo', 'cantidad' , 'unidad']