diff --git a/conntrackt/views.py b/conntrackt/views.py --- a/conntrackt/views.py +++ b/conntrackt/views.py @@ -1,4 +1,6 @@ -from django.views.generic import TemplateView +from django.views.generic import TemplateView, DetailView + +from conntrackt.models import Project, Entity class IndexView(TemplateView): """ @@ -7,3 +9,16 @@ class IndexView(TemplateView): template_name = 'conntrackt/index.html' + 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)