from django.db import models
from json import dumps, decoder

# Create your models here.



class Category(models.Model):
    name_obj = models.CharField(max_length=255)


class Brand(models.Model):
    name_obj = models.CharField(max_length=255)

    def __str__(self):
        return f'{self.name_obj} (id:f{self.id})'


class Color(models.Model):
    name_obj = models.CharField(max_length=255)


class Url(models.Model):
    name_obj = models.CharField(max_length=255)


class Product(models.Model):
    '''
    Результаты которые будут получены от скрапи многие в
    типа list и нужно их переформатировать прежде чем
    записывать в базу.

    Варианта было два. И я реализовал оба.
    1) Запись в базу списка как строка.
    2) Где объекты повторяются связь
    many-to-many
    '''
    category = models.ManyToManyField(Category)
    brand = models.ManyToManyField(Brand)
    brand_model = models.CharField(max_length=255)
    color = models.ManyToManyField(Color)
    size = models.CharField(max_length=255) #json<->string
    imgs = models.TextField()
    description = models.TextField() #func set_desc
    link_to_product = models.ForeignKey(Url, on_delete=models.CASCADE)

    def list_to_string(self, value):
        return dumps(value)

    def string_to_list(self, value):
        Decoder = decoder.JSONDecoder()
        return Decoder.decode(value)

    def set_description(self, array):
        return ''.join(array)