bf29a0b21b
- Added and properties to model. - Exposed these URLs in . - Updated driver dashboard to show 'Navigate to Pickup' and 'Navigate to Delivery' buttons for active shipments.
421 lines
20 KiB
Python
421 lines
20 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 Profile(models.Model):
|
|
ROLE_CHOICES = (
|
|
('shipper', _('Shipper')),
|
|
('car_owner', _('Car Owner')),
|
|
)
|
|
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
|
|
is_approved = models.BooleanField(_('Approved'), default=False, help_text=_("Designates whether this user is approved to use the platform (mainly for drivers)."))
|
|
|
|
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."))
|
|
|
|
@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'))
|
|
|
|
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') |