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


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

class  FormArticulo(forms.ModelForm):

    class Meta:
        model = Articulo
        fields = [
            "articulo",
            "cantidad",
            "precio",
            "unidad_medida",
            "serial",
            "marca",
            "modelo",
            "fecha_registro",
            "description",
        ]
        
    def clean_articulo(self):
        articulo = self.cleaned_data.get('articulo')
        if Articulo.objects.filter(articulo=articulo).exists():
            raise forms.ValidationError("El artículo ya existe.")
        return articulo


class FormEnvio(forms.ModelForm):
    class Meta:
        model = ArticuloUnidad
        fields = [
            "articulo",
            "cantidad",
            "unidad",
            "comprobante",
        ]


class FormComprobante(forms.ModelForm):
    class Meta:
        model = Comprobante
        fields = [
            "grupo",
            "subgrupo",
            "conceptos",
            "observaciones",
            "numero_movimiento",
            "concepto_movimiento",
            "e_o_s",
        ]