openmedialibrary/oml/item/handlers.py

207 lines
7.4 KiB
Python
Raw Normal View History

2014-05-19 22:50:41 +00:00
# -*- coding: utf-8 -*-
# vi:si:et:sw=4:sts=4:ts=4
2014-09-02 22:32:44 +00:00
2014-05-19 22:50:41 +00:00
2014-08-12 08:16:57 +00:00
from datetime import datetime
import mimetypes
2014-05-19 22:50:41 +00:00
import os
2014-10-31 11:48:20 +00:00
from urllib.request import quote
2014-05-19 22:50:41 +00:00
import zipfile
2015-06-06 17:54:44 +00:00
import ox
from .models import Item, File
from user.models import List
from .scan import add_file
2014-08-09 16:33:59 +00:00
import db
2014-08-12 08:16:57 +00:00
import settings
import tornado.web
2015-06-06 17:54:44 +00:00
import tornado.gen
import tornado.concurrent
from oxtornado import json_dumps, json_response
2014-08-12 08:16:57 +00:00
2015-06-06 17:54:44 +00:00
from media import get_id
import state
2015-06-06 17:54:44 +00:00
import logging
2015-11-29 14:56:38 +00:00
logger = logging.getLogger(__name__)
2014-05-19 22:50:41 +00:00
class OMLHandler(tornado.web.RequestHandler):
2014-08-09 16:33:59 +00:00
def initialize(self):
pass
2014-05-19 22:50:41 +00:00
class EpubHandler(OMLHandler):
def get(self, id, filename):
2014-08-09 16:33:59 +00:00
with db.session():
2014-05-19 22:50:41 +00:00
item = Item.get(id)
path = item.get_path()
2015-12-24 15:11:47 +00:00
if not item or item.info['extension'] != 'epub' or not path:
self.set_status(404)
self.write('')
else:
z = zipfile.ZipFile(path)
if filename == '':
self.write('<br>\n'.join([f.filename for f in z.filelist]))
elif filename not in [f.filename for f in z.filelist]:
2014-05-19 22:50:41 +00:00
self.set_status(404)
self.write('')
else:
2015-12-24 15:11:47 +00:00
content_type = {
'xpgt': 'application/vnd.adobe-page-template+xml'
}.get(filename.split('.')[0], mimetypes.guess_type(filename)[0]) or 'text/plain'
self.set_header('Content-Type', content_type)
self.write(z.read(filename))
2014-05-19 22:50:41 +00:00
2014-10-31 11:48:20 +00:00
def serve_static(handler, path, mimetype, include_body=True, disposition=None):
2014-05-19 22:50:41 +00:00
handler.set_header('Content-Type', mimetype)
2015-02-22 11:08:06 +00:00
size = os.stat(path).st_size
handler.set_header('Accept-Ranges', 'bytes')
2014-10-31 11:48:20 +00:00
if disposition:
handler.set_header('Content-Disposition', "attachment; filename*=UTF-8''%s" % quote(disposition.encode('utf-8')))
2014-05-25 20:32:00 +00:00
if include_body:
2015-02-22 11:08:06 +00:00
if 'Range' in handler.request.headers:
handler.set_status(206)
r = handler.request.headers.get('Range').split('=')[-1].split('-')
start = int(r[0])
2017-05-30 19:03:14 +00:00
end = int(r[1]) if r[1] else (size - 1)
2015-02-22 12:08:58 +00:00
length = end - start + 1
2015-02-22 11:08:06 +00:00
handler.set_header('Content-Length', str(length))
2015-02-22 12:08:58 +00:00
handler.set_header('Content-Range', 'bytes %s-%s/%s' % (start, end, size))
2015-02-22 11:08:06 +00:00
with open(path, 'rb') as fd:
fd.seek(start)
handler.write(fd.read(length))
else:
handler.set_header('Content-Length', str(size))
with open(path, 'rb') as fd:
handler.write(fd.read())
else:
handler.set_header('Content-Length', str(size))
2014-05-19 22:50:41 +00:00
return
class FileHandler(OMLHandler):
2014-12-13 20:01:54 +00:00
def initialize(self, attachment=False):
self._attachment = attachment
2014-05-25 20:32:00 +00:00
def head(self, id):
self.get(id, include_body=False)
def get(self, id, include_body=True):
2014-08-09 16:33:59 +00:00
with db.session():
2014-05-19 22:50:41 +00:00
item = Item.get(id)
2014-05-25 20:32:00 +00:00
path = item.get_path() if item else None
if not item or not path:
2014-05-19 22:50:41 +00:00
self.set_status(404)
return
mimetype = {
2015-03-14 07:35:15 +00:00
'cbr': 'application/x-cbr',
'cbz': 'application/x-cbz',
2014-05-19 22:50:41 +00:00
'epub': 'application/epub+zip',
'pdf': 'application/pdf',
2014-05-21 23:06:08 +00:00
'txt': 'text/plain',
2014-05-19 22:50:41 +00:00
}.get(path.split('.')[-1], None)
if mimetype == 'text/plain':
try:
open(path, 'rb').read().decode('utf-8')
mimetype = 'text/plain; charset=utf-8'
except:
mimetype = 'text/plain; charset=latin-1'
2014-12-13 20:01:54 +00:00
if self._attachment:
disposition = os.path.basename(path)
else:
disposition = None
2015-12-24 15:11:47 +00:00
return serve_static(self, path, mimetype, include_body, disposition=disposition)
2014-05-19 22:50:41 +00:00
class ReaderHandler(OMLHandler):
def get(self, id):
2014-08-09 16:33:59 +00:00
with db.session():
2014-05-19 22:50:41 +00:00
item = Item.get(id)
if not item:
self.set_status(404)
return
2015-03-14 07:35:15 +00:00
if item.info['extension'] in ('cbr', 'cbz'):
html = 'html/cbr.html'
elif item.info['extension'] == 'epub':
2014-05-19 22:50:41 +00:00
html = 'html/epub.html'
elif item.info['extension'] == 'pdf':
html = 'html/pdf.html'
elif item.info['extension'] == 'txt':
html = 'html/txt.html'
else:
self.set_status(404)
return
item.accessed = datetime.utcnow()
item.timesaccessed = (item.timesaccessed or 0) + 1
item.update_sort()
2014-05-19 22:50:41 +00:00
item.save()
2015-12-24 15:11:47 +00:00
path = os.path.join(settings.static_path, html)
return serve_static(self, path, 'text/html')
2015-06-06 17:54:44 +00:00
class UploadHandler(tornado.web.RequestHandler):
def initialize(self, context=None):
self._context = context
def get(self):
self.write('use POST')
@tornado.web.asynchronous
@tornado.gen.coroutine
def post(self):
if 'origin' in self.request.headers and self.request.host not in self.request.headers['origin']:
logger.debug('reject cross site attempt to access api %s', self.request)
self.set_status(403)
self.write('')
return
2016-06-24 10:27:05 +00:00
2015-06-06 17:54:44 +00:00
def save_files(context, request, callback):
listname = request.arguments.get('list', None)
if listname:
listname = listname[0]
if isinstance(listname, bytes):
listname = listname.decode('utf-8')
with context():
prefs = settings.preferences
ids = []
for upload in request.files.get('files', []):
filename = upload.filename
id = get_id(data=upload.body)
ids.append(id)
file = File.get(id)
if not file:
2016-01-20 07:19:19 +00:00
prefix_books = os.path.join(os.path.expanduser(prefs['libraryPath']), 'Books' + os.sep)
prefix_imported = os.path.join(prefix_books, '.import' + os.sep)
2015-06-06 17:54:44 +00:00
ox.makedirs(prefix_imported)
import_name = os.path.join(prefix_imported, filename)
n = 1
while os.path.exists(import_name):
n += 1
name, extension = filename.rsplit('.', 1)
if extension == 'kepub':
extension = 'epub'
2015-06-06 17:54:44 +00:00
import_name = os.path.join(prefix_imported, '%s [%d].%s' % (name, n, extension))
with open(import_name, 'wb') as fd:
fd.write(upload.body)
file = add_file(id, import_name, prefix_books)
2016-01-15 08:32:13 +00:00
file.move()
else:
user = state.user()
file.item.add_user(user)
2015-06-06 17:54:44 +00:00
if listname and ids:
l = List.get(settings.USER_ID, listname)
if l:
l.add_items(ids)
response = json_response({'ids': ids})
callback(response)
response = yield tornado.gen.Task(save_files, self._context, self.request)
2016-06-24 10:27:05 +00:00
if 'status' not in response:
2015-06-06 17:54:44 +00:00
response = json_response(response)
response = json_dumps(response)
self.set_header('Content-Type', 'application/json')
self.write(response)