Rename catalogue to documents.

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

diff --git a/src/documents/management/commands/mark_final.py b/src/documents/management/commands/mark_final.py
new file mode 100644 (file)
index 0000000..b2c316c
--- /dev/null
+++ b/src/documents/management/commands/mark_final.py
@@ -0,0 +1,42 @@
+# 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 django.core.management import BaseCommand
+
+from documents.models import Book, Chunk
+
+
+class Command(BaseCommand):
+    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.')
+            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)
+            for chunk in book.chunk_set.all():
+                src = chunk.head.materialize()
+                chunk.commit(
+                    text=src,
+                    author=user,
+                    description=u'Ostateczna akceptacja merytoryczna przez kierownika literackiego.',
+                    tags=[Chunk.tag_model.objects.get(slug='editor-proofreading')],
+                    publishable=True
+                )
+            print('committed %s' % book.slug)