mirror/krunner-vscode
1
Fork 0
mirror of https://github.com/j1g5awi/krunner-vscode.git synced 2024-11-22 15:19:30 +00:00
Code Activity

Provide accurate values for relevance

Browse source
This commit is contained in:
Jake Howard 2023-05-14 17:57:22 +01:00
parent e2cbbcd1f4
commit 102abf38e1
Signed by: jake
GPG key ID: 57AFB45680EDD477
1 changed files with 20 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

23
krunner_vscode/__main__.py
View file

@ -1,3 +1,4 @@
import difflib
import json import json
import os import os
import sqlite3 import sqlite3
@ -31,6 +32,21 @@ class Match(NamedTuple):
properties: dict properties: dict
def get_matches(paths, query):
"""
Equivalent to `difflib.get_close_matches`, but returning the ratio too
"""
matches = []
s = difflib.SequenceMatcher()
s.set_seq2(query)
for path in paths:
s.set_seq1(path)
matches.append((s.ratio(), path))
return matches
# Read path_list from database # Read path_list from database
def get_path_list(): def get_path_list():
paths = set() paths = set()
@ -72,17 +88,18 @@ class Runner(dbus.service.Object):
@dbus.service.method(iface, in_signature="s", out_signature="a(sssida{sv})") @dbus.service.method(iface, in_signature="s", out_signature="a(sssida{sv})")
def Match(self, query: str): def Match(self, query: str):
# data, display text, icon, type (Plasma::QueryType), relevance (0-1), properties (subtext, category and urls) # data, display text, icon, type (Plasma::QueryType), relevance (0-1), properties (subtext, category and urls)
matches = get_matches(get_path_list(), query)
return [ return [
Match( Match(
path, path,
Path(path).name, Path(path).name,
"com.visualstudio.code.oss", "com.visualstudio.code.oss",
100, 100,
1.0, ratio,
{"subtext": path}, {"subtext": path},
) )
for path in get_path_list() for ratio, path in matches
if query.lower() in Path(path).name.lower()
] ]
@dbus.service.method(iface, out_signature="a(sss)") @dbus.service.method(iface, out_signature="a(sss)")