Skip to content

services

django_spire.file.services

logger = logging.getLogger(__name__) module-attribute

COPY_BATCH_SIZE_MAX = 200 module-attribute

copy_files_to_instance

Source code in django_spire/file/services.py 
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
def copy_files_to_instance(
    source_files: models.QuerySet,
    target: models.Model,
) -> list[File]:
    if target.pk is None:
        message = 'Cannot copy files to an unsaved model instance.'
        raise ValueError(message)

    if not source_files.exists():
        return []

    source_count = source_files.count()

    if source_count > COPY_BATCH_SIZE_MAX:
        message = (
            f'Cannot copy more than {COPY_BATCH_SIZE_MAX} '
            f'files at once ({source_count} requested).'
        )
        raise ValueError(message)

    target_content_type = ContentType.objects.get_for_model(target)

    path_builder = FilePathBuilder(
        base_folder=settings.BASE_FOLDER_NAME,
        app_name=target._meta.app_label,
    )

    copies = []

    for source_file in source_files:
        target_file = File(
            content_type=target_content_type,
            object_id=target.pk,
            name=source_file.name,
            size=source_file.size,
            type=source_file.type,
            related_field=source_file.related_field,
        )

        path = path_builder.build(
            source_file.name or 'unnamed',
            source_file.type or 'bin',
            source_file.related_field,
        )

        source_file.file.open('rb')

        try:
            target_file.file.save(path, source_file.file, save=False)
        finally:
            source_file.file.close()

        copies.append(target_file)

    return File.objects.bulk_create(copies)