summaryrefslogtreecommitdiff
path: root/indoteknik_custom/models/keywords.py
blob: 5faa94a664d13b4ec5c8b4d5bcde27bfdceb58f2 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
from itertools import product
from odoo import fields, models, api, tools, _
import logging
import re
import pysolr
from odoo.exceptions import UserError
import base64
import xlrd, xlwt
import io
from bs4 import BeautifulSoup


_logger = logging.getLogger(__name__)
# solr = pysolr.Solr('http://10.148.0.5:8983/solr/searchkey/', always_commit=True, timeout=30)
solr = pysolr.Solr('http://127.0.0.1:8983/solr/searchkey/', always_commit=True, timeout=30)

class Keywords(models.Model):
    _name = 'keywords'
    _order= 'id desc'

    category_id = fields.Many2one('product.public.category', string='Category', required=True)
    keywords = fields.Char('Keywords', required=True)
    product_ids = fields.Many2many(
        'product.product',
        'keywords_product_rel',
        'keyword_id',
        'product_id',
        string='Products'
    )
    name = fields.Char('Name', compute="_compute_name")
    skip = fields.Boolean('Skip Generate Product', default=False)
    url = fields.Char('Website URL', compute="_compute_url")
    sum = fields.Integer('Total Product', compute="_compute_total_product", readonly=True)

    @api.depends('product_ids')
    def _compute_total_product(self):
        for record in self:
            record.sum = len(record.product_ids)

    @api.depends('keywords')
    def _compute_url(self):
        prefix = "https://indoteknik.com/searchkey/"
        for record in self:
            if record.keywords:
                slug = re.sub(r'[^a-zA-Z0-9]+', '-', record.keywords.strip().lower())
                slug = slug.strip('-')
                record.url = prefix + slug
            else:
                record.url = False

    def _compute_name(self):
        for record in self:
            if not record.name:
                record.name = record.keywords

    # @api.constrains('keywords')
    # def check_already_exist(self):
    #     model = self.env['keywords']
    #     for record in self:
    #         match = model.search([
    #             ('name', 'ilike', record.name)
    #         ])
    #         if match:
    #             raise UserError("Tidak bisa create/write/duplicate karena keywords sudah dipakai")


    def generate_products(self):
        for record in self:
            if not record.keywords:
                continue

            keyword = "%" + record.keywords.strip() + "%"

            # Query dasar
            sql = """
                SELECT pp.id
                FROM product_product pp
                JOIN product_template pt ON pt.id = pp.product_tmpl_id
                LEFT JOIN product_public_category_product_template_rel rel
                    ON rel.product_template_id = pt.id
                WHERE (
                    pt.name ILIKE %s
                    OR pt.website_description ILIKE %s
                    AND pt.unpublished = False
                    AND pt.product_rating >= 8
                )
            """

            params = [keyword, keyword]

            # Filter kategori ke child
            if record.category_id:
                child_categs = self.env['product.public.category'].search([
                    ('id', 'child_of', record.category_id.id)
                ])
                sql += " AND rel.product_public_category_id = ANY(%s)"
                params.append(child_categs.ids)

            # Exec SQL
            self.env.cr.execute(sql, params)
            rows = self.env.cr.fetchall()

            product_ids = [r[0] for r in rows]

            if not product_ids:
                raise UserError("Tidak berhasil menemukan barang")

            record.with_context(skip_generate=True).write({
                'product_ids': [(6, 0, product_ids)]
            })

            _logger.info(
                "Product Found: Found %s products for keyword '%s'",
                len(product_ids),
                record.keywords
            )

    def sync_solr(self):
        # solr_model = self.env['apache.solr']
        documents = []
        data = {}
        for keyword in self.search([]):
            searchkey = (keyword.keywords or '').strip().lower().replace(' ', '-')
            try:
                doc = {
                    'id': keyword.id,
                    'category_id_i': keyword.category_id.id,
                    'keywords_s': searchkey,
                    # 'searchkey_t': searchkey,
                    'url_s': keyword.url,
                    'product_ids_is': [p.product_tmpl_id.id for p in keyword.product_ids]
                }
                documents.append(doc)
                data = doc
            except Exception as e:
                _logger.error('failed %s', e)
                _logger.error('doc data: %s', data)
        solr.add(documents)
        return True

    @api.model
    def create(self, vals):
        record = super().create(vals)
        record.generate_products()
        return record

    def write(self, vals):
        result = super().write(vals)
        # if not self.env.context.get("skip_generate") and not self.skip:
            # self.generate_products()
        return result