File uploads and downloads

# -*- coding: utf-8 -*-

# Zato
from zato.server.service import Service

class DownloadReport(Service):
    """ Serves a PDF report as a downloadable attachment.
    """
    name = 'demo.rest.download-report'

    input = 'report_id'

    def handle(self) -> 'None':

        report_id = self.request.input.report_id

        # The report's bytes come from another service ..
        content = self.invoke('reports.generate', report_id=report_id)

        # .. and the headers make the response a named download.
        self.response.payload = content
        self.response.content_type = 'application/pdf'

        filename = f'report-{report_id}.pdf'
        self.response.headers['Content-Disposition'] = f'attachment; filename={filename}'

# -*- coding: utf-8 -*-

# Zato
from zato.server.service import Service

class DownloadCSV(Service):
    name = 'demo.rest.download-csv'

    def handle(self) -> 'None':

        lines = [
            'id,name,email',
            '1,John,john@example.com',
            '2,Jane,jane@example.com',
        ]
        csv_data = '\n'.join(lines)

        self.response.payload = csv_data
        self.response.headers['Content-Disposition'] = 'attachment; filename=export.csv'
        self.response.content_type = 'text/csv'


class DownloadExcel(Service):
    name = 'demo.rest.download-excel'

    def handle(self) -> 'None':

        # The workbook bytes come from a service using a library such as openpyxl
        excel_bytes = self.invoke('reports.build-excel')

        self.response.payload = excel_bytes
        self.response.headers['Content-Disposition'] = 'attachment; filename=export.xlsx'

        # The standard MIME type for .xlsx files
        xlsx_mime = 'application/vnd.openxmlformats-officedocument.spreadsheetml.sheet'
        self.response.content_type = xlsx_mime

# -*- coding: utf-8 -*-

# Zato
from zato.common.exception import BadRequest
from zato.server.service import Service

class UploadDocument(Service):
    """ Accepts a document upload and stores it.
    """
    name = 'demo.rest.upload-document'

    def handle(self) -> 'None':

        # The multipart form, files included ..
        form_data = self.request.http.get_form_data()

        # .. a request without a file is rejected with 400 ..
        uploaded_file = form_data.get('file')
        if not uploaded_file:
            raise BadRequest(self.cid, 'No file provided')

        # .. and the file's name, bytes and type go to storage.
        filename = uploaded_file.filename
        content = uploaded_file.read()
        content_type = uploaded_file.content_type

        self.invoke('storage.save',
            filename=filename,
            content=content,
            content_type=content_type)

        self.response.payload = {
            'status': 'uploaded',
            'filename': filename,
            'size': len(content)
        }

# -*- coding: utf-8 -*-

# Zato
from zato.server.service import Service

class FetchJiraAttachment(Service):
    """ Serves a Jira attachment to the caller.
    """
    name = 'demo.rest.fetch-jira-attachment'

    input = 'attachment_id'

    def handle(self) -> 'None':

        attachment_id = self.request.input.attachment_id

        # The connection's URL path is /rest/api/3/attachment/content/{attachment_id},
        # with Jira credentials attached through its security definition
        conn = self.rest['Jira']
        response = conn.get(self.cid, params={'attachment_id': attachment_id})

        # The attachment's bytes pass through to our own caller
        self.response.payload = response.content
        self.response.content_type = 'application/octet-stream'

# -*- coding: utf-8 -*-

# stdlib
import base64

# Zato
from zato.server.service import Service

class GetFileAsBase64(Service):
    name = 'demo.rest.get-file-base64'

    input = 'file_id'

    def handle(self) -> 'None':

        file_id = self.request.input.file_id

        # The binary content from storage ..
        content = self.invoke('storage.get', file_id=file_id)

        # .. becomes text that JSON can carry.
        encoded = base64.b64encode(content).decode('utf-8')

        self.response.payload = {
            'file_id': file_id,
            'content_base64': encoded,
            'size': len(content)
        }