You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
80 lines
3.6 KiB
80 lines
3.6 KiB
# -*- coding: utf-8 -*-
|
|
##############################################################################
|
|
#
|
|
# Cybrosys Technologies Pvt. Ltd.
|
|
#
|
|
# Copyright (C) 2024-TODAY Cybrosys Technologies(<https://www.cybrosys.com>)
|
|
# Author: Aysha Shalin (odoo@cybrosys.com)
|
|
#
|
|
# You can modify it under the terms of the GNU AFFERO
|
|
# GENERAL PUBLIC LICENSE (AGPL v3), Version 3.
|
|
#
|
|
# This program 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 (AGPL v3) for more details.
|
|
#
|
|
# You should have received a copy of the GNU AFFERO GENERAL PUBLIC LICENSE
|
|
# (AGPL v3) along with this program.
|
|
# If not, see <http://www.gnu.org/licenses/>.
|
|
#
|
|
##############################################################################
|
|
from odoo.exceptions import AccessError
|
|
from odoo.http import request
|
|
from odoo import _
|
|
from odoo.addons.mail.controllers.attachment import AttachmentController
|
|
from werkzeug.exceptions import NotFound
|
|
from odoo.addons.mail.tools.discuss import Store
|
|
|
|
|
|
class DiscussController(AttachmentController):
|
|
def mail_attachment_upload(self, ufile, thread_id, thread_model,
|
|
is_pending=False, **kwargs):
|
|
""" Shows warning if the attachment size exceeds the maximum size allowed """
|
|
thread = request.env[thread_model].search([("id", "=", thread_id)])
|
|
if not thread:
|
|
raise NotFound()
|
|
set_restriction = request.env.user.set_restriction
|
|
max_size = request.env.user.max_size * 1024 * 1024
|
|
if (thread_model == "discuss.channel" and not thread.allow_public_upload
|
|
and not request.env.user._is_internal()):
|
|
raise AccessError(
|
|
_("You are not allowed to upload attachments on this channel."))
|
|
vals = {
|
|
"name": ufile.filename,
|
|
"raw": ufile.read(),
|
|
"res_id": int(thread_id),
|
|
"res_model": thread_model,
|
|
}
|
|
if is_pending and is_pending != "false":
|
|
# Add this point, the message related to the uploaded file does
|
|
# not exist yet, so we use those placeholder values instead.
|
|
vals.update(
|
|
{
|
|
"res_id": 0,
|
|
"res_model": "mail.compose.message",
|
|
}
|
|
)
|
|
if request.env.user.share:
|
|
# Only generate the access token if absolutely necessary
|
|
# (= not for internal user).
|
|
vals["access_token"] = request.env[
|
|
"ir.attachment"]._generate_access_token()
|
|
try:
|
|
# sudo: ir.attachment - posting a new attachment on an
|
|
# accessible thread
|
|
attachment = request.env["ir.attachment"].sudo().create(vals)
|
|
attachment._post_add_create(**kwargs)
|
|
attachmentData = {"data": Store(attachment, extra_fields=["access_token"]).get_result()}
|
|
if attachment.access_token:
|
|
attachmentData["accessToken"] = attachment.access_token
|
|
if set_restriction:
|
|
if attachmentData['data']['ir.attachment'][0]['size'] > max_size:
|
|
attachmentData = {'error': _('Attachment'
|
|
' size cannot exceed %s MB.'
|
|
) % request.env.user.max_size}
|
|
attachment.unlink()
|
|
except AccessError:
|
|
attachmentData = {
|
|
"error": _("You are not allowed to upload an attachment here.")}
|
|
return request.make_json_response(attachmentData)
|
|
|