63 lines
2.7 KiB
Python
63 lines
2.7 KiB
Python
# -*- coding: utf-8 -*-
|
|
# Part of Odoo. See LICENSE file for full copyright and licensing details.
|
|
import io
|
|
import zipfile
|
|
|
|
from odoo import http, _
|
|
from odoo.http import request, content_disposition
|
|
|
|
|
|
def _get_headers(filename, filetype, content):
|
|
return [
|
|
('Content-Type', filetype),
|
|
('Content-Length', len(content)),
|
|
('Content-Disposition', content_disposition(filename)),
|
|
('X-Content-Type-Options', 'nosniff'),
|
|
]
|
|
|
|
|
|
def _build_zip_from_data(docs_data):
|
|
buffer = io.BytesIO()
|
|
with zipfile.ZipFile(buffer, 'w', compression=zipfile.ZIP_DEFLATED) as zipfile_obj:
|
|
for doc_data in docs_data:
|
|
zipfile_obj.writestr(doc_data['filename'], doc_data['content'])
|
|
return buffer.getvalue()
|
|
|
|
|
|
class AccountDocumentDownloadController(http.Controller):
|
|
|
|
@http.route('/account/download_invoice_attachments/<models("ir.attachment"):attachments>', type='http', auth='user')
|
|
def download_invoice_attachments(self, attachments):
|
|
attachments.check_access('read')
|
|
assert all(attachment.res_id and attachment.res_model == 'account.move' for attachment in attachments)
|
|
if len(attachments) == 1:
|
|
headers = _get_headers(attachments.name, attachments.mimetype, attachments.raw)
|
|
return request.make_response(attachments.raw, headers)
|
|
else:
|
|
inv_ids = attachments.mapped('res_id')
|
|
if len(set(inv_ids)) == 1:
|
|
invoice = request.env['account.move'].browse(inv_ids[0])
|
|
filename = invoice._get_invoice_report_filename(extension='zip')
|
|
else:
|
|
filename = _('invoices') + '.zip'
|
|
content = attachments._build_zip_from_attachments()
|
|
headers = _get_headers(filename, 'zip', content)
|
|
return request.make_response(content, headers)
|
|
|
|
@http.route('/account/download_invoice_documents/<models("account.move"):invoices>/<string:filetype>', type='http', auth='user')
|
|
def download_invoice_documents_filetype(self, invoices, filetype, allow_fallback=True):
|
|
invoices.check_access('read')
|
|
docs_data = []
|
|
for invoice in invoices:
|
|
doc_data = invoice._get_invoice_legal_documents(filetype, allow_fallback=allow_fallback)
|
|
if doc_data:
|
|
docs_data.append(doc_data)
|
|
if len(docs_data) == 1:
|
|
doc_data = docs_data[0]
|
|
headers = _get_headers(**doc_data)
|
|
return request.make_response(doc_data['content'], headers)
|
|
elif len(docs_data) > 1:
|
|
zip_content = _build_zip_from_data(docs_data)
|
|
headers = _get_headers(_('invoices') + '.zip', 'zip', zip_content)
|
|
return request.make_response(zip_content, headers)
|