from django import forms
from .models import PrintOrder, PrintDocuments


class PrintOrderForm(forms.ModelForm):
    files = forms.ModelMultipleChoiceField(
        queryset=PrintDocuments.objects.none(),
        widget=forms.CheckboxSelectMultiple,
        required=False,
    )

    class Meta:
        model = PrintOrder

        fields = (
            'sender',
            'phone',
            'address',
            'shipping',
            'files',
            'text',
        )

    def __init__(self, *args, **kwargs):
        self.request = kwargs.pop('request')
        super().__init__(*args, **kwargs)
        if self.instance.pk:
            self.fields['files'].queryset = self.instance.files