mirror of
https://github.com/pacnpal/thrillwiki_django_no_react.git
synced 2025-12-27 09:27:04 -05:00
62 lines
2.1 KiB
Python
62 lines
2.1 KiB
Python
from django.db import models
|
|
from django.conf import settings
|
|
from django.contrib.contenttypes.fields import GenericForeignKey
|
|
from django.contrib.contenttypes.models import ContentType
|
|
from apps.core.history import TrackedModel
|
|
from apps.core.choices import RichChoiceField
|
|
import pghistory
|
|
|
|
@pghistory.track()
|
|
class UserList(TrackedModel):
|
|
user = models.ForeignKey(
|
|
settings.AUTH_USER_MODEL,
|
|
on_delete=models.CASCADE,
|
|
related_name="user_lists",
|
|
help_text="User who created this list",
|
|
)
|
|
title = models.CharField(max_length=100, help_text="Title of the list")
|
|
category = RichChoiceField(
|
|
choice_group="top_list_categories",
|
|
domain="accounts",
|
|
max_length=2,
|
|
help_text="Category of items in this list",
|
|
)
|
|
description = models.TextField(blank=True, help_text="Description of the list")
|
|
is_public = models.BooleanField(default=True, help_text="Whether this list is visible to others")
|
|
|
|
class Meta(TrackedModel.Meta):
|
|
verbose_name = "User List"
|
|
verbose_name_plural = "User Lists"
|
|
ordering = ["-updated_at"]
|
|
|
|
def __str__(self):
|
|
return f"{self.user.username}'s {self.category} List: {self.title}"
|
|
|
|
|
|
@pghistory.track()
|
|
class ListItem(TrackedModel):
|
|
user_list = models.ForeignKey(
|
|
UserList,
|
|
on_delete=models.CASCADE,
|
|
related_name="items",
|
|
help_text="List this item belongs to",
|
|
)
|
|
content_type = models.ForeignKey(
|
|
ContentType,
|
|
on_delete=models.CASCADE,
|
|
help_text="Type of item (park, ride, etc.)",
|
|
)
|
|
object_id = models.PositiveIntegerField(help_text="ID of the item")
|
|
content_object = GenericForeignKey("content_type", "object_id")
|
|
rank = models.PositiveIntegerField(help_text="Position in the list")
|
|
notes = models.TextField(blank=True, help_text="User's notes about this item")
|
|
|
|
class Meta(TrackedModel.Meta):
|
|
verbose_name = "List Item"
|
|
verbose_name_plural = "List Items"
|
|
ordering = ["rank"]
|
|
unique_together = [["user_list", "rank"]]
|
|
|
|
def __str__(self):
|
|
return f"#{self.rank} in {self.user_list.title}"
|