network: fix host update

network/forms.py

@@ -50,6 +50,7 @@ class HostForm(ModelForm):
                         'description',
                         'description',
                         'location',
                         'location',
                         'comment',
                         'comment',
+                        'owner',
                         # 'created_at',
                         # 'created_at',
                         # 'modified_at',
                         # 'modified_at',
                         # 'id'
                         # 'id'

network/views.py

 from django.views.generic import TemplateView
 from django.views.generic import TemplateView
 from django.views.generic import UpdateView
 from django.views.generic import UpdateView
+from django.core.urlresolvers import reverse_lazy
 
 
 from django_tables2 import SingleTableView
 from django_tables2 import SingleTableView
 
 
@@ -23,6 +24,10 @@ class HostDetail(UpdateView):
     template_name = "network/host-edit.html"
     template_name = "network/host-edit.html"
     form_class = HostForm
     form_class = HostForm
 
 
+    def get_success_url(self):
+        if 'pk' in self.kwargs:
+            return reverse_lazy('network.host', kwargs=self.kwargs)
+
 
 
 class VlanList(SingleTableView):
 class VlanList(SingleTableView):
     model = Vlan
     model = Vlan
@@ -40,3 +45,5 @@ class VlanDetail(UpdateView):
         q = Host.objects.filter(vlan=self.object).all()
         q = Host.objects.filter(vlan=self.object).all()
         context['host_list'] = SmallHostTable(q)
         context['host_list'] = SmallHostTable(q)
         return context
         return context
+
+    success_url = reverse_lazy('network.vlan_list')