ptitlutins-backend/voyage/models.py
2025-09-14 04:54:02 +02:00

96 lines
3.9 KiB
Python

from django.db import models
from django.contrib.auth.models import User
import uuid
class Voyage(models.Model):
id = models.UUIDField(primary_key=True, default=uuid.uuid4, unique=True, editable=False)
title = models.CharField(max_length=255)
description = models.TextField(blank=True)
creator = models.ForeignKey(User, on_delete=models.CASCADE, related_name="voyages", blank=True, null=True)
collaborators = models.ManyToManyField(User, related_name="voyages_collaborators", through="VoyagePermission", blank=True)
created_at = models.DateTimeField(auto_now_add=True)
def __str__(self):
return self.title
class VoyagePermission(models.Model):
view = models.BooleanField(default=False)
change = models.BooleanField(default=False)
user = models.ForeignKey(User, on_delete=models.CASCADE, related_name="voyage_permissions", blank=True, null=True)
voyage = models.ForeignKey(Voyage, on_delete=models.CASCADE, related_name="collaborator_permissions", blank=True, null=True)
def update_permission(self, permission_data):
self.view = permission_data.get('view', self.view)
self.change = permission_data.get('change', self.change)
self.save()
class VoyageLink(models.Model):
id = models.UUIDField(primary_key=True, default=uuid.uuid4, unique=True, editable=False)
voyage = models.ForeignKey(Voyage, on_delete=models.CASCADE, related_name="links")
permission = models.OneToOneField(VoyagePermission, on_delete=models.CASCADE, related_name="link")
primary = models.BooleanField(default=False)
def __str__(self):
return f"Link for {self.voyage.title}"
class Location(models.Model):
voyage = models.ForeignKey(Voyage, on_delete=models.CASCADE, related_name="locations")
name = models.CharField(max_length=255, blank=True)
postal_address = models.CharField(max_length=255, blank=True)
latitude = models.FloatField(null=True, blank=True)
longitude = models.FloatField(null=True, blank=True)
osm_id = models.CharField(max_length=100, blank=True) # optional: OpenStreetMap reference
def __str__(self):
return self.name or self.postal_address or "Unnamed location"
class Event(models.Model):
voyage = models.ForeignKey(Voyage, on_delete=models.CASCADE, related_name="events")
title = models.CharField(max_length=255)
start_date = models.DateTimeField()
end_date = models.DateTimeField()
location = models.ForeignKey(Location, on_delete=models.SET_NULL, null=True, blank=True)
note = models.TextField(blank=True)
parent = models.ForeignKey("self", on_delete=models.CASCADE, null=True, blank=True, related_name="sub_events")
participants = models.ManyToManyField(User, related_name="events_participated", blank=True)
def __str__(self):
return self.title
class Need(models.Model):
item = models.CharField(max_length=255)
proposed_by = models.ForeignKey(User, on_delete=models.SET_NULL, null=True, blank=True)
voyage = models.ForeignKey(Voyage, on_delete=models.CASCADE, related_name="needs")
event = models.ForeignKey(Event, on_delete=models.CASCADE, related_name="needs", null=True, blank=True)
def __str__(self):
return self.item
class TransportMode(models.TextChoices):
CAR = 'car', 'Car'
TRAIN = 'train', 'Train'
PLANE = 'plane', 'Plane'
BIKE = 'bike', 'Bike'
WALK = 'walk', 'Walk'
BUS = 'bus', 'Bus'
TRAM = 'tram', 'Tram'
METRO = 'metro', 'Metro'
TAXI = 'taxi', 'Taxi'
BOAT = 'boat', 'Boat'
OTHER = 'other', 'Other'
class Trip(models.Model):
voyage = models.ForeignKey(Voyage, on_delete=models.CASCADE, related_name="trips")
event_start = models.ForeignKey(Event, on_delete=models.CASCADE, related_name="trip_start")
event_end = models.ForeignKey(Event, on_delete=models.CASCADE, related_name="trip_end")
mode = models.CharField(max_length=100, blank=True, choices=TransportMode.choices)
itinerary = models.JSONField(blank=True, null=True)