~mhall119/ubuntu-website/django-foundations

# Modified copy of django.contrib.admin.widgets.RelatedFieldWidgetWrapper

# Removed reverences to admin_site, and instead accept a popup_url

 

from django import forms

from django.conf import settings

from django.utils.translation import ugettext as _

from django.utils.safestring import mark_safe

import copy

 

class PopupRelatedFieldWidgetWrapper(forms.Widget):

    """

    This class is a wrapper to a given widget to add the add icon for the

    admin interface.

    """

    def __init__(self, widget, popup_url):

        self.is_hidden = widget.is_hidden

        self.needs_multipart_form = widget.needs_multipart_form

        self.attrs = widget.attrs

        self.choices = widget.choices

        self.widget = widget

        self.popup_url = popup_url

 

    def __deepcopy__(self, memo):

        obj = copy.copy(self)

        obj.widget = copy.deepcopy(self.widget, memo)

        obj.attrs = self.widget.attrs

        memo[id(self)] = obj

        return obj

 

    def _media(self):

        wm = self.widget.media

        wm.add_js(['%sjs/admin/RelatedObjectLookups.js'%settings.ADMIN_MEDIA_PREFIX,])

        return wm

    media = property(_media)

 

    def render(self, name, value, *args, **kwargs):

 

        self.widget.choices = self.choices

        output = [self.widget.render(name, value, *args, **kwargs)]

        output.append(u'<a href="%s" class="add-another" id="add_id_%s" onclick="return showAddAnotherPopup(this);"> ' % \

            (self.popup_url, name))

        output.append(u'<img src="%simg/admin/icon_addlink.gif" width="10" height="10" alt="%s"/></a>' % (settings.ADMIN_MEDIA_PREFIX, _('Add Another')))

        return mark_safe(u''.join(output))

 

    def build_attrs(self, extra_attrs=None, **kwargs):

        "Helper function for building an attribute dictionary."

        self.attrs = self.widget.build_attrs(extra_attrs=None, **kwargs)

        return self.attrs

 

    def value_from_datadict(self, data, files, name):

        return self.widget.value_from_datadict(data, files, name)

 

    def _has_changed(self, initial, data):

        return self.widget._has_changed(initial, data)

 

    def id_for_label(self, id_):

        return self.widget.id_for_label(id_)