umschool/python_code_check_system/views.py - master - therealyou/python-code-check-system

umschool/python_code_check_system/views.py
79 строк2 KB
Artem_dev
added some functional in tasks (#24)
03 янв 2024, 22:53
03 янв 2024, 22:53e9da24d
from django.shortcuts import render, redirect
from .models import Task, Student
from django.views.generic import ListView, CreateView, TemplateView, DetailView
from .forms import TaskForm


class TaskHomeListView(ListView):
    model = Task
    template_name = 'python_code_check_system/tasks.html'
    context_object_name = 'all_tasks'

    def get_context_data(self, *, object_list=None, **kwargs):
        context = super(TaskHomeListView, self).get_context_data(**kwargs)
        context['main'] = 'Задания'
        context['title'] = 'Задания'
        return context

    def get(self, request, *args, **kwargs):
        tasks = Task.objects.all()
        return render(request, 'python_code_check_system/tasks.html', {'all_tasks' : tasks})


class StudentHomeListView(ListView):
    model = Student
    template_name = 'python_code_check_system/profile.html'
    context_object_name = 'profile'

    def get_queryset(self):
        return Student.objects.get(pk=1)

    def get_context_data(self, *, object_list=None, **kwargs):
        context = super().get_context_data()
        context['main'] = 'Профиль'
        context['title'] = 'Профиль'
        return context

    def get(self, request, *args, **kwargs):
        return render(request, 'python_code_check_system/profile.html')


class AddTask(CreateView):
    model = Task
    form = TaskForm()
    fields = ['name', 'complexity', 'description']
    template_name = 'python_code_check_system/add_to_db_page.html'
    success_url = 'tasks/'
    extra_context = {
        'form': form
    }

    def get_context_data(self, *, object_list=None, **kwargs):
        context = super().get_context_data()
        context['main'] = 'Задания'
        context['title'] = 'Задания'
        return context

    def post(self, request, *args, **kwargs):
        form = TaskForm(request.POST)
        if form.is_valid():
            form.save()
            return redirect('/tasks/', permanent=True)


class HomeTemplateView(TemplateView):
    template_name = 'python_code_check_system/index.html'


class ContactsTemplateView(TemplateView):
    template_name = 'python_code_check_system/contacts.html'


class AboutTemplateView(TemplateView):
    template_name = 'python_code_check_system/about.html'


class TaskDetailView(DetailView):
    model = Task
    template_name = 'python_code_check_system/task_detail_view.html'
    context_object_name = 'task'