This provides ability to let all of the other fields of a communication

        instance to be editable.

        """

        return "Edit"

    # 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", "get"):

            context[key] = value

        else:

    return context

@register.simple_tag

def iptables(communication):

      request.

    """

    matches = current_url_equals(context, url_name, **kwargs)

    return return_value if matches else ''

def current_url_equals(context, url_name, **kwargs):

    """

    Helper function for checking if the specified URL corresponds to the current

    request path in the context.

    Arguments:

            kwarg = kwargs.get(key)

            resolved_kwarg = resolved.kwargs.get(key)

            if not resolved_kwarg or kwarg != resolved_kwarg:

                return False

    return matches

        comm = Communication.objects.get(pk=1)

        expected = "Test Entity 2 -> Test Entity 1 (TCP:22)"

        self.assertEqual(expected, str(comm))

:INPUT ACCEPT [0:0]

:OUTPUT ACCEPT [0:0]

:POSTROUTING ACCEPT [0:0]

COMMIT

"""

        self.assertEqual(generated, expected)

        Tests if projects are shown or not.

        """

        self.client.login(username="fullperms", password="fullperms")

        response = self.client.get(reverse("index"))

        self.assertQuerysetEqual(response.context["projects"], ["<Project: Test Project 1>", "<Project: Test Project 2>"])

        self.assertContains(response, "Test Project 1")

        self.assertContains(response, "Test Project 2")

class ProjectViewTest(ViewTest):

        self.assertContains(response, "Test Project 1")

        self.assertContains(response, "Test Location 1")

        self.assertContains(response, "Test Location 2")

class EntityView(ViewTest):

    def test_permission_denied(self):

        """

        Tests if permission will be denied for client without sufficient privileges.

        """

        self.client.login(username="noperms", password="noperms")

        response = self.client.get(reverse("entity_iptables", args=(1,)))

        self.assertContains(response, ":INPUT")

        self.assertContains(response, ":OUTPUT")

        self.assertContains(response, ":FORWARD")

class ProjectIptablesTest(ViewTest):

    def test_permission_denied(self):

        """

        Tests if permission will be denied for client without sufficient privileges.

        self.client.login(username="fullperms", password="fullperms")

        response = self.client.get(reverse("project_iptables", args=(200,)))

        self.assertEqual(response.status_code, 404)

        self.assertEqual(response.status_code, 404)

    def test_invalid_location(self):

        """

        Tests if a 404 is returned if invalid location is specified.

        """

        self.client.login(username="fullperms", password="fullperms")

        self.assertEqual(response.status_code, 404)

    def test_content_type(self):

        """

        Test if correct content type is being returned by the response.

        """

        self.client.login(username="fullperms", password="fullperms")

        response = self.client.get(reverse("project_iptables", args=(1,)))

        self.assertEqual(response['Content-Disposition'], 'attachment; filename="test_project_1-iptables.zip"')

        self.assertEqual(response['Content-Disposition'], 'attachment; filename="test_project_1-test_location_1-iptables.zip"')

    def test_project_entities_show(self):

        """

        Test if the project's iptables are being shown or not.

        """

        buff = StringIO(response.content)

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

        expected_zip_files = ["test_entity_1-iptables.conf",

                              "test_entity_2-iptables.conf",

                              "test_entity_3-iptables.conf",

        self.assertEqual(len(zipped_iptables.namelist()), 4)

        self.assertEqual(zipped_iptables.namelist(), expected_zip_files)

        for filename in expected_zip_files:

            iptables_file = zipped_iptables.read(filename)

        """

        Test if the project location's iptables are being shown or not.

        """

        self.client.login(username="fullperms", password="fullperms")

        buff = StringIO(response.content)

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

        expected_zip_files = ["test_entity_1-iptables.conf",

                              "test_entity_2-iptables.conf"]

            iptables_file = zipped_iptables.read(filename)

            self.assertIn(":INPUT", iptables_file)

            self.assertIn(":OUTPUT", iptables_file)

            self.assertIn(":FORWARD", iptables_file)

        zipped_iptables.close()

    # 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"),

)

    filter.add_chain(iptables.Chain("FORWARD", "DROP"))

    # Construct the iptables file using the two tables.

    content = "%s%s" % (filter, nat)

    return content

from braces.views import MultiplePermissionsRequiredMixin

# Application imports.

from .models import Project, Entity, Location

from .utils import generate_entity_iptables

class IndexView(MultiplePermissionsRequiredMixin, TemplateView):

    """

    Custom view used for rendering the index page.

    """

    template_name = 'conntrackt/index.html'

        # Add the rendered iptables rules to the context.

        context['entity_iptables'] = generate_entity_iptables(self.object)

        return context

def entity_iptables(request, pk):

    """

    Custom view that returns response containing iptables rules generated for an

    entity.

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

    # Fetch the entity, and construct the response with iptables rules as

    # content.

    entity = get_object_or_404(Entity, pk=pk)

    content = generate_entity_iptables(entity)

    response = HttpResponse(content, mimetype='text/plain')

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

    # browser to download the file with suggested filename.

    return response

def 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:

    # 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