diff --git a/cps/templates/feed.xml b/cps/templates/feed.xml index 6f79a3aa..fce047d9 100644 --- a/cps/templates/feed.xml +++ b/cps/templates/feed.xml @@ -64,8 +64,7 @@ {{entry.name}} {{ url_for(folder, book_id=entry.id) }} - - + {% endfor %} diff --git a/cps/templates/index.xml b/cps/templates/index.xml index 228637c2..f0a6e2e0 100644 --- a/cps/templates/index.xml +++ b/cps/templates/index.xml @@ -12,64 +12,57 @@ {{_('Hot Books')}} - - + {{url_for('feed_hot')}} {{_('Popular publications from this catalog based on Downloads.')}} {{_('Best rated Books')}} - - + {{url_for('feed_best_rated')}} {{_('Popular publications from this catalog based on Rating.')}} {{_('New Books')}} - - + {{url_for('feed_new')}} {{_('The latest Books')}} {{_('Random Books')}} - - + {{url_for('feed_discover')}} {{_('Show Random Books')}} +{% if not current_user.is_anonymous() %} {{_('Read Books')}} - - + {{url_for('feed_read_books')}} {{_('Read Books')}} {{_('Unread Books')}} - - + {{url_for('feed_unread_books')}} {{_('Unread Books')}} +{% endif %} {{_('Authors')}} - - + {{url_for('feed_authorindex')}} {{_('Books ordered by Author')}} {{_('Category list')}} - - + {{url_for('feed_categoryindex')}} {{_('Books ordered by category')}} {{_('Series list')}} - - + {{url_for('feed_seriesindex')}} {{_('Books ordered by series')}} diff --git a/cps/ub.py b/cps/ub.py index a559db43..3d589c64 100644 --- a/cps/ub.py +++ b/cps/ub.py @@ -6,6 +6,7 @@ from sqlalchemy import exc from sqlalchemy.ext.declarative import declarative_base from sqlalchemy.orm import * from flask_login import AnonymousUserMixin +import sys import os import logging from werkzeug.security import generate_password_hash @@ -387,7 +388,11 @@ class Config: (self.config_default_role & ROLE_DELETE_BOOKS == ROLE_DELETE_BOOKS)) def mature_content_tags(self): - return list(map(unicode.lstrip, self.config_mature_content_tags.split(","))) + if (sys.version_info > (3, 0)): #Python3 str, Python2 unicode + lstrip = str.lstrip + else: + lstrip = unicode.lstrip + return list(map(lstrip, self.config_mature_content_tags.split(","))) def get_Log_Level(self): ret_value="" diff --git a/cps/web.py b/cps/web.py index 35096070..32af945a 100755 --- a/cps/web.py +++ b/cps/web.py @@ -203,6 +203,7 @@ class ReverseProxied(object): mimetypes.init() mimetypes.add_type('application/xhtml+xml', '.xhtml') mimetypes.add_type('application/epub+zip', '.epub') +mimetypes.add_type('application/fb2+zip', '.fb2') mimetypes.add_type('application/x-mobipocket-ebook', '.mobi') mimetypes.add_type('application/x-mobipocket-ebook', '.prc') mimetypes.add_type('application/vnd.amazon.ebook', '.azw') @@ -631,7 +632,7 @@ def before_request(): def feed_index(): xml = render_title_template('index.xml') response = make_response(xml) - response.headers["Content-Type"] = "application/xml" + response.headers["Content-Type"] = "application/atom+xml; charset=utf-8" return response @@ -640,7 +641,7 @@ def feed_index(): def feed_osd(): xml = render_title_template('osd.xml', lang='de-DE') response = make_response(xml) - response.headers["Content-Type"] = "application/xml" + response.headers["Content-Type"] = "application/xml; charset=utf-8" return response @@ -670,7 +671,7 @@ def feed_search(term): else: xml = render_title_template('feed.xml', searchterm="") response = make_response(xml) - response.headers["Content-Type"] = "application/xml" + response.headers["Content-Type"] = "application/atom+xml; charset=utf-8" return response @@ -684,7 +685,7 @@ def feed_new(): db.Books, True, db.Books.timestamp.desc()) xml = render_title_template('feed.xml', entries=entries, pagination=pagination) response = make_response(xml) - response.headers["Content-Type"] = "application/xml" + response.headers["Content-Type"] = "application/atom+xml; charset=utf-8" return response @@ -696,7 +697,7 @@ def feed_discover(): pagination = Pagination(1, config.config_books_per_page, int(config.config_books_per_page)) xml = render_title_template('feed.xml', entries=entries, pagination=pagination) response = make_response(xml) - response.headers["Content-Type"] = "application/xml" + response.headers["Content-Type"] = "application/atom+xml; charset=utf-8" return response @@ -710,7 +711,7 @@ def feed_best_rated(): db.Books, db.Books.ratings.any(db.Ratings.rating > 9), db.Books.timestamp.desc()) xml = render_title_template('feed.xml', entries=entries, pagination=pagination) response = make_response(xml) - response.headers["Content-Type"] = "application/xml" + response.headers["Content-Type"] = "application/atom+xml; charset=utf-8" return response @@ -738,7 +739,7 @@ def feed_hot(): pagination = Pagination((int(off) / (int(config.config_books_per_page)) + 1), config.config_books_per_page, numBooks) xml = render_title_template('feed.xml', entries=entries, pagination=pagination) response = make_response(xml) - response.headers["Content-Type"] = "application/xml" + response.headers["Content-Type"] = "application/atom+xml; charset=utf-8" return response @@ -754,7 +755,7 @@ def feed_authorindex(): len(db.session.query(db.Authors).all())) xml = render_title_template('feed.xml', listelements=entries, folder='feed_author', pagination=pagination) response = make_response(xml) - response.headers["Content-Type"] = "application/xml" + response.headers["Content-Type"] = "application/atom+xml; charset=utf-8" return response @@ -768,7 +769,7 @@ def feed_author(book_id): db.Books, db.Books.authors.any(db.Authors.id == book_id), db.Books.timestamp.desc()) xml = render_title_template('feed.xml', entries=entries, pagination=pagination) response = make_response(xml) - response.headers["Content-Type"] = "application/xml" + response.headers["Content-Type"] = "application/atom+xml; charset=utf-8" return response @@ -784,7 +785,7 @@ def feed_categoryindex(): len(db.session.query(db.Tags).all())) xml = render_title_template('feed.xml', listelements=entries, folder='feed_category', pagination=pagination) response = make_response(xml) - response.headers["Content-Type"] = "application/xml" + response.headers["Content-Type"] = "application/atom+xml; charset=utf-8" return response @@ -798,7 +799,7 @@ def feed_category(book_id): db.Books, db.Books.tags.any(db.Tags.id == book_id), db.Books.timestamp.desc()) xml = render_title_template('feed.xml', entries=entries, pagination=pagination) response = make_response(xml) - response.headers["Content-Type"] = "application/xml" + response.headers["Content-Type"] = "application/atom+xml; charset=utf-8" return response @@ -814,7 +815,7 @@ def feed_seriesindex(): len(db.session.query(db.Series).all())) xml = render_title_template('feed.xml', listelements=entries, folder='feed_series', pagination=pagination) response = make_response(xml) - response.headers["Content-Type"] = "application/xml" + response.headers["Content-Type"] = "application/atom+xml; charset=utf-8" return response @@ -828,7 +829,7 @@ def feed_series(book_id): db.Books, db.Books.series.any(db.Series.id == book_id),db.Books.series_index) xml = render_title_template('feed.xml', entries=entries, pagination=pagination) response = make_response(xml) - response.headers["Content-Type"] = "application/xml" + response.headers["Content-Type"] = "application/atom+xml; charset=utf-8" return response @@ -873,7 +874,11 @@ def get_opds_download_link(book_id, book_format): file_name = helper.get_valid_filename(file_name) headers = Headers() headers["Content-Disposition"] = "attachment; filename*=UTF-8''%s.%s" % (quote(file_name.encode('utf8')), book_format) - app.logger.info(time.time()-startTime) + try: + headers["Content-Type"] = mimetypes.types_map['.' + book_format] + except KeyError: + headers["Content-Type"] = "application/octet-stream" + app.logger.info(time.time() - startTime) startTime = time.time() if config.config_use_google_drive: app.logger.info(time.time() - startTime) @@ -1712,15 +1717,21 @@ def feed_get_cover(book_id): def render_read_books(page, are_read, as_xml=False): - readBooks = ub.session.query(ub.ReadBook).filter(ub.ReadBook.user_id == int(current_user.id)).filter(ub.ReadBook.is_read == True).all() - readBookIds = [x.book_id for x in readBooks] - if are_read: - db_filter = db.Books.id.in_(readBookIds) + if not current_user.is_anonymous(): + readBooks = ub.session.query(ub.ReadBook).filter(ub.ReadBook.user_id == int(current_user.id)).filter(ub.ReadBook.is_read == True).all() + readBookIds = [x.book_id for x in readBooks] + if are_read: + db_filter = db.Books.id.in_(readBookIds) + else: + db_filter = ~db.Books.id.in_(readBookIds) + + entries, random, pagination = fill_indexpage(page, db.Books, + db_filter, db.Books.timestamp.desc()) else: - db_filter = ~db.Books.id.in_(readBookIds) + entries = [] + random = False + pagination = Pagination(page, 1, 0) - entries, random, pagination = fill_indexpage(page, db.Books, - db_filter, db.Books.timestamp.desc()) if as_xml: xml = render_title_template('feed.xml', entries=entries, pagination=pagination) response = make_response(xml)