from django.contrib import messages from django.contrib.auth import authenticate, login from django.contrib.auth.decorators import login_required, user_passes_test from django.contrib.auth.forms import AuthenticationForm from django.contrib.auth.hashers import make_password from django.db.models import Exists, OuterRef from django.http import JsonResponse from django.shortcuts import render, redirect, get_object_or_404 from django.views.decorators.http import require_POST from .models import Issue, IssueCategory, User, Vote, Comment, Department from .forms import CitizenRegistrationForm, IssueForm, CommentForm def home(request): total_issues = Issue.objects.count() resolved_issues = Issue.objects.filter(status='resolved').count() active_users = User.objects.filter(is_active=True).count() # Get recently reported issues (last 3 issues) recent_issues = Issue.objects.all().order_by('-created_at')[:3] for issue in recent_issues: issue.user_has_voted = issue.has_user_voted(request.user) if request.user.is_authenticated else False # Get municipal departments count (assuming you have a Department model) # If you don't have one yet, you can use a placeholder or create the model municipal_departments = 5 # Placeholder - replace with actual count when you have the model context = { 'total_issues': total_issues, 'resolved_issues': resolved_issues, 'active_users': active_users, 'municipal_departments': municipal_departments, 'recent_issues': recent_issues, } return render(request, 'core/index.html', context) @login_required def citizen_dashboard(request): if not request.user.is_citizen: messages.error(request, 'Access denied. Citizen role required.') return redirect('home') # Get only basic data for now all_user_issues = Issue.objects.filter(reporter=request.user) user_issues_display = all_user_issues.order_by('-created_at')[:5] resolved_count = all_user_issues.filter(status='resolved').count() categories = IssueCategory.objects.all() context = { 'user_issues': user_issues_display, 'resolved_count': resolved_count, 'categories': categories, 'issue_form': IssueForm(), } return render(request, 'dashboard/citizen_dashboard.html', context) @login_required def report_issue(request): if not request.user.is_citizen: messages.error(request, 'Access denied. Citizen role required.') return redirect('home') if request.method == 'POST': form = IssueForm(request.POST, request.FILES) if form.is_valid(): issue = form.save(commit=False) issue.reporter = request.user issue.save() messages.success(request, 'Issue reported successfully!') return redirect('citizen_dashboard') else: messages.error(request, 'Please correct the errors below.') else: form = IssueForm() return render(request, 'core/report_issue.html', {'form': form}) @login_required def view_all_issues(request): if not request.user.is_citizen: messages.error(request, 'Access denied. Citizen role required.') return redirect('home') # Efficiently annotate whether the current user has voted user_vote_subq = Vote.objects.filter(user=request.user, issue_id=OuterRef('pk')) issues = ( Issue.objects .select_related('category', 'reporter') .annotate(user_has_voted=Exists(user_vote_subq)) .order_by('-created_at') ) categories = IssueCategory.objects.all() # Optional filters (status/category) if you wired the dropdowns status = request.GET.get('status') or '' category_id = request.GET.get('category') or '' if status: issues = issues.filter(status=status) if category_id: issues = issues.filter(category_id=category_id) return render(request, 'issues/view_all_issues.html', { 'issues': issues, 'categories': categories, 'selected_status': status, 'selected_category': category_id, }) def register(request): if request.method == 'POST': form = CitizenRegistrationForm(request.POST) if form.is_valid(): user = form.save() messages.success(request, 'Registration successful! Please login.') return redirect('login') else: form = CitizenRegistrationForm() return render(request, 'core/register.html', {'form': form}) def custom_login(request): if request.method == 'POST': form = AuthenticationForm(request, data=request.POST) if form.is_valid(): username = form.cleaned_data.get('username') password = form.cleaned_data.get('password') user = authenticate(username=username, password=password) if user is not None: login(request, user) messages.success(request, f'Welcome back, {username}!') return redirect('home') else: messages.error(request, 'Invalid username or password.') else: messages.error(request, 'Invalid username or password.') else: form = AuthenticationForm() return render(request, 'core/login.html', {'form': form}) @login_required @require_POST def vote_issue(request, issue_id): try: issue = Issue.objects.get(id=issue_id) vote, created = Vote.objects.get_or_create(user=request.user, issue=issue) if not created: # User already voted, so remove the vote (toggle) vote.delete() voted = False else: voted = True return JsonResponse({ 'success': True, 'voted': voted, 'vote_count': issue.vote_count() }) except Issue.DoesNotExist: return JsonResponse({'success': False, 'error': 'Issue not found'}) except Exception as e: return JsonResponse({'success': False, 'error': str(e)}) @login_required def add_comment(request, pk, parent_id=None): issue = get_object_or_404(Issue, pk=pk) parent = get_object_or_404(Comment, pk=parent_id) if parent_id else None if request.method == "POST": form = CommentForm(request.POST) if form.is_valid(): comment = form.save(commit=False) comment.issue = issue comment.user = request.user comment.parent = parent comment.save() return redirect("issue_detail", pk=pk) def issue_detail(request, pk): issue = get_object_or_404(Issue, pk=pk) comments = issue.comments.filter(parent__isnull=True) # top-level comments return render(request, "issues/issue_detail.html", {"issue": issue, "comments": comments}) def add_comment(request, pk, parent_id=None): if request.method == "POST" and request.user.is_authenticated: issue = get_object_or_404(Issue, pk=pk) text = request.POST.get("text") content = request.POST.get("content") if content: parent = Comment.objects.get(pk=parent_id) if parent_id else None Comment.objects.create(issue=issue, user=request.user, content=content, parent=parent) return redirect("issue_detail", pk=pk) def superadmin_check(user): return user.is_superuser @login_required @user_passes_test(superadmin_check) def superadmin_dashboard(request): return render(request, "dashboard/superadmin_dashboard.html") @login_required @user_passes_test(superadmin_check) def manage_departments(request): departments = Department.objects.all().order_by("name") if request.method == "POST": name = request.POST.get("name") description = request.POST.get("description") if name: Department.objects.create(name=name, description=description) return redirect("manage_departments") return render(request, "department/manage_departments.html", { "departments": departments }) @login_required @user_passes_test(superadmin_check) def department_detail(request, pk): department = get_object_or_404(Department, pk=pk) users = department.users.all() if request.method == "POST": username = request.POST.get("username") email = request.POST.get("email") password = request.POST.get("password") if username and password: user = User.objects.create( username=username, email=email, password=make_password(password), # hash the password is_staff=True # mark as staff ) department.users.add(user) return redirect("department_detail", pk=department.id) return render(request, "department/department_detail.html", { "department": department, "users": users, })