views.py 804 Bytes
from django.shortcuts import render
from django.views.generic import TemplateView
from django.contrib.auth.mixins import LoginRequiredMixin

from courses.models import Course

class AppIndex(LoginRequiredMixin, TemplateView):
	template_name = "home_professor.html"

	def render_to_response(self, context, **response_kwargs):
		context = {}

		if self.request.user.type_profile == 2:
			template = "home_student.html"
			context['courses'] = Course.objects.filter(user = self.request.user)
		else:
			template = self.get_template_names()
			context['courses'] = Course.objects.filter(user = self.request.user)

		context['title'] = 'Amadeus'

		return self.response_class(request = self.request, template = template, context = context, using = self.template_engine, **response_kwargs)