From f0284bb8a6f7fd0d816a1c3e9b1675607ceb3b70 Mon Sep 17 00:00:00 2001
From: Flancian <0@flancia.org>
Date: Mon, 16 Nov 2020 13:43:15 +0100
Subject: [PATCH] Enabled fuzzy matching by default.

---
 app/db.py | 7 ++++---
 1 file changed, 4 insertions(+), 3 deletions(-)

diff --git a/app/db.py b/app/db.py
index 46f8da0..ff5a3ce 100644
--- a/app/db.py
+++ b/app/db.py
@@ -17,6 +17,7 @@ import re
 import os
 from . import config
 from collections import defaultdict
+from fuzzywuzzy import fuzz
 from operator import attrgetter
 
 RE_WIKILINKS = re.compile('\[\[(.*?)\]\]')
@@ -146,10 +147,10 @@ def nodes_by_wikilink(wikilink):
     nodes = [node for node in all_nodes() if node.wikilink == wikilink]
     return nodes
 
-def subnodes_by_wikilink(wikilink, fuzzy=True):
-    if fuzzy:
+def subnodes_by_wikilink(wikilink, fuzzy_matching=True):
+    if fuzzy_matching:
         # TODO
-        subnodes = [subnode for subnode in all_subnodes() if subnode.wikilink == wikilink]
+        subnodes = [subnode for subnode in all_subnodes() if fuzz.ratio(subnode.wikilink, wikilink) > 90]
     else:
         subnodes = [subnode for subnode in all_subnodes() if subnode.wikilink == wikilink]
     return subnodes