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
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
|
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
from django.contrib.auth.models import User
from django.contrib.auth.mixins import UserPassesTestMixin
from django.http import HttpResponseRedirect
from datetime import datetime, timedelta
from django.utils import formats
from dateutil import rrule
def home(request):
return render(request,"dispatch/index.html")
def get_week_dates(date=None):
if date == None:
date = formats.date_format(datetime.now(), "SHORT_DATE_FORMAT")
dt = datetime.strptime(date, '%m/%d/%Y')
weekday = dt.weekday()
if weekday == 6:
start_date = dt
else:
weekday = weekday + 1
start_date = dt - timedelta(days=weekday)
end_date = start_date + timedelta(days=6)
next_week = end_date + timedelta(days=1)
previous_week = start_date - timedelta(days=1)
return start_date, end_date, next_week, previous_week
def split_loads_by_day(loads,start_date,end_date):
split_loads = {}
for date in rrule.rrule(rrule.DAILY,dtstart=start_date, until=end_date):
str_date = formats.date_format(date, "SHORT_DATE_FORMAT")
if str_date not in split_loads:
split_loads[str_date] = loads.filter(date=date)
return split_loads
class LoadDateSort(DetailView):
def get_context_data(self, **kwargs):
context = super(DetailView, self).get_context_data(**kwargs)
start_date, end_date, next_week, previous_week = get_week_dates(self.request.GET.get('date',None))
loads = self.get_object().load_set.filter(date__range=(start_date, end_date)).prefetch_related('company')
context['loads'] = split_loads_by_day(loads,start_date,end_date)
context['start_date'] = formats.date_format(start_date, "SHORT_DATE_FORMAT")
context['end_date'] = formats.date_format(end_date, "SHORT_DATE_FORMAT")
context['next_week'] = formats.date_format(next_week, "SHORT_DATE_FORMAT")
context['previous_week'] = formats.date_format(previous_week, "SHORT_DATE_FORMAT")
return context
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
# Driver CRUD
class DriverList(UserPassesTestMixin, ListView):
template_name = "dispatch/drivers/list.html"
model = User
def test_func(self):
return self.request.user.is_superuser
class DriverDetail(LoadDateSort):
template_name = "dispatch/drivers/detail.html"
model = User
def test_func(self):
return self.request.user.is_superuser
class DriverUpdate(UpdateView):
template_name = "dispatch/drivers/edit.html"
model = User
success_url = reverse_lazy('driver_list')
fields = ['username', 'first_name','last_name','email','groups', 'is_active']
def test_func(self):
return self.request.user.is_superuser
# Company CRUD
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(LoadDateSort):
template_name = "dispatch/companies/detail.html"
model = Company
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')
def test_func(self):
return self.request.user.is_superuser
# Load CRUD
class LoadList(FilteredListView):
template_name = "dispatch/loads/list.html"
model = Load
def get_queryset(self):
base_qs = super(LoadList, self).get_queryset()
start_date, end_date, next_week, previous_week = get_week_dates(self.request.GET.get('date',None))
return base_qs.filter(date__range=(start_date, end_date))
def get_context_data(self, **kwargs):
context = super(LoadList, self).get_context_data(**kwargs)
start_date, end_date, next_week, previous_week = get_week_dates(self.request.GET.get('date',None))
context['start_date'] = formats.date_format(start_date, "SHORT_DATE_FORMAT")
context['end_date'] = formats.date_format(end_date, "SHORT_DATE_FORMAT")
context['next_week'] = formats.date_format(next_week, "SHORT_DATE_FORMAT")
context['previous_week'] = formats.date_format(previous_week, "SHORT_DATE_FORMAT")
context['loads'] = split_loads_by_day(self.object_list,start_date,end_date)
return context
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','load_number','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')
|
