fastled 1.2.70__py3-none-any.whl → 1.2.73__py3-none-any.whl

fastled/string_diff.py

@@ -3,6 +3,24 @@ from pathlib import Path
 from rapidfuzz import fuzz
 
 
+def _filter_out_obvious_bad_choices(
+    input_str: str, string_list: list[str]
+) -> list[str]:
+    """
+    Filter out strings that are too different from the input string.
+    This is a heuristic and may not be perfect.
+    """
+    input_chars = set(input_str)
+    filtered_list = []
+    for s in string_list:
+        # Check if at least half of the input characters are in the string
+        s_chars = set(s)
+        common_chars = input_chars.intersection(s_chars)
+        if len(common_chars) >= len(input_chars) / 2:
+            filtered_list.append(s)
+    return filtered_list
+
+
 # Returns the min distance strings. If there is a tie, it returns
 # all the strings that have the same min distance.
 # Returns a tuple of index and string.
@@ -24,6 +42,10 @@ def string_diff(
         string_list = [s.lower() for s in string_list]
         input_string = input_string.lower()
 
+    # Apply set membership filtering for queries with 3+ characters
+    if len(input_string) >= 3:
+        string_list = _filter_out_obvious_bad_choices(input_string, string_list)
+
     is_substring = False
     for s in string_list:
         if input_string in s: