def _getParams(self): super()._getParams() self._attachment = Attachment.get(int(self._pathParams['res'])) if not self._attachment: raise HTTPAPIError("File not found", 404)
def _process(self): from indico_conversion.plugin import ConversionPlugin try: payload = secure_serializer.loads(request.form['directory'], salt='pdf-conversion') except BadData: ConversionPlugin.logger.exception('Received invalid payload (%s)', request.form['directory']) return jsonify(success=False) attachment = Attachment.get(payload['attachment_id']) if not attachment or attachment.is_deleted or attachment.folder.is_deleted: ConversionPlugin.logger.info('Attachment has been deleted: %s', attachment) return jsonify(success=True) elif request.form['status'] != '1': ConversionPlugin.logger.error('Received invalid status %s for %s', request.form['status'], attachment) return jsonify(success=False) name, ext = os.path.splitext(attachment.file.filename) title = get_pdf_title(attachment) pdf_attachment = Attachment(folder=attachment.folder, user=attachment.user, title=title, description=attachment.description, type=AttachmentType.file, protection_mode=attachment.protection_mode, acl=attachment.acl) # TODO: remove first case when Conversion Server is fully on new version if 'content' in request.form: # handling of legacy API data = BytesIO(base64.decodestring(request.form['content'])) else: filepdf = request.files['content'] data = filepdf.stream.read() pdf_attachment.file = AttachmentFile(user=attachment.file.user, filename='{}.pdf'.format(name), content_type='application/pdf') pdf_attachment.file.save(data) db.session.add(pdf_attachment) db.session.flush() cache.set(unicode(attachment.id), 'finished', timedelta(minutes=15)) ConversionPlugin.logger.info('Added PDF attachment %s for %s', pdf_attachment, attachment) signals.attachments.attachment_created.send(pdf_attachment, user=None) return jsonify(success=True)
def _process(self): from indico_conversion.plugin import ConversionPlugin try: payload = secure_serializer.loads(request.form['directory'], salt='pdf-conversion') except BadData: ConversionPlugin.logger.exception('Received invalid payload (%s)', request.form['directory']) return jsonify(success=False) attachment = Attachment.get(payload['attachment_id']) if not attachment or attachment.is_deleted or attachment.folder.is_deleted: ConversionPlugin.logger.info('Attachment has been deleted: %s', attachment) return jsonify(success=True) elif request.form['status'] != '1': ConversionPlugin.logger.error('Received invalid status %s for %s', request.form['status'], attachment) return jsonify(success=False) name, ext = os.path.splitext(attachment.file.filename) title = get_pdf_title(attachment) pdf_attachment = Attachment(folder=attachment.folder, user=attachment.user, title=title, description=attachment.description, type=AttachmentType.file, protection_mode=attachment.protection_mode, acl=attachment.acl) data = request.files['content'].stream.read() pdf_attachment.file = AttachmentFile(user=attachment.file.user, filename=f'{name}.pdf', content_type='application/pdf') pdf_attachment.file.save(data) db.session.add(pdf_attachment) db.session.flush() pdf_state_cache.set(str(attachment.id), 'finished', timeout=timedelta(minutes=15)) ConversionPlugin.logger.info('Added PDF attachment %s for %s', pdf_attachment, attachment) signals.attachments.attachment_created.send(pdf_attachment, user=None) return jsonify(success=True)
def _getParams(self): super(FileHook, self)._getParams() self._attachment = Attachment.get(int(self._pathParams['res'])) if not self._attachment: raise HTTPAPIError("File not found", 404)