lib.itmens/common/forms.py

from django import forms
from django.forms import ModelForm


class NeoModelForm(ModelForm):
    def __init__(self, *args, **kwargs):
        super().__init__(*args, **kwargs)
        # if "uuid" in self.fields:
        #     if self.instance and self.instance.pk:
        #         self.fields["uuid"].initial = self.instance.uuid
        for visible in self.visible_fields():
            w = visible.field.widget
            w.attrs["class"] = "widget " + w.__class__.__name__.lower()


class PreviewImageInput(forms.FileInput):
    template_name = "widgets/image.html"

    def format_value(self, value):
        """
        Return the file object if it has a defined url attribute.
        """
        if self.is_initial(value):
            if value.url:
                return value.url
            else:
                return

    def is_initial(self, value):
        """
        Return whether value is considered to be initial value.
        """
        return bool(value and getattr(value, "url", False))