Code layout change.

[wolnelektury.git] / src / catalogue / import_utils.py

diff --git a/src/catalogue/import_utils.py b/src/catalogue/import_utils.py
new file mode 100644 (file)
index 0000000..ca0c6ef
--- /dev/null
+++ b/src/catalogue/import_utils.py
@@ -0,0 +1,17 @@
+# -*- coding: utf-8 -*-
+# This file is part of Wolnelektury, licensed under GNU Affero GPLv3 or later.
+# Copyright © Fundacja Nowoczesna Polska. See NOTICE for more information.
+#
+from librarian import DocProvider
+
+class ORMDocProvider(DocProvider):
+    """Used for getting books' children."""
+
+    def __init__(self, book):
+        self.book = book
+
+    def by_slug(self, slug):
+        if slug == self.book.slug:
+            return open(self.book.xml_file.path)
+        else:
+            return type(self.book).objects.get(slug=slug).xml_file