from odoo import fields, models, api
import logging
from datetime import datetime, timedelta

_logger = logging.getLogger(__name__)


class StockVendor(models.Model):
    _name = "stock.vendor"
    _description = "Stock Vendor"
    _rec_name = "product_variant_id"

    product_variant_id = fields.Many2one("product.product", string="Product Variant")
    quantity = fields.Integer(string="Quantity")
    cache_reset_status = fields.Selection([
        ('reset', 'Reset'),
        ('done', 'Done')
    ], string="Cache Reset")

    def cache_reset(self):
        stocks = self.env['stock.vendor'].search([
            ('cache_reset_status', '=', 'reset'),
        ])

        templates = []
        for stock in stocks:
            templates.append(stock.product_variant_id.product_tmpl_id)
            # stock.cache_reset_status = 'done'
        templates = list(dict.fromkeys(templates))

        for template in templates:
            if template.solr_flag == 1:
                template.solr_flag = 2
                _logger.info('Update Solr Flag to 2 %s' % template.name)

    def update_product_template(self):
        updated_virtual_qty_product_template = self.env['ir.config_parameter'].search([('key', '=', 'updated_virtual_qty_product_template')], limit=1)
        updated_virtual_qty_product_template_value = int(updated_virtual_qty_product_template.value)

        limit = 10000
        query = [('active', '=', True), ('type', '=', 'product')]
        templates = self.env['product.template'].search(query, limit=limit, offset=updated_virtual_qty_product_template_value)
        template_count = self.env['product.template'].search_count(query)

        if (updated_virtual_qty_product_template_value + limit) > template_count:
            updated_virtual_qty_product_template.value = '0'
        else:
            updated_virtual_qty_product_template.value = str(limit + updated_virtual_qty_product_template_value)

        for template in templates:
            template.virtual_qty = template.qty_stock_vendor or 0
            _logger.info("Update Stock Product Template %s : %s" % (template.id, template.virtual_qty))

    # @api.onchange('quantity')
    # def update_solr_flag(self):
    #     for stock in self:
    #         if stock.product_variant_id.product_tmpl_id.solr_flag == 1:
    #             stock.product_variant_id.product_tmpl_id.solr_flag = 2