import os
import odooly
import argparse
import pygsheets
import pandas as pd

def update_attributes_in_product(odoo_items, id, value_ids):
    if not id:
        return
    
    product_id = odoo_items.browse(int(id))

    if not product_id.exists():
        print(f'el producto {product_id} no existe')
        return
    
    product_tmp = product_id.product_tmpl_id

    if len(product_tmp.product_variant_ids) > 1:
        print(f'Plantilla {product_tmp.name} #{product_tmp.id}: más de una variante.')
        return
    
    attribute_lines = {line.attribute_id.id: line for line in product_tmp.attribute_line_ids}

    for product_attribute in value_ids:
        if not product_attribute:
            print(f'Revisar el producto {product_tmp.default_code}, No se encuentra el valor de algun atributo')
            continue
        
        attribute_id = product_attribute[0]['attribute_id'][0]
        attribute_value_id = product_attribute[0]['id']

        if attribute_id in attribute_lines:
            product_with_attribute = attribute_lines[attribute_id]
            value_exist = attribute_value_id in product_with_attribute.value_ids[0].ids

            if value_exist:
                continue  

            product_with_attribute.value_ids = [(6, 0, [product_attribute[0]['id']])]

        else:
            product_tmp.attribute_line_ids = [(0, 0, {
                'attribute_id': product_attribute[0]['attribute_id'][0],
                'value_ids': [(4, product_attribute[0]['id'], 0)]
            })]

def main(odoo, filename, args):

    gc = pygsheets.authorize(service_file=os.path.expanduser('config/0a4abc985039aa.json'))
    ss = gc.open(filename)
    ws = ss.worksheet_by_title('Atributos')

    Value = odoo.env['product.attribute.value'] 
    values = Value.search_read([], ['id', 'name', 'attribute_id']) # Obtiene todos los valores creados en Odoo

    df = ws.get_as_df()
    attributes_df = df.iloc[:, int(args.num_atributo_primer):int(args.num_atributo_ultimo)] # df de atributos
    product_attributes_df = df[(attributes_df != '').any(axis=1)] # df de los productos filtrados donde almenos tengan un atributo

    product_ids = product_attributes_df.iloc[:, int(args.num_product_ids)] # product_id de los productos con almenos un atributo
    attributes_list = product_attributes_df.iloc[:, int(args.num_atributo_primer):int(args.num_atributo_ultimo)].values #valores del productos en tuplas

    # Recorre cada producto con su valores y actualiza el campo

    for product_id, product_attributes in zip(product_ids, attributes_list):
        value_ids = [list(filter(lambda l: l['attribute_id'][1] == attribute and l['name'] == str(value), values)) 
                     for attribute, value in zip(list(attributes_df.columns), product_attributes) if value]
        update_attributes_in_product(odoo.env['product.product'], product_id, value_ids)

if __name__ == "__main__":
    parser = argparse.ArgumentParser(description='Escribir Atributos y valores a los productos')
    parser.add_argument('--num_product_ids', required=True, help='Numero de la columna donde se encuentra el id de los productos, Ejem: 1')
    parser.add_argument('--num_atributo_primer', required=True, help='Numero de la primera columna de los atributos, Ejem: 6')
    parser.add_argument('--num_atributo_ultimo', required=True, help='Numero de la primera columna de los atributos, Ejem: 18')
    args = parser.parse_args()

    filename = 'Maestro de Productos'

    odooly.Client._config_file = os.path.expanduser('config/odooly.ini')
    odoo = odooly.Client.from_config('odootest25')
    main(odoo, filename, args)