# Django-specific imports

from django.core.exceptions import ValidationError

<dt>Project</dt><dd>{% html_link project.name 'project'  project.id  %}</dd>

<dt>Location</dt><dd>{% html_link location.name 'location' location.id %}</dd>

<dt>iptables rules</dt><dd><pre>{% include "conntrackt/entity_iptables.html" with entity=entity %}</pre>{% html_link 'Download' 'entity_iptables' entity.id class="btn btn-primary" %}</dd>

{% load url from future %}

<a href="{{url}}"{% if title %} title="{{title}}"{% endif %}{% if id %} id="{{id}}"{% endif %}{%if class %} class="{{class}}"{% endif %}>{{text}}</a>

<hr>

<div class="row">

  <div class="span4">

    <div class="well">{% include "conntrackt/project_widget.html" %}</div>

  </div>

</div>

{% load conntrackt %}

  <table class="table table-striped">

    <tr><td><strong>{{location.name}}</strong></td><td>{% html_link '<i class="icon-book"></i>' 'project_location_iptables' project.id location.id class="btn btn-link" %}</td></tr>

{% for entity in entities %}

    <tr><td>{% html_link entity.name 'entity' entity.id class="btn btn-link" %}</td><td>{% html_link '<i class="icon-list"></i>' 'entity_iptables' entity.id class="btn btn-link" %}</td></tr>

{% endfor %}

  </table>

<div>{{project.description}}</div>

<hr>

{% if location_entities %}

<div class="row">

  {% for location, entities in location_entities %}

  <div class="span4">

    <div class="well">{% include "conntrackt/location_widget.html" with location=location entities=entities %}</div>

  </div>

  {% endfor %}

</div>

{% load conntrackt %}

  <table class="table table-striped">

    <tr><td><strong>{% html_link project.name 'project' project.id class="btn btn-link" title=project.description %}</strong></td><td>{% html_link '<i class="icon-book"></i>' 'project_iptables' project.id class="btn btn-link" %}</td></tr>

{% for entity in project.entity_set.all %}

    <tr><td>{% html_link entity.name 'entity' entity.id class="btn btn-link" %}</td><td>{% html_link '<i class="icon-list"></i>' 'entity_iptables' entity.id class="btn btn-link" %}</td></tr>

{% endfor %}

  </table>

          {% block header_title %}{% html_link 'Conntrackt' 'index' class="brand" %}{% endblock %}

@register.inclusion_tag('conntrackt/html_link.html')

def html_link(text, view, *args, **kwargs):

    A small wrapper for showing HTML linnks.

    Positional arguments:

        text - Text that should be used as a link.

        view - View name for which the URL should be shown

        args - Additional positional arguments that will be passed to resolver

        for creating the URL.

    Keyword arguments:

        id - Identifier for the <a> HTML element.

        class - Class(es) for the <a> HTML element.

        title - Title for the HTML <a> element.

    # We'll return a context for rendering with the template. Add the text right

    # away.

    context = {'text': text}

    # Generate the URL by using the supplied view name and arguments that should

    # be passed to the view.

    context['url'] = urlresolvers.reverse(view, args=args)

    # Iterate over keyword arguments, and if we support them, just assign them

    # to the context as is. Raise an exception if a keyword is unsupported.

    for key, value in kwargs.items():

        if key in ("class", "title", "id"):

            context[key] = value

        else:

            raise template.TemplateSyntaxError("Unknown argument for 'advhtml_link' tag: %r" % key )

    return context

from conntrackt.views import IndexView, IptablesView, ProjectView, get_project_iptables

    # View for rendering zip file with iptables rules for all entities in a project.

    url(r'^project/(?P<project_id>\d+)/iptables/$', get_project_iptables, name="project_iptables"),

    # View for rendering zip file with iptables rules for all entities in a project for a specific location.

    url(r'^project/(?P<project_id>\d+)/location/(?P<location_id>\d+)/iptables/$', get_project_iptables, name="project_location_iptables"),

# For generating ZIP files.

from StringIO import StringIO

from zipfile import ZipFile, ZIP_DEFLATED

# Django-specific imports.

from django.http import HttpResponse

from django.shortcuts import render_to_response, get_object_or_404

# Application-specific imports.

