1 files changed, 36 insertions, 21 deletions

diff --git a/app/dispatch/views.py b/app/dispatch/views.py
index 916c04b..600a746 100644
--- a/app/dispatch/views.py
+++ b/app/dispatch/views.py
@@ -1,30 +1,45 @@
 from django.shortcuts import render, redirect
 from django.http import HttpResponse
 from django.template import loader
-
-from django.contrib.auth.decorators import login_required
-# from django.contrib.auth import authenticate, login, logout
+from django.core.urlresolvers import reverse
 import django.contrib.auth as auth
-
+from django.conf import settings
 # Create your views here.
+from django.views.generic import TemplateView,ListView
+from django.views.generic.detail import DetailView
+from django.views.generic.edit import CreateView, UpdateView, DeleteView
+from django.core.urlresolvers import reverse_lazy
+from dispatch.models import Company
 
-
-@login_required(login_url="login/")
 def home(request):
     return render(request,"dispatch/index.html")
 
-def login(request):
-    username = request.POST['username']
-    password = request.POST['password']
-    user = auth.authenticate(request, username=username, password=password)
-    if user is not None:
-        auth.login(request, user)
-        return redirect('/dispatch/')
-    else:
-        template = loader.get_template('dispatch/login.html')
-        context = {}
-        return HttpResponse(template.render(context, request))
-
-def logout(request):
-    auth.logout(request)
-    return redirect('/dispatch/login', permanent=False)
+
+class CompanyList(ListView):
+    template_name = "dispatch/companies/list.html"
+    model = Company
+
+class CompanyCreate(CreateView):
+    template_name = "dispatch/companies/create.html"
+    model = Company
+    success_url = reverse_lazy('company_list')
+    fields = ['name', 'address', 'phone_number','email_address']
+
+class CompanyDetail(DetailView):
+    template_name = "dispatch/companies/detail.html"
+    model = Company
+
+    def get_context_data(self, **kwargs):
+        context = super(CompanyDetail, self).get_context_data(**kwargs)
+        return context
+
+class CompanyUpdate(UpdateView):
+    template_name = "dispatch/companies/edit.html"
+    model = Company
+    success_url = reverse_lazy('company_list')
+    fields = ['name', 'address', 'phone_number','email_address']
+
+class CompanyDelete(DeleteView):
+    template_name = "dispatch/companies/delete.html"
+    model = Company
+    success_url = reverse_lazy('company_list')
\ No newline at end of file