GlitchTip/files/tasks.py

from datetime import timedelta

from celery import shared_task
from django.conf import settings
from django.utils.timezone import now

from organizations_ext.models import Organization

from .assemble import assemble_artifacts
from .models import FileBlob


@shared_task
def assemble_artifacts_task(org_id, version, checksum, chunks, **kwargs):
    """
    Creates release files from an uploaded artifact bundle.
    """
    organization = Organization.objects.get(pk=org_id)
    assemble_artifacts(organization, version, checksum, chunks)


@shared_task
def cleanup_old_files():
    """
    Delete files in both the database and media storage

    Deletion only occurs when the creation date of the following are older than max file life days
    - FileBlob
    - Any related File objects
    - Any related events attached to the project, release, release file

    This operation has minor risk of deleting a file that is still desired,
    if it hasn't been used for a long time and have no recent event data
    """
    days_ago = now() - timedelta(days=settings.GLITCHTIP_MAX_FILE_LIFE_DAYS)
    for file_blob in (
        FileBlob.objects.filter(created__lt=days_ago)
        .exclude(file__created__gte=days_ago)
        .exclude(file__releasefile__release__projects__issue__created__gte=days_ago)
    ):
        file_blob.blob.delete()  # Delete actual file
        file_blob.delete()