2017-10-31 03:40:49 +00:00
|
|
|
#!/usr/bin/env python
|
|
|
|
# -*- coding: utf-8 -*-
|
|
|
|
|
|
|
|
# This file is part of Mylar.
|
|
|
|
#
|
|
|
|
# Mylar is free software: you can redistribute it and/or modify
|
|
|
|
# it under the terms of the GNU General Public License as published by
|
|
|
|
# the Free Software Foundation, either version 3 of the License, or
|
|
|
|
# (at your option) any later version.
|
|
|
|
#
|
|
|
|
# Mylar is distributed in the hope that it will be useful,
|
|
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
|
|
# GNU General Public License for more details.
|
|
|
|
#
|
|
|
|
# You should have received a copy of the GNU General Public License
|
|
|
|
# along with Mylar. If not, see <http://www.gnu.org/licenses/>.
|
|
|
|
|
|
|
|
import mylar
|
|
|
|
from mylar import db, mb, importer, search, PostProcessor, versioncheck, logger
|
|
|
|
import simplejson as simplejson
|
|
|
|
import cherrypy
|
|
|
|
import os
|
|
|
|
import urllib2
|
2017-11-01 02:47:34 +00:00
|
|
|
from urllib import urlencode, quote_plus
|
2017-10-31 03:40:49 +00:00
|
|
|
import cache
|
|
|
|
import imghdr
|
|
|
|
from operator import itemgetter
|
|
|
|
from cherrypy.lib.static import serve_file, serve_download
|
|
|
|
import datetime
|
|
|
|
from mylar.webserve import serve_template
|
|
|
|
|
2017-11-01 02:47:34 +00:00
|
|
|
cmd_list = ['root', 'Publishers', 'AllTitles', 'StoryArcs', 'ReadList']
|
2017-10-31 03:40:49 +00:00
|
|
|
|
|
|
|
class OPDS(object):
|
|
|
|
|
|
|
|
def __init__(self):
|
|
|
|
self.cmd = None
|
|
|
|
self.img = None
|
|
|
|
self.file = None
|
|
|
|
self.filename = None
|
|
|
|
self.kwargs = None
|
|
|
|
self.data = None
|
|
|
|
|
|
|
|
def checkParams(self, *args, **kwargs):
|
|
|
|
|
|
|
|
|
|
|
|
if 'cmd' not in kwargs:
|
|
|
|
self.cmd = 'root'
|
|
|
|
|
|
|
|
if not mylar.CONFIG.OPDS_ENABLE:
|
|
|
|
self.data = self._error_with_message('OPDS not enabled')
|
|
|
|
return
|
|
|
|
|
|
|
|
if not self.cmd:
|
|
|
|
if kwargs['cmd'] not in cmd_list:
|
|
|
|
self.data = self._error_with_message('Unknown command: %s' % kwargs['cmd'])
|
|
|
|
return
|
|
|
|
else:
|
|
|
|
self.cmd = kwargs.pop('cmd')
|
|
|
|
|
|
|
|
self.kwargs = kwargs
|
|
|
|
self.data = 'OK'
|
|
|
|
|
|
|
|
def fetchData(self):
|
|
|
|
|
|
|
|
if self.data == 'OK':
|
|
|
|
logger.fdebug('Recieved OPDS command: ' + self.cmd)
|
|
|
|
methodToCall = getattr(self, "_" + self.cmd)
|
|
|
|
result = methodToCall(**self.kwargs)
|
|
|
|
if self.img:
|
|
|
|
return serve_file(path=self.img, content_type='image/jpeg')
|
|
|
|
if self.file and self.filename:
|
|
|
|
return serve_download(path=self.file, name=self.filename)
|
|
|
|
if isinstance(self.data, basestring):
|
|
|
|
return self.data
|
|
|
|
else:
|
|
|
|
cherrypy.response.headers['Content-Type'] = "application/atom+xml"
|
|
|
|
return serve_template(templatename="opds.html", title=self.data['title'], opds=self.data)
|
|
|
|
else:
|
|
|
|
return self.data
|
|
|
|
|
|
|
|
def _error_with_message(self, message):
|
2017-10-31 16:57:26 +00:00
|
|
|
error = '<feed><error>%s</error></feed>' % message
|
2017-10-31 03:40:49 +00:00
|
|
|
cherrypy.response.headers['Content-Type'] = "text/xml"
|
2017-10-31 16:57:26 +00:00
|
|
|
return error
|
2017-10-31 03:40:49 +00:00
|
|
|
|
|
|
|
def _root(self, **kwargs):
|
|
|
|
myDB = db.DBConnection()
|
|
|
|
feed = {}
|
|
|
|
feed['title'] = 'Mylar OPDS'
|
|
|
|
feed['id'] = 'OPDSRoot'
|
|
|
|
feed['updated'] = mylar.helpers.now()
|
|
|
|
links = []
|
|
|
|
entries=[]
|
2017-10-31 16:57:26 +00:00
|
|
|
links.append(getLink(href='/opds',type='application/atom+xml;profile=opds-catalog;kind=navigation', rel='start', title='Home'))
|
|
|
|
links.append(getLink(href='/opds',type='application/atom+xml;profile=opds-catalog;kind=navigation',rel='self'))
|
|
|
|
links.append(getLink(href='/opds?cmd=search', type='application/opensearchdescription+xml',rel='search',title='Search'))
|
2017-10-31 03:40:49 +00:00
|
|
|
publishers = myDB.select("SELECT ComicPublisher from comics GROUP BY ComicPublisher")
|
|
|
|
if len(publishers) > 0:
|
|
|
|
count = len(publishers)
|
|
|
|
entries.append(
|
|
|
|
{
|
|
|
|
'title': 'Publishers (%s)' % count,
|
|
|
|
'id': 'Publishers',
|
|
|
|
'updated': mylar.helpers.now(),
|
|
|
|
'content': 'List of Comic Publishers',
|
|
|
|
'href': '/opds?cmd=Publishers',
|
|
|
|
'kind': 'navigation'
|
|
|
|
}
|
|
|
|
)
|
|
|
|
comics = mylar.helpers.havetotals()
|
|
|
|
count = 0
|
|
|
|
for comic in comics:
|
|
|
|
if comic['haveissues'] > 0:
|
|
|
|
count += 1
|
|
|
|
if count > -1:
|
|
|
|
entries.append(
|
|
|
|
{
|
|
|
|
'title': 'All Titles (%s)' % count,
|
|
|
|
'id': 'AllTitles',
|
|
|
|
'updated': mylar.helpers.now(),
|
|
|
|
'content': 'List of All Comics',
|
|
|
|
'href': '/opds?cmd=AllTitles',
|
|
|
|
'kind': 'navigation'
|
|
|
|
}
|
|
|
|
)
|
2017-10-31 16:57:26 +00:00
|
|
|
storyArcs = mylar.helpers.listStoryArcs()
|
|
|
|
logger.debug(storyArcs)
|
|
|
|
if len(storyArcs) > 0:
|
|
|
|
entries.append(
|
|
|
|
{
|
|
|
|
'title': 'Story Arcs (%s)' % len(storyArcs),
|
|
|
|
'id': 'StoryArcs',
|
|
|
|
'updated': mylar.helpers.now(),
|
|
|
|
'content': 'List of Story Arcs',
|
|
|
|
'href': '/opds?cmd=StoryArcs',
|
|
|
|
'kind': 'navigation'
|
|
|
|
}
|
|
|
|
)
|
2017-11-01 02:47:34 +00:00
|
|
|
readList = myDB.select("SELECT * from readlist")
|
|
|
|
if len(readList) > 0:
|
|
|
|
entries.append(
|
|
|
|
{
|
|
|
|
'title': 'Read List (%s)' % len(readList),
|
|
|
|
'id': 'ReadList',
|
|
|
|
'updated': mylar.helpers.now(),
|
|
|
|
'content': 'Current Read List',
|
|
|
|
'href': '/opds?cmd=ReadList',
|
|
|
|
'kind': 'navigation'
|
|
|
|
|
|
|
|
}
|
|
|
|
)
|
|
|
|
|
|
|
|
feed['links'] = links
|
|
|
|
feed['entries'] = entries
|
|
|
|
self.data = feed
|
|
|
|
return
|
2017-10-31 16:57:26 +00:00
|
|
|
|
2017-11-01 02:47:34 +00:00
|
|
|
def _Publishers(self, **kwargs):
|
2017-11-01 13:47:29 +00:00
|
|
|
index = 0
|
|
|
|
if 'index' in kwargs:
|
|
|
|
index = int(kwargs['index'])
|
2017-11-01 02:47:34 +00:00
|
|
|
myDB = db.DBConnection()
|
|
|
|
feed = {}
|
|
|
|
feed['title'] = 'Mylar OPDS - Publishers'
|
|
|
|
feed['id'] = 'Publishers'
|
|
|
|
feed['updated'] = mylar.helpers.now()
|
|
|
|
links = []
|
|
|
|
entries=[]
|
|
|
|
links.append(getLink(href='/opds',type='application/atom+xml;profile=opds-catalog;kind=navigation', rel='start', title='Home'))
|
|
|
|
links.append(getLink(href='/opds?cmd=Publishers',type='application/atom+xml;profile=opds-catalog;kind=navigation',rel='self'))
|
|
|
|
publishers = myDB.select("SELECT ComicPublisher from comics GROUP BY ComicPublisher")
|
|
|
|
comics = mylar.helpers.havetotals()
|
|
|
|
for publisher in publishers:
|
|
|
|
totaltitles = 0
|
|
|
|
for comic in comics:
|
2017-11-01 03:22:50 +00:00
|
|
|
if comic['ComicPublisher'] == publisher['ComicPublisher'] and comic['haveissues'] > 0:
|
2017-11-01 02:47:34 +00:00
|
|
|
totaltitles += 1
|
|
|
|
if totaltitles > 0:
|
|
|
|
entries.append(
|
|
|
|
{
|
2017-11-01 03:25:47 +00:00
|
|
|
'title': publisher['ComicPublisher'],
|
2017-11-01 03:24:32 +00:00
|
|
|
'id': 'publisher:%s' % publisher['ComicPublisher'],
|
2017-11-01 02:47:34 +00:00
|
|
|
'updated': mylar.helpers.now(),
|
2017-11-01 03:24:32 +00:00
|
|
|
'content': publisher['ComicPublisher'],
|
|
|
|
'href': '/opds?cmd=Publisher&id=%s' % quote_plus(publisher['ComicPublisher']),
|
2017-11-01 02:47:34 +00:00
|
|
|
'kind': 'navigation',
|
|
|
|
}
|
|
|
|
)
|
2017-11-01 13:47:29 +00:00
|
|
|
if len(entries) > (index + 30):
|
|
|
|
links.append(
|
|
|
|
getLink(href='/opds?cmd=Publishers&index=%s' % (index+30), type='application/atom+xml;profile=opds-catalog;kind=navigation', rel='next'))
|
|
|
|
if index >= 30:
|
|
|
|
links.append(
|
|
|
|
getLink(href='/opds?cmd=Publishers&index=%s' % (index-30), type='application/atom+xml;profile=opds-catalog;kind=navigation', rel='previous'))
|
|
|
|
|
2017-10-31 03:40:49 +00:00
|
|
|
feed['links'] = links
|
2017-11-01 13:47:29 +00:00
|
|
|
feed['entries'] = entries[index:(index+30)]
|
2017-10-31 03:40:49 +00:00
|
|
|
self.data = feed
|
2017-10-31 16:57:26 +00:00
|
|
|
return
|
|
|
|
|
2017-11-01 02:47:34 +00:00
|
|
|
|
2017-10-31 16:57:26 +00:00
|
|
|
def getLink(href=None, type=None, rel=None, title=None):
|
|
|
|
link = {}
|
|
|
|
if href:
|
|
|
|
link['href'] = href
|
|
|
|
if type:
|
|
|
|
link['type'] = type
|
|
|
|
if rel:
|
|
|
|
link['rel'] = rel
|
|
|
|
if title:
|
|
|
|
link['title'] = title
|
|
|
|
return link
|