Django: Cómo crear un formulario con un campo para imágenes o documentos

Answer

1. 

   davy.ai

   July 31, 2023 at 8:53 pm

   Para crear un formulario que detecte el tipo de carga y utilice el tipo de campo correcto, puedes utilizar el tipo de Multipurpose Internet Mail Extensions (extensiones de correo de Internet multipropósito, MIME) del archivo cargado.

   En primer lugar, importa el módulo de formularios de Django:

   from django import forms
   

   Luego, crea un formulario que incluya tanto un campo FileField como un campo ImageField:

   class UploadForm(forms.Form):
       file = forms.FileField()
       image = forms.ImageField()
   

   A continuación, define una función que valide el archivo cargado y devuelva el tipo de campo relevante:

   import magic
   
   def validate_file_type(file):
       mime = magic.Magic(mime=True)
       file_type = mime.from_file(file.temporary_file_path())
       if 'image' in file_type:
           return 'image'
       elif 'pdf' in file_type:
           return 'pdf'
       else:
           return None
   

   Esta función utiliza el paquete python-magic para determinar el tipo MIME del archivo cargado. Si el tipo MIME contiene “image”, entonces el archivo es una imagen; de lo contrario, si contiene “pdf”, entonces el archivo es un documento. Si el tipo MIME no coincide con ninguno de los criterios, se devuelve None.

   Por último, modifica el método clean_file() del formulario para detectar el tipo de archivo y utilizar el campo relevante:

   class UploadForm(forms.Form):
       file = forms.FileField()
       image = forms.ImageField()
   
       def clean_file(self):
           uploaded_file = self.cleaned_data['file']
           file_type = validate_file_type(uploaded_file)
           if file_type == 'image':
               return self.cleaned_data['image']
           elif file_type == 'pdf':
               return self.cleaned_data['file']
           else:
               raise forms.ValidationError("Tipo de archivo no válido")
   

   Aquí, anulamos el método clean_file() del formulario, que es el responsable de validar y limpiar el campo file. El método obtiene primero el objeto de archivo cargado del diccionario de datos limpios. Luego llama a la función validate_file_type() para determinar el tipo MIME del archivo. Si el archivo es una imagen, se devuelve el campo image. Si es un documento, se devuelve el campo file. Por último, si el tipo de archivo no coincide con ninguno de los criterios, se genera un error de validación.

   Con este código, cuando los usuarios envíen un formulario con un archivo cargado, el formulario detectará si el archivo es una imagen o un documento y utilizará el tipo de campo apropiado.

Comments are closed.