37794-vm/core/models.py
2026-02-02 03:35:13 +00:00

563 lines
27 KiB
Python

from django.db import models
from django.contrib.auth.models import User
from django.utils.translation import gettext_lazy as _
from django.utils.translation import get_language
from django.db.models.signals import post_save
from django.dispatch import receiver
from django.utils import timezone
from django.core.exceptions import ValidationError
import uuid
from decimal import Decimal
class Country(models.Model):
name_en = models.CharField(_('Name (English)'), max_length=100)
name_ar = models.CharField(_('Name (Arabic)'), max_length=100)
phone_code = models.CharField(_('Phone Code'), max_length=10, blank=True, help_text=_("e.g. +968"))
@property
def name(self):
if get_language() == 'ar':
return self.name_ar
return self.name_en
def __str__(self):
return f"{self.name} ({self.phone_code})" if self.phone_code else self.name
class Meta:
verbose_name = _('Country')
verbose_name_plural = _('Countries')
class Governate(models.Model):
country = models.ForeignKey(Country, on_delete=models.CASCADE, verbose_name=_('Country'))
name_en = models.CharField(_('Name (English)'), max_length=100)
name_ar = models.CharField(_('Name (Arabic)'), max_length=100)
@property
def name(self):
if get_language() == 'ar':
return self.name_ar
return self.name_en
def __str__(self):
return f"{self.name} ({self.country.name})"
class Meta:
verbose_name = _('Governate')
verbose_name_plural = _('Governates')
class City(models.Model):
governate = models.ForeignKey(Governate, on_delete=models.CASCADE, verbose_name=_('Governate'))
name_en = models.CharField(_('Name (English)'), max_length=100)
name_ar = models.CharField(_('Name (Arabic)'), max_length=100)
@property
def name(self):
if get_language() == 'ar':
return self.name_ar
return self.name_en
def __str__(self):
return f"{self.name} ({self.governate.name})"
class Meta:
verbose_name = _('City')
verbose_name_plural = _('Cities')
class ParcelType(models.Model):
name_en = models.CharField(_('Name (English)'), max_length=100)
name_ar = models.CharField(_('Name (Arabic)'), max_length=100)
@property
def name(self):
if get_language() == 'ar':
return self.name_ar
return self.name_en
def __str__(self):
return self.name
class Meta:
verbose_name = _('Parcel Type')
verbose_name_plural = _('Parcel Types')
class Profile(models.Model):
ROLE_CHOICES = (
("shipper", _("Shipper")),
("car_owner", _("Car Owner")),
)
DRIVER_GRADE_CHOICES = (
("none", _("No Grade")),
("bronze_3", _("Bronze III")),
("bronze_2", _("Bronze II")),
("bronze_1", _("Bronze I")),
("silver", _("Silver")),
("gold", _("Gold")),
)
user = models.OneToOneField(User, on_delete=models.CASCADE, verbose_name=_('User'))
role = models.CharField(_('Role'), max_length=20, choices=ROLE_CHOICES, default='shipper')
phone_number = models.CharField(_('Phone Number'), max_length=20, blank=True)
profile_picture = models.ImageField(_('Profile Picture'), upload_to='profile_pics/', blank=True, null=True)
address = models.CharField(_('Address'), max_length=255, blank=True)
# Driver specific fields
license_front_image = models.ImageField(_('License Front Image'), upload_to='licenses/', blank=True, null=True)
license_back_image = models.ImageField(_('License Back Image'), upload_to='licenses/', blank=True, null=True)
car_plate_number = models.CharField(_('Car Plate Number'), max_length=20, blank=True)
bank_account_number = models.CharField(_('Bank Account Number'), max_length=50, blank=True)
country = models.ForeignKey(Country, on_delete=models.SET_NULL, null=True, blank=True, verbose_name=_('Country'))
governate = models.ForeignKey(Governate, on_delete=models.SET_NULL, null=True, blank=True, verbose_name=_('Governate'))
city = models.ForeignKey(City, on_delete=models.SET_NULL, null=True, blank=True, verbose_name=_('City'))
# Approval Status
# Driver Assessment
driver_grade = models.CharField(_("Driver Grade"), max_length=20, choices=DRIVER_GRADE_CHOICES, default="none")
is_recommended = models.BooleanField(_("Recommended by Shippers"), default=False)
is_approved = models.BooleanField(_('Approved'), default=False, help_text=_("Designates whether this user is approved to use the platform (mainly for drivers)."))
language = models.CharField(_("Language"), max_length=10, choices=[("en", _("English")), ("ar", _("Arabic"))], default="ar")
# Ban Status
is_banned = models.BooleanField(_('Banned'), default=False)
ban_reason = models.TextField(_('Ban Reason'), blank=True)
def __str__(self):
return f"{self.user.username} - {self.get_role_display()}"
def get_average_rating(self):
if self.role != 'car_owner':
return None
ratings = self.user.received_ratings.all()
if not ratings:
return 0
return sum(r.rating for r in ratings) / len(ratings)
def get_rating_count(self):
if self.role != 'car_owner':
return 0
return self.user.received_ratings.count()
class Meta:
verbose_name = _('Profile')
verbose_name_plural = _('Profiles')
@receiver(post_save, sender=User)
def create_user_profile(sender, instance, created, **kwargs):
if created:
Profile.objects.get_or_create(user=instance)
@receiver(post_save, sender=User)
def save_user_profile(sender, instance, **kwargs):
if hasattr(instance, 'profile'):
instance.profile.save()
class PricingRule(models.Model):
min_distance = models.DecimalField(_('Min Distance (km)'), max_digits=10, decimal_places=2)
max_distance = models.DecimalField(_('Max Distance (km)'), max_digits=10, decimal_places=2)
min_weight = models.DecimalField(_('Min Weight (kg)'), max_digits=10, decimal_places=2)
max_weight = models.DecimalField(_('Max Weight (kg)'), max_digits=10, decimal_places=2)
price = models.DecimalField(_('Price (OMR)'), max_digits=10, decimal_places=3)
def __str__(self):
return f"{self.min_distance}-{self.max_distance}km | {self.min_weight}-{self.max_weight}kg = {self.price} OMR"
class Meta:
verbose_name = _('Pricing Rule')
verbose_name_plural = _('Pricing Rules')
ordering = ['min_distance', 'min_weight']
class PlatformProfile(models.Model):
name = models.CharField(_('Platform Name'), max_length=100)
logo = models.ImageField(_('Logo'), upload_to='platform_logos/', blank=True, null=True)
favicon = models.ImageField(_('Favicon'), upload_to='platform_logos/', blank=True, null=True, help_text=_("Upload a favicon (e.g., .ico or .png)"))
admin_panel_logo = models.ImageField(_('Admin Panel Logo'), upload_to='platform_logos/', blank=True, null=True, help_text=_("Logo for the Admin Panel (top left)."))
slogan = models.CharField(_('Slogan'), max_length=255, blank=True)
address = models.TextField(_('Address'), blank=True)
phone_number = models.CharField(_('Phone Number'), max_length=50, blank=True)
registration_number = models.CharField(_('Registration Number'), max_length=100, blank=True)
vat_number = models.CharField(_('VAT Number'), max_length=100, blank=True)
# Financial Configuration
platform_fee_percentage = models.DecimalField(_('Platform Fee (%)'), max_digits=5, decimal_places=2, default=Decimal('0.00'), help_text=_("Percentage deducted from total trip price."))
# Integrations
google_maps_api_key = models.CharField(_('Google Maps API Key'), max_length=255, blank=True, help_text=_("API Key for Google Maps (Distance Matrix, Maps JS)."))
# Bilingual Policies
privacy_policy_en = models.TextField(_('Privacy Policy (English)'), blank=True)
privacy_policy_ar = models.TextField(_('Privacy Policy (Arabic)'), blank=True)
terms_conditions_en = models.TextField(_('Terms and Conditions (English)'), blank=True)
terms_conditions_ar = models.TextField(_('Terms and Conditions (Arabic)'), blank=True)
# WhatsApp Configuration (Wablas Gateway)
whatsapp_access_token = models.TextField(_('Wablas API Token'), blank=True, help_text=_("Your Wablas API Token."))
whatsapp_business_phone_number_id = models.CharField(_('Wablas Domain'), max_length=100, blank=True, default="https://deu.wablas.com", help_text=_("The Wablas API domain (e.g., https://deu.wablas.com)."))
whatsapp_app_secret = models.CharField(_('Wablas Secret Key'), max_length=255, blank=True, help_text=_("Your Wablas API Secret Key (if required)."))
# Payment Configuration
enable_payment = models.BooleanField(_('Enable Payment'), default=True, help_text=_("Toggle to enable or disable payments on the platform."))
# Testing / Development
auto_mark_paid = models.BooleanField(_('Test Mode: Auto-Paid'), default=False, help_text=_("If enabled, newly created parcels will automatically be marked as 'Paid' for testing."))
# Maintenance / Availability
accepting_shipments = models.BooleanField(_("Accepting Shipments"), default=True, help_text=_("Toggle to allow or stop receiving new parcel shipments."))
maintenance_message_en = models.TextField(_("Maintenance Message (English)"), blank=True, help_text=_("Message to show when shipments are stopped."))
maintenance_message_ar = models.TextField(_("Maintenance Message (Arabic)"), blank=True, help_text=_("Message to show when shipments are stopped."))
# Driver Rejection / Auto-Ban
# Driver Warning / Auto-Ban
enable_auto_ban_on_warnings = models.BooleanField(_("Enable Auto-Ban on Warnings"), default=False, help_text=_("Automatically ban drivers who exceed a certain number of warnings."))
max_warnings_before_ban = models.PositiveIntegerField(_("Max Warnings Before Ban"), default=3, help_text=_("Number of warnings allowed before auto-ban."))
auto_ban_on_rejections = models.BooleanField(_("Enable Auto-Ban on Rejections"), default=False, help_text=_("Automatically ban drivers who exceed a certain number of rejections."))
rejection_limit = models.PositiveIntegerField(_("Rejection Limit"), default=5, help_text=_("Number of rejections allowed before auto-ban."))
# Live Activity Ticker
ticker_limit = models.PositiveIntegerField(_("Ticker Shipment Limit"), default=10, help_text=_("Number of recent shipments to show in the live activity ticker."))
ticker_bg_color = models.CharField(_("Ticker Background Color"), max_length=20, default="#FFFFFF", help_text=_("Background color for the live activity ticker (e.g. #FFFFFF or white)."))
ticker_text_color = models.CharField(_("Ticker Text Color"), max_length=20, default="#1A1A1D", help_text=_("Text color for the live activity ticker."))
@property
def maintenance_message(self):
if get_language() == "ar":
return self.maintenance_message_ar or _("Service is temporarily suspended. Please try again later.")
return self.maintenance_message_en or _("Service is temporarily suspended. Please try again later.")
@property
def privacy_policy(self):
if get_language() == 'ar':
return self.privacy_policy_ar
return self.privacy_policy_en
@property
def terms_conditions(self):
if get_language() == 'ar':
return self.terms_conditions_ar
return self.terms_conditions_en
def save(self, *args, **kwargs):
# Auto-clean whitespace from credentials
if self.whatsapp_access_token:
self.whatsapp_access_token = self.whatsapp_access_token.strip()
if self.whatsapp_business_phone_number_id:
val = self.whatsapp_business_phone_number_id.strip()
# Remove common path suffixes if user pasted full URL
for suffix in ['/api/send-message', '/api/v2/send-message']:
if val.endswith(suffix):
val = val[:-len(suffix)]
if val.endswith('/'):
val = val[:-1]
self.whatsapp_business_phone_number_id = val
if self.whatsapp_app_secret:
self.whatsapp_app_secret = self.whatsapp_app_secret.strip()
super().save(*args, **kwargs)
def __str__(self):
return self.name
class Meta:
verbose_name = _('Platform Profile')
verbose_name_plural = _('Platform Profile')
class Parcel(models.Model):
STATUS_CHOICES = (
('pending', _('Pending Pickup')),
('picked_up', _('Picked Up')),
('in_transit', _('In Transit')),
('delivered', _('Delivered')),
('cancelled', _('Cancelled')),
)
PAYMENT_STATUS_CHOICES = (
('pending', _('Pending')),
('paid', _('Paid')),
('failed', _('Failed')),
)
tracking_number = models.CharField(_('Tracking Number'), max_length=20, unique=True, blank=True)
shipper = models.ForeignKey(User, on_delete=models.CASCADE, related_name='sent_parcels', verbose_name=_('Shipper'))
carrier = models.ForeignKey(User, on_delete=models.SET_NULL, null=True, blank=True, related_name='carried_parcels', verbose_name=_('Carrier'))
parcel_type = models.ForeignKey(ParcelType, on_delete=models.SET_NULL, null=True, blank=True, verbose_name=_('Parcel Type'))
description = models.TextField(_('Description'))
weight = models.DecimalField(_('Weight (kg)'), max_digits=5, decimal_places=2, help_text=_("Weight in kg"))
price = models.DecimalField(_('Total Price (OMR)'), max_digits=10, decimal_places=3, default=Decimal('0.000'))
# Financial Breakdown
platform_fee = models.DecimalField(_('Platform Fee (OMR)'), max_digits=10, decimal_places=3, default=Decimal('0.000'))
platform_fee_percentage = models.DecimalField(_('Fee Percentage (%)'), max_digits=5, decimal_places=2, default=Decimal('0.00'))
driver_amount = models.DecimalField(_('Driver Amount (OMR)'), max_digits=10, decimal_places=3, default=Decimal('0.000'))
# Trip Info
distance_km = models.DecimalField(_('Distance (km)'), max_digits=10, decimal_places=2, default=Decimal('0.00'))
# Pickup Location
pickup_country = models.ForeignKey(Country, on_delete=models.SET_NULL, null=True, blank=True, related_name='pickup_parcels', verbose_name=_('Pickup Country'))
pickup_governate = models.ForeignKey(Governate, on_delete=models.SET_NULL, null=True, blank=True, related_name='pickup_parcels', verbose_name=_('Pickup Governate'))
pickup_city = models.ForeignKey(City, on_delete=models.SET_NULL, null=True, blank=True, related_name='pickup_parcels', verbose_name=_('Pickup City'))
pickup_address = models.CharField(_('Pickup Address'), max_length=255)
pickup_lat = models.DecimalField(_('Pickup Latitude'), max_digits=20, decimal_places=16, null=True, blank=True)
pickup_lng = models.DecimalField(_('Pickup Longitude'), max_digits=20, decimal_places=16, null=True, blank=True)
# Delivery Location
delivery_country = models.ForeignKey(Country, on_delete=models.SET_NULL, null=True, blank=True, related_name='delivery_parcels', verbose_name=_('Delivery Country'))
delivery_governate = models.ForeignKey(Governate, on_delete=models.SET_NULL, null=True, blank=True, related_name='delivery_parcels', verbose_name=_('Delivery Governate'))
delivery_city = models.ForeignKey(City, on_delete=models.SET_NULL, null=True, blank=True, related_name='delivery_city_parcels', verbose_name=_('Delivery City'))
delivery_address = models.CharField(_('Delivery Address'), max_length=255)
delivery_lat = models.DecimalField(_('Delivery Latitude'), max_digits=20, decimal_places=16, null=True, blank=True)
delivery_lng = models.DecimalField(_('Delivery Longitude'), max_digits=20, decimal_places=16, null=True, blank=True)
receiver_name = models.CharField(_('Receiver Name'), max_length=100)
receiver_phone = models.CharField(_('Receiver Phone'), max_length=20)
status = models.CharField(_('Status'), max_length=20, choices=STATUS_CHOICES, default='pending')
payment_status = models.CharField(_('Payment Status'), max_length=20, choices=PAYMENT_STATUS_CHOICES, default='pending')
thawani_session_id = models.CharField(_('Thawani Session ID'), max_length=255, blank=True, null=True)
created_at = models.DateTimeField(_('Created At'), auto_now_add=True)
updated_at = models.DateTimeField(_('Updated At'), auto_now=True)
def save(self, *args, **kwargs):
is_new = self.pk is None
# Test Mode Logic: Auto-Mark Paid
if is_new and self.payment_status == 'pending':
try:
# Use filter().first() to avoid error if table is empty
profile = PlatformProfile.objects.first()
if profile and profile.auto_mark_paid:
self.payment_status = 'paid'
except Exception:
# Fail gracefully (e.g. during migrations)
pass
if not self.tracking_number:
self.tracking_number = str(uuid.uuid4().hex[:10]).upper()
# Calculate Distance and Price if Lat/Lng provided and price is 0 (or always update to ensure accuracy)
# We only recalculate if status is pending or if it's a new object to avoid changing history for completed trips
if (is_new or self.status == 'pending') and self.pickup_lat and self.pickup_lng and self.delivery_lat and self.delivery_lng:
# Local import to avoid circular dependency
from .pricing import calculate_haversine_distance, get_pricing_breakdown
# Calculate Distance
dist = calculate_haversine_distance(
self.pickup_lat, self.pickup_lng,
self.delivery_lat, self.delivery_lng
)
self.distance_km = dist
# Calculate Price
breakdown = get_pricing_breakdown(self.distance_km, self.weight)
if not breakdown.get('error'):
self.price = breakdown['price']
self.platform_fee = breakdown['platform_fee']
self.platform_fee_percentage = breakdown['platform_fee_percentage']
self.driver_amount = breakdown['driver_amount']
super().save(*args, **kwargs)
@property
def pickup_google_maps_url(self):
if self.pickup_lat and self.pickup_lng:
return f"https://www.google.com/maps/search/?api=1&query={self.pickup_lat},{self.pickup_lng}"
return f"https://www.google.com/maps/search/?api=1&query={self.pickup_address}"
@property
def delivery_google_maps_url(self):
if self.delivery_lat and self.delivery_lng:
return f"https://www.google.com/maps/search/?api=1&query={self.delivery_lat},{self.delivery_lng}"
return f"https://www.google.com/maps/search/?api=1&query={self.delivery_address}"
def __str__(self):
return f"Parcel {self.tracking_number} - {self.status}"
class Meta:
verbose_name = _('Parcel')
verbose_name_plural = _('Parcels')
class OTPVerification(models.Model):
PURPOSE_CHOICES = (
('profile_update', _('Profile Update')),
('password_reset', _('Password Reset')),
('registration', _('Registration')),
('login', _('Login')),
)
user = models.ForeignKey(User, on_delete=models.CASCADE)
code = models.CharField(max_length=6)
purpose = models.CharField(max_length=20, choices=PURPOSE_CHOICES, default='profile_update')
created_at = models.DateTimeField(auto_now_add=True)
is_verified = models.BooleanField(default=False)
def is_valid(self):
# OTP valid for 10 minutes
return self.created_at >= timezone.now() - timezone.timedelta(minutes=10)
class Testimonial(models.Model):
name_en = models.CharField(_('Name (English)'), max_length=100)
name_ar = models.CharField(_('Name (Arabic)'), max_length=100)
role_en = models.CharField(_('Role (English)'), max_length=100)
role_ar = models.CharField(_('Role (Arabic)'), max_length=100)
content_en = models.TextField(_('Testimony (English)'))
content_ar = models.TextField(_('Testimony (Arabic)'))
image = models.ImageField(_('Image'), upload_to='testimonials/', blank=True, null=True)
is_active = models.BooleanField(_('Active'), default=True)
created_at = models.DateTimeField(auto_now_add=True)
updated_at = models.DateTimeField(auto_now=True)
@property
def name(self):
return self.name_ar if get_language() == 'ar' else self.name_en
@property
def role(self):
return self.role_ar if get_language() == 'ar' else self.role_en
@property
def content(self):
return self.content_ar if get_language() == 'ar' else self.content_en
def __str__(self):
return self.name
class Meta:
verbose_name = _('Testimonial')
verbose_name_plural = _('Testimonials')
ordering = ['-created_at']
class DriverRating(models.Model):
parcel = models.OneToOneField(Parcel, on_delete=models.CASCADE, related_name='rating', verbose_name=_('Parcel'))
driver = models.ForeignKey(User, on_delete=models.CASCADE, related_name='received_ratings', verbose_name=_('Driver'))
shipper = models.ForeignKey(User, on_delete=models.CASCADE, related_name='given_ratings', verbose_name=_('Shipper'))
rating = models.PositiveSmallIntegerField(_('Rating'), choices=[(i, str(i)) for i in range(1, 6)])
comment = models.TextField(_('Comment'), blank=True)
created_at = models.DateTimeField(_('Created At'), auto_now_add=True)
def __str__(self):
return f"Rating {self.rating} for {self.driver.username} by {self.shipper.username}"
class Meta:
verbose_name = _('Driver Rating')
verbose_name_plural = _('Driver Ratings')
class NotificationTemplate(models.Model):
KEY_CHOICES = (
('otp_registration', 'OTP Registration'),
('otp_login', 'OTP Login'),
('otp_profile_update', 'OTP Profile Update'),
('shipment_created_shipper', 'Shipment Created (Shipper)'),
('payment_success_shipper', 'Payment Success (Shipper)'),
('shipment_visible_receiver', 'Shipment Visible (Receiver)'),
('driver_pickup_shipper', 'Driver Pickup (Shipper)'),
('driver_pickup_receiver', 'Driver Pickup (Receiver)'),
('driver_pickup_driver', 'Driver Pickup (Driver/Carrier)'),
('shipment_status_update', 'Shipment Status Update'),
('admin_alert_driver_accept', 'Admin Alert: Driver Accepted'),
('contact_form_admin', 'Contact Form (Admin)'),
)
key = models.CharField(max_length=50, choices=KEY_CHOICES, unique=True)
description = models.CharField(max_length=255, help_text="Description of where this notification is used.")
available_variables = models.TextField(help_text="Comma-separated list of variables available in this template (e.g. {{ code }}, {{ name }}).", blank=True)
# Email
subject_en = models.CharField(max_length=255, blank=True, verbose_name="Email Subject (EN)")
subject_ar = models.CharField(max_length=255, blank=True, verbose_name="Email Subject (AR)")
email_body_en = models.TextField(blank=True, verbose_name="Email Body (EN)", help_text="HTML allowed.")
email_body_ar = models.TextField(blank=True, verbose_name="Email Body (AR)", help_text="HTML allowed.")
# WhatsApp
whatsapp_body_en = models.TextField(blank=True, verbose_name="WhatsApp Message (EN)")
whatsapp_body_ar = models.TextField(blank=True, verbose_name="WhatsApp Message (AR)")
def __str__(self):
return f"{self.get_key_display()}"
class Meta:
verbose_name = _('Notification Template')
verbose_name_plural = _('Notification Templates')
class DriverReport(models.Model):
STATUS_CHOICES = (
('pending', _('Pending Investigation')),
('investigating', _('Investigating')),
('resolved', _('Resolved')),
('dismissed', _('Dismissed')),
)
REASON_CHOICES = (
('unprofessional', _('Unprofessional Behavior')),
('reckless_driving', _('Reckless Driving')),
('delayed_delivery', _('Significant Delay')),
('item_damaged', _('Item Damaged')),
('item_missing', _('Item Missing')),
('other', _('Other')),
)
reporter = models.ForeignKey(User, on_delete=models.CASCADE, related_name='filed_reports', verbose_name=_('Reporter'))
driver = models.ForeignKey(User, on_delete=models.CASCADE, related_name='reports_received', verbose_name=_('Driver'))
parcel = models.ForeignKey('Parcel', on_delete=models.SET_NULL, null=True, blank=True, related_name='reports', verbose_name=_('Related Parcel'))
reason = models.CharField(_('Reason'), max_length=50, choices=REASON_CHOICES)
description = models.TextField(_('Detailed Description'))
status = models.CharField(_('Status'), max_length=20, choices=STATUS_CHOICES, default='pending')
admin_note = models.TextField(_('Admin Internal Note'), blank=True)
created_at = models.DateTimeField(_('Created At'), auto_now_add=True)
updated_at = models.DateTimeField(_('Updated At'), auto_now=True)
def __str__(self):
return f"Report against {self.driver.username} by {self.reporter.username} - {self.get_status_display()}"
class Meta:
verbose_name = _('Driver Report')
verbose_name_plural = _('Driver Reports')
ordering = ['-created_at']
class DriverRejection(models.Model):
driver = models.ForeignKey(User, on_delete=models.CASCADE, related_name='rejections', verbose_name=_('Driver'))
parcel = models.ForeignKey(Parcel, on_delete=models.CASCADE, related_name='rejections', verbose_name=_('Parcel'))
reason = models.TextField(_('Reason for Rejection'))
created_at = models.DateTimeField(_('Created At'), auto_now_add=True)
def __str__(self):
return f"Rejection by {self.driver.username} for {self.parcel.tracking_number}"
class Meta:
verbose_name = _('Driver Rejection')
verbose_name_plural = _('Driver Rejections')
ordering = ['-created_at']
class DriverWarning(models.Model):
driver = models.ForeignKey(User, on_delete=models.CASCADE, related_name='warnings', verbose_name=_('Driver'))
reason = models.TextField(_('Reason for Warning'))
created_at = models.DateTimeField(_('Created At'), auto_now_add=True)
def __str__(self):
return f"Warning for {self.driver.username} on {self.created_at.strftime('%Y-%m-%d')}"
def save(self, *args, **kwargs):
is_new = self.pk is None
super().save(*args, **kwargs)
if is_new:
# Check for auto-ban
try:
platform_profile = PlatformProfile.objects.first()
if platform_profile and platform_profile.enable_auto_ban_on_warnings:
warning_count = DriverWarning.objects.filter(driver=self.driver).count()
if warning_count >= platform_profile.max_warnings_before_ban:
profile = self.driver.profile
profile.is_banned = True
profile.ban_reason = _("Automatically banned due to exceeding warning limit (%(limit)d warnings).") % {
'limit': platform_profile.max_warnings_before_ban
}
profile.save()
except Exception:
pass
class Meta:
verbose_name = _('Driver Warning')
verbose_name_plural = _('Driver Warnings')
ordering = ['-created_at']