from odoo import models, api
from datetime import datetime
from pytz import timezone
from typing import Type
import pysolr


class PromotionProgram(models.Model):
    _inherit = 'promotion.program'
    _solr_schema = 'promotion_programs'

    def solr(self) -> Type[pysolr.Solr]:
        return self.env['apache.solr'].connect(self._solr_schema)

    def _create_solr_queue(self, function_name: str):
        for rec in self:
            self.env['apache.solr.queue'].create_unique({
                'res_model': self._name,
                'res_id': rec.id,
                'function_name': function_name
            })

    def _sync_to_solr(self):
        ir_attachment = self.env['ir.attachment']
        solr_model = self.env['apache.solr']

        for rec in self:
            document = solr_model.get_doc(self._solr_schema, rec.id)
            document.update({
                'id': rec.id,
                'name_s': rec.name,
                'banner_s': ir_attachment.api_image(self._name, 'banner', rec.id) if rec.banner else '',
                'image_s': ir_attachment.api_image(self._name, 'image', rec.id) if rec.image else '',
                'keywords': [x.name for x in rec.keyword_ids],
                'line_ids': [x.id for x in rec.program_line],
                'start_time_s': self._time_format(rec.start_time),
                'end_time_s': self._time_format(rec.end_time),
                'applies_to_s': rec.applies_to,
                'icon_s': ir_attachment.api_image(self._name, 'icon', rec.id) if rec.icon else '',
                'icon_top_s': ir_attachment.api_image(self._name, 'icon_top', rec.id) if rec.icon_top else '',
                'icon_bottom_s': ir_attachment.api_image(self._name, 'icon_bottom', rec.id) if rec.icon_bottom else '',
            })

            self.solr().add([document])

        self.solr().commit()

    def _time_format(self, object) -> str:
        time = ''
        tz_jakarta = timezone('Asia/Jakarta')
        if isinstance(object, datetime):
            time = object.astimezone(tz_jakarta).strftime("%Y-%m-%d %H:%M:%S")
        return time

    @api.model
    def create(self, vals):
        self._create_solr_queue('_sync_to_solr')
        return super(PromotionProgram, self).create(vals)
    
    def write(self, vals):
        self._create_solr_queue('_sync_to_solr')
        return super(PromotionProgram, self).write(vals)
    
    @api.constrains('program_line')
    def constrains_program_line(self):
        for rec in self:
            for line in rec.program_line:
                line._create_solr_queue('_sync_to_solr')
                
    def solr_flag_to_queue(self, limit=500):
        domain = [
            ('solr_flag', '=', 2),
            ('active', 'in', [True, False])
        ]
        records = self.search(domain, limit=limit)
        for record in records:
            record._create_solr_queue('_sync_to_solr')
            record.solr_flag = 1

    def action_sync_to_solr(self):
        rec_ids = self.env.context.get('active_ids', [])
        recs = self.search([('id', 'in', rec_ids)])
        recs._create_solr_queue('_sync_to_solr')