scripts/dump-book.py

   1 #!/usr/bin/env python
   2 # -*- coding: utf-8 -*-
   3 # This file is part of Wolnelektury, licensed under GNU Affero GPLv3 or later.
   4 # Copyright © Fundacja Nowoczesna Polska. See NOTICE for more information.
   5 #
   6 import sys
   7 sys.path.append('.')
   8 sys.path.append('./apps')
   9 sys.path.append('./lib')
  10
  11 from django.core.management import setup_environ
  12 from redakcja import settings
  13
  14 setup_environ(settings)
  15
  16 from catalogue.models import Book, Chunk
  17 import re
  18
  19 if len(sys.argv) != 3:
  20     print "dump-book slug-part filename"
  21     sys.exit(-1)
  22
  23 term = sys.argv[1]
  24
  25 books = Book.objects.filter(slug__contains=term)
  26 if len(books) > 1:
  27     for b in range(len(books)):
  28         print "%d) %s" % (b+1, books[b].slug)
  29     print "Which one? "
  30     ch = int(raw_input())
  31     book = books[ch-1]
  32 else:
  33     book = books[0]
  34
  35 open(sys.argv[2], "w").write(book.materialize().encode('utf-8'))
  36