Added arbitrary files and links to boxes

This commit is contained in:
2026-01-01 14:50:07 +01:00
parent acde0cb2f8
commit a4f9274da4
8 changed files with 673 additions and 15 deletions

View 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')