obook/product_import/models/product_template.py

64 lines
2.6 KiB
Python

# -*- coding: utf-8 -*-
#############################################################################
#
# Cybrosys Technologies Pvt. Ltd.
#
# Copyright (C) 2019-TODAY Cybrosys Technologies(<https://www.cybrosys.com>).
# Author: Mohammed Shahil MP @cybrosys(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 base64
import certifi
import urllib3
from odoo import api, fields, models, _
from odoo.exceptions import UserError
class ProductTemplate(models.Model):
"""Inherit the model to add fields and function"""
_inherit = 'product.template'
image_url = fields.Char(string='Image URL', help='Image URL or Path')
image_added = fields.Binary("Image (1920x1920)",
compute='_compute_image_added', store=True)
@api.depends('image_url')
def _compute_image_added(self):
""" Function to load an image from URL or local file path """
image = False
if self.image_url:
if self.image_url.startswith(('http://', 'https://')):
# Load image from URL
try:
http = urllib3.PoolManager(cert_reqs='CERT_REQUIRED',
ca_certs=certifi.where())
image_response = http.request('GET', self.image_url)
image = base64.b64encode(image_response.data)
except Exception as e:
# Handle URL loading errors
raise UserError(
_(f"Error loading image from URL: {str(e)}"))
else:
# Load image from local file path
try:
with open(self.image_url, 'rb') as image_file:
image = base64.b64encode(image_file.read())
except Exception as e:
# Handle local file loading errors
raise UserError(
_(f"Error loading image from local path: {str(e)}"))
image_added = image
if image_added:
self.image_1920 = image_added