{% extends "conntrackt/base.html" %}

{# For html_link. #}

{% load conntrackt_tags %}

{% block content %}

{# Use a bit shorter variable names. #}

{% with project=entity.project location=entity.location %}

{% if entity %}

<dl>

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

  <dt>Location</dt><dd>{{location.name}}</dd>

  <dt>Incoming communications</dt><dd><ul class="unstyled">{% for interface in entity.interface_set.all %}

    {% for communication in interface.destination_set.all %}

    <li>{{communication.source}} - {{communication.protocol}}: {{communication.port}}</li>

    {% endfor %}{% endfor %}</ul></dd>

    <dt>Outgoing communications</dt><dd><ul class="unstyled">{% for interface in entity.interface_set.all %}

  {% for communication in interface.source_set.all %}

      <li>{{communication.destination}} - {{communication.protocol}}: {{communication.port}}</li>

{% endfor %}{% endfor %}</ul></dd>

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

{% load conntrackt_tags %}

  <table class="table table-striped">

{% for entity in entities %}

{% endfor %}

    {% with project_id=project.id|slugify location_id=location.id|slugify %}

    {% endwith %}

  </table>

# Django imports.

from django.conf.urls import patterns, url

from django.contrib.auth.views import login, logout

# Application imports.

from .views import IndexView, EntityView, entity_iptables, project_iptables

from .views import ProjectView, ProjectCreateView, ProjectUpdateView, ProjectDeleteView

from .views import LocationCreateView, LocationUpdateView, LocationDeleteView

urlpatterns = patterns(

    'conntrackt.views',

    # Homepage/index view.

    url(r'^$', IndexView.as_view(), name="index"),

    # View for showing information about a project.

    url(r'^project/(?P<pk>\d+)/$', ProjectView.as_view(),

        name='project'),

    # View for creating a new project.

    url(r'^project/add/$', ProjectCreateView.as_view(), name="project_create"),

    url(r'^location/add/$', LocationCreateView.as_view(), name="location_create"),

    # View for updating an existing location.

    url(r'^location/(?P<pk>\d+)/edit/$', LocationUpdateView.as_view(), name="location_update"),

    # View for deleting a location.

    url(r'^location/(?P<pk>\d+)/remove/$', LocationDeleteView.as_view(), name="location_delete"),

    # View for showing information about an entity.

    url(r'^entity/(?P<pk>\d+)/$', EntityView.as_view(),

        name='entity'),

    # View for creating a new entity.

    url(r'^entity/add/$', EntityCreateView.as_view(), name="entity_create"),

    # View for rendering iptables rules for a specific entity.

    url(r'^entity/(?P<pk>\d+)/iptables/$', entity_iptables, name="entity_iptables"),

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

    url(r'^project/(?P<project_id>\d+)/iptables/$', 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/$', project_iptables, name="project_location_iptables"),

    # Views for logging-in/out the users.

    url(r'^login/$', login, {'template_name': 'conntrackt/login.html'}, name="login"),

    url(r'^logout/$', logout, name="logout"),

)

# Standard library imports.

from StringIO import StringIO

from zipfile import ZipFile, ZIP_DEFLATED

# Django imports.

from django.contrib.auth.decorators import permission_required

from django.contrib import messages

from django.http import HttpResponse

from django.shortcuts import render_to_response, get_object_or_404

from django.views.generic import TemplateView, DetailView, CreateView, UpdateView, DeleteView

# Third-party application imports.

from braces.views import MultiplePermissionsRequiredMixin

# Application imports.

from .forms import EntityForm

from .models import Project, Entity, Location

from .utils import generate_entity_iptables

        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.

        context = super(ProjectView, self).get_context_data(**kwargs)

        # 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.

            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.

        return context

class EntityView(MultiplePermissionsRequiredMixin, DetailView):

    """

    def get_initial(self):

        """

        Returns initial values that should be pre-selected (if they were

        specified through a GET parameter).

        """

        initial = super(EntityCreateView, self).get_initial()

        initial["project"] = self.request.GET.get("project", None)

        initial["location"] = self.request.GET.get("location", None)

        return initial