X-Git-Url: https://git.mdrn.pl/redakcja.git/blobdiff_plain/e01dee1eda68946acf73e43f2c54b3c730514a21..0327b33c1e7816f8a30269f8798bff5e216ccd7c:/apps/catalogue/ebook_utils.py diff --git a/apps/catalogue/ebook_utils.py b/apps/catalogue/ebook_utils.py index 5961b637..c16b2958 100644 --- a/apps/catalogue/ebook_utils.py +++ b/apps/catalogue/ebook_utils.py @@ -1,14 +1,19 @@ # -*- coding: utf-8 -*- from catalogue.models import Book -from librarian import DocProvider +from librarian import DocProvider, IOFile 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 IOFile.from_string(Book.objects.get(dc_slug=slug + ).materialize(publishable=self.publishable + ).encode('utf-8')) def serve_file(file_path, name, mime_type):