196 lines
7.6 KiB
Python
196 lines
7.6 KiB
Python
from django.shortcuts import render, redirect, get_object_or_404
|
|
from django.contrib.auth.decorators import login_required
|
|
from django.contrib.auth import login, authenticate
|
|
from django.utils import timezone
|
|
from .models import Profile, Truck, Shipment, Bid, Message
|
|
from .forms import TruckForm, ShipmentForm, BidForm, UserRegistrationForm
|
|
from django.contrib import messages
|
|
from django.utils.translation import gettext as _
|
|
from django.db.models import Q
|
|
from django.contrib.auth.models import User
|
|
|
|
def home(request):
|
|
"""Render the landing screen for MASAR CARGO."""
|
|
context = {
|
|
"deployment_timestamp": timezone.now().timestamp(),
|
|
}
|
|
return render(request, "core/index.html", context)
|
|
|
|
def register(request):
|
|
if request.method == 'POST':
|
|
form = UserRegistrationForm(request.POST)
|
|
if form.is_valid():
|
|
user = form.save()
|
|
profile = user.profile
|
|
profile.role = form.cleaned_data.get('role')
|
|
profile.phone_number = form.cleaned_data.get('phone_number')
|
|
profile.save()
|
|
login(request, user)
|
|
messages.success(request, _("Registration successful. Welcome!"))
|
|
return redirect('dashboard')
|
|
else:
|
|
messages.error(request, _("Please correct the errors below."))
|
|
else:
|
|
form = UserRegistrationForm()
|
|
return render(request, 'registration/register.html', {'form': form})
|
|
|
|
@login_required
|
|
def dashboard(request):
|
|
profile, created = Profile.objects.get_or_create(user=request.user)
|
|
if profile.role == 'SHIPPER':
|
|
my_shipments = Shipment.objects.filter(shipper=request.user).order_by('-created_at')
|
|
return render(request, 'core/shipper_dashboard.html', {'shipments': my_shipments})
|
|
elif profile.role == 'TRUCK_OWNER':
|
|
approved_trucks = Truck.objects.filter(owner=request.user, is_approved=True)
|
|
pending_trucks = Truck.objects.filter(owner=request.user, is_approved=False)
|
|
my_bids = Bid.objects.filter(truck_owner=request.user).order_by('-created_at')
|
|
return render(request, 'core/truck_owner_dashboard.html', {
|
|
'trucks': approved_trucks,
|
|
'pending_trucks': pending_trucks,
|
|
'bids': my_bids
|
|
})
|
|
elif profile.role == 'ADMIN' or request.user.is_superuser:
|
|
pending_trucks = Truck.objects.filter(is_approved=False).order_by('-created_at')
|
|
approved_trucks = Truck.objects.filter(is_approved=True).order_by('-created_at')
|
|
context = {
|
|
'total_users': User.objects.count(),
|
|
'total_trucks': Truck.objects.count(),
|
|
'total_shipments': Shipment.objects.count(),
|
|
'total_bids': Bid.objects.count(),
|
|
'pending_trucks': pending_trucks,
|
|
'approved_trucks': approved_trucks,
|
|
}
|
|
return render(request, 'core/admin_dashboard.html', context)
|
|
else:
|
|
# Fallback for undefined roles
|
|
return redirect('/')
|
|
|
|
@login_required
|
|
def truck_register(request):
|
|
if request.user.profile.role != 'TRUCK_OWNER':
|
|
return redirect('dashboard')
|
|
|
|
if request.method == 'POST':
|
|
form = TruckForm(request.POST, request.FILES)
|
|
if form.is_valid():
|
|
truck = form.save(commit=False)
|
|
truck.owner = request.user
|
|
truck.is_approved = False # Ensure it stays false on new reg
|
|
truck.save()
|
|
messages.success(request, _("Truck registered successfully! It will be visible after admin approval."))
|
|
return redirect('dashboard')
|
|
else:
|
|
messages.error(request, _("There was an error in your registration. Please check the form."))
|
|
else:
|
|
form = TruckForm()
|
|
|
|
return render(request, 'core/truck_register.html', {'form': form})
|
|
|
|
@login_required
|
|
def approve_truck(request, truck_id):
|
|
if not (request.user.profile.role == 'ADMIN' or request.user.is_superuser):
|
|
return redirect('dashboard')
|
|
|
|
truck = get_object_or_404(Truck, id=truck_id)
|
|
truck.is_approved = True
|
|
truck.save()
|
|
messages.success(request, _("Truck approved successfully!"))
|
|
return redirect('dashboard')
|
|
|
|
@login_required
|
|
def suspend_truck(request, truck_id):
|
|
if not (request.user.profile.role == 'ADMIN' or request.user.is_superuser):
|
|
return redirect('dashboard')
|
|
|
|
truck = get_object_or_404(Truck, id=truck_id)
|
|
truck.is_approved = False
|
|
truck.save()
|
|
messages.warning(request, _("Truck has been suspended."))
|
|
return redirect('dashboard')
|
|
|
|
@login_required
|
|
def post_shipment(request):
|
|
if request.user.profile.role != 'SHIPPER':
|
|
return redirect('dashboard')
|
|
|
|
if request.method == 'POST':
|
|
form = ShipmentForm(request.POST)
|
|
if form.is_valid():
|
|
shipment = form.save(commit=False)
|
|
shipment.shipper = request.user
|
|
shipment.save()
|
|
messages.success(request, _("Shipment posted successfully!"))
|
|
return redirect('dashboard')
|
|
else:
|
|
messages.error(request, _("Please correct the errors in the form."))
|
|
else:
|
|
form = ShipmentForm()
|
|
|
|
return render(request, 'core/post_shipment.html', {'form': form})
|
|
|
|
@login_required
|
|
def marketplace(request):
|
|
if request.user.profile.role != 'TRUCK_OWNER':
|
|
return redirect('dashboard')
|
|
|
|
shipments = Shipment.objects.filter(status='OPEN').order_by('-created_at')
|
|
return render(request, 'core/marketplace.html', {'shipments': shipments})
|
|
|
|
@login_required
|
|
def place_bid(request, shipment_id):
|
|
shipment = get_object_or_404(Shipment, id=shipment_id)
|
|
if request.user.profile.role != 'TRUCK_OWNER':
|
|
return redirect('dashboard')
|
|
|
|
# Optional: Only allow bidding if user has at least one approved truck
|
|
if not Truck.objects.filter(owner=request.user, is_approved=True).exists():
|
|
messages.warning(request, _("You must have at least one approved truck to place a bid."))
|
|
return redirect('dashboard')
|
|
|
|
if request.method == 'POST':
|
|
form = BidForm(request.POST, user=request.user)
|
|
if form.is_valid():
|
|
bid = form.save(commit=False)
|
|
bid.truck_owner = request.user
|
|
bid.shipment = shipment
|
|
bid.save()
|
|
messages.success(request, _("Bid placed successfully!"))
|
|
return redirect('marketplace')
|
|
else:
|
|
messages.error(request, _("Error placing bid. Please check the form."))
|
|
else:
|
|
form = BidForm(user=request.user)
|
|
|
|
return render(request, 'core/place_bid.html', {'form': form, 'shipment': shipment})
|
|
|
|
@login_required
|
|
def shipment_detail(request, shipment_id):
|
|
shipment = get_object_or_404(Shipment, id=shipment_id)
|
|
# Security: check if user is shipper or a truck owner who bid
|
|
if shipment.shipper != request.user and not Bid.objects.filter(shipment=shipment, truck_owner=request.user).exists():
|
|
if request.user.profile.role != 'ADMIN' and not request.user.is_superuser:
|
|
return redirect('dashboard')
|
|
|
|
bids = shipment.bids.all()
|
|
return render(request, 'core/shipment_detail.html', {'shipment': shipment, 'bids': bids})
|
|
|
|
@login_required
|
|
def accept_bid(request, bid_id):
|
|
bid = get_object_or_404(Bid, id=bid_id)
|
|
if bid.shipment.shipper != request.user:
|
|
return redirect('dashboard')
|
|
|
|
# Accept this bid
|
|
bid.status = 'ACCEPTED'
|
|
bid.save()
|
|
|
|
# Reject others
|
|
bid.shipment.bids.exclude(id=bid_id).update(status='REJECTED')
|
|
|
|
# Update shipment
|
|
bid.shipment.status = 'IN_PROGRESS'
|
|
bid.shipment.assigned_truck = bid.truck
|
|
bid.shipment.save()
|
|
|
|
messages.success(request, _("Bid accepted! Shipment is now in progress."))
|
|
return redirect('shipment_detail', shipment_id=bid.shipment.id) |