chore: Update Django server configuration and models for Adventure and Trip
This commit is contained in:
@@ -2,6 +2,13 @@ from django.db import models
|
||||
|
||||
from django.contrib.auth import get_user_model
|
||||
from django.contrib.postgres.fields import ArrayField
|
||||
from django.forms import ValidationError
|
||||
|
||||
ADVENTURE_TYPES = [
|
||||
('visited', 'Visited'),
|
||||
('planned', 'Planned'),
|
||||
('featured', 'Featured')
|
||||
]
|
||||
|
||||
|
||||
# Assuming you have a default user ID you want to use
|
||||
@@ -14,7 +21,7 @@ class Adventure(models.Model):
|
||||
id = models.AutoField(primary_key=True)
|
||||
user_id = models.ForeignKey(
|
||||
User, on_delete=models.CASCADE, default=default_user_id)
|
||||
type = models.CharField(max_length=100)
|
||||
type = models.CharField(max_length=100, choices=ADVENTURE_TYPES)
|
||||
name = models.CharField(max_length=200)
|
||||
location = models.CharField(max_length=200, blank=True, null=True)
|
||||
activity_types = ArrayField(models.CharField(
|
||||
@@ -24,10 +31,36 @@ class Adventure(models.Model):
|
||||
link = models.URLField(blank=True, null=True)
|
||||
image = models.ImageField(null=True, blank=True, upload_to='images/')
|
||||
date = models.DateField(blank=True, null=True)
|
||||
trip_id = models.IntegerField(blank=True, null=True)
|
||||
is_public = models.BooleanField(default=False)
|
||||
longitude = models.DecimalField(max_digits=9, decimal_places=6, null=True, blank=True)
|
||||
latitude = models.DecimalField(max_digits=9, decimal_places=6, null=True, blank=True)
|
||||
trip = models.ForeignKey('Trip', on_delete=models.CASCADE, blank=True, null=True)
|
||||
|
||||
def clean(self):
|
||||
if self.trip:
|
||||
if self.is_public and not self.trip.is_public:
|
||||
raise ValidationError('Public adventures must be associated with a public trip')
|
||||
if self.trip.is_public and not self.is_public:
|
||||
raise ValidationError('Adventures associated with a public trip must be public')
|
||||
if self.user_id != self.trip.user_id:
|
||||
raise ValidationError('Adventures must be associated with trips owned by the same user')
|
||||
elif self.is_public:
|
||||
raise ValidationError('Public adventures must be associated with a trip')
|
||||
|
||||
class Trip(models.Model):
|
||||
id = models.AutoField(primary_key=True)
|
||||
user_id = models.ForeignKey(
|
||||
User, on_delete=models.CASCADE, default=default_user_id)
|
||||
name = models.CharField(max_length=200)
|
||||
type = models.CharField(max_length=100, choices=ADVENTURE_TYPES)
|
||||
location = models.CharField(max_length=200, blank=True, null=True)
|
||||
date = models.DateField(blank=True, null=True)
|
||||
is_public = models.BooleanField(default=False)
|
||||
|
||||
def clean(self):
|
||||
if self.is_public:
|
||||
if self.adventures.filter(is_public=False).exists():
|
||||
raise ValidationError('Public trips cannot have private adventures')
|
||||
|
||||
def __str__(self):
|
||||
return self.name
|
||||
|
||||
Reference in New Issue
Block a user