37 lines
1.5 KiB
Python
37 lines
1.5 KiB
Python
from pathlib import Path
|
|
import os
|
|
|
|
from django.core.management.base import BaseCommand
|
|
from photologue.models import Gallery
|
|
from django.conf import settings
|
|
|
|
|
|
class Command(BaseCommand):
|
|
help = 'Rename uploaded media file to match gallery and photo names'
|
|
|
|
def add_arguments(self, parser):
|
|
parser.add_argument('--apply', action='store_true')
|
|
|
|
def handle(self, *args, **options):
|
|
media_dir = Path(settings.MEDIA_ROOT)
|
|
for gallery in Gallery.objects.all():
|
|
# Create gallery directory
|
|
gallery_year = str(gallery.extended.date_start.year)
|
|
gallery_dir = Path('photos') / gallery_year / gallery.slug
|
|
gallery_path = media_dir / gallery_dir
|
|
if not gallery_path.exists():
|
|
self.stdout.write(f"Creating {gallery_dir}")
|
|
if options["apply"]:
|
|
gallery_path.mkdir(parents=True)
|
|
|
|
# Move photos in gallery folder
|
|
for photo in gallery.photos.all():
|
|
photo_name = str(gallery_dir / photo.image.name.split("/")[-1])
|
|
if photo.image.name == photo_name:
|
|
continue
|
|
self.stdout.write(f" Moving {photo.image.name} -> {photo_name}")
|
|
if options["apply"]:
|
|
if not (media_dir / photo_name).exists():
|
|
os.rename(photo.image.path, media_dir / photo_name)
|
|
photo.image.name = photo_name
|
|
photo.save()
|