diff options
| author | sinanmohd <sinan@sinanmohd.com> | 2023-10-08 14:36:33 +0530 |
|---|---|---|
| committer | sinanmohd <sinan@sinanmohd.com> | 2023-10-08 14:36:33 +0530 |
| commit | f04e77faa68095bfe7fbe8883aad59aff157da02 (patch) | |
| tree | 4baabb13b078bcc09d368cf4f53db92a1619fc4c /lib | |
| parent | 48700c12403ea1fc77d4fa6bd84ce1a208b876b0 (diff) | |
server/opensubtitles/search_ohash: use the new fuzzy search algorithm
Diffstat (limited to 'lib')
| -rw-r--r-- | lib/attr.lua | 22 |
1 files changed, 14 insertions, 8 deletions
diff --git a/lib/attr.lua b/lib/attr.lua index 8c1613a..565f4c1 100644 --- a/lib/attr.lua +++ b/lib/attr.lua @@ -32,6 +32,8 @@ local build_dlim = function (name, attrs) } local series = { '[Ss]%d%d?' .. dlim .. '[Ee]%d%d?', + '[Ss]%d%d?' .. dlim .. '[Ee][Pp]' .. dlim .. '%d%d?', + 'Season' .. dlim .. '%d%d?' .. dlim .. 'Episode' .. dlim .. '%d%d?' } local sizes = { '%d%d%d' .. dlim .. '[Mm][Bb]', @@ -171,9 +173,9 @@ local weigh = function (a1, a2) local key_score, score key_score = { - name = 10, - season = 10, - episode = 10, + season = 15, + episode = 15, + title = 10, source = 7, scene = 5, vcodec = 3, @@ -184,16 +186,20 @@ local weigh = function (a1, a2) score = 0 for k, v in pairs(a1) do - if not a2[k] then + if not a2[k] or k == 'episode' then goto continue end - if k == 'name' then - for _, name in pairs(v) do - if util.array_search(a2.name, name) then - score = score + key_score.name + if k == 'title' then + for _, title in pairs(v) do + if util.array_search(a2.title, title) then + score = score + key_score.title end end + elseif k == 'season' then + if a1.season == a2.season and a1.episode == a2.episode then + score = score + key_score.season + key_score.episode + end else if v == a2[k] then score = score + (key_score[k] or key_score.default) |