feat(client): sort list command by query id

ddkasa · Nov 12, 2024 · 40dfffc · 40dfffc
1 parent bc9c748
commit 40dfffc
Showing 1 changed file with 14 additions and 1 deletion.
diff --git a/ulauncher_toggl_extension/commands/client.py b/ulauncher_toggl_extension/commands/client.py
@@ -15,6 +15,7 @@
     EDIT_IMG,
     REFRESH_IMG,
 )
+from ulauncher_toggl_extension.utils import get_distance
 
 from .meta import QueryResults, SubCommand
 
@@ -42,7 +43,19 @@ def get_models(self, query: Query, **kwargs: Any) -> list[TogglClient]:
             self.handle_error(err)
             clients = endpoint.collect()
 
-        clients.sort(key=lambda x: x.timestamp, reverse=query.sort_order)
+        if isinstance(query.id, int):
+            clients.sort(
+                key=lambda x: get_distance(query.id, x.id),
+                reverse=query.sort_order,
+            )
+        elif isinstance(query.id, str):
+            clients.sort(
+                key=lambda x: get_distance(query.id, x.name),
+                reverse=query.sort_order,
+            )
+        else:
+            clients.sort(key=lambda x: x.timestamp, reverse=query.sort_order)
+
         return clients
 
     def get_model(