mirror of
https://github.com/seanmorley15/AdventureLog.git
synced 2025-07-21 22:09:36 +02:00
60 lines
1.6 KiB
Python
60 lines
1.6 KiB
Python
|
from django.db import models
|
||
|
|
from django.contrib.auth import get_user_model
|
||
|
|
|
||
|
|
|
||
|
|
User = get_user_model()
|
||
|
|
|
||
|
|
default_user_id = 1 # Replace with an actual user ID
|
||
|
|
|
||
|
|
class Country(models.Model):
|
||
|
|
AFRICA = 'AF'
|
||
|
|
ANTARCTICA = 'AN'
|
||
|
|
ASIA = 'AS'
|
||
|
|
EUROPE = 'EU'
|
||
|
|
NORTH_AMERICA = 'NA'
|
||
|
|
OCEANIA = 'OC'
|
||
|
|
SOUTH_AMERICA = 'SA'
|
||
|
|
|
||
|
|
CONTINENT_CHOICES = [
|
||
|
|
(AFRICA, 'Africa'),
|
||
|
|
(ANTARCTICA, 'Antarctica'),
|
||
|
|
(ASIA, 'Asia'),
|
||
|
|
(EUROPE, 'Europe'),
|
||
|
|
(NORTH_AMERICA, 'North America'),
|
||
|
|
(OCEANIA, 'Oceania'),
|
||
|
|
(SOUTH_AMERICA, 'South America'),
|
||
|
|
]
|
||
|
|
|
||
|
|
id = models.AutoField(primary_key=True)
|
||
|
|
name = models.CharField(max_length=100)
|
||
|
|
country_code = models.CharField(max_length=2)
|
||
|
|
continent = models.CharField(
|
||
|
|
max_length=2,
|
||
|
|
choices=CONTINENT_CHOICES,
|
||
|
|
default=AFRICA
|
||
|
|
)
|
||
|
|
|
||
|
|
class Meta:
|
||
|
|
verbose_name = "Country"
|
||
|
|
verbose_name_plural = "Countries"
|
||
|
|
|
||
|
|
def __str__(self):
|
||
|
|
return self.name
|
||
|
|
|
||
|
|
class Region(models.Model):
|
||
|
|
id = models.CharField(primary_key=True)
|
||
|
|
name = models.CharField(max_length=100)
|
||
|
|
country = models.ForeignKey(Country, on_delete=models.CASCADE)
|
||
|
|
|
||
|
|
def __str__(self):
|
||
|
|
return self.name
|
||
|
|
|
||
|
|
class VisitedRegion(models.Model):
|
||
|
|
id = models.AutoField(primary_key=True)
|
||
|
|
user_id = models.ForeignKey(
|
||
|
|
User, on_delete=models.CASCADE, default=default_user_id)
|
||
|
|
region = models.ForeignKey(Region, on_delete=models.CASCADE)
|
||
|
|
|
||
|
|
def __str__(self):
|
||
|
|
return f'{self.region.name} ({self.region.country.country_code}) visited by: {self.user_id.username}'
|