X-Git-Url: https://git.mdrn.pl/redakcja.git/blobdiff_plain/e01dee1eda68946acf73e43f2c54b3c730514a21..28228fc438edfd3275ce50a90bd3f87180344036:/apps/catalogue/ebook_utils.py diff --git a/apps/catalogue/ebook_utils.py b/apps/catalogue/ebook_utils.py index 5961b637..d9c99227 100644 --- a/apps/catalogue/ebook_utils.py +++ b/apps/catalogue/ebook_utils.py @@ -1,4 +1,5 @@ # -*- coding: utf-8 -*- +from StringIO import StringIO from catalogue.models import Book from librarian import DocProvider from django.http import HttpResponse @@ -7,8 +8,13 @@ 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 Book.objects.get(dc_slug=slug).xml_file + return StringIO(Book.objects.get(dc_slug=slug + ).materialize(publishable=self.publishable + ).encode('utf-8')) def serve_file(file_path, name, mime_type):