blob: 77d780522c759bd9d034939a7c01ebfdd602fd2a (
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
|
from math import ceil
from queue import Queue
from typing import List
from szurubooru import model, search
_search_executor_config = search.configs.PostSearchConfig()
_search_executor = search.Executor(_search_executor_config)
def find_similar_posts(source_post: model.Post, limit: int) -> List[model.Post]:
results = []
# Sort tags in order of increasing post count, i.e. least to most popular
# This will help yield results quicker
source_tags = sorted(source_post.tags, key=lambda t: t.post_count)
source_tag_count = len(source_tags)
max_removals = ceil(source_tag_count / 2) # remove at most 50% of tags
tags = source_tags
for x in range(max_removals + 1):
# prepare the current search, remove known results
query = ' '.join([t.first_name for t in tags])
query += ' -id:%d' % source_post.post_id
for r in results:
query += ' -id:%d' % r.post_id
# execute
_, posts = _search_executor.execute(query, 0, limit - len(results))
# update results
for p in posts:
results.append(p)
if len(results) >= limit:
break
# remove the least popular tag
if len(tags) <= 1:
break
tags = tags[1:]
return results
|
