from django.shortcuts import render
from django.contrib.auth.mixins import PermissionRequiredMixin
from django.views.generic import ListView, DetailView, CreateView
from django.contrib.auth.models import Permission
from django.contrib.auth.decorators import permission_required

from .models import Project
from users.models import CustomUser
from .forms import ProjectForm


class ProjectListView(ListView):
    model = Project

    def get_context_data(self, **kwargs):
        context = super(ProjectListView, self).get_context_data(**kwargs)
        return context

class ProjectDetailView(DetailView):
    model = Project



class ProjectCreateView(PermissionRequiredMixin, CreateView):
    model = Project
    permission_required = ['projects.add_project']
    form_class = ProjectForm
    success_url = '/projects/'