X-Git-Url: https://git.mdrn.pl/redakcja.git/blobdiff_plain/e01dee1eda68946acf73e43f2c54b3c730514a21..2dadb147794c8bc2da8c062e3452ddaf992fa7f0:/apps/catalogue/ebook_utils.py diff --git a/apps/catalogue/ebook_utils.py b/apps/catalogue/ebook_utils.py index 5961b637..1fcf8d33 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,12 @@ 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)) def serve_file(file_path, name, mime_type):