gistfile1.py

class AllImagesCheckPermissionForObjectsNode(Node):
    def __init__(self, obj, name):
        self.obj = Variable(obj)
        self.name = name

    def render(self, context):
        check_published = True
        if context['request'].user.is_staff or \
                context['request'].user.is_superuser:
            check_published = False

        context[self.name] = self.obj.resolve(context).all_images(
            check_published)
        return ''


@register.tag(name='all_images_check_permission')
def all_images_check_permission(parser, token):
    """
    {% all_images_check_permission object as images %}
    """
    try:
        parans = token.split_contents()
    except ValueError:
        raise TemplateSyntaxError(
            _('tag requires exactly two arguments'))
    if len(parans) != 4:
        raise TemplateSyntaxError(
            _('tag requires exactly three arguments'))
    if parans[2] != 'as':
        raise TemplateSyntaxError(
            _("second argument to tag must be 'as'"))
    return AllImagesCheckPermissionForObjectsNode(parans[1], parans[3])