apps/catalogue/management/commands/__init__.py

   1 # -*- coding: utf-8 -*-
   2 #
   3 # This file is part of FNP-Redakcja, licensed under GNU Affero GPLv3 or later.
   4 # Copyright © Fundacja Nowoczesna Polska. See NOTICE for more information.
   5 #
   6 import sys
   7 from optparse import make_option
   8 from django.contrib.auth.models import User
   9 from django.core.management.base import BaseCommand
  10 from catalogue.models import Book
  11
  12
  13 class XmlUpdaterCommand(BaseCommand):
  14     """Base class for creating massive XML-updating commands.
  15
  16     In a subclass, provide an XmlUpdater class in the `updater' attribute.
  17     """
  18     option_list = BaseCommand.option_list + (
  19         make_option(
  20             '-q', '--quiet', action='store_false', dest='verbose',
  21             default=True, help='Less output'),
  22         make_option(
  23             '-d', '--dry-run', action='store_true', dest='dry_run',
  24             default=False, help="Don't actually touch anything"),
  25         make_option(
  26             '-u', '--username', dest='username', metavar='USER',
  27             help='Assign commits to this user (required, preferably yourself).'),
  28     )
  29     args = "[slug]..."
  30
  31     updater = NotImplemented
  32
  33     def handle(self, *args, **options):
  34         verbose = options.get('verbose')
  35         dry_run = options.get('dry_run')
  36         username = options.get('username')
  37
  38         if username:
  39             user = User.objects.get(username=username)
  40         else:
  41             print 'Please provide a username.'
  42             sys.exit(1)
  43
  44         books = Book.objects.filter(slug__in=args) if args else None
  45
  46         updater = self.updater()
  47         updater.run(user, verbose=verbose, dry_run=dry_run, books=books)
  48         updater.print_results()