from conntrackt.stripper import StripperMiddleware

        """

        Returns the context data that should be used for rendering of the

        template.

        Adds the 'projects' context object containing all of the projects

        available sorted aplhabetically by name.

        """

        # Set the context using the parent class.

    """

    Custom view for presenting the project information.

    """

        """

        Returns the context data that should be used for rendering of the

        template.

        Adds the 'location_entities' context object that contains tuples of the

        form '(location, entities)', where location is an instance of a

        Location, and entities is a query set of entities that belong to that

        particular location, and to related project.

        """

        # Set the context using the parent class.

        # Set-up an array that will contaion (location, entities) tuples.

        location_entities = []

        # Add the (location, entities) tuple for each location that has entities

        # belonging to the related project.

        for location in Location.objects.filter(entity__project = self.object).distinct():

            entities = Entity.objects.filter(project = self.object, location = location)

            location_entities.append((location, entities))

        # Add the (location, entities) tuples to context.

        context['location_entities'] = location_entities

        # Finally return the context.

    """

    Custom view for rendering iptables rules for a specific entity, and starting

    automatic download by browser.

    """

        """

        Renders the template with the given context to response. The response

        will be an iptables configuration file that can be used with

        iptables-restore.

        Makes sure to set the Content-Disposition of a response in order to

        signal the browser it should start download of this view's response

        immediately. Also sets the suggested filename for it.

        """

        # Call the parent class method first. This will render the template. Set

        # the content type to text/plain.

        # Add the Content-Disposition information for the browser, telling the

        # browser to download the file with suggested filename.

        response['Content-Disposition']="attachment; filename=%s-iptables.conf" % self.object.name.lower().replace(" ", "_")

        # Return the modified response.

def get_project_iptables(request, project_id, location_id = None):

    """

    Custom view for obtaining iptables for all entities of a project or project

    location in a single ZIP file.

    Arguments:

        request - Request object.

        project_id - Unique ID of the project for whose entities the iptables

        rules should be generated.

        location_id - Optional unique ID of the project location for whose

        entities the iptables rules should be generated. Default is None, which

        means generate rules for _all_ entities in a project.

    Returns:

        Response object that contains the ZIP file and Content-Disposition

        information.

    """

    # Fetch the project.

    project = get_object_or_404(Project, pk = project_id)

    # Set-up a string IO object to which we'll write the ZIP file (in-memory).

    buff = StringIO()

    # Create a new ZIP file in-memory.

    zipped_iptables = ZipFile(buff, "w", ZIP_DEFLATED)

    # Create the response object, setting the mime type so browser could offer

    # to open the file with program as well.

    response = HttpResponse(mimetype='application/zip')

    # Stripper middleware has to be used in order to remove the excess blank

    # lines from rendering (this has to be done manually).

    stripper_middleware = StripperMiddleware()

    # If specific location was specified, get the entities that are part of that

    # project location only, otherwise fetch all of the project's entities. Also

    # set-up the filename that will be suggested to the browser.

    if location_id:

        location = get_object_or_404(Location, pk=location_id)

        entities = project.entity_set.filter(location = location)

        filename = '%s_%s-iptables.zip' % (project.name, location.name)

    else:

        entities = project.entity_set.all()

        filename = '%s-iptables.zip' % (project.name)

    # Lower-case the filename, and replace spaces with underscores (_).

    filename = filename.lower().replace(" ", "_")

    # Render iptables rules for each entity, placing them in the ZIP archive.

    for entity in entities:

        entity_iptables = stripper_middleware.process_response(request,render_to_response('conntrackt/entity_iptables.html', {'entity': entity}, mimetype="text/plain"))

        zipped_iptables.writestr("%s-iptables.conf" % entity.name.lower().replace(" ", "_"), entity_iptables.content)

    # Close the archive, and flush the buffer.

    zipped_iptables.close()

    buff.flush()

    # Write the contents of our buffer (ZIP archive) to response content, and

    # close the IO string.

    response.write(buff.getvalue())

    buff.close()

    # Set the Content-Disposition so the browser would know it should download

    # the archive, and suggest the filename.

    response['Content-Disposition'] = 'attachment; filename="%s"' % filename

    # Finally return the response object.

    return response