class Image(Model):
    path = f'emp/'
    owner = ForeignKey("elite.Employee", on_delete=SET_NULL, null=True, blank=True, related_name='images_owner')

    info = CharField(max_length=20, verbose_name='Описание', blank=True, null=True)
    type = CharField(max_length=6, choices=[
        ('tg', 'Телеграм'),
        ('site', 'Сайт'),
        ('avatar', 'Аватар')], verbose_name='Тип фотографии', blank=True, null=True)

    def get_file_path(self, filename):
        if self.path:
            ext = filename.split('.')[-1]
            filename = f"{datetime.now().strftime('%Y-%m-%d_%H-%M-%S')}.{ext}"

            if self.type == 'tg':
                return os.path.join(self.path + f"{self.owner.employer.id}/images/telegram_main/", filename)

            if self.type == 'site':
                return os.path.join(self.path + f"{self.owner.employer.id}/images/site/", filename)

            if self.type == 'avatar':
                return os.path.join(self.path + f"{self.owner.employer.id}/images/avatar/", filename)
        else:
            return None

    image = ImageField(upload_to=get_file_path, null=True, blank=True)

    class Meta:
        verbose_name = 'Фотографии'
        verbose_name_plural = 'Фотографии'
        unique_together = ['owner', 'type']