Как переименовывать изображения при загрузке?

Question

[Wiki-fan]

Модель такая:

class Picture(models.Model):
	ID = models.IntegerField(primary_key=True)
	name = models.CharField(max_length=256)
	pic_file = models.ImageField(upload_to='media/images')

Форма такая:

class PictureUploadForm(forms.ModelForm):
	name = forms.CharField(max_length=256, help_text="Enter picture name:")
	pic_file = forms.ImageField(help_text='Choose your file:')
	
	class Meta:
		model = Picture
		fields = ('name', 'pic_file',) #Ещё один вопрос мимоходом. Здесь указываются поля, содержимое которых переносится в модель автоматически, так?

Вид, на всякий случай, такой:

def add_picture(request):
	if request.method == 'POST':
		form = PictureUploadForm(request.POST, request.FILES)

		if form.is_valid():
			form.save(commit=True)
		else:
			print form.errors
	else:
		form = PictureUploadForm()

	return render(request, 'add_picture.html', {'form': form})

Картинка сохраняется с тем же именем, которое было у неё при загрузке. То есть, как у пользователя. А мне нужно, чтобы ей выдавалось имя вида < ID>.<расширение>, т. к. у меня ID является primary_key и не повторяется. Как это сделать?

Answer 1

[Yuri Shikanov]

Обычно используют параметр upload_to поля FileField от которого наследуется ImageField.
Но, т.к. вам нужно сначала сохранить инстанс, чтобы знать его ID (в случае добавления), придется использовать сигнал модели post_save в обработчике которого уже переименовывать файл и сохранять значение поля в базе данных.

Answer 2

[un1t]

Нужно сделать свой FileField или ImageField и перопределить метод generate_filename
https://github.com/django/django/blob/master/djang...
До модели можно добраться соотвественно через self.instance.

Comments

[Wiki-fan]

В модель дописал

class PictureFileStorage(FileSystemStorage):
	def get_available_name(self, name):
		#тут я переопределяю
picture_storage = FileSystemStorage(location=MEDIA_ROOT+"/images", base_url=MEDIA_URL+"images/")

В location и base_url я должен по-хорошему использовать os.path.join(MEDIA_ROOT, "images") и os.path.join(MEDIA_URL,"images")+'/' , да?

Но как передать ID в функцию get_available_name?

[un1t]

Хороший вопрос. Я обновил ответ.

[Wiki-fan]

class MyImageField(models.ImageField):
	def generate_filename(self, instance, filename):
		return os.path.join(self.get_directory_name(), models.IntegerField.value_to_string(instance.ID))

TypeError at /add_picture/
unbound method value_to_string() must be called with IntegerField instance as first argument (got NoneType instance instead)
Мне кажется, что MyImageField создаётся здесь раньше, чем ID. Нужно, похоже, использовать post_save, с которым я ещё не разобрался.

[un1t]

С post_save подозреваю тоже костыльно будет. Можешь во вьюхе попробовать сначала сохранить без картинки, а потом ее прикрепить.

[Wiki-fan]

Пытаюсь сделать, как здесь написано:

if form.is_valid():
	instance = Picture(name=request.POST['name'])
	instance.pic_file = request.FILES['pic_file']
	instance.save()

Теперь у меня ещё один вопрос. Раньше ID django инициализировал сам, а теперь мне надо делать это самостоятельно, и чтобы это происходило раньше, чем добавление файла. Как мне узнать, на какой цифре он остановился?

[un1t]

if form.is_valid():
  instance = Picture(name=request.POST['name'])
  instance.save()
  print instance.id
  instance.pic_file = request.FILES['pic_file']
  instance.save()

[Wiki-fan]

Действительно, так всё работает. Благодарю. Ещё я зря писал ID = models.IntegerField(primary_key=True)
Надо было ID = models.AutoField(primary_key=True). Но django автоматически добавляет id = models.AutoField(primary_key=True), если не находит явно указанный primary_key, так что проще вообще этот мой ID убрать, и оставить джанговский