fixes
[redakcja.git] / src / documents / ebook_utils.py
1 # This file is part of FNP-Redakcja, licensed under GNU Affero GPLv3 or later.
2 # Copyright © Fundacja Nowoczesna Polska. See NOTICE for more information.
3 #
4 from io import BytesIO
5 from .models import Book
6 from librarian import DocProvider
7 from django.http import HttpResponse
8
9
10 class RedakcjaDocProvider(DocProvider):
11     """Used for getting books' children."""
12
13     def __init__(self, publishable):
14         self.publishable = publishable
15
16     def by_slug(self, slug):
17         return BytesIO(Book.objects.get(catalogue_book_id=slug
18                     ).materialize(publishable=self.publishable
19                     ).encode('utf-8'))
20
21
22 def serve_file(file_path, name, mime_type):
23     def read_chunks(f, size=8192):
24         chunk = f.read(size)
25         while chunk:
26             yield chunk
27             chunk = f.read(size)
28
29     response = HttpResponse(content_type=mime_type)
30     response['Content-Disposition'] = 'attachment; filename=%s' % name
31     with open(file_path, 'rb') as f:
32         for chunk in read_chunks(f):
33             response.write(chunk)
34     return response