kopia lustrzana https://github.com/OpenDroneMap/WebODM
63 wiersze
2.5 KiB
Python
63 wiersze
2.5 KiB
Python
|
import os
|
||
|
|
import shutil
|
||
|
|
from django.core.management.base import BaseCommand
|
||
|
|
from django.core.management import call_command
|
||
|
|
from app.models import Project
|
||
|
|
from webodm import settings
|
||
|
|
|
||
|
|
class Command(BaseCommand):
|
||
|
|
requires_system_checks = []
|
||
|
|
|
||
|
|
def add_arguments(self, parser):
|
||
|
|
parser.add_argument("action", type=str, choices=['projects'])
|
||
|
|
parser.add_argument("--dry-run", action='store_true', required=False, help="Don't actually delete folders")
|
||
|
|
parser.add_argument("--only-empty", action='store_true', required=False, help="Only delete folders if there's no data")
|
||
|
|
|
||
|
|
super(Command, self).add_arguments(parser)
|
||
|
|
|
||
|
|
def handle(self, **options):
|
||
|
|
if options.get('action') == 'projects':
|
||
|
|
projects_root = os.path.join(settings.MEDIA_ROOT, "project")
|
||
|
|
|
||
|
|
print("Enumerating orphaned project folders")
|
||
|
|
db_project_ids = {p.id for p in Project.objects.all()}
|
||
|
|
fs_project_ids = {int(i) for i in os.listdir(projects_root) if i.isnumeric()}
|
||
|
|
|
||
|
|
print(f"Filesystem: {len(fs_project_ids)}")
|
||
|
|
print(f"Database: {len(db_project_ids)}")
|
||
|
|
|
||
|
|
orphaned_ids = fs_project_ids - db_project_ids
|
||
|
|
|
||
|
|
def rm(folder):
|
||
|
|
if not options.get('dry_run'):
|
||
|
|
try:
|
||
|
|
shutil.rmtree(folder)
|
||
|
|
print(f"R {folder}")
|
||
|
|
except Exception as e:
|
||
|
|
print(f"Error while removing {folder}: {str(e)}")
|
||
|
|
else:
|
||
|
|
print(f"R {folder}")
|
||
|
|
|
||
|
|
for orphaned_id in orphaned_ids:
|
||
|
|
orphaned_folder = os.path.join(settings.MEDIA_ROOT, "project", str(orphaned_id))
|
||
|
|
|
||
|
|
if options.get('only_empty'):
|
||
|
|
entries = os.listdir(orphaned_folder)
|
||
|
|
if len(entries) == 0:
|
||
|
|
rm(orphaned_folder)
|
||
|
|
elif len(entries) == 1 and entries[0] == "task":
|
||
|
|
task_entries = os.listdir(os.path.join(orphaned_folder, "task"))
|
||
|
|
if len(task_entries) == 0:
|
||
|
|
rm(orphaned_folder)
|
||
|
|
else:
|
||
|
|
print(f"WARNING: orphaned folder with data inside: {orphaned_folder}")
|
||
|
|
else:
|
||
|
|
print(f"WARNING: orphaned folder with data inside: {orphaned_folder}")
|
||
|
|
else:
|
||
|
|
rm(orphaned_folder)
|
||
|
|
|
||
|
|
else:
|
||
|
|
print("Invalid action")
|
||
|
|
|
||
|
|
|