1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
|
from django.shortcuts import render, redirect
from django.http import HttpResponse
from django.template import loader
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, Load
def home(request):
return render(request,"dispatch/index.html")
class FilteredListView(ListView):
def get_queryset(self):
base_qs = super(FilteredListView, self).get_queryset()
if not self.request.user.is_superuser:
return base_qs.filter(user=self.request.user)
return base_qs
class FilteredUpdateView(UpdateView):
def get_queryset(self):
base_qs = super(FilteredUpdateView, self).get_queryset()
if not self.request.user.is_superuser:
return base_qs.filter(user=self.request.user)
return base_qs
class FilteredDeleteView(DeleteView):
def get_queryset(self):
base_qs = super(FilteredDeleteView, self).get_queryset()
if not self.request.user.is_superuser:
return base_qs.filter(user=self.request.user)
return base_qs
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/loads/edit.html"
model = Company
success_url = reverse_lazy('company_list')
fields = ['name', 'address', 'phone_number','email_address']
class CompanyDelete(DeleteView):
template_name = "dispatch/loads/delete.html"
model = Company
success_url = reverse_lazy('company_list')
# Load CRUD
class LoadList(FilteredListView):
template_name = "dispatch/loads/list.html"
model = Load
class LoadCreate(CreateView):
template_name = "dispatch/loads/create.html"
model = Load
success_url = reverse_lazy('load_list')
fields = ['date', 'company','description','amount']
def get(self,request):
if request.user.is_superuser:
self.fields.insert(1,'user')
return super(LoadCreate, self).get(request)
def form_valid(self, form):
if not self.request.user.is_superuser:
load = form.save(commit=False)
load.user = self.request.user
return super(LoadCreate, self).form_valid(form)
class LoadDetail(DetailView):
template_name = "dispatch/loads/detail.html"
model = Load
def get_context_data(self, **kwargs):
context = super(LoadDetail, self).get_context_data(**kwargs)
return context
class LoadUpdate(FilteredUpdateView):
template_name = "dispatch/loads/edit.html"
model = Load
success_url = reverse_lazy('load_list')
fields = ['date', 'company','description','amount']
def get(self,request,pk):
if request.user.is_superuser:
self.fields.insert(1,'user')
return super(LoadUpdate, self).get(request)
def form_valid(self, form):
if not self.request.user.is_superuser:
load = form.save(commit=False)
load.user = self.request.user
return super(LoadUpdate, self).form_valid(form)
class LoadDelete(FilteredDeleteView):
template_name = "dispatch/loads/delete.html"
model = Load
success_url = reverse_lazy('load_list')
|
