ca0c6efe5c3460dd96699df8982fe6f104751de9
[wolnelektury.git] / apps / 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 class ORMDocProvider(DocProvider):
8     """Used for getting books' children."""
9
10     def __init__(self, book):
11         self.book = book
12
13     def by_slug(self, slug):
14         if slug == self.book.slug:
15             return open(self.book.xml_file.path)
16         else:
17             return type(self.book).objects.get(slug=slug).xml_file