65 lines
2.7 KiB
Python
65 lines
2.7 KiB
Python
from odoo.addons.portal.controllers.portal import CustomerPortal
|
|
from odoo.http import request, route
|
|
from odoo.exceptions import AccessError, MissingError
|
|
from odoo import _
|
|
|
|
|
|
class DocumentCustomerPortal(CustomerPortal):
|
|
|
|
def _prepare_home_portal_values(self, counters):
|
|
rtn = super()._prepare_home_portal_values(counters)
|
|
domain = self._prepare_documents_domain()
|
|
rtn['documents_count'] = request.env['documents.document'].search_count(domain)
|
|
return rtn
|
|
|
|
@route('/my/documents', type='http', auth='user', website=True)
|
|
def portal_my_documents(self, **kwargs):
|
|
values = self._prepare_portal_layout_values()
|
|
Documents = request.env['documents.document']
|
|
domain = self._prepare_documents_domain()
|
|
documents_count = Documents.search_count(domain)
|
|
documents = Documents.search(domain)
|
|
values.update({
|
|
'documents_count': documents_count,
|
|
'documents': documents.sudo(),
|
|
'default_url': '/my/documents',
|
|
'page_name': 'my_documents',
|
|
})
|
|
return request.render("bemade_documents_portal.portal_my_documents", values)
|
|
|
|
def _prepare_documents_domain(self):
|
|
partner = request.env.user.partner_id
|
|
user = request.env.user
|
|
"""Helper method intended to be overridden for future modules."""
|
|
return ['|',
|
|
('partner_id', '=', partner.id),
|
|
('owner_id', '=', user.id),
|
|
]
|
|
|
|
def _render_record_template(self, values):
|
|
""" Override this method to apply a different template for a single document
|
|
record on the portal. """
|
|
return request.render("bemade_documents_portal.document_portal_template", values)
|
|
|
|
@route('/my/documents/<int:document_id>', type='http', auth='user', website=True)
|
|
def portal_document_page(self, document_id, download=False, **kwargs):
|
|
document = request.env['documents.document'].browse(document_id)
|
|
if not document:
|
|
raise MissingError(_('This document does not exist.'))
|
|
if download:
|
|
return self._download_attachment(document)
|
|
values={
|
|
'document': document,
|
|
'page_name': 'my_documents',
|
|
'action': document._get_portal_return_action(),
|
|
}
|
|
return self._render_record_template(values)
|
|
|
|
def _download_attachment(self, document):
|
|
attachment = document.attachment_id
|
|
headers = [
|
|
('content-type', attachment.mimetype),
|
|
('content-length', attachment.file_size),
|
|
('content-disposition', f'attachment; filename="{document.name}"')
|
|
]
|
|
return request.make_response(attachment.raw, headers)
|