diff --git a/photologue_custom/management/commands/duplicate.py b/photologue_custom/management/commands/duplicate.py index 4719375..c9fd70a 100644 --- a/photologue_custom/management/commands/duplicate.py +++ b/photologue_custom/management/commands/duplicate.py @@ -30,13 +30,22 @@ class Command(BaseCommand): print('error') # Find duplicates in all galleries for gallery in galleries: - duplicate = find_duplicate(gallery) + duplicates = find_duplicate(gallery) self.stdout.write('Gallery {} :'.format(gallery.slug)) - print(duplicate) - for (original, copies) in duplicate: + for (original, copies) in duplicates: + self.stdout.write( + ' {} has following duplicate(s) :'.format(original.slug)) for copy in copies: - self.stdout.write('{} is duplicate of {}'.format( - copy.slug, original.slug)) + self.stdout.write(' {}'.format(copy.slug)) + # Delete them if --delete + if options['delete']: + self.stdout.write( + ' Deleting duplicate in {} :'.format(gallery.slug)) + for (_original, copies) in duplicates: + for copy in copies: + self.stdout.write( + ' Deleting {}...'.format(copy.slug)) + copy.delete() def find_duplicate(gallery):