from django.db import models

from user.models import User


class Brand(models.Model):
    name = models.CharField(max_length=20)

    def __str__(self):
        return self.name


class Product(models.Model):
    name = models.CharField(max_length=50)
    brand = models.ForeignKey(
        Brand, on_delete=models.CASCADE, related_name="products"
    )
    release_date = models.DateField(blank=True, null=True)

    class Meta:
        ordering = ["-release_date"]

    def __str__(self):
        return self.name


class Post(models.Model):
    STATUS_CHOICES = (
        ("s", "selling"),
        ("r", "reserved"),
        ("d", "done"),
    )
    product = models.ForeignKey(
        Product, on_delete=models.CASCADE, related_name="posts"
    )
    price = models.IntegerField()
    text = models.TextField()
    author = models.ForeignKey(User, on_delete=models.SET_NULL, null=True)
    status = models.CharField(
        max_length=1, choices=STATUS_CHOICES, default="s"
    )
    written_at = models.DateTimeField(auto_now_add=True)

    class Meta:
        ordering = ["-written_at"]

    def __str__(self):
        return f"{self.product} : {self.author} : {self.written_at}"


class Image(models.Model):
    post = models.ForeignKey(
        Post, on_delete=models.CASCADE, related_name="images"
    )
    image = models.ImageField(upload_to="images", blank=True, null=True)

    class Meta:
        ordering = ["pk"]

    def __str__(self):
        return f"{self.post}"