from django.db import models
from django.utils import timezone
from sorl.thumbnail import ImageField
from users.models import User


class WorkSell(models.Model):
    name = models.CharField(max_length=255)
    description = models.TextField(blank=True)
    img = ImageField(upload_to='worksell/worksell')
    price = models.DecimalField(max_digits=10, decimal_places=0, default=0)
    contractor = models.ForeignKey(User, related_name='work_sell')
    created = models.DateTimeField(default=timezone.now)

    def __str__(self):
        return self.name

    def is_author_for_work(self):
        pass

    class Meta:
        ordering = ['-created']
        verbose_name = 'Готовая работа'
        verbose_name_plural = 'Готовые работы'


class WorkSellPhoto(models.Model):
    img = ImageField(upload_to='worksell/worksell')
    worksell = models.ForeignKey(WorkSell, related_name='photos')

    def __str__(self):
        return self.worksell.name

    class Meta:
        verbose_name = 'Изображение Готовая работа'
        verbose_name_plural = 'Изображения Готовые работы'