Amélioration recherche

chef_mou.py

@@ -6,11 +6,35 @@ import aiohttp
 import asyncio
 from bs4 import BeautifulSoup
 import random
+import re
+from difflib import SequenceMatcher
+from heapq import nlargest as _nlargest
 
 class ChefMou(Plugin):
     async def start(self) -> None:
         self.on_external_config_update()
 
+    def get_close_matches_indexes(self, word, possibilities, n=3, cutoff=0.6):
+        if not n >  0:
+            raise ValueError("n must be > 0: %r" % (n,))
+        if not 0.0 <= cutoff <= 1.0:
+            raise ValueError("cutoff must be in [0.0, 1.0]: %r" % (cutoff,))
+        result = []
+        s = SequenceMatcher()
+        s.set_seq2(word)
+        for idx, x in enumerate(possibilities):
+            s.set_seq1(x)
+            if s.real_quick_ratio() >= cutoff and \
+                s.quick_ratio() >= cutoff and \
+                s.ratio() >= cutoff:
+                result.append((s.ratio(), idx))
+
+        # Move the best scorers to head of list
+        result = _nlargest(n, result)
+
+        # Strip scores for the best n matches
+        return [x for score, x in result]
+
 
     async def get_site_content(self):
         async with aiohttp.ClientSession() as session:
@@ -49,10 +73,24 @@ class ChefMou(Plugin):
                     loop.close()
                     return
                 soup = await self.get_site_content()
-                s = soup.find('th', string=chunks[2])
-                if not s :
-                    await evt.respond("Morceau non trouvé")
-                    return
+                # Récuperer liste morceaux
+                s = soup.select('th', background='/fond3.gif')
+                # Object bts en str
+                parsedList=[]
+                originalList=[]
+                for ts in s:
+                    fullTitre = ts.text
+                    #Supression des () 
+                    parsedList.append(re.sub("[\(\[].*?[\)\]]", "", fullTitre))
+                    #On garde une copie pour la recherche
+                    originalList.append(ts.text)
+            
+                # Fonction qui retourne l'index du string le plus proche
+                idxTitre = self.get_close_matches_indexes(chunks[2], parsedList, cutoff=0.4)
+                # Si trouvé le chercher sinon erreur
+                if len(idxTitre) > 0 :
+                    #Et on recherche dans la liste originale avec le titre original
+                    s = soup.find('th', string=originalList[idxTitre[0]])
                     t = s.parent.findAll('a')
                     idx = 9
                     if chunks[1].lower() == "souba" or chunks[1].lower() == "soubassophone" or chunks[1].lower() == "sb":
@@ -71,8 +109,13 @@ class ChefMou(Plugin):
                         await evt.respond("Pas trouvé cet instru")
                         loop.close()
                         return
+                    try :
+                        await evt.respond("["+originalList[idxTitre[0]]+ " " + chunks[1]+"]"+"(<https://partoches.pustule.org"+t[idx].get("href")+">)")
+                    except IndexError:
+                        await evt.respond("Le morceau "+ strTitre[0]  + " existe mais pas ta partoche")
 
-                await evt.respond("["+chunks[2]+ " " + chunks[1]+"]"+"(<https://partoches.pustule.org"+t[idx].get("href")+">)")
+                else:
+                    await evt.respond("Morceau non trouvé")
             else:
                 await evt.respond("Nope !")
         loop.close()