mirror of
https://github.com/camptocamp/odoo-cloud-platform.git
synced 2026-06-23 18:04:34 +00:00
Yannick Payot
66 lines
2.3 KiB
Python
66 lines
2.3 KiB
Python
# Copyright 2016-2019 Camptocamp SA
|
|
# Copyright 2021 Open Source Integrators
|
|
# License AGPL-3.0 or later (http://www.gnu.org/licenses/agpl.html)
|
|
from odoo import models
|
|
from odoo.http import Stream
|
|
|
|
|
|
class IrBinary(models.AbstractModel):
|
|
_inherit = "ir.binary"
|
|
_description = "File streaming helper model for controllers"
|
|
|
|
def _azure_stream(self, attachment):
|
|
# we will create or own tream and return it
|
|
stream_data = self.env["ir.attachment"]._store_file_read(attachment.store_fname)
|
|
azurestream = Stream(
|
|
type="data",
|
|
data=stream_data,
|
|
path=None,
|
|
url=None,
|
|
mimetype=attachment.mimetype or None,
|
|
download_name=attachment.name,
|
|
size=len(stream_data),
|
|
etag=attachment.checksum,
|
|
)
|
|
return azurestream
|
|
|
|
def _record_to_stream(self, record, field_name):
|
|
"""
|
|
Low level method responsible for the actual conversion from a
|
|
model record to a stream. This method is an extensible hook for
|
|
other modules. It is not meant to be directly called from
|
|
outside or the ir.binary model.
|
|
|
|
:param record: the record where to load the data from.
|
|
:param str field_name: the binary field where to load the data
|
|
from.
|
|
:rtype: odoo.http.Stream
|
|
"""
|
|
if (
|
|
record._name == "ir.attachment"
|
|
and record.store_fname
|
|
and record.store_fname.startswith("azure://")
|
|
):
|
|
# we will create or own tream and return it
|
|
return self._azure_stream(record)
|
|
elif (
|
|
record._name == "documents.document"
|
|
and record.attachment_id
|
|
and record.attachment_id.store_fname
|
|
and record.attachment_id.store_fname.startswith("azure://")
|
|
):
|
|
return self._azure_stream(record.attachment_id)
|
|
|
|
else:
|
|
return super()._record_to_stream(record, field_name)
|
|
|
|
|
|
# This part is used if the customer install tne enterprise module documents
|
|
try:
|
|
from odoo.addons import documents
|
|
|
|
documents.models.ir_binary.IrBinary._record_to_stream = IrBinary._record_to_stream
|
|
except ImportError: # pylint: disable=except-pass
|
|
# document enterprise module if not installed, we just ignore
|
|
pass
|