mirror of
https://github.com/nottinghamtec/PyRIGS.git
synced 2026-01-17 21:42:14 +00:00
30 lines
830 B
Python
30 lines
830 B
Python
from django.conf import settings
|
|
from django.contrib.auth import get_user_model
|
|
from django.db.models import Q
|
|
from django.shortcuts import render
|
|
from django.views import generic
|
|
from training import models
|
|
|
|
|
|
# Create your views here.
|
|
|
|
|
|
class UserTrainingRecordView(generic.DetailView):
|
|
model = get_user_model()
|
|
template_name = 'training/profile_detail.html'
|
|
|
|
def get_context_data(self, **kwargs):
|
|
context = super(UserTrainingRecordView, self).get_context_data(**kwargs)
|
|
context['categories'] = models.TrainingCategory.objects.all()
|
|
return context
|
|
|
|
|
|
class SelfUserTrainingRecordView(UserTrainingRecordView):
|
|
template_name = 'training/index.html'
|
|
|
|
def get_queryset(self):
|
|
pk = self.request.user.id
|
|
self.kwargs['pk'] = pk
|
|
|
|
return self.model.objects.filter(pk=pk)
|