X-Git-Url: https://git.mdrn.pl/wolnelektury.git/blobdiff_plain/46705d393ae76a1d8e0eb2dc8bf121269f6adb67..9f5432c9c13b7b44d9f28d98807ff41311c6cce8:/apps/catalogue/utils.py diff --git a/apps/catalogue/utils.py b/apps/catalogue/utils.py index 02e5b6d93..368f96d20 100644 --- a/apps/catalogue/utils.py +++ b/apps/catalogue/utils.py @@ -49,13 +49,13 @@ class ExistingFile(UploadedFile): class BookImportDocProvider(DocProvider): - """ used for joined EPUBs """ + """Used for joined EPUB and PDF files.""" def __init__(self, book): self.book = book - + def by_slug(self, slug): if slug == self.book.slug: return self.book.xml_file else: - return Book.objects.get(slug=slug).xml_file + return type(self.book).objects.get(slug=slug).xml_file