Example-TCG-Site/store/models.py

from django.db import models
from django.conf import settings

class Game(models.Model):
    name = models.CharField(max_length=100, unique=True)
    slug = models.SlugField(unique=True)

    def __str__(self):
        return self.name

class Set(models.Model):
    game = models.ForeignKey(Game, on_delete=models.CASCADE, related_name='sets')
    name = models.CharField(max_length=200)
    code = models.CharField(max_length=20, blank=True)
    release_date = models.DateField(null=True, blank=True)

    def __str__(self):
        return f"{self.game.name} - {self.name}"

class Card(models.Model):
    set = models.ForeignKey(Set, on_delete=models.CASCADE, related_name='cards')
    name = models.CharField(max_length=200)
    rarity = models.CharField(max_length=50, blank=True)
    image_url = models.URLField(max_length=500, blank=True)
    scryfall_id = models.CharField(max_length=100, blank=True, null=True)
    tcgplayer_id = models.CharField(max_length=100, blank=True, null=True)
    collector_number = models.CharField(max_length=50, blank=True)

    def __str__(self):
        return self.name

class CardListing(models.Model):
    CONDITION_CHOICES = (
        ('NM', 'Near Mint'),
        ('LP', 'Lightly Played'),
        ('MP', 'Moderately Played'),
        ('HP', 'Heavily Played'),
    )

    card = models.ForeignKey(Card, on_delete=models.CASCADE, related_name='listings')
    condition = models.CharField(max_length=2, choices=CONDITION_CHOICES, default='NM')
    price = models.DecimalField(max_digits=10, decimal_places=2)
    quantity = models.PositiveIntegerField(default=0)
    market_price = models.DecimalField(max_digits=10, decimal_places=2, null=True, blank=True)
    is_foil = models.BooleanField(default=False)

    def __str__(self):
        return f"{self.card.name} ({self.condition}) - ${self.price}"

class PackListing(models.Model):
    game = models.ForeignKey(Game, on_delete=models.CASCADE, related_name='pack_listings')
    name = models.CharField(max_length=200)
    price = models.DecimalField(max_digits=10, decimal_places=2)
    image_url = models.URLField(max_length=500, blank=True)

    def __str__(self):
        return f"{self.name} - ${self.price}"

class VirtualPack(models.Model):
    STATUS_CHOICES = (
        ('sealed', 'Sealed'),
        ('opened', 'Opened'),
    )
    listing = models.ForeignKey(PackListing, on_delete=models.CASCADE, related_name='packs')
    cards = models.ManyToManyField(Card, related_name='packs')
    status = models.CharField(max_length=10, choices=STATUS_CHOICES, default='sealed')
    owner = models.ForeignKey(settings.AUTH_USER_MODEL, on_delete=models.SET_NULL, null=True, blank=True, related_name='packs')
    created_at = models.DateTimeField(auto_now_add=True)

    def __str__(self):
        return f"{self.listing.name} ({self.get_status_display()})"

class Order(models.Model):
    STATUS_CHOICES = (
        ('pending', 'Pending'),
        ('paid', 'Paid'),
        ('shipped', 'Shipped'),
        ('cancelled', 'Cancelled'),
    )

    user = models.ForeignKey(settings.AUTH_USER_MODEL, on_delete=models.CASCADE, related_name='orders')
    status = models.CharField(max_length=20, choices=STATUS_CHOICES, default='pending')
    created_at = models.DateTimeField(auto_now_add=True)
    updated_at = models.DateTimeField(auto_now=True)
    total_price = models.DecimalField(max_digits=12, decimal_places=2, default=0)
    stripe_payment_intent = models.CharField(max_length=100, blank=True)
    shipping_address = models.TextField(blank=True)
    insurance_purchased = models.BooleanField(default=False)
    proxy_service = models.BooleanField(default=False)

    def __str__(self):
        return f"Order #{self.id} - {self.user.username}"

class OrderItem(models.Model):
    order = models.ForeignKey(Order, on_delete=models.CASCADE, related_name='items')
    listing = models.ForeignKey(CardListing, on_delete=models.SET_NULL, null=True, blank=True)
    pack_listing = models.ForeignKey(PackListing, on_delete=models.SET_NULL, null=True, blank=True)
    price_at_purchase = models.DecimalField(max_digits=10, decimal_places=2)
    quantity = models.PositiveIntegerField(default=1)

    def __str__(self):
        if self.pack_listing:
            return f"{self.quantity}x {self.pack_listing.name}"
        return f"{self.quantity}x {self.listing.card.name if self.listing else 'Deleted Listing'}"

class Cart(models.Model):
    user = models.OneToOneField(settings.AUTH_USER_MODEL, on_delete=models.CASCADE, related_name='cart', null=True, blank=True)
    created_at = models.DateTimeField(auto_now_add=True)
    insurance = models.BooleanField(default=False)
    
    @property
    def total_price(self):
        base_total = sum(item.total_price for item in self.items.all())
        if self.insurance:
            return base_total + 5  # Flat fee
        return base_total

class CartItem(models.Model):
    cart = models.ForeignKey(Cart, on_delete=models.CASCADE, related_name='items')
    listing = models.ForeignKey(CardListing, on_delete=models.CASCADE, null=True, blank=True)
    pack_listing = models.ForeignKey(PackListing, on_delete=models.CASCADE, null=True, blank=True)
    quantity = models.PositiveIntegerField(default=1)

    @property
    def total_price(self):
        if self.listing:
            return self.listing.price * self.quantity
        if self.pack_listing:
            return self.pack_listing.price * self.quantity
        return 0

class Bounty(models.Model):
    card = models.ForeignKey(Card, on_delete=models.CASCADE, related_name='bounties')
    target_price = models.DecimalField(max_digits=10, decimal_places=2)
    quantity_wanted = models.PositiveIntegerField(default=1)
    is_active = models.BooleanField(default=True)
    created_at = models.DateTimeField(auto_now_add=True)

    def __str__(self):
        return f"WANTED: {self.card.name} @ ${self.target_price}"

class VaultItem(models.Model):
    user = models.ForeignKey(settings.AUTH_USER_MODEL, on_delete=models.CASCADE, related_name='vault_items')
    card = models.ForeignKey(Card, on_delete=models.CASCADE, related_name='vault_items')
    quantity = models.PositiveIntegerField(default=1)
    added_at = models.DateTimeField(auto_now_add=True)

    class Meta:
        unique_together = ('user', 'card')

    def __str__(self):
        return f"{self.user.username}'s {self.card.name} ({self.quantity})"