Make management command compatible with latest Django and Python 3

chazapis · Jun 25, 2021 · 7cbacca · 7cbacca
1 parent 2d49a8c
commit 7cbacca
Showing 1 changed file with 11 additions and 9 deletions.
diff --git a/chunked_upload/management/commands/delete_expired_uploads.py b/chunked_upload/management/commands/delete_expired_uploads.py
@@ -18,13 +18,15 @@ class Command(BaseCommand):
 
     help = 'Deletes chunked uploads that have already expired.'
 
-    option_list = BaseCommand.option_list + (
-        make_option('--interactive',
-                    action='store_true',
-                    dest='interactive',
-                    default=False,
-                    help='Prompt confirmation before each deletion.'),
-    )
+    def add_arguments(self, parser):
+        super().add_arguments(parser)
+        parser.add_argument(
+            '--interactive',
+            action='store_true',
+            dest='interactive',
+            default=False,
+            help='Prompt confirmation before each deletion.',
+        )
 
     def handle(self, *args, **options):
         interactive = options.get('interactive')
@@ -36,9 +38,9 @@ def handle(self, *args, **options):
         for chunked_upload in qs:
             if interactive:
                 prompt = prompt_msg.format(obj=chunked_upload) + u' (y/n): '
-                answer = raw_input(prompt).lower()
+                answer = input(prompt).lower()
                 while answer not in ('y', 'n'):
-                    answer = raw_input(prompt).lower()
+                    answer = input(prompt).lower()
                 if answer == 'n':
                     continue