58 lines
2.0 KiB
Python
58 lines
2.0 KiB
Python
from pkg_resources import parse_version
|
|
|
|
from django_select2.forms import ModelSelect2Widget, HeavySelect2Widget
|
|
from django_hstore.forms import DictionaryFieldWidget
|
|
|
|
from django import get_version
|
|
from django.urls import reverse
|
|
from django.conf import settings
|
|
from django.utils.safestring import mark_safe
|
|
from django.template import Context
|
|
from django.template.loader import get_template
|
|
from django.contrib.admin.widgets import AdminTextareaWidget
|
|
|
|
|
|
class ItemSelectWidget(ModelSelect2Widget):
|
|
def __init__(self, *args, **kwargs):
|
|
kwargs["data_view"] = "item-complete"
|
|
super(ItemSelectWidget, self).__init__(*args, **kwargs)
|
|
|
|
def label_from_instance(self, obj):
|
|
return obj.name
|
|
|
|
|
|
class PropsSelectWidget(DictionaryFieldWidget):
|
|
def __init__(self, *args, **kwargs):
|
|
super().__init__(*args, **kwargs)
|
|
|
|
def render(self, name, value, attrs=None):
|
|
if attrs is None:
|
|
attrs = {}
|
|
# it's called "original" because it will be replaced by a copy
|
|
attrs["class"] = "hstore-original-textarea"
|
|
w = HeavySelect2Widget(
|
|
data_view="prop-complete", attrs={"data-tags": "true", "class": "hs-key"}
|
|
)
|
|
|
|
# get default HTML from AdminTextareaWidget
|
|
html = AdminTextareaWidget.render(self, name, value, attrs)
|
|
# prepare template context
|
|
template_context = {
|
|
"field_name": name,
|
|
"STATIC_URL": settings.STATIC_URL,
|
|
"use_svg": parse_version(get_version())
|
|
>= parse_version("1.9"), # use svg icons if django >= 1.9
|
|
"ajax_url": reverse("prop-complete"),
|
|
"w": w.build_attrs(base_attrs=w.attrs),
|
|
}
|
|
# get template object
|
|
template = get_template("hstore_%s_widget.html" % self.admin_style)
|
|
# render additional html
|
|
additional_html = template.render(template_context)
|
|
|
|
# append additional HTML and mark as safe
|
|
html = html + additional_html
|
|
html = mark_safe(html)
|
|
|
|
return html
|