57 lines
1.4 KiB
Python
57 lines
1.4 KiB
Python
from django.urls import reverse_lazy, reverse
|
|
from django.views.generic import ListView, DetailView, CreateView, UpdateView, DeleteView
|
|
|
|
from web.models import Activity, User, ActivityLog
|
|
|
|
|
|
class UserListView(ListView):
|
|
model = User
|
|
|
|
|
|
class UserCreateView(CreateView):
|
|
model = User
|
|
fields = ['name', 'email']
|
|
success_url = reverse_lazy('users')
|
|
|
|
|
|
class UserDetailView(DetailView):
|
|
model = User
|
|
context_object_name = 'user'
|
|
pk_url_kwarg = 'user_id'
|
|
|
|
def get_context_data(self, **kwargs):
|
|
context = super().get_context_data(**kwargs)
|
|
context['activity_logs'] = ActivityLog.objects.filter(user=self.object).order_by('start_time')
|
|
return context
|
|
|
|
|
|
class ActivityLogCreateView(CreateView):
|
|
model = ActivityLog
|
|
fields = ['activity', 'user', 'start_time', 'end_time']
|
|
|
|
def get_success_url(self):
|
|
return reverse('user', kwargs={
|
|
'user_id': self.object.user.id,
|
|
})
|
|
|
|
|
|
class ActivityLogUpdateView(UpdateView):
|
|
model = ActivityLog
|
|
fields = ['activity', 'user', 'start_time', 'end_time']
|
|
pk_url_kwarg = 'activity_log_id'
|
|
|
|
def get_success_url(self):
|
|
return reverse('user', kwargs={
|
|
'user_id': self.object.user.id,
|
|
})
|
|
|
|
|
|
class ActivityLogDeleteView(DeleteView):
|
|
model = ActivityLog
|
|
pk_url_kwarg = 'activity_log_id'
|
|
|
|
def get_success_url(self):
|
|
return reverse('user', kwargs={
|
|
'user_id': self.object.user.id,
|
|
})
|