91 lines
3.5 KiB
Python
91 lines
3.5 KiB
Python
# main/views.py
|
|
from django.shortcuts import render, redirect, get_object_or_404
|
|
from django.contrib.auth import authenticate, login, logout
|
|
from django.contrib.auth.decorators import login_required
|
|
from .models import Project, Task, RegisteredUser
|
|
from .forms import ProjectForm, TaskForm
|
|
from rest_framework import status
|
|
from rest_framework.views import APIView
|
|
from rest_framework.response import Response
|
|
|
|
|
|
class UserRegistrationView(APIView):
|
|
def post(self, request):
|
|
serializer = UserRegistrationSerializer(data=request.data)
|
|
if serializer.is_valid():
|
|
user = serializer.save()
|
|
return Response({"id": user.id, "username": user.username}, status=status.HTTP_201_CREATED)
|
|
return Response(serializer.errors, status=status.HTTP_400_BAD_REQUEST)
|
|
|
|
class UserLoginView(APIView):
|
|
def post(self, request):
|
|
username = request.data.get('username')
|
|
password = request.data.get('password')
|
|
user = authenticate(request, username=username, password=password)
|
|
if user is not None:
|
|
login(request, user)
|
|
return Response({"message": "Login successful"}, status=status.HTTP_200_OK)
|
|
return Response({error: "Invalid credentials"}, status=status.HTTP_401_UNAUTHORIZED)
|
|
|
|
class UserLogoutView(APIView):
|
|
def post(self, request):
|
|
logout(request)
|
|
return Response({"message": "Logout successful"}, status=status.HTTP_200_OK)
|
|
|
|
# create project
|
|
@login_required
|
|
def create_project(request):
|
|
if request.method == 'POST':
|
|
form = ProjectForm(request.POST)
|
|
if form.is_valid():
|
|
project = form.save(commit=False)
|
|
project.initiator = request.user
|
|
project.save()
|
|
# Notify PMO Admins
|
|
pmo_admins = User.objects.filter(profile__role='PMO Admin')
|
|
for admin in pmo_admins:
|
|
send_mail(
|
|
'New Project Created',
|
|
f'A new project "{project.name}" has been created and requires a project manager assignment.',
|
|
'admin@example.com',
|
|
[admin.email],
|
|
fail_silently=False,
|
|
)
|
|
return redirect('project_dashboard', project_id=project.id)
|
|
else:
|
|
form = ProjectForm()
|
|
return render(request, 'create_project.html', {'form': form})
|
|
|
|
# Load project as current focus
|
|
@login_required
|
|
def load_project(request):
|
|
if request.method == 'POST':
|
|
project_id = request.POST['project_id']
|
|
return redirect('project_dashboard', project_id=project_id)
|
|
projects = Project.objects.all()
|
|
return render(request, 'load_project.html', {'projects': projects})
|
|
|
|
|
|
# Active projects
|
|
@login_required
|
|
def project_dashboard(request, project_id):
|
|
project = get_object_or_404(Project, id=project_id)
|
|
tasks = project.tasks.all()
|
|
return render(request, 'project_dashboard.html', {'project': project, 'tasks': tasks})
|
|
|
|
# Create task in current focus project - if no project chosen default project for user is focus
|
|
@login_required
|
|
def create_task(request, project_id):
|
|
project = get_object_or_404(Project, id=project_id)
|
|
if request.method == 'POST':
|
|
form = TaskForm(request.POST)
|
|
if form.is_valid():
|
|
task = form.save(commit=False)
|
|
task.project = project
|
|
task.save()
|
|
return redirect('project_dashboard', project_id=project.id)
|
|
else:
|
|
form = TaskForm()
|
|
return render(request, 'create_task.html', {'form': form, 'project': project})
|
|
|