# -*- 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 binascii
import tempfile
import certifi
import urllib3
import xlrd
from odoo.exceptions import ValidationError
from odoo import models, fields, _


class ProductImport(models.Model):
    """Model to import the product"""
    _name = 'product.import'
    _description = 'Product Import'

    file = fields.Binary(string="Upload File")
    file_name = fields.Char(string="File Name")
    option = fields.Selection([('csv', 'CSV'), ('xlsx', 'XLSX')],
                              default='csv')

    def import_file(self):
        """Function to import product details from csv and xlsx file """
        if self.option == 'csv':
            try:
                product_temp_data = self.env['product.template'].search([])
                file = base64.b64decode(self.file)
                file_string = file.decode('utf-8')
                file_string = file_string.split('\n')
                http = urllib3.PoolManager(cert_reqs='CERT_REQUIRED',
                                           ca_certs=certifi.where())
            except Exception:
                raise ValidationError(_("Please choose the correct file!"))
            firstline = True
            for file_item in file_string:
                if firstline:
                    firstline = False
                    continue
                product_temp = self.env['product.template'].search(
                    [('name', '=', file_item.split(",")[0])], limit=0)
                if not product_temp.id:
                    file_parts = file_item.split(",")
                    if len(file_parts) >= 5:
                        name, detailed_type, barcode, list_price, file_path_or_url = file_parts[:5]
                        product_name = {
                            'name': name,
                            'detailed_type': detailed_type,
                            'barcode': barcode,
                            'list_price': list_price,
                        }
                        if (file_path_or_url.startswith("http://") or
                                file_path_or_url.startswith("https://")):
                            image_response = http.request('GET',
                                                          file_path_or_url)
                            image_thumbnail = base64.b64encode(
                                image_response.data)
                            product_name['image_1920'] = image_thumbnail
                        elif file_path_or_url.startswith('/home'):
                            with open(file_path_or_url, 'rb') as file:
                                data = base64.b64encode(file.read())
                                product_name['image_1920'] = data
                        product_temp_data.create(product_name)
                else:
                    raise ValidationError(
                        _("Add the product which is not available in products")
                    )
        elif self.option == 'xlsx':
            try:
                product_temp_data = self.env['product.template']
                file_string = tempfile.NamedTemporaryFile(suffix=".xlsx")
                file_string.write(binascii.a2b_base64(self.file))
                book = xlrd.open_workbook(file_string.name)
                sheet = book.sheet_by_index(0)
                http = urllib3.PoolManager(cert_reqs='CERT_REQUIRED',
                                           ca_certs=certifi.where())
            except:
                raise ValidationError(_("Please choose the correct file"))
            startline = True
            for i in range(sheet.nrows):
                if startline:
                    startline = False
                else:
                    line = list(sheet.row_values(i))
                    product_temp = self.env['product.template'].search(
                        [('name', '=', line[0])], limit=0)
                    if product_temp.id:
                        raise ValidationError(
                            _("Add the product which is not available in"
                              " products"))
                    if line[0]:
                        if "http://" in line[4] or "https://" in line[4]:
                            link = line[4]
                            image_response = http.request('GET', link)
                            image_thumbnail = base64.b64encode(
                                image_response.data)
                        elif "/home" in line[4]:
                            with open(line[4], 'rb') as file:
                                image_thumbnail = base64.b64encode(file.read())
                        else:
                            image_thumbnail = False  # or None
                        product_name = {
                            'name': line[0],
                            'detailed_type': line[1],
                            'barcode': line[2],
                            'list_price': line[3],
                            'image_1920': image_thumbnail,
                        }
                        product_temp_data.create(product_name)