src/catalogue/import_utils.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 librarian import DocProvider
   6
   7
   8 class ORMDocProvider(DocProvider):
   9     """Used for getting books' children."""
  10
  11     def __init__(self, book):
  12         self.book = book
  13
  14     def by_slug(self, slug):
  15         if slug == self.book.slug:
  16             return open(self.book.xml_file.path)
  17         else:
  18             return type(self.book).objects.get(slug=slug).xml_file