Rename catalogue to documents.

[redakcja.git] / src / documents / ebook_utils.py

diff --git a/src/documents/ebook_utils.py b/src/documents/ebook_utils.py
new file mode 100644 (file)
index 0000000..e75ed77
--- /dev/null
+++ b/src/documents/ebook_utils.py
@@ -0,0 +1,34 @@
+# This file is part of FNP-Redakcja, licensed under GNU Affero GPLv3 or later.
+# Copyright © Fundacja Nowoczesna Polska. See NOTICE for more information.
+#
+from io import BytesIO
+from .models import Book
+from librarian import DocProvider
+from django.http import HttpResponse
+
+
+class RedakcjaDocProvider(DocProvider):
+    """Used for getting books' children."""
+
+    def __init__(self, publishable):
+        self.publishable = publishable
+
+    def by_slug(self, slug):
+        return BytesIO(Book.objects.get(dc_slug=slug
+                    ).materialize(publishable=self.publishable
+                    ).encode('utf-8'))
+
+
+def serve_file(file_path, name, mime_type):
+    def read_chunks(f, size=8192):
+        chunk = f.read(size)
+        while chunk:
+            yield chunk
+            chunk = f.read(size)
+
+    response = HttpResponse(content_type=mime_type)
+    response['Content-Disposition'] = 'attachment; filename=%s' % name
+    with open(file_path, 'rb') as f:
+        for chunk in read_chunks(f):
+            response.write(chunk)
+    return response