from django.views.generic import TemplateView, DetailView

from conntrackt.models import Project, Entity

    def get_context_data(self, **kwargs):

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

        context['projects'] = Project.objects.all().order_by('name')

        return context

class IptablesView(DetailView):

    model = Entity

    template_name = 'conntrackt/entity_iptables.html'

    def render_to_response(self, context, **kwargs):

        return super(IptablesView, self).render_to_response(context,

                                                            content_type='text/plain', **kwargs)