source

upload_가 포함된 Django FileField는 런타임에 결정됩니다.

bestscript 2023. 7. 19. 22:48

upload_가 포함된 Django FileField는 런타임에 결정됩니다.

모든 사용자의 파일이 MEDIA_ROOT로 이동하는 대신 사용자 Joe가 파일을 업로드하면 MEDIA_ROOT/joe로 이동하도록 업로드를 설정하려고 합니다.문제는 이것을 모델에 정의하는 방법을 모른다는 것입니다.현재 상태는 다음과 같습니다.

class Content(models.Model):
    name = models.CharField(max_length=200)
    user = models.ForeignKey(User)
    file = models.FileField(upload_to='.')

그래서 제가 원하는 것은 upload_to의 '.' 대신 사용자의 이름을 사용하는 것입니다.

Django 1.0부터는 업로드_to를 처리할 수 있는 자신의 함수를 정의할 수 있는 것으로 알고 있습니다만, 그 함수 역시 사용자가 누구인지 전혀 모르기 때문에 저는 좀 당황스럽습니다.

도와주셔서 감사합니다!

설명서를 읽어 본 적이 있을 것이므로 다음은 이해하기 쉬운 예입니다.

def content_file_name(instance, filename):
    return '/'.join(['content', instance.user.username, filename])

class Content(models.Model):
    name = models.CharField(max_length=200)
    user = models.ForeignKey(User)
    file = models.FileField(upload_to=content_file_name)

보시다시피, 지정된 파일 이름을 사용할 필요도 없습니다. 원한다면 업로드 파일 이름을 호출할 수도 있습니다.

이것은 정말 도움이 되었습니다.좀 더 간략하게 설명하기 위해, 제 경우 람다를 사용하기로 결정했습니다.

file = models.FileField(
    upload_to=lambda instance, filename: '/'.join(['mymodel', str(instance.pk), filename]),
)

'instance' 개체의 pk 값 사용에 대한 참고 사항입니다.설명서에 따르면:

대부분의 경우 이 개체는 아직 데이터베이스에 저장되지 않았으므로 기본 AutoField를 사용하는 경우 기본 키 필드에 대한 값이 아직 없을 수 있습니다.

따라서 pk 사용의 유효성은 특정 모형이 정의되는 방식에 따라 달라집니다.

마이그레이션에 문제가 있는 경우 사용해야 합니다.@deconstructible장식가

import datetime
import os
import unicodedata

from django.core.files.storage import default_storage
from django.utils.deconstruct import deconstructible
from django.utils.encoding import force_text, force_str


@deconstructible
class UploadToPath(object):
    def __init__(self, upload_to):
        self.upload_to = upload_to

    def __call__(self, instance, filename):
        return self.generate_filename(filename)

    def get_directory_name(self):
        return os.path.normpath(force_text(datetime.datetime.now().strftime(force_str(self.upload_to))))

    def get_filename(self, filename):
        filename = default_storage.get_valid_name(os.path.basename(filename))
        filename = force_text(filename)
        filename = unicodedata.normalize('NFKD', filename).encode('ascii', 'ignore').decode('ascii')
        return os.path.normpath(filename)

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

용도:

class MyModel(models.Model):
    file = models.FileField(upload_to=UploadToPath('files/%Y/%m/%d'), max_length=255)

런타임에 업로드 경로를 변경하고 싶었으나 이러한 요구에 적합한 솔루션이 없었습니다.

이게 제가 한 일입니다.

class Content(models.Model):
    name = models.CharField(max_length=200)
    user = models.ForeignKey(User)
    file = models.FileField(upload_to=DynamicUploadPath.get_file_path)


class ContentSerializer(serializers.ModelSerializer):
    class Meta:
        model = Content
        fields = '__all__'


class UploadDir(models.TextChoices):
    PRODUCT = 'PRD', _('Product')
    USER_PROFILE = 'UP', _('User Profile')


class DynamicUploadPath:
    dir: UploadDir = None

    @classmethod
    def get_file_path(cls, instance, filename):
        return str(cls.dir.name.lower() + '/' + filename)


def set_DynamicUploadPath(dir: UploadDir):
    DynamicUploadPath.dir = dir


class UploadFile(APIView):
    parser_classes = (MultiPartParser, FormParser)

    def post(self, request):
        # file save path: MEDIA_ROOT/product/filename
        set_DynamicUploadPath(UploadDir.PRODUCT)

        # file save path: MEDIA_ROOT/user_profile/filename
        # set_DynamicUploadPath(UploadDir.USER_PROFILE)

        serializer = ContentSerializer(data=request.data)
        serializer.is_valid(raise_exception=True)
        serializer.save()

        return Response(serializer.data, status=status.HTTP_200_OK)

사용자 인스턴스가 있는 경우 생성할 빠른 설정이 있어야 합니다.

<model-slug>/<username>-<first_name>-<last_name>/filename-random.png

예:/medias/content/ft0004-john-doe/filename-lkl9237.png


def upload_directory_name(instance, filename):

    user = getattr(instance, 'user', None)
    if user:
        name = f"{user.username}-{user.get_full_name().replace(' ', '-')}"
    else:
        name=str(instance)
    model_name = instance._meta.verbose_name.replace(' ', '-')
    return str(os.path.pathsep).join([model_name, name, filename])


class Content(models.Model):
    name = models.CharField(max_length=200)
    user = models.ForeignKey(User)
    file = models.FileField(upload_to=upload_directory_name)


[@SmileyChris의 수정 버전]

언급URL : https://stackoverflow.com/questions/1190697/django-filefield-with-upload-to-determined-at-runtime