From 8abb94a88e40fde249b562a841a5b0398582717e Mon Sep 17 00:00:00 2001 From: Duda Dornelles Date: Wed, 21 Jan 2015 18:12:25 -0200 Subject: #224 App is working without klein - migration to twisted "complete" --- .../pixelated/resources/attachments_resource.py | 63 ++++++++++++++++++++++ 1 file changed, 63 insertions(+) create mode 100644 service/pixelated/resources/attachments_resource.py (limited to 'service/pixelated/resources/attachments_resource.py') diff --git a/service/pixelated/resources/attachments_resource.py b/service/pixelated/resources/attachments_resource.py new file mode 100644 index 00000000..0ab214b9 --- /dev/null +++ b/service/pixelated/resources/attachments_resource.py @@ -0,0 +1,63 @@ +# +# Copyright (c) 2014 ThoughtWorks, Inc. +# +# Pixelated is free software: you can redistribute it and/or modify +# it under the terms of the GNU Affero General Public License as published by +# the Free Software Foundation, either version 3 of the License, or +# (at your option) any later version. +# +# Pixelated 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 Affero General Public License for more details. +# +# You should have received a copy of the GNU Affero General Public License +# along with Pixelated. If not, see . + +import io + +import re +from twisted.protocols.basic import FileSender +from twisted.python.log import err +from twisted.web.resource import Resource + + +class AttachmentResource(Resource): + def __init__(self, attachment_id, querier): + Resource.__init__(self) + self.attachment_id = attachment_id + self.querier = querier + + def render_GET(self, request): + encoding = request.args.get('encoding', [None])[0] + filename = request.args.get('filename', [self.attachment_id])[0] + attachment = self.querier.attachment(self.attachment_id, encoding) + + request.setHeader(b'Content-Type', b'application/force-download') + request.setHeader(b'Content-Disposition', bytes('attachment; filename=' + filename)) + bytes_io = io.BytesIO(attachment['content']) + d = FileSender().beginFileTransfer(bytes_io, request) + + def cb_finished(_): + bytes_io.close() + request.finish() + + d.addErrback(err).addCallback(cb_finished) + + return d + + def _extract_mimetype(self, content_type): + match = re.compile('([A-Za-z-]+\/[A-Za-z-]+)').search(content_type) + return match.group(1) + + +class AttachmentsResource(Resource): + + isLeaf = True + + def __init__(self, querier): + Resource.__init__(self) + self.querier = querier + + def getChild(self, attachment_id, request): + return AttachmentResource(attachment_id, self.querier) -- cgit v1.2.3