Minor upgrades.

[redakcja.git] / src / catalogue / management / commands / mark_final.py

diff --git a/src/catalogue/management/commands/mark_final.py b/src/catalogue/management/commands/mark_final.py
index cdfaab9..11d63a3 100644 (file)
--- a/src/catalogue/management/commands/mark_final.py
+++ b/src/catalogue/management/commands/mark_final.py
@@ -1,43 +1,35 @@
-# -*- coding: utf-8 -*-
-#
 # This file is part of FNP-Redakcja, licensed under GNU Affero GPLv3 or later.
 # Copyright © Fundacja Nowoczesna Polska. See NOTICE for more information.
 #
 import sys
 from django.contrib.auth.models import User
-from optparse import make_option
-
 from django.core.management import BaseCommand
 
 from catalogue.models import Book, Chunk
 
 
 class Command(BaseCommand):
-    option_list = BaseCommand.option_list + (
-        # make_option('-q', '--quiet', action='store_false', dest='verbose',
-        #     default=True, help='Less output'),
-        # make_option('-d', '--dry-run', action='store_true', dest='dry_run',
-        #     default=False, help="Don't actually touch anything"),
-        make_option(
-            '-u', '--username', dest='username', metavar='USER',
-            help='Assign commits to this user (required).'),
-    )
     args = 'slug_file'
 
+    def add_arguments(self, parser):
+        self.add_argument(
+            '-u', '--username', dest='username', metavar='USER',
+            help='Assign commits to this user (required).')
+
     def handle(self, slug_file, **options):
         username = options.get('username')
 
         if username:
             user = User.objects.get(username=username)
         else:
-            print 'Please provide a username.'
+            print('Please provide a username.')
             sys.exit(1)
 
         slugs = [line.strip() for line in open(slug_file)]
         books = Book.objects.filter(slug__in=slugs)
 
         for book in books:
-            print 'processing %s' % book.slug
+            print('processing %s' % book.slug)
             for chunk in book.chunk_set.all():
                 src = chunk.head.materialize()
                 chunk.commit(
@@ -47,4 +39,4 @@ class Command(BaseCommand):
                     tags=[Chunk.tag_model.objects.get(slug='editor-proofreading')],
                     publishable=True
                 )
-            print 'committed %s' % book.slug
+            print('committed %s' % book.slug)