from django.contrib.admin.templatetags.admin_list import items_for_result, result_headers
from django.template import Library
from django.utils.safestring import mark_safe
register = Library()

def results(cl, additional_links):
    """
    Rewrite of original function to add additional columns after each result
    in the change list.
    """
    for res in cl.result_list:
        rl = list(items_for_result(cl,res))
        for link in additional_links:
            rl.append(mark_safe(link['url_template'] % res.id))
        yield rl

def extended_result_list(cl, additional_links):
    """
    Rewrite of original function to add an additional columns after each result
    in the change list.
    """
    headers = list(result_headers(cl))
    for header in additional_links:
        headers.append(header)

    return {
        'cl': cl,
        'result_headers': headers,
        'results': list(results(cl, additional_links))
    }

# This function is an example template tag for use in an overridden change_list.html template.
def person_result_list(cl):
    additional_links = (
        { 'text': 'All Registrations',
          'sortable': False,
          'url_template': '<td><a href="/ce/admin/registrations/registration/?person__id__exact=%s">All Registrations</a></td>'
        },
        { 'text': 'New Registration',
          'sortable': False,
          'url_template': '<td><a href="/ce/admin/registrations/registration/add/?person_id=%s">New Registration</a></td>',
        },
    )

    return extended_result_list(cl, additional_links)
person_result_list = register.inclusion_tag("admin/change_list_results.html")(person_result_list)

# Modified section of change_list.html template.
{# Override the original result_list templatetag to add a registration link for each person #}
{# {% block result_list %}{% result_list cl %}{% endblock %} #}
{% load admin_extras %}
{% block result_list %}{% person_result_list cl %}{% endblock %}