import logging import meilisearch from django.conf import settings from django.db.models.signals import post_save, post_delete import types INDEX_NAME = 'items' SEARCHABLE_ATTRIBUTES = ['title', 'orig_title', 'other_title', 'subtitle', 'artist', 'author', 'translator', 'developer', 'director', 'actor', 'playwright', 'pub_house', 'company', 'publisher', 'isbn', 'imdb_code'] INDEXABLE_DIRECT_TYPES = ['BigAutoField', 'BooleanField', 'CharField', 'PositiveIntegerField', 'PositiveSmallIntegerField', 'TextField', 'ArrayField'] INDEXABLE_TIME_TYPES = ['DateTimeField'] INDEXABLE_DICT_TYPES = ['JSONField'] INDEXABLE_FLOAT_TYPES = ['DecimalField'] # NONINDEXABLE_TYPES = ['ForeignKey', 'FileField',] SEARCH_PAGE_SIZE = 20 logger = logging.getLogger(__name__) def item_post_save_handler(sender, instance, created, **kwargs): if not created and settings.SEARCH_INDEX_NEW_ONLY: return Indexer.replace_item(instance) def item_post_delete_handler(sender, instance, **kwargs): Indexer.delete_item(instance) def tag_post_save_handler(sender, instance, **kwargs): pass def tag_post_delete_handler(sender, instance, **kwargs): pass class Indexer: class_map = {} _instance = None @classmethod def instance(self): if self._instance is None: self._instance = meilisearch.Client(settings.MEILISEARCH_SERVER, settings.MEILISEARCH_KEY).index(INDEX_NAME) return self._instance @classmethod def init(self): meilisearch.Client(settings.MEILISEARCH_SERVER, settings.MEILISEARCH_KEY).create_index(INDEX_NAME, {'primaryKey': '_id'}) self.update_settings() @classmethod def update_settings(self): self.instance().update_searchable_attributes(SEARCHABLE_ATTRIBUTES) self.instance().update_filterable_attributes(['_class', 'tags', 'source_site']) self.instance().update_settings({'displayedAttributes': ['_id', '_class', 'id', 'title', 'tags']}) @classmethod def get_stats(self): return self.instance().get_stats() @classmethod def busy(self): return self.instance().get_stats()['isIndexing'] @classmethod def update_model_indexable(self, model): if settings.SEARCH_BACKEND is None: return self.class_map[model.__name__] = model model.indexable_fields = ['tags'] model.indexable_fields_time = [] model.indexable_fields_dict = [] model.indexable_fields_float = [] for field in model._meta.get_fields(): type = field.get_internal_type() if type in INDEXABLE_DIRECT_TYPES: model.indexable_fields.append(field.name) elif type in INDEXABLE_TIME_TYPES: model.indexable_fields_time.append(field.name) elif type in INDEXABLE_DICT_TYPES: model.indexable_fields_dict.append(field.name) elif type in INDEXABLE_FLOAT_TYPES: model.indexable_fields_float.append(field.name) post_save.connect(item_post_save_handler, sender=model) post_delete.connect(item_post_delete_handler, sender=model) @classmethod def obj_to_dict(self, obj): pk = f'{obj.__class__.__name__}-{obj.id}' item = { '_id': pk, '_class': obj.__class__.__name__, # 'id': obj.id } for field in obj.__class__.indexable_fields: item[field] = getattr(obj, field) for field in obj.__class__.indexable_fields_time: item[field] = getattr(obj, field).timestamp() for field in obj.__class__.indexable_fields_float: item[field] = float(getattr(obj, field)) if getattr(obj, field) else None for field in obj.__class__.indexable_fields_dict: d = getattr(obj, field) if d.__class__ is dict: item.update(d) item = {k: v for k, v in item.items() if v} return item @classmethod def replace_item(self, obj): try: self.instance().add_documents([self.obj_to_dict(obj)]) except Exception as e: logger.error(f"replace item error: \n{e}") @classmethod def replace_batch(self, objects): try: self.instance().update_documents(documents=objects) except Exception as e: logger.error(f"replace batch error: \n{e}") @classmethod def delete_item(self, obj): pk = f'{obj.__class__.__name__}-{obj.id}' try: self.instance().delete_document(pk) except Exception as e: logger.error(f"delete item error: \n{e}") @classmethod def patch_item(self, obj, fields): pk = f'{obj.__class__.__name__}-{obj.id}' data = {} for f in fields: data[f] = getattr(obj, f) try: self.instance().update_documents(documents=[data], primary_key=[pk]) except Exception as e: logger.error(f"patch item error: \n{e}") @classmethod def search(self, q, page=1, category=None, tag=None, sort=None): if category or tag: f = [] if category == 'music': f.append("(_class = 'Album' OR _class = 'Song')") elif category: f.append(f"_class = '{category}'") if tag: t = tag.replace("'", "\'") f.append(f"tags = '{t}'") filter = ' AND '.join(f) else: filter = None options = { 'offset': (page - 1) * SEARCH_PAGE_SIZE, 'limit': SEARCH_PAGE_SIZE, 'filter': filter, 'facetsDistribution': ['_class'], 'sort': None } try: r = self.instance().search(q, options) except Exception as e: logger.error(f"MeiliSearch error: \n{e}") r = {'nbHits': 0, 'hits': []} # print(r) results = types.SimpleNamespace() results.items = list([x for x in map(lambda i: self.item_to_obj(i), r['hits']) if x is not None]) results.num_pages = (r['nbHits'] + SEARCH_PAGE_SIZE - 1) // SEARCH_PAGE_SIZE # print(results) return results @classmethod def item_to_obj(self, item): try: return self.class_map[item['_class']].objects.get(id=item['id']) except Exception as e: logger.error(f"unable to load search result item from db:\n{item}") return None