def item_enclosures(self, item): mime = mime_detector.fmt(item['format']) enclosure = [ opdsEnclosure(reverse("opds_catalog:download", kwargs={"book_id": item['id'], "zip_flag": 0}), mime, "http://opds-spec.org/acquisition/open-access"), ] if not item['format'] in settings.NOZIP_FORMATS: mimezip = Mimetype.FB2_ZIP if mime==Mimetype.FB2 else "%s+zip"%mime enclosure += [opdsEnclosure(reverse("opds_catalog:download", kwargs={"book_id": item['id'], "zip_flag": 1}), mimezip, "http://opds-spec.org/acquisition/open-access")] enclosure += [opdsEnclosure(reverse("opds_catalog:cover", kwargs={"book_id": item['id']}), "image/jpeg", "http://opds-spec.org/image"), opdsEnclosure(reverse("opds_catalog:thumb", kwargs={"book_id": item['id']}), "image/jpeg", "http://opds-spec.org/thumbnail"), ] if (config.SOPDS_FB2TOEPUB != "") and (item['format'] == 'fb2'): enclosure += [ opdsEnclosure(reverse("opds_catalog:convert", kwargs={"book_id": item['id'], "convert_type": "epub"}), Mimetype.EPUB, "http://opds-spec.org/acquisition/open-access")] if (config.SOPDS_FB2TOMOBI != "") and (item['format'] == 'fb2'): enclosure += [ opdsEnclosure(reverse("opds_catalog:convert", kwargs={"book_id": item['id'], "convert_type": "mobi"}), Mimetype.MOBI, "http://opds-spec.org/acquisition/open-access")] return enclosure
def item_enclosures(self, item): mime = mime_detector.fmt(item['format']) enclosure = [ opdsEnclosure(reverse("opds_catalog:download", kwargs={"book_id": item['id'], "zip_flag": 0}), mime, "http://opds-spec.org/acquisition/open-access"), ] if not item['format'] in settings.NOZIP_FORMATS: mimezip = Mimetype.FB2_ZIP if mime==Mimetype.FB2 else "%s+zip"%mime enclosure += [opdsEnclosure(reverse("opds_catalog:download", kwargs={"book_id": item['id'], "zip_flag": 1}), mimezip, "http://opds-spec.org/acquisition/open-access")] enclosure += [opdsEnclosure(reverse("opds_catalog:cover", kwargs={"book_id": item['id']}), "image/jpeg", "http://opds-spec.org/image"), opdsEnclosure(reverse("opds_catalog:thumb", kwargs={"book_id": item['id']}), "image/jpeg", "http://opds-spec.org/thumbnail"), ] if (config.SOPDS_FB2TOEPUB != "") and (item['format'] == 'fb2'): enclosure += [ opdsEnclosure(reverse("opds_catalog:convert", kwargs={"book_id": item['id'], "convert_type": "epub"}), Mimetype.EPUB, "http://opds-spec.org/acquisition/open-access")] if (config.SOPDS_FB2TOMOBI != "") and (item['format'] == 'fb2'): enclosure += [ opdsEnclosure(reverse("opds_catalog:convert", kwargs={"book_id": item['id'], "convert_type": "mobi"}), Mimetype.MOBI, "http://opds-spec.org/acquisition/open-access")] return enclosure
def ConvertFB2(request, book_id, convert_type): """ Выдача файла книги после конвертации в EPUB или mobi """ book = Book.objects.get(id=book_id) if book.format != 'fb2': raise Http404 if config.SOPDS_AUTH and request.user.is_authenticated: bookshelf.objects.get_or_create(user=request.user, book=book) full_path = os.path.join(config.SOPDS_ROOT_LIB, book.path) if book.cat_type == opdsdb.CAT_INP: # Убираем из пути INPX и INP файл inp_path, zip_name = os.path.split(full_path) inpx_path, inp_name = os.path.split(inp_path) path, inpx_name = os.path.split(inpx_path) full_path = os.path.join(path, zip_name) if config.SOPDS_TITLE_AS_FILENAME: transname = utils.translit(book.title + '.' + book.format) else: transname = utils.translit(book.filename) transname = utils.to_ascii(transname) (n, e) = os.path.splitext(transname) dlfilename = "%s.%s" % (n, convert_type) if convert_type == 'epub': converter_path = config.SOPDS_FB2TOEPUB elif convert_type == 'mobi': converter_path = config.SOPDS_FB2TOMOBI content_type = mime_detector.fmt(convert_type) if book.cat_type == opdsdb.CAT_NORMAL: tmp_fb2_path = None file_path = os.path.join(full_path, book.filename) elif book.cat_type in [opdsdb.CAT_ZIP, opdsdb.CAT_INP]: try: fz = codecs.open(full_path, "rb") except FileNotFoundError: raise Http404 z = zipfile.ZipFile(fz, 'r', allowZip64=True) z.extract(book.filename, config.SOPDS_TEMP_DIR) tmp_fb2_path = os.path.join(config.SOPDS_TEMP_DIR, book.filename) file_path = tmp_fb2_path tmp_conv_path = os.path.join(config.SOPDS_TEMP_DIR, dlfilename) popen_args = ("\"%s\" \"%s\" \"%s\"" % (converter_path, file_path, tmp_conv_path)) proc = subprocess.Popen(popen_args, shell=True, stdout=subprocess.PIPE) #proc = subprocess.Popen((converter_path.encode('utf8'),file_path.encode('utf8'),tmp_conv_path.encode('utf8')), shell=True, stdout=subprocess.PIPE) out = proc.stdout.readlines() if os.path.isfile(tmp_conv_path): fo = codecs.open(tmp_conv_path, "rb") s = fo.read() # HTTP Header response = HttpResponse() response["Content-Type"] = '%s; name="%s"' % (content_type, dlfilename) response["Content-Disposition"] = 'attachment; filename="%s"' % ( dlfilename) response["Content-Transfer-Encoding"] = 'binary' response["Content-Length"] = str(len(s)) response.write(s) fo.close() else: raise Http404 try: if tmp_fb2_path: os.remove(tmp_fb2_path) except: pass try: os.remove(tmp_conv_path) except: pass return response
def Download(request, book_id, zip_flag): """ Загрузка файла книги """ book = Book.objects.get(id=book_id) if config.SOPDS_AUTH and request.user.is_authenticated: bookshelf.objects.get_or_create(user=request.user, book=book) full_path = os.path.join(config.SOPDS_ROOT_LIB, book.path) if book.cat_type == opdsdb.CAT_INP: # Убираем из пути INPX и INP файл inp_path, zip_name = os.path.split(full_path) inpx_path, inp_name = os.path.split(inp_path) path, inpx_name = os.path.split(inpx_path) full_path = os.path.join(path, zip_name) if config.SOPDS_TITLE_AS_FILENAME: transname = utils.translit(book.title + '.' + book.format) else: transname = utils.translit(book.filename) transname = utils.to_ascii(transname) if zip_flag == '1': dlfilename = transname + '.zip' content_type = Mimetype.FB2_ZIP if book.format == 'fb2' else Mimetype.ZIP else: dlfilename = transname content_type = mime_detector.fmt(book.format) response = HttpResponse() response["Content-Type"] = '%s; name="%s"' % (content_type, dlfilename) response["Content-Disposition"] = 'attachment; filename="%s"' % ( dlfilename) response["Content-Transfer-Encoding"] = 'binary' z = None fz = None s = None book_size = book.filesize if book.cat_type == opdsdb.CAT_NORMAL: file_path = os.path.join(full_path, book.filename) book_size = os.path.getsize(file_path) try: fo = codecs.open(file_path, "rb") except FileNotFoundError: raise Http404 s = fo.read() elif book.cat_type in [opdsdb.CAT_ZIP, opdsdb.CAT_INP]: try: fz = codecs.open(full_path, "rb") except FileNotFoundError: raise Http404 z = zipfile.ZipFile(fz, 'r', allowZip64=True) book_size = z.getinfo(book.filename).file_size fo = z.open(book.filename) s = fo.read() if zip_flag == '1': dio = io.BytesIO() zo = zipfile.ZipFile(dio, 'w', zipfile.ZIP_DEFLATED) zo.writestr(transname, s) zo.close() buf = dio.getvalue() response["Content-Length"] = str(len(buf)) response.write(buf) else: response["Content-Length"] = str(book_size) response.write(s) fo.close() if z: z.close() if fz: fz.close() return response
def ConvertFB2(request, book_id, convert_type): """ Выдача файла книги после конвертации в EPUB или mobi """ book = Book.objects.get(id=book_id) if book.format!='fb2': raise Http404 if config.SOPDS_AUTH and request.user.is_authenticated: bookshelf.objects.get_or_create(user=request.user, book=book) full_path=os.path.join(config.SOPDS_ROOT_LIB,book.path) if book.cat_type==opdsdb.CAT_INP: # Убираем из пути INPX и INP файл inp_path, zip_name = os.path.split(full_path) inpx_path, inp_name = os.path.split(inp_path) path, inpx_name = os.path.split(inpx_path) full_path = os.path.join(path,zip_name) if config.SOPDS_TITLE_AS_FILENAME: transname=utils.translit(book.title+'.'+book.format) else: transname=utils.translit(book.filename) transname = utils.to_ascii(transname) (n,e)=os.path.splitext(transname) dlfilename="%s.%s"%(n,convert_type) if convert_type=='epub': converter_path=config.SOPDS_FB2TOEPUB elif convert_type=='mobi': converter_path=config.SOPDS_FB2TOMOBI content_type=mime_detector.fmt(convert_type) if book.cat_type==opdsdb.CAT_NORMAL: tmp_fb2_path=None file_path=os.path.join(full_path, book.filename) elif book.cat_type in [opdsdb.CAT_ZIP, opdsdb.CAT_INP]: try: fz=codecs.open(full_path, "rb") except FileNotFoundError: raise Http404 z = zipfile.ZipFile(fz, 'r', allowZip64=True) z.extract(book.filename,config.SOPDS_TEMP_DIR) tmp_fb2_path=os.path.join(config.SOPDS_TEMP_DIR,book.filename) file_path=tmp_fb2_path tmp_conv_path=os.path.join(config.SOPDS_TEMP_DIR,dlfilename) popen_args = ("\"%s\" \"%s\" \"%s\""%(converter_path,file_path,tmp_conv_path)) proc = subprocess.Popen(popen_args, shell=True, stdout=subprocess.PIPE) #proc = subprocess.Popen((converter_path.encode('utf8'),file_path.encode('utf8'),tmp_conv_path.encode('utf8')), shell=True, stdout=subprocess.PIPE) out = proc.stdout.readlines() if os.path.isfile(tmp_conv_path): fo=codecs.open(tmp_conv_path, "rb") s=fo.read() # HTTP Header response = HttpResponse() response["Content-Type"]='%s; name="%s"'%(content_type,dlfilename) response["Content-Disposition"] = 'attachment; filename="%s"'%(dlfilename) response["Content-Transfer-Encoding"]='binary' response["Content-Length"] = str(len(s)) response.write(s) fo.close() else: raise Http404 try: if tmp_fb2_path: os.remove(tmp_fb2_path) except: pass try: os.remove(tmp_conv_path) except: pass return response
def Download(request, book_id, zip_flag): """ Загрузка файла книги """ book = Book.objects.get(id=book_id) if config.SOPDS_AUTH and request.user.is_authenticated: bookshelf.objects.get_or_create(user=request.user, book=book) full_path=os.path.join(config.SOPDS_ROOT_LIB,book.path) if book.cat_type==opdsdb.CAT_INP: # Убираем из пути INPX и INP файл inp_path, zip_name = os.path.split(full_path) inpx_path, inp_name = os.path.split(inp_path) path, inpx_name = os.path.split(inpx_path) full_path = os.path.join(path,zip_name) if config.SOPDS_TITLE_AS_FILENAME: transname=utils.translit(book.title+'.'+book.format) else: transname=utils.translit(book.filename) transname = utils.to_ascii(transname) if zip_flag == '1': dlfilename=transname+'.zip' content_type= Mimetype.FB2_ZIP if book.format=='fb2' else Mimetype.ZIP else: dlfilename=transname content_type = mime_detector.fmt(book.format) response = HttpResponse() response["Content-Type"]='%s; name="%s"'%(content_type,dlfilename) response["Content-Disposition"] = 'attachment; filename="%s"'%(dlfilename) response["Content-Transfer-Encoding"]='binary' z = None fz = None s = None book_size = book.filesize if book.cat_type==opdsdb.CAT_NORMAL: file_path=os.path.join(full_path, book.filename) book_size=os.path.getsize(file_path) try: fo=codecs.open(file_path, "rb") except FileNotFoundError: raise Http404 s=fo.read() elif book.cat_type in [opdsdb.CAT_ZIP, opdsdb.CAT_INP]: try: fz=codecs.open(full_path, "rb") except FileNotFoundError: raise Http404 z = zipfile.ZipFile(fz, 'r', allowZip64=True) book_size=z.getinfo(book.filename).file_size fo= z.open(book.filename) s=fo.read() if zip_flag=='1': dio = io.BytesIO() zo = zipfile.ZipFile(dio, 'w', zipfile.ZIP_DEFLATED) zo.writestr(transname,s) zo.close() buf = dio.getvalue() response["Content-Length"] = str(len(buf)) response.write(buf) else: response["Content-Length"] = str(book_size) response.write(s) fo.close() if z: z.close() if fz: fz.close() return response