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.
112 lines
4.6 KiB
112 lines
4.6 KiB
# -*- coding: utf-8 -*-
|
|
################################################################################
|
|
#
|
|
# Cybrosys Technologies Pvt. Ltd.
|
|
#
|
|
# Copyright (C) 2024-TODAY Cybrosys Technologies(<https://www.cybrosys.com>).
|
|
# Author: Nikhil M (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/>.
|
|
################################################################################
|
|
import io
|
|
import base64
|
|
from io import BytesIO
|
|
|
|
import pandas as pd
|
|
from docx import Document as DocxDocument
|
|
import qrcode
|
|
from odoo import api, fields, models
|
|
|
|
|
|
class IrAttachment(models.Model):
|
|
"""Extended Attachment Model (Inherited from 'ir.attachment')
|
|
|
|
This class represents an extension of the original 'ir.attachment' model in
|
|
Odoo. The 'ir.attachment' model is a built-in Odoo model that handles file
|
|
attachments to various records, such as documents, emails, or notes.
|
|
|
|
In this custom model, we extend the functionality of 'ir.attachment' to add
|
|
new fields and custom methods to cater to specific requirements of our
|
|
application.
|
|
"""
|
|
_inherit = 'ir.attachment'
|
|
|
|
tags_ids = fields.Many2many(comodel_name='ir.attachment.tag',
|
|
string='Tags', help="Tags for attachments")
|
|
|
|
@api.model
|
|
def decode_content(self, attach_id, doc_type):
|
|
"""Decode XLSX or DOC File Data.
|
|
This method takes a binary file data from an attachment and decodes
|
|
the content of the file for XLSX and DOC file formats.
|
|
:param int attach_id: id of attachment.
|
|
:param str doc_type: the type of the given attachment either 'xlsx' or
|
|
'doc'
|
|
:return: return the decoded data."""
|
|
attachment = self.sudo().browse(attach_id)
|
|
xlsx_data = base64.b64decode(attachment.datas)
|
|
if doc_type in ['xlsx', 'xls', 'docx']:
|
|
try:
|
|
if doc_type == 'xlsx':
|
|
content = pd.read_excel(BytesIO(xlsx_data),
|
|
engine='openpyxl',
|
|
converters={'A': str})
|
|
elif doc_type == 'xls':
|
|
content = pd.read_excel(BytesIO(xlsx_data), engine='xlrd',
|
|
converters={'A': str})
|
|
elif doc_type == 'docx':
|
|
doc = DocxDocument(io.BytesIO(xlsx_data))
|
|
paragraphs = [p.text for p in doc.paragraphs]
|
|
return paragraphs
|
|
else:
|
|
raise ValueError("Unsupported file format")
|
|
html_table = content.to_html(index=False)
|
|
return html_table
|
|
except TypeError:
|
|
return ("<p style='padding-top:8px;color:red;'>"
|
|
"No preview available</p>")
|
|
text = "Cant Preview"
|
|
return text
|
|
|
|
@api.model
|
|
def save_edited_image(self, attachment_id, myImage):
|
|
"""The image is replaced by image from Toast image editor
|
|
:param int attach_id: id of attachment.
|
|
:param str image: new image data
|
|
:return file containing image
|
|
"""
|
|
file = self.sudo().browse(attachment_id)
|
|
file.write({'datas': myImage.strip('data:image/png;base64')})
|
|
return file
|
|
|
|
@api.model
|
|
def generate_qr_code(self, attach_id):
|
|
"""Generate qr code for attachment tha allow anyone to download it."""
|
|
base_url = self.env['ir.config_parameter'].sudo().get_param(
|
|
'web.base.url')
|
|
data = {}
|
|
download_url = f"mail/channel/1/attachment/{attach_id}?download=true"
|
|
if qrcode and base64:
|
|
attach_qr = qrcode.QRCode(
|
|
version=3,
|
|
error_correction=qrcode.constants.ERROR_CORRECT_L,
|
|
box_size=4, border=4)
|
|
attach_qr.add_data(base_url + download_url)
|
|
attach_qr.make(fit=True)
|
|
img = attach_qr.make_image()
|
|
temp = BytesIO()
|
|
img.save(temp, format="PNG")
|
|
qr_image = base64.b64encode(temp.getvalue())
|
|
data.update({'image': qr_image,
|
|
'company': self.env.company.name})
|
|
return data
|
|
|