Added arbitrary files and links to boxes
This commit is contained in:
79
boxes/management/commands/clean_orphaned_files.py
Normal file
79
boxes/management/commands/clean_orphaned_files.py
Normal file
@@ -0,0 +1,79 @@
|
||||
import os
|
||||
from django.conf import settings
|
||||
from django.core.management.base import BaseCommand
|
||||
from boxes.models import ThingFile
|
||||
|
||||
|
||||
class Command(BaseCommand):
|
||||
help = 'Clean up orphaned files from deleted things'
|
||||
|
||||
def add_arguments(self, parser):
|
||||
parser.add_argument(
|
||||
'--dry-run',
|
||||
action='store_true',
|
||||
dest='dry_run',
|
||||
help='Show what would be deleted without actually deleting',
|
||||
)
|
||||
|
||||
def handle(self, *args, **options):
|
||||
dry_run = options.get('dry_run', False)
|
||||
|
||||
if dry_run:
|
||||
self.stdout.write(self.style.WARNING('DRY RUN - No files will be deleted'))
|
||||
|
||||
self.stdout.write('Finding orphaned files...')
|
||||
|
||||
media_root = settings.MEDIA_ROOT
|
||||
things_files_root = os.path.join(media_root, 'things', 'files')
|
||||
|
||||
if not os.path.exists(things_files_root):
|
||||
self.stdout.write(self.style.WARNING('No things/files directory found'))
|
||||
return
|
||||
|
||||
valid_paths = set()
|
||||
for thing_file in ThingFile.objects.all():
|
||||
if thing_file.file:
|
||||
file_path = thing_file.file.path
|
||||
if os.path.exists(file_path):
|
||||
valid_paths.add(os.path.relpath(file_path, things_files_root))
|
||||
|
||||
self.stdout.write(f'Found {len(valid_paths)} valid files in database')
|
||||
|
||||
deleted_count = 0
|
||||
empty_dirs_removed = 0
|
||||
|
||||
for root, dirs, files in os.walk(things_files_root, topdown=False):
|
||||
for filename in files:
|
||||
file_path = os.path.join(root, filename)
|
||||
relative_path = os.path.relpath(file_path, things_files_root)
|
||||
|
||||
if relative_path not in valid_paths:
|
||||
deleted_count += 1
|
||||
if dry_run:
|
||||
self.stdout.write(f'Would delete: {file_path}')
|
||||
else:
|
||||
try:
|
||||
os.remove(file_path)
|
||||
self.stdout.write(f'Deleted: {file_path}')
|
||||
except OSError as e:
|
||||
self.stdout.write(self.style.ERROR(f'Failed to delete {file_path}: {e}'))
|
||||
|
||||
for dirname in dirs:
|
||||
dir_path = os.path.join(root, dirname)
|
||||
if not os.listdir(dir_path):
|
||||
if dry_run:
|
||||
self.stdout.write(f'Would remove empty directory: {dir_path}')
|
||||
else:
|
||||
try:
|
||||
os.rmdir(dir_path)
|
||||
self.stdout.write(f'Removed empty directory: {dir_path}')
|
||||
empty_dirs_removed += 1
|
||||
except OSError as e:
|
||||
self.stdout.write(self.style.ERROR(f'Failed to remove {dir_path}: {e}'))
|
||||
|
||||
if dry_run:
|
||||
self.stdout.write(self.style.WARNING(f'\nDry run complete. Would delete {deleted_count} files'))
|
||||
self.stdout.write(f'Would remove {empty_dirs_removed} empty directories')
|
||||
else:
|
||||
self.stdout.write(self.style.SUCCESS(f'\nCleanup complete! Deleted {deleted_count} orphaned files'))
|
||||
self.stdout.write(f'Removed {empty_dirs_removed} empty directories')
|
||||
Reference in New Issue
Block a user