From 2df1bc4d50fc6051c7b61cd903d24cd7b0a42b1d Mon Sep 17 00:00:00 2001 From: aeltheos Date: Sun, 14 Nov 2021 01:06:03 +0100 Subject: [PATCH] Added deletion of duplicate if --delete is specified --- .../management/commands/duplicate.py | 19 ++++++++++++++----- 1 file changed, 14 insertions(+), 5 deletions(-) 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):