6ae2b9a2f3aab13739b812005b1c702470670f43
[wolnelektury.git] / src / catalogue / management / commands / checkintegrity.py
1 # -*- coding: utf-8 -*-
2 # This file is part of Wolnelektury, licensed under GNU Affero GPLv3 or later.
3 # Copyright © Fundacja Nowoczesna Polska. See NOTICE for more information.
4 #
5 from optparse import make_option
6 from django.core.management.base import BaseCommand
7
8 from catalogue.models import Book
9
10
11 class Command(BaseCommand):
12     option_list = BaseCommand.option_list + (
13         make_option('-q', '--quiet', action='store_false', dest='verbose', default=True,
14             help='Suppress output'),
15         make_option('-d', '--dry-run', action='store_true', dest='dry_run', default=False,
16             help="Just check for problems, don't fix them"),
17     )
18     help = 'Checks integrity of catalogue data.'
19
20     def handle(self, **options):
21         from django.db import transaction
22
23         verbose = options['verbose']
24
25         with transaction.commit_on_success():
26             for book in Book.objects.all().iterator():
27                 try:
28                     info = book.wldocument().book_info
29                 except:
30                     if verbose:
31                         print "ERROR! Bad XML for book:", book.slug
32                         print "To resolve: republish."
33                         print
34                 else:
35                     should_be = [p.slug for p in info.parts]
36                     is_now = [p.slug for p in book.children.all().order_by('parent_number')]
37                     if should_be != is_now:
38                         if verbose:
39                             print "ERROR! Wrong children for book:", book.slug
40                             print "Is:       ", is_now
41                             print "Should be:", should_be
42                             print "To resolve: republish parent book."
43                             print
44
45                 # Check for ancestry.
46                 parents = []
47                 parent = book.parent
48                 while parent:
49                     parents.append(parent)
50                     parent = parent.parent
51                 ancestors = list(book.ancestor.all())
52                 if set(ancestors) != set(parents):
53                     if options['verbose']:
54                         print "Wrong ancestry for book:", book
55                         print "Is:       ", ", ".join(ancestors)
56                         print "Should be:", ", ".join(parents)
57                     if not options['dry_run']:
58                         book.repopulate_ancestors()
59                         if options['verbose']:
60                             print "Fixed."
61                     if options['verbose']:
62                         print
63
64                 # TODO: check metadata tags, reset counters