from django import forms

from django.conf import settings

from django.utils.html import escape

from django.utils.safestring import mark_safe

from django.utils.encoding import force_unicode

from django.contrib.admin.util import flatten_fieldsets

from django.contrib.contenttypes.models import ContentType

from django.utils.translation import ugettext_lazy as _

ACTION_CHECKBOX_NAME = '_selected_action'

class ActionForm(forms.Form):

    action = forms.ChoiceField(label=_('Action:'))

checkbox = forms.CheckboxInput({'class': 'action-select'}, lambda value: False)

class AdminForm(object):

    def __init__(self, form, fieldsets, prepopulated_fields):

        self.form, self.fieldsets = form, normalize_fieldsets(fieldsets)

        self.prepopulated_fields = [{

            'field': form[field_name],

            'dependencies': [form[f] for f in dependencies]

        } for field_name, dependencies in prepopulated_fields.items()]

    def __iter__(self):

        for name, options in self.fieldsets:

            yield Fieldset(self.form, name, **options)

    def first_field(self):

        try:

            fieldset_name, fieldset_options = self.fieldsets[0]

            field_name = fieldset_options['fields'][0]

            if not isinstance(field_name, basestring):

                field_name = field_name[0]

            return self.form[field_name]

        except (KeyError, IndexError):

            pass

        try:

            return iter(self.form).next()

        except StopIteration:

            return None

    def _media(self):

        media = self.form.media

        for fs in self:

            media = media + fs.media

        return media

    media = property(_media)

class Fieldset(object):

    def __init__(self, form, name=None, fields=(), classes=(), description=None):

        self.form = form

        self.name, self.fields = name, fields

        self.classes = u' '.join(classes)

        self.description = description

    def _media(self):

        if 'collapse' in self.classes:

            return forms.Media(js=['%sjs/admin/CollapsedFieldsets.js' % settings.ADMIN_MEDIA_PREFIX])

        return forms.Media()

    media = property(_media)

    def __iter__(self):

        for field in self.fields:

            yield Fieldline(self.form, field)

class Fieldline(object):

    def __init__(self, form, field):

        self.form = form # A django.forms.Form instance

        if isinstance(field, basestring):

            self.fields = [field]

        else:

            self.fields = field

    def __iter__(self):

        for i, field in enumerate(self.fields):

            yield AdminField(self.form, field, is_first=(i == 0))

    def errors(self):

        return mark_safe(u'\n'.join([self.form[f].errors.as_ul() for f in self.fields]).strip('\n'))

class AdminField(object):

    def __init__(self, form, field, is_first):

        self.field = form[field] # A django.forms.BoundField instance

        self.is_first = is_first # Whether this field is first on the line

        self.is_checkbox = isinstance(self.field.field.widget, forms.CheckboxInput)

    def label_tag(self):

        classes = []

        if self.is_checkbox:

            classes.append(u'vCheckboxLabel')

            contents = force_unicode(escape(self.field.label))

        else:

            contents = force_unicode(escape(self.field.label)) + u':'

        if self.field.field.required:

            classes.append(u'required')

        if not self.is_first:

            classes.append(u'inline')

        attrs = classes and {'class': u' '.join(classes)} or {}

        return self.field.label_tag(contents=contents, attrs=attrs)

class InlineAdminFormSet(object):

    """

    A wrapper around an inline formset for use in the admin system.

    """

    def __init__(self, inline, formset, fieldsets):

        self.opts = inline

        self.formset = formset

        self.fieldsets = fieldsets

    def __iter__(self):

        for form, original in zip(self.formset.initial_forms, self.formset.get_queryset()):

            yield InlineAdminForm(self.formset, form, self.fieldsets, self.opts.prepopulated_fields, original)

        for form in self.formset.extra_forms:

            yield InlineAdminForm(self.formset, form, self.fieldsets, self.opts.prepopulated_fields, None)

    def fields(self):

        fk = getattr(self.formset, "fk", None)

        for field_name in flatten_fieldsets(self.fieldsets):

            if fk and fk.name == field_name:

                continue

            yield self.formset.form.base_fields[field_name]

    def _media(self):

        media = self.opts.media + self.formset.media

        for fs in self:

            media = media + fs.media

        return media

    media = property(_media)

class InlineAdminForm(AdminForm):

    """

    A wrapper around an inline form for use in the admin system.

    """

    def __init__(self, formset, form, fieldsets, prepopulated_fields, original):

        self.formset = formset

        self.original = original

        if original is not None:

            self.original_content_type_id = ContentType.objects.get_for_model(original).pk

        self.show_url = original and hasattr(original, 'get_absolute_url')

        super(InlineAdminForm, self).__init__(form, fieldsets, prepopulated_fields)

    def __iter__(self):

        for name, options in self.fieldsets:

            yield InlineFieldset(self.formset, self.form, name, **options)

    def has_auto_field(self):

        if self.form._meta.model._meta.has_auto_field:

            return True

        # Also search any parents for an auto field.

        for parent in self.form._meta.model._meta.get_parent_list():

            if parent._meta.has_auto_field:

                return True

        return False

    def field_count(self):

        # tabular.html uses this function for colspan value.

        num_of_fields = 0

        if self.has_auto_field():

            num_of_fields += 1

        num_of_fields += len(self.fieldsets[0][1]["fields"])

        if self.formset.can_order:

            num_of_fields += 1

        if self.formset.can_delete:

            num_of_fields += 1

        return num_of_fields

    def pk_field(self):

        return AdminField(self.form, self.formset._pk_field.name, False)

    def fk_field(self):

        fk = getattr(self.formset, "fk", None)

        if fk:

            return AdminField(self.form, fk.name, False)

        else:

            return ""

    def deletion_field(self):

        from django.forms.formsets import DELETION_FIELD_NAME

        return AdminField(self.form, DELETION_FIELD_NAME, False)

    def ordering_field(self):

        from django.forms.formsets import ORDERING_FIELD_NAME

        return AdminField(self.form, ORDERING_FIELD_NAME, False)

class InlineFieldset(Fieldset):

    def __init__(self, formset, *args, **kwargs):

        self.formset = formset

        super(InlineFieldset, self).__init__(*args, **kwargs)

    def __iter__(self):

        fk = getattr(self.formset, "fk", None)

        for field in self.fields:

            if fk and fk.name == field:

                continue

            yield Fieldline(self.form, field)

class AdminErrorList(forms.util.ErrorList):

    """

    Stores all errors for the form/formsets in an add/change stage view.

    """

    def __init__(self, form, inline_formsets):

        if form.is_bound:

            self.extend(form.errors.values())

            for inline_formset in inline_formsets:

                self.extend(inline_formset.non_form_errors())

                for errors_in_inline_form in inline_formset.errors:

                    self.extend(errors_in_inline_form.values())

def normalize_fieldsets(fieldsets):

    """

    Make sure the keys in fieldset dictionaries are strings. Returns the

    normalized data.

    """

    result = []

    for name, options in fieldsets:

        result.append((name, normalize_dictionary(options)))

    return result

def normalize_dictionary(data_dict):

    """

    Converts all the keys in "data_dict" to strings. The keys must be

    convertible using str().

    """

    for key, value in data_dict.items():

        if not isinstance(key, str):

            del data_dict[key]

            data_dict[str(key)] = value

    return data_dict