NODEDC_TASKMANAGER/plane-src/apps/api/plane/bgtasks/file_asset_task.py

45 lines
1.5 KiB
Python

# Copyright (c) 2023-present Plane Software, Inc. and contributors
# SPDX-License-Identifier: AGPL-3.0-only
# See the LICENSE file for details.
# Python imports
import os
from datetime import timedelta
# Django imports
from django.conf import settings
from django.utils import timezone
from django.db.models import Q
# Third party imports
from celery import shared_task
# Module imports
from plane.db.models import FileAsset
from plane.settings.storage import S3Storage
from plane.utils.file_dedup import hard_delete_file_asset, release_file_asset_blob
@shared_task
def delete_unuploaded_file_asset():
"""This task deletes unuploaded file assets older than a certain number of days."""
stale_assets = FileAsset.objects.filter(
Q(created_at__lt=timezone.now() - timedelta(days=int(os.environ.get("UNUPLOADED_ASSET_DELETE_DAYS", "7"))))
& Q(is_uploaded=False)
)
for asset in stale_assets.iterator():
release_file_asset_blob(asset, delete_untracked_object=True)
asset.delete(soft=False)
@shared_task
def delete_expired_file_asset():
"""Hard delete soft-deleted file assets after the restore retention window."""
days = int(os.environ.get("FILE_ASSET_HARD_DELETE_AFTER_DAYS", settings.HARD_DELETE_AFTER_DAYS))
cutoff = timezone.now() - timedelta(days=days)
storage = S3Storage()
expired_assets = FileAsset.all_objects.filter(deleted_at__lt=cutoff)
for asset in expired_assets.iterator():
hard_delete_file_asset(asset, storage=storage)