attachments; previews (not real, just working)
[redakcja.git] / apps / catalogue / ebook_utils.py
1 # -*- coding: utf-8 -*-
2 from catalogue.models import Book
3 from librarian import DocProvider, IOFile
4 from django.http import HttpResponse
5
6
7 class RedakcjaDocProvider(DocProvider):
8     """Used for getting books' children."""
9
10     def __init__(self, publishable):
11         self.publishable = publishable
12
13     def by_slug(self, slug):
14         return IOFile.from_string(Book.objects.get(dc_slug=slug
15                     ).materialize(publishable=self.publishable
16                     ).encode('utf-8'))
17
18
19 def serve_file(file_path, name, mime_type):
20     def read_chunks(f, size=8192):
21         chunk = f.read(size)
22         while chunk:
23             yield chunk
24             chunk = f.read(size)
25
26     response = HttpResponse(mimetype=mime_type)
27     response['Content-Disposition'] = 'attachment; filename=%s' % name
28     with open(file_path) as f:
29         for chunk in read_chunks(f):
30             response.write(chunk)
31     return response