from django import forms
from .models import Asset, Category


class AssetForm(forms.ModelForm):
    class Meta:
        model = Asset
        fields = [
            'asset_number', 'name', 'category', 'brand', 'model', 'asset_value', 'serial_number',
            'location', 'cabinet', 'cabinet_position', 'bmc_address', 'ip_address',
            'has_gpu', 'gpu_type', 'gpu_count',
            'purchase_date', 'warranty_expire', 'supplier',
            'responsible_person', 'department', 'user', 'business_type', 'status', 'remark',
        ]
        widgets = {
            'purchase_date': forms.DateInput(attrs={'type': 'date'}),
            'warranty_expire': forms.DateInput(attrs={'type': 'date'}),
            'remark': forms.Textarea(attrs={'rows': 3}),
            'status': forms.Select(attrs={'class': 'form-select'}),
            'category': forms.Select(attrs={'class': 'form-select'}),
        }


class AssetImportForm(forms.Form):
    excel_file = forms.FileField(
        label='Excel文件',
        help_text='支持 .xlsx 格式',
        widget=forms.FileInput(attrs={'accept': '.xlsx,.xls'})
    )


class CategoryForm(forms.ModelForm):
    class Meta:
        model = Category
        fields = ['name', 'description']