Как не повторять код функции upload_to?

Question

[NerVik]

Здравствуйте, есть модель:

class Image(models.Model):
    image = models.ImageField(upload_to=image, blank=True)
    image_xs = models.ImageField(upload_to=image_xs, blank=True)
    image_sm = models.ImageField(upload_to=image_sm, blank=True)
    image_md = models.ImageField(upload_to=image_md, blank=True)
    am = models.ForeignKey(AnotherModel)

в upload_to функции:

def image(instance, filename):
    return 'image/{0}/{1}'.format(instance.am_id, filename)


def image_xs(instance, filename):
    return 'image/{0}/xs/{1}'.format(instance.am_id, filename)


def image_sm(instance, filename):
    return 'image/{0}/sm/{1}'.format(instance.am_id, filename)


def image_md(instance, filename):
    return 'image/{0}/md/{1}'.format(instance.am_id, filename)

как видно, функции почти одинаковые, отличаются буквально парой символов.

Вопрос: как 4 функции превратить в одну универсальную?

Answer 1

[Oscar Django]

def image(instance, filename, image_size_type):
  if image_size_type:
    size = '/{}'.format(image_size_type)
  else:
    size = ''
  return 'image/{0}{1}/{2}'.format(instance.am_id, size, filename)

class MyImageField(ImageField):
  def __init__(self, image_size_type='', **kwargs):
    self.image_size_type = image_size_type
    super().__init__(**kwargs)

  def generate_filename(self, instance, filename):
    if callable(self.upload_to):
      directory_name, filename = os.path.split(self.upload_to(instance, filename, self.image_size_type))
      filename = self.storage.get_valid_name(filename)
      return os.path.normpath(os.path.join(directory_name, filename))

    return os.path.join(self.get_directory_name(), self.get_filename(filename))

class Image(models.Model):
    image = MyImageField(upload_to=image, blank=True)
    image_xs = MyImageField(upload_to=image, blank=True, image_size_type='xs')
    image_sm = MyImageField(upload_to=image, blank=True, image_size_type='sm')
    image_md = MyImageField(upload_to=image, blank=True, image_size_type='md')
    am = models.ForeignKey(AnotherModel)

Comments

[NerVik]

о, переопределить ImageField я не додумался, спасибо. пробовать буду. Только непонятно, а где используется функция generate_filename ?

[Oscar Django]

В save

Answer 2

[sim3x]

Использовать плагин для ресайза картинок https://djangopackages.org/grids/g/thumbnails/
Использовать os.path.join для